Netty 处理简单HTTP请求的例子

淩亂°似流年 2022-07-18 00:28 281阅读 0赞

#### 废话不多说 上代码 ####

--------------------

**HttpHelloWorldServerInitializer.java**

import io.netty.channel.ChannelInitializer;
      import io.netty.channel.ChannelPipeline;
      import io.netty.channel.socket.SocketChannel;
      import io.netty.handler.codec.http.HttpServerCodec;
      import io.netty.handler.ssl.SslContext;
    
      public class HttpHelloWorldServerInitializer extends ChannelInitializer<SocketChannel> { 
    
          private final SslContext sslCtx;
    
          public HttpHelloWorldServerInitializer(SslContext sslCtx) {
             this.sslCtx = sslCtx;
          }
    
          @Override
          public void initChannel(SocketChannel ch) {
              ChannelPipeline p = ch.pipeline();
              if (sslCtx != null) {
                  p.addLast(sslCtx.newHandler(ch.alloc()));
              }
              p.addLast(new HttpServerCodec());
              p.addLast(new HttpHelloWorldServerHandler());
          }
      }

--------------------

**HttpHelloWorldServerHandler.java**

import io.netty.buffer.Unpooled;
    import io.netty.channel.ChannelFutureListener;
    import io.netty.channel.ChannelHandlerAdapter;
    import io.netty.channel.ChannelHandlerContext;
    import io.netty.handler.codec.http.DefaultFullHttpResponse;
    import io.netty.handler.codec.http.FullHttpResponse;
    import io.netty.handler.codec.http.HttpHeaderUtil;
    import io.netty.handler.codec.http.HttpHeaderValues;
    import io.netty.handler.codec.http.HttpRequest;
    
    import static io.netty.handler.codec.http.HttpHeaderNames.*;
    import static io.netty.handler.codec.http.HttpResponseStatus.*;
    import static io.netty.handler.codec.http.HttpVersion.*;
    
    
      public class HttpHelloWorldServerHandler extends ChannelHandlerAdapter { 
          private static final byte[] CONTENT = { 'H', 'e', 'l', 'l', 'o', ' ', 'W', 'o', 'r', 'l', 'd' };
    
          @Override
          public void channelReadComplete(ChannelHandlerContext ctx) {
              ctx.flush();
          }
    
          @Override
          public void channelRead(ChannelHandlerContext ctx, Object msg) {
              if (msg instanceof HttpRequest) {
                  HttpRequest req = (HttpRequest) msg;
    
                  if (HttpHeaderUtil.is100ContinueExpected(req)) {
                      ctx.write(new DefaultFullHttpResponse(HTTP_1_1, CONTINUE));
                  }
                  boolean keepAlive = HttpHeaderUtil.isKeepAlive(req);
                  FullHttpResponse response = new DefaultFullHttpResponse(HTTP_1_1, OK, Unpooled.wrappedBuffer(CONTENT));
                  response.headers().set(CONTENT_TYPE, "text/plain");
                  response.headers().setInt(CONTENT_LENGTH, response.content().readableBytes());
    
                  if (!keepAlive) {
                      ctx.write(response).addListener(ChannelFutureListener.CLOSE);
                  } else {
                      response.headers().set(CONNECTION, HttpHeaderValues.KEEP_ALIVE);
                      ctx.write(response);
                  }
              }
          }
    
          @Override
          public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) {
              cause.printStackTrace();
              ctx.close();
          }
      }

--------------------

**HttpHelloWorldServer.java**

import io.netty.bootstrap.ServerBootstrap;
    import io.netty.channel.Channel;
    import io.netty.channel.ChannelOption;
    import io.netty.channel.EventLoopGroup;
    import io.netty.channel.nio.NioEventLoopGroup;
    import io.netty.channel.socket.nio.NioServerSocketChannel;
    import io.netty.handler.logging.LogLevel;
    import io.netty.handler.logging.LoggingHandler;
    import io.netty.handler.ssl.SslContext;
    import io.netty.handler.ssl.util.SelfSignedCertificate;
    
    /** * An HTTP server that sends back the content of the received HTTP request * in a pretty plaintext form. */
    public final class HttpHelloWorldServer { 
    
          static final boolean SSL = System.getProperty("ssl") != null;
          static final int PORT = Integer.parseInt(System.getProperty("port", SSL? "8443" : "8080"));
    
          public static void main(String[] args) throws Exception {
              // Configure SSL.
              final SslContext sslCtx;
              if (SSL) {
                  SelfSignedCertificate ssc = new SelfSignedCertificate();
                  sslCtx = SslContext.newServerContext(ssc.certificate(), ssc.privateKey());
              } else {
                  sslCtx = null;
              }
    
              // Configure the server.
              EventLoopGroup bossGroup = new NioEventLoopGroup(1);
              EventLoopGroup workerGroup = new NioEventLoopGroup();
              try {
                  ServerBootstrap b = new ServerBootstrap();
                  b.option(ChannelOption.SO_BACKLOG, 1024);
                  b.group(bossGroup, workerGroup)
                   .channel(NioServerSocketChannel.class)
                   .handler(new LoggingHandler(LogLevel.INFO))
                   .childHandler(new HttpHelloWorldServerInitializer(sslCtx));
    
                  Channel ch = b.bind(PORT).sync().channel();
    
                  System.err.println("Open your web browser and navigate to " +
                          (SSL? "https" : "http") + "://127.0.0.1:" + PORT + '/');
    
                  ch.closeFuture().sync();
              } finally {
                  bossGroup.shutdownGracefully();
                  workerGroup.shutdownGracefully();
              }
          }
    
    
    }