82 lines
2.3 KiB
Java
82 lines
2.3 KiB
Java
|
/**
|
||
|
|
* Copyright (C) 2009-2013 Typesafe Inc. <http://www.typesafe.com>
|
||
|
|
*/
|
||
|
|
|
||
|
|
package docs.io.japi;
|
||
|
|
|
||
|
|
import java.net.InetSocketAddress;
|
||
|
|
|
||
|
|
import akka.actor.ActorRef;
|
||
|
|
import akka.actor.Props;
|
||
|
|
import akka.actor.SupervisorStrategy;
|
||
|
|
import akka.actor.UntypedActor;
|
||
|
|
import akka.event.Logging;
|
||
|
|
import akka.event.LoggingAdapter;
|
||
|
|
import akka.io.Tcp;
|
||
|
|
import akka.io.Tcp.Bind;
|
||
|
|
import akka.io.Tcp.Bound;
|
||
|
|
import akka.io.Tcp.CommandFailed;
|
||
|
|
import akka.io.Tcp.Connected;
|
||
|
|
import akka.io.TcpMessage;
|
||
|
|
|
||
|
|
public class EchoManager extends UntypedActor {
|
||
|
|
|
||
|
|
final LoggingAdapter log = Logging
|
||
|
|
.getLogger(getContext().system(), getSelf());
|
||
|
|
|
||
|
|
final Class<?> handlerClass;
|
||
|
|
|
||
|
|
public EchoManager(Class<?> handlerClass) {
|
||
|
|
this.handlerClass = handlerClass;
|
||
|
|
}
|
||
|
|
|
||
|
|
@Override
|
||
|
|
public SupervisorStrategy supervisorStrategy() {
|
||
|
|
return SupervisorStrategy.stoppingStrategy();
|
||
|
|
}
|
||
|
|
|
||
|
|
@Override
|
||
|
|
public void preStart() throws Exception {
|
||
|
|
//#manager
|
||
|
|
final ActorRef tcpManager = Tcp.get(getContext().system()).manager();
|
||
|
|
//#manager
|
||
|
|
tcpManager.tell(
|
||
|
|
TcpMessage.bind(getSelf(), new InetSocketAddress("localhost", 0), 100),
|
||
|
|
getSelf());
|
||
|
|
}
|
||
|
|
|
||
|
|
@Override
|
||
|
|
public void postRestart(Throwable arg0) throws Exception {
|
||
|
|
// do not restart
|
||
|
|
getContext().stop(getSelf());
|
||
|
|
}
|
||
|
|
|
||
|
|
@Override
|
||
|
|
public void onReceive(Object msg) throws Exception {
|
||
|
|
if (msg instanceof Bound) {
|
||
|
|
log.info("listening on [{}]", ((Bound) msg).localAddress());
|
||
|
|
} else if (msg instanceof Tcp.CommandFailed) {
|
||
|
|
final CommandFailed failed = (CommandFailed) msg;
|
||
|
|
if (failed.cmd() instanceof Bind) {
|
||
|
|
log.warning("cannot bind to [{}]", ((Bind) failed.cmd()).endpoint());
|
||
|
|
getContext().stop(getSelf());
|
||
|
|
} else {
|
||
|
|
log.warning("unknown command failed [{}]", failed.cmd());
|
||
|
|
}
|
||
|
|
} else
|
||
|
|
if (msg instanceof Connected) {
|
||
|
|
final Connected conn = (Connected) msg;
|
||
|
|
log.info("received connection from [{}]", conn.remoteAddress());
|
||
|
|
final ActorRef connection = getSender();
|
||
|
|
final ActorRef handler = getContext().actorOf(
|
||
|
|
Props.create(handlerClass, connection, conn.remoteAddress()));
|
||
|
|
//#echo-manager
|
||
|
|
connection.tell(TcpMessage.register(handler,
|
||
|
|
true, // <-- keepOpenOnPeerClosed flag
|
||
|
|
true), getSelf());
|
||
|
|
//#echo-manager
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|