过滤

tangrong · tangrong · commit b639c46c36b3 · 2020-11-05T01:18:25.000+08:00
diff --git a/02nio/nio02/src/main/java/io/github/kimmking/gateway/filter/TokenFilter.java b/02nio/nio02/src/main/java/io/github/kimmking/gateway/filter/TokenFilter.java
@@ -0,0 +1,33 @@
+package io.github.kimmking.gateway.filter;
+
+import cn.hutool.core.convert.Convert;
+import io.netty.buffer.Unpooled;
+import io.netty.channel.ChannelFutureListener;
+import io.netty.channel.ChannelHandlerContext;
+import io.netty.handler.codec.http.*;
+
+import static io.netty.handler.codec.http.HttpResponseStatus.OK;
+import static io.netty.handler.codec.http.HttpVersion.HTTP_1_1;
+
+public class TokenFilter implements HttpRequestFilter{
+    @Override
+    public void filter(FullHttpRequest fullRequest, ChannelHandlerContext ctx) {
+        HttpHeaders headers = fullRequest.trailingHeaders();
+
+        if (!headers.contains("token")) {
+            FullHttpResponse response = null;
+            byte[] bytes = "deny access".getBytes();
+            response = new DefaultFullHttpResponse(HTTP_1_1, OK, Unpooled.wrappedBuffer(bytes));
+            response.headers().set("Content-Type", "application/json");
+            response.setStatus(HttpResponseStatus.FORBIDDEN);
+            response.headers().setInt("Content-Length", Convert.toInt(bytes.length));
+            if (fullRequest != null) {
+                if (!HttpUtil.isKeepAlive(fullRequest)) {
+                    ctx.write(response).addListener(ChannelFutureListener.CLOSE);
+                } else {
+                    ctx.write(response);
+                }
+            }
+        }
+    }
+}
diff --git a/02nio/nio02/src/main/java/io/github/kimmking/gateway/inbound/HttpInboundHandler.java b/02nio/nio02/src/main/java/io/github/kimmking/gateway/inbound/HttpInboundHandler.java
@@ -1,5 +1,6 @@
 package io.github.kimmking.gateway.inbound;
 
+import io.github.kimmking.gateway.filter.TokenFilter;
 import io.github.kimmking.gateway.outbound.httpclient4.HttpOutboundHandler;
 import io.github.kimmking.gateway.outbound.okhttp.HutoolhttpOutboundHandler;
 import io.github.kimmking.gateway.router.HttpEndpointRouter;
@@ -34,19 +35,20 @@ public void channelRead(ChannelHandlerContext ctx, Object msg) {
         try {
 
             FullHttpRequest fullRequest = (FullHttpRequest) msg;
+            TokenFilter tokenFilter = new TokenFilter();
+            //过滤
+            tokenFilter.filter(fullRequest,ctx);
 
             String url = fullRequest.uri();
-            HttpEndpointRouter router = new HttpEndpointRouter() {
-                @Override
-                public String route(List<String> endpoints, List<String> proxyServers) {
-                    if (endpoints.stream().anyMatch(e->e.startsWith("/user"))) {
-                        return proxyServers.get(0);
-                    } else    if (endpoints.stream().anyMatch(e->e.startsWith("/order"))) {
-                        return proxyServers.get(1);
-                    }
-                    return null;
+            HttpEndpointRouter router = (endpoints, proxyServers) -> {
+                if (endpoints.stream().anyMatch(e->e.startsWith("/user"))) {
+                    return proxyServers.get(0);
+                } else    if (endpoints.stream().anyMatch(e->e.startsWith("/order"))) {
+                    return proxyServers.get(1);
                 }
+                return null;
             };
+            //路由
             String proxy = router.route(Arrays.asList(url), Arrays.asList(this.proxyServer.split(",").clone()));
             HutoolhttpOutboundHandler handler = new HutoolhttpOutboundHandler(proxy);
             handler.handle(fullRequest, ctx);
