Java 类com.facebook.nifty.core.ThriftServerDefBuilder 实例源码

private ThriftServerDefBuilder getThriftServerDefBuilder() {
  return new ThriftServerDefBuilder()
      .listen(0)
      .withProcessor(new scribe.Processor<>(new scribe.Iface() {
        @Override
        public ResultCode Log(List<LogEntry> messages) throws TException {
          RequestContext context = RequestContexts.getCurrentContext();
          for (LogEntry message : messages) {
            log.info("[Client: {}] {}: {}",
                context.getConnectionContext().getRemoteAddress(),
                message.getCategory(),
                message.getMessage());
          }
          return ResultCode.OK;
        }
      }));
}

@Override
public void start() throws STException {
    if (server == null) {
        ThriftServerDef thriftServerDef = new ThriftServerDefBuilder().listen(getServerPort())
                .withProcessor(getProcessor()).build();
        server = nettyServerConfig == null ? new NettyServerTransport(thriftServerDef)
                : new NettyServerTransport(thriftServerDef, nettyServerConfig, channelGroup);
    }
    server.start();
    LOG.debug("Server start .");
}

public static void startServer() {
    // Create the handler
    //ThriftTestService.Iface serviceInterface = 
   //   MyService.Iface serviceInterface = new MyServiceHandler();

    // Create the processor
    //TProcessor processor = new MyService.Processor<>(serviceInterface);

    // Create the processor
    //TProcessor processor = new ThriftTestService.Processor<>(new InMemoryScribe());

    InMemoryScribe inMemoryScribe = new InMemoryScribeImpl();
    TProtocolFactory protocolFactory  = new TBinaryProtocol.Factory();
    ThriftCodecManager thriftCodecManager = new ThriftCodecManager();
     List list  = new ArrayList<>();
     list.add(inMemoryScribe);

    ThriftServiceProcessor processor = new ThriftServiceProcessor(thriftCodecManager, Arrays.<ThriftEventHandler>asList(), inMemoryScribe);

    // Build the server definition
    ThriftServerDef serverDef = new ThriftServerDefBuilder().withProcessor(processor)
                                                            .build();

    // Create the server transport
    final NettyServerTransport server = new NettyServerTransport(serverDef  );

    // Create netty boss and executor thread pools
    ExecutorService bossExecutor = Executors.newCachedThreadPool();
    ExecutorService workerExecutor = Executors.newCachedThreadPool();

    // Start the server
    //server.start(bossExecutor, workerExecutor);
    server.start();
    // Arrange to stop the server at shutdown
    Runtime.getRuntime().addShutdownHook(new Thread() {
        @Override
        public void run() {
            try {
                server.stop();
            } catch (InterruptedException e) {
                Thread.currentThread().interrupt();
            }
        }
    });
}

public static void main(String[] args) {
    ThriftServiceProcessor processor = new ThriftServiceProcessor(
               new ThriftCodecManager(),
               ImmutableList.<ThriftEventHandler>of(),
               new ThirdPartyCollectionServiceImpl()
       );



    // Build the server definition
    ThriftServerDef serverDef = new ThriftServerDefBuilder()
            .listen(8899)
            .withProcessor(processor)
            .build();

    // Create the server transport
    final NettyServerTransport server = new NettyServerTransport(serverDef  );

    // Create netty boss and executor thread pools
    ExecutorService bossExecutor = Executors.newCachedThreadPool();
    ExecutorService workerExecutor = Executors.newCachedThreadPool();

    // Start the server
    //server.start(bossExecutor, workerExecutor);
    server.start();
    // Arrange to stop the server at shutdown
    Runtime.getRuntime().addShutdownHook(new Thread() {
        @Override
        public void run() {
            try {
                server.stop();
            } catch (InterruptedException e) {
                Thread.currentThread().interrupt();
            }
        }
    });

    /**ThreadPool taskWorkerExecutor = newFixedThreadPool(1);

       ThriftServerDef serverDef = ThriftServerDef.newBuilder()
               .listen(8899)
               .withProcessor(processor)
               .using(taskWorkerExecutor)
               .build();

       bossExecutor = newCachedThreadPool();
       ioWorkerExecutor = newCachedThreadPool();

       NettyServerConfig serverConfig = NettyServerConfig.newBuilder()
               .setBossThreadExecutor(bossExecutor)
               .setWorkerThreadExecutor(ioWorkerExecutor)
               .build();

       server = new ThriftServer(serverConfig, serverDef);
       server.start();**/

}

public static void main(String[] args) throws Exception {

        EventLoopGroup workerGroup = new NioEventLoopGroup();
        EventLoopGroup bossGroup = new NioEventLoopGroup();

        try {
            ServerBootstrap b = new ServerBootstrap();
            b.group(bossGroup, workerGroup);
            b.channel(NioServerSocketChannel.class);
            b.handler(new LoggingHandler(LogLevel.DEBUG));
            b.childHandler(new ChannelInitializer<SocketChannel>() {

                @Override
                public void initChannel(SocketChannel ch) throws Exception {

                    TMultiplexedProcessor multiprocessor = new TMultiplexedProcessor();

                    multiprocessor.registerProcessor("Calculator", new Calculator.Processor(new CalculatorHandler()));

                    multiprocessor.registerProcessor("Scribe", new scribe.Processor<scribe.Iface>(new scribe.Iface() {
                        @Override
                        public ResultCode Log(List<LogEntry> messages) throws TException {
                            for (LogEntry message : messages) {
                                log.info("{}: {}", message.getCategory(), message.getMessage());
                            }
                            return ResultCode.OK;
                        }
                    }));

                    ThriftServerDef def = new ThriftServerDefBuilder().withProcessor(multiprocessor).build();

                    ChannelPipeline pipeline = ch.pipeline();

                    pipeline.addLast("frameDecoder", new ThriftFrameDecoder(def.getMaxFrameSize(), def.getInProtocolFactory()));

                    pipeline.addLast("dispatcher", new NiftyDispatcher(def));

                }
            });
            b.option(ChannelOption.SO_BACKLOG, 128);
            b.childOption(ChannelOption.SO_KEEPALIVE, true);
            log.debug("configuration serverBootstrap");

            if (log.isInfoEnabled()) {
                log.info("Start server with port: {} ", 9090);
            } else if (log.isWarnEnabled()) {
                log.warn("Start server with port: {} ", 9090);
            } else if (log.isErrorEnabled()) {
                log.error("Start server with port: {} ", 9090);
            }
            Channel serverChannel = b.bind(9090).sync().channel().closeFuture().sync().channel();

        } finally {
            workerGroup.shutdownGracefully();
            bossGroup.shutdownGracefully();
        }
    }

private void startServer(final ThriftServerDefBuilder thriftServerDefBuilder) {
  server = new NettyServerTransport(thriftServerDefBuilder.build(), NettyServerConfig.newBuilder().build(), new DefaultChannelGroup());
  server.start();
  port = ((InetSocketAddress) server.getServerChannel().getLocalAddress()).getPort();
}