move multi-node-testkit into its own module, see #2981

2013-02-10 16:04:59 -05:00 · 2013-02-10 16:04:59 -05:00 · c3b363c9b3
commit c3b363c9b3
parent 29b274b9c0
18 changed files with 103 additions and 33 deletions
--- a/akka-multi-node-testkit/src/main/scala/akka/remote/testconductor/RemoteConnection.scala
+++ b/akka-multi-node-testkit/src/main/scala/akka/remote/testconductor/RemoteConnection.scala
@ -0,0 +1,77 @@
+/**
+ *  Copyright (C) 2009-2011 Typesafe Inc. <http://www.typesafe.com>
+ */
+package akka.remote.testconductor
+
+import org.jboss.netty.channel.{ Channel, ChannelPipeline, ChannelPipelineFactory, ChannelUpstreamHandler, SimpleChannelUpstreamHandler, DefaultChannelPipeline }
+import org.jboss.netty.channel.socket.nio.{ NioClientSocketChannelFactory, NioServerSocketChannelFactory }
+import org.jboss.netty.bootstrap.{ ClientBootstrap, ServerBootstrap }
+import org.jboss.netty.handler.codec.frame.{ LengthFieldBasedFrameDecoder, LengthFieldPrepender }
+import org.jboss.netty.handler.codec.compression.{ ZlibDecoder, ZlibEncoder }
+import org.jboss.netty.handler.codec.protobuf.{ ProtobufDecoder, ProtobufEncoder }
+import org.jboss.netty.handler.timeout.{ ReadTimeoutHandler, ReadTimeoutException }
+import java.net.InetSocketAddress
+import java.util.concurrent.Executors
+import akka.event.Logging
+import akka.util.Helpers
+
+/**
+ * INTERNAL API.
+ */
+private[akka] class TestConductorPipelineFactory(handler: ChannelUpstreamHandler) extends ChannelPipelineFactory {
+  def getPipeline: ChannelPipeline = {
+    val encap = List(new LengthFieldPrepender(4), new LengthFieldBasedFrameDecoder(10000, 0, 4, 0, 4))
+    val proto = List(new ProtobufEncoder, new ProtobufDecoder(TestConductorProtocol.Wrapper.getDefaultInstance))
+    val msg = List(new MsgEncoder, new MsgDecoder)
+    (encap ::: proto ::: msg ::: handler :: Nil).foldLeft(new DefaultChannelPipeline) {
+      (pipe, handler) ⇒ pipe.addLast(Logging.simpleName(handler.getClass), handler); pipe
+    }
+  }
+}
+
+/**
+ * INTERNAL API.
+ */
+private[akka] sealed trait Role
+/**
+ * INTERNAL API.
+ */
+private[akka] case object Client extends Role
+/**
+ * INTERNAL API.
+ */
+private[akka] case object Server extends Role
+
+/**
+ * INTERNAL API.
+ */
+private[akka] object RemoteConnection {
+  def apply(role: Role, sockaddr: InetSocketAddress, poolSize: Int, handler: ChannelUpstreamHandler): Channel = {
+    role match {
+      case Client ⇒
+        val socketfactory = new NioClientSocketChannelFactory(Executors.newCachedThreadPool, Executors.newCachedThreadPool,
+          poolSize)
+        val bootstrap = new ClientBootstrap(socketfactory)
+        bootstrap.setPipelineFactory(new TestConductorPipelineFactory(handler))
+        bootstrap.setOption("tcpNoDelay", true)
+        bootstrap.connect(sockaddr).getChannel
+      case Server ⇒
+        val socketfactory = new NioServerSocketChannelFactory(Executors.newCachedThreadPool, Executors.newCachedThreadPool,
+          poolSize)
+        val bootstrap = new ServerBootstrap(socketfactory)
+        bootstrap.setPipelineFactory(new TestConductorPipelineFactory(handler))
+        bootstrap.setOption("reuseAddress", !Helpers.isWindows)
+        bootstrap.setOption("child.tcpNoDelay", true)
+        bootstrap.bind(sockaddr)
+    }
+  }
+
+  def getAddrString(channel: Channel) = channel.getRemoteAddress match {
+    case i: InetSocketAddress ⇒ i.toString
+    case _                    ⇒ "[unknown]"
+  }
+
+  def shutdown(channel: Channel) = {
+    channel.getFactory.releaseExternalResources()
+  }
+}