cyy 1、提交NettyRequestFilter

cutie · cutie · commit a1b94758df1f · 2020-11-04T23:50:25.000+08:00
diff --git a/02nio/nio02/src/main/java/io/github/kimmking/gateway/filter/NettyRequestFilter.java b/02nio/nio02/src/main/java/io/github/kimmking/gateway/filter/NettyRequestFilter.java
@@ -0,0 +1,20 @@
+package io.github.kimmking.gateway.filter;
+
+import io.netty.channel.ChannelHandlerContext;
+import io.netty.channel.ChannelInboundHandlerAdapter;
+import io.netty.handler.codec.http.FullHttpRequest;
+
+public class NettyRequestFilter extends ChannelInboundHandlerAdapter implements HttpRequestFilter{
+
+    @Override
+    public void filter(FullHttpRequest fullRequest, ChannelHandlerContext ctx) {
+        fullRequest.headers().set("nio", "cyy");
+        ctx.fireChannelRead(fullRequest);
+    }
+
+    @Override
+    public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception {
+        FullHttpRequest fullRequest = (FullHttpRequest) msg;
+        this.filter(fullRequest, ctx);
+    }
+}
diff --git a/02nio/nio02/src/main/java/io/github/kimmking/gateway/inbound/HttpInboundInitializer.java b/02nio/nio02/src/main/java/io/github/kimmking/gateway/inbound/HttpInboundInitializer.java
@@ -1,5 +1,6 @@
 package io.github.kimmking.gateway.inbound;
 
+import io.github.kimmking.gateway.filter.NettyRequestFilter;
 import io.netty.channel.ChannelInitializer;
 import io.netty.channel.ChannelPipeline;
 import io.netty.channel.socket.SocketChannel;
@@ -23,6 +24,7 @@ public void initChannel(SocketChannel ch) {
 		p.addLast(new HttpServerCodec());
 		//p.addLast(new HttpServerExpectContinueHandler());
 		p.addLast(new HttpObjectAggregator(1024 * 1024));
+		p.addLast(new NettyRequestFilter());
 		p.addLast(new HttpInboundHandler(this.proxyServer));
 	}
 }

Original file line number	Diff line number	Diff line change
`@@ -1,5 +1,6 @@`
`1`	`1`	`package io.github.kimmking.gateway.inbound;`
`2`	`2`
	`3`	`+import io.github.kimmking.gateway.filter.NettyRequestFilter;`
`3`	`4`	`import io.netty.channel.ChannelInitializer;`
`4`	`5`	`import io.netty.channel.ChannelPipeline;`
`5`	`6`	`import io.netty.channel.socket.SocketChannel;`
`@@ -23,6 +24,7 @@ public void initChannel(SocketChannel ch) {`
`23`	`24`	`p.addLast(new HttpServerCodec());`
`24`	`25`	`//p.addLast(new HttpServerExpectContinueHandler());`
`25`	`26`	`p.addLast(new HttpObjectAggregator(1024 * 1024));`
	`27`	`+ p.addLast(new NettyRequestFilter());`
`26`	`28`	`p.addLast(new HttpInboundHandler(this.proxyServer));`
`27`	`29`	`}`
`28`	`30`	`}`