core/src/main/java/com/google/net/stubby/spdy/netty/SpdyServer.java

package com.google.net.stubby.spdy.netty;

import com.google.net.stubby.RequestRegistry;
import com.google.net.stubby.Session;

import io.netty.bootstrap.ServerBootstrap;
import io.netty.buffer.UnpooledByteBufAllocator;
import io.netty.channel.Channel;
import io.netty.channel.ChannelFuture;
import io.netty.channel.ChannelInitializer;
import io.netty.channel.ChannelOption;
import io.netty.channel.EventLoopGroup;
import io.netty.channel.nio.NioEventLoopGroup;
import io.netty.channel.socket.SocketChannel;
import io.netty.channel.socket.nio.NioServerSocketChannel;
import io.netty.handler.codec.spdy.SpdyFrameCodec;
import io.netty.handler.codec.spdy.SpdyVersion;

/**
 * Simple server connection startup that attaches a {@link Session} implementation to
 * a connection.
 */
public class SpdyServer implements Runnable {
    private final int port;
    private final Session session;
    private final RequestRegistry operations;
    private Channel channel;

    public SpdyServer(int port, Session session, RequestRegistry operations) {
      this.port = port;
      this.session = session;
      this.operations = operations;
    }

    @Override
    public void run() {
      EventLoopGroup bossGroup = new NioEventLoopGroup(); // (1)
      EventLoopGroup workerGroup = new NioEventLoopGroup();
      try {
        ServerBootstrap b = new ServerBootstrap(); // (2)
        // TODO(user): Evaluate use of pooled allocator
        b.childOption(ChannelOption.ALLOCATOR, UnpooledByteBufAllocator.DEFAULT);
        b.group(bossGroup, workerGroup)
            .channel(NioServerSocketChannel.class) // (3)
            .childHandler(new ChannelInitializer<SocketChannel>() { // (4)
              @Override
              public void initChannel(SocketChannel ch) throws Exception {
                ch.pipeline().addLast(
                    new SpdyFrameCodec(SpdyVersion.SPDY_3_1),
                    new SpdyCodec(session, operations));
              }
            })
            .option(ChannelOption.SO_BACKLOG, 128)          // (5)
            .childOption(ChannelOption.SO_KEEPALIVE, true); // (6)

        // Bind and startContext to accept incoming connections.
        ChannelFuture f = b.bind(port).sync(); // (7)

        // Wait until the server socket is closed.
        channel = f.channel();
        channel.closeFuture().sync();
      } catch (Exception e) {
        e.printStackTrace();
      } finally {
        workerGroup.shutdownGracefully();
        bossGroup.shutdownGracefully();
      }
    }

    public void stop() throws Exception {
      if (channel != null) {
        channel.close().get();
      }
    }
}