monad.rpc.LocalRpcServerModule.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of monad-rpc Show documentation
The newest version!
// Copyright 2014,2015 the original author or authors. All rights reserved.
// site: http://www.ganshane.com
package monad.rpc

import monad.rpc.internal.NettyRpcServerImpl
import monad.rpc.protocol.CommandProto.BaseCommand
import monad.rpc.services._
import org.apache.tapestry5.ioc.annotations.ServiceId
import org.apache.tapestry5.ioc.services.{ChainBuilder, PipelineBuilder, RegistryShutdownHub}
import org.apache.tapestry5.ioc.{ObjectLocator, OrderedConfiguration}
import org.slf4j.Logger

/**
 * local rpc module
 */
object LocalRpcServerModule {
  /*
  def bind(binder: ServiceBinder) {
    binder.bind(classOf[RpcServer], classOf[NettyRpcServerImpl]).
      withId("RpcServer")
  }
  */
  //@Contribute(classOf[RegistryStartup])
  def contributeRegistryStartup(registryShutdownHub: RegistryShutdownHub, orderedConfiguration: OrderedConfiguration[Runnable], objectLocator: ObjectLocator): Unit = {
    val server = objectLocator.autobuild(classOf[NettyRpcServerImpl])
    //最后启动
    orderedConfiguration.add("rpc-server", new Runnable() {
      override def run(): Unit = {
        server.start(registryShutdownHub)
      }
    }, "after:*")
    //最先关闭
    orderedConfiguration.add("rpc-server-closer", new Runnable {
      override def run(): Unit = {
        server.registryShutdownListener(registryShutdownHub)
      }
    }, "before:*")

  }


  def buildRpcServerListener(chainBuilder: ChainBuilder, configuration: java.util.List[RpcServerListener]): RpcServerListener = {
    chainBuilder.build(classOf[RpcServerListener], configuration)
  }

  @ServiceId("RpcServerMessageHandler")
  def buildRpcServerMessageHandler(pipelineBuilder: PipelineBuilder, logger: Logger,
                                   configuration: java.util.List[RpcServerMessageFilter])
  : RpcServerMessageHandler = {
    val terminator = new RpcServerMessageHandler {
      /**
       * @param commandRequest message command
       * @return handled if true .
       */
      override def handle(commandRequest: BaseCommand, response: CommandResponse): Boolean = false
    }
    pipelineBuilder.build(logger, classOf[RpcServerMessageHandler], classOf[RpcServerMessageFilter], configuration, terminator)
  }
}