From 570794b45497a3da4bdc91ac9690e450c216b660 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=94=90=E6=A6=95?= <solike_club@sina.com>
Date: Wed, 4 Nov 2020 19:18:27 +0800
Subject: [PATCH 1/3] =?UTF-8?q?Week03=20=E4=BD=9C=E4=B8=9A=E9=A2=98?=
 =?UTF-8?q?=E7=9B=AE=EF=BC=88=E5=91=A8=E5=9B=9B=EF=BC=89?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 02nio/nio02/pom.xml                           | 14 +++-
 .../gateway/NettyServerApplication.java       |  2 +-
 .../gateway/inbound/HttpInboundHandler.java   |  5 +-
 .../httpclient4/HttpOutboundHandler.java      |  3 +-
 .../okhttp/HutoolhttpOutboundHandler.java     | 81 +++++++++++++++++++
 .../okhttp/OkhttpOutboundHandler.java         |  4 -
 6 files changed, 100 insertions(+), 9 deletions(-)
 create mode 100644 02nio/nio02/src/main/java/io/github/kimmking/gateway/outbound/okhttp/HutoolhttpOutboundHandler.java
 delete mode 100644 02nio/nio02/src/main/java/io/github/kimmking/gateway/outbound/okhttp/OkhttpOutboundHandler.java

diff --git a/02nio/nio02/pom.xml b/02nio/nio02/pom.xml
index 6cbbeffd..960654ba 100644
--- a/02nio/nio02/pom.xml
+++ b/02nio/nio02/pom.xml
@@ -52,7 +52,19 @@
             <artifactId>httpasyncclient</artifactId>
             <version>4.1.4</version>
         </dependency>
-    
+
+        <dependency>
+            <groupId>org.apache.httpcomponents</groupId>
+            <artifactId>httpasyncclient</artifactId>
+            <version>4.1.4</version>
+        </dependency>
+
+        <!-- 国人开发的Java工具包-->
+        <dependency>
+            <groupId>cn.hutool</groupId>
+            <artifactId>hutool-all</artifactId>
+            <version>5.3.9</version>
+        </dependency>
         <!--
         <dependency>
             <groupId>org.springframework.boot</groupId>
diff --git a/02nio/nio02/src/main/java/io/github/kimmking/gateway/NettyServerApplication.java b/02nio/nio02/src/main/java/io/github/kimmking/gateway/NettyServerApplication.java
index 870b2d4f..3e1e5496 100644
--- a/02nio/nio02/src/main/java/io/github/kimmking/gateway/NettyServerApplication.java
+++ b/02nio/nio02/src/main/java/io/github/kimmking/gateway/NettyServerApplication.java
@@ -9,7 +9,7 @@ public class NettyServerApplication {
     public final static String GATEWAY_VERSION = "1.0.0";
     
     public static void main(String[] args) {
-        String proxyServer = System.getProperty("proxyServer","http://localhost:8088");
+        String proxyServer = System.getProperty("proxyServer","http://localhost:8080");
         String proxyPort = System.getProperty("proxyPort","8888");
         
           //  http://localhost:8888/api/hello  ==> gateway API
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
index 22fb2525..e4d6d2f2 100644
--- 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,6 +1,7 @@
 package io.github.kimmking.gateway.inbound;
 
 import io.github.kimmking.gateway.outbound.httpclient4.HttpOutboundHandler;
+import io.github.kimmking.gateway.outbound.okhttp.HutoolhttpOutboundHandler;
 import io.netty.channel.ChannelHandlerContext;
 import io.netty.channel.ChannelInboundHandlerAdapter;
 import io.netty.handler.codec.http.FullHttpRequest;
@@ -12,11 +13,11 @@ public class HttpInboundHandler extends ChannelInboundHandlerAdapter {
 
     private static Logger logger = LoggerFactory.getLogger(HttpInboundHandler.class);
     private final String proxyServer;
-    private HttpOutboundHandler handler;
+    private HutoolhttpOutboundHandler handler;
     
     public HttpInboundHandler(String proxyServer) {
         this.proxyServer = proxyServer;
-        handler = new HttpOutboundHandler(this.proxyServer);
+        handler = new HutoolhttpOutboundHandler(this.proxyServer);
     }
     
     @Override
diff --git a/02nio/nio02/src/main/java/io/github/kimmking/gateway/outbound/httpclient4/HttpOutboundHandler.java b/02nio/nio02/src/main/java/io/github/kimmking/gateway/outbound/httpclient4/HttpOutboundHandler.java
index 856dc168..93475ddf 100644
--- a/02nio/nio02/src/main/java/io/github/kimmking/gateway/outbound/httpclient4/HttpOutboundHandler.java
+++ b/02nio/nio02/src/main/java/io/github/kimmking/gateway/outbound/httpclient4/HttpOutboundHandler.java
@@ -1,6 +1,7 @@
 package io.github.kimmking.gateway.outbound.httpclient4;
 
 
+import cn.hutool.core.convert.Convert;
 import io.netty.buffer.Unpooled;
 import io.netty.channel.ChannelFutureListener;
 import io.netty.channel.ChannelHandlerContext;
@@ -103,7 +104,7 @@ private void handleResponse(final FullHttpRequest fullRequest, final ChannelHand
     
             response = new DefaultFullHttpResponse(HTTP_1_1, OK, Unpooled.wrappedBuffer(body));
             response.headers().set("Content-Type", "application/json");
-            response.headers().setInt("Content-Length", Integer.parseInt(endpointResponse.getFirstHeader("Content-Length").getValue()));
+            response.headers().setInt("Content-Length", Convert.toInt(body.length));
     
 //            for (Header e : endpointResponse.getAllHeaders()) {
 //                //response.headers().set(e.getName(),e.getValue());
diff --git a/02nio/nio02/src/main/java/io/github/kimmking/gateway/outbound/okhttp/HutoolhttpOutboundHandler.java b/02nio/nio02/src/main/java/io/github/kimmking/gateway/outbound/okhttp/HutoolhttpOutboundHandler.java
new file mode 100644
index 00000000..8f6a07ed
--- /dev/null
+++ b/02nio/nio02/src/main/java/io/github/kimmking/gateway/outbound/okhttp/HutoolhttpOutboundHandler.java
@@ -0,0 +1,81 @@
+package io.github.kimmking.gateway.outbound.okhttp;
+
+import cn.hutool.core.convert.Convert;
+import cn.hutool.http.HttpRequest;
+import cn.hutool.http.HttpResponse;
+import io.github.kimmking.gateway.outbound.httpclient4.NamedThreadFactory;
+import io.netty.buffer.Unpooled;
+import io.netty.channel.ChannelFutureListener;
+import io.netty.channel.ChannelHandlerContext;
+import io.netty.handler.codec.http.DefaultFullHttpResponse;
+import io.netty.handler.codec.http.FullHttpRequest;
+import io.netty.handler.codec.http.FullHttpResponse;
+import io.netty.handler.codec.http.HttpUtil;
+
+import java.util.concurrent.ArrayBlockingQueue;
+import java.util.concurrent.ExecutorService;
+import java.util.concurrent.ThreadPoolExecutor;
+import java.util.concurrent.TimeUnit;
+
+import static io.netty.handler.codec.http.HttpResponseStatus.NO_CONTENT;
+import static io.netty.handler.codec.http.HttpResponseStatus.OK;
+import static io.netty.handler.codec.http.HttpVersion.HTTP_1_1;
+
+public class HutoolhttpOutboundHandler {
+    private final String backendUrl;
+    private final ExecutorService proxyService;
+
+    public HutoolhttpOutboundHandler(String backendUrl) {
+        this.backendUrl = backendUrl.endsWith("/") ? backendUrl.substring(0, backendUrl.length() - 1) : backendUrl;
+        int cores = Runtime.getRuntime().availableProcessors() * 2;
+        long keepAliveTime = 1000;
+        int queueSize = 2048;
+
+        proxyService = new ThreadPoolExecutor(cores, cores,
+                keepAliveTime, TimeUnit.MILLISECONDS, new ArrayBlockingQueue<>(queueSize),
+                new NamedThreadFactory("proxyService"), new ThreadPoolExecutor.CallerRunsPolicy());
+
+    }
+
+    public void handle(final FullHttpRequest fullRequest, final ChannelHandlerContext ctx) {
+        final String url = this.backendUrl + fullRequest.uri();
+        proxyService.submit(() -> fetchGet(fullRequest, ctx, url));
+    }
+
+    private void fetchGet(final FullHttpRequest inbound, final ChannelHandlerContext ctx, final String url) {
+        HttpResponse response = new HttpRequest(url).keepAlive(true).executeAsync();
+        handleResponse(inbound, ctx, response);
+    }
+
+    private void handleResponse(final FullHttpRequest fullRequest, final ChannelHandlerContext ctx, final HttpResponse endpointResponse) {
+        FullHttpResponse response = null;
+        try {
+
+            byte[] body = endpointResponse.bodyBytes();
+
+            response = new DefaultFullHttpResponse(HTTP_1_1, OK, Unpooled.wrappedBuffer(body));
+            response.headers().set("Content-Type", "application/json");
+            response.headers().setInt("Content-Length", Convert.toInt(body.length));
+
+        } catch (Exception e) {
+            e.printStackTrace();
+            response = new DefaultFullHttpResponse(HTTP_1_1, NO_CONTENT);
+            exceptionCaught(ctx, e);
+        } finally {
+            if (fullRequest != null) {
+                if (!HttpUtil.isKeepAlive(fullRequest)) {
+                    ctx.write(response).addListener(ChannelFutureListener.CLOSE);
+                } else {
+                    ctx.write(response);
+                }
+            }
+            ctx.flush();
+        }
+
+    }
+
+    public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) {
+        cause.printStackTrace();
+        ctx.close();
+    }
+}
diff --git a/02nio/nio02/src/main/java/io/github/kimmking/gateway/outbound/okhttp/OkhttpOutboundHandler.java b/02nio/nio02/src/main/java/io/github/kimmking/gateway/outbound/okhttp/OkhttpOutboundHandler.java
deleted file mode 100644
index 5f194588..00000000
--- a/02nio/nio02/src/main/java/io/github/kimmking/gateway/outbound/okhttp/OkhttpOutboundHandler.java
+++ /dev/null
@@ -1,4 +0,0 @@
-package io.github.kimmking.gateway.outbound.okhttp;
-
-public class OkhttpOutboundHandler {
-}

From 1e78fd4f7898d3f54ab41e10a8a78efe189be154 Mon Sep 17 00:00:00 2001
From: tangrong <solike_club@sina.com>
Date: Thu, 5 Nov 2020 01:01:41 +0800
Subject: [PATCH 2/3] =?UTF-8?q?=E8=B7=AF=E7=94=B1?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../gateway/NettyServerApplication.java       |  5 +++-
 .../gateway/inbound/HttpInboundHandler.java   | 27 ++++++++++++++-----
 .../gateway/router/HttpEndpointRouter.java    |  5 ++--
 3 files changed, 26 insertions(+), 11 deletions(-)

diff --git a/02nio/nio02/src/main/java/io/github/kimmking/gateway/NettyServerApplication.java b/02nio/nio02/src/main/java/io/github/kimmking/gateway/NettyServerApplication.java
index 3e1e5496..44131626 100644
--- a/02nio/nio02/src/main/java/io/github/kimmking/gateway/NettyServerApplication.java
+++ b/02nio/nio02/src/main/java/io/github/kimmking/gateway/NettyServerApplication.java
@@ -1,15 +1,18 @@
 package io.github.kimmking.gateway;
 
 
+import cn.hutool.core.collection.CollUtil;
 import io.github.kimmking.gateway.inbound.HttpInboundServer;
 
+import java.util.Arrays;
+
 public class NettyServerApplication {
     
     public final static String GATEWAY_NAME = "NIOGateway";
     public final static String GATEWAY_VERSION = "1.0.0";
     
     public static void main(String[] args) {
-        String proxyServer = System.getProperty("proxyServer","http://localhost:8080");
+        String proxyServer = CollUtil.join(Arrays.asList("http://locallhost:8080","http://locallhost:8081"),",");
         String proxyPort = System.getProperty("proxyPort","8888");
         
           //  http://localhost:8888/api/hello  ==> gateway API
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
index e4d6d2f2..81b58433 100644
--- 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
@@ -2,6 +2,7 @@
 
 import io.github.kimmking.gateway.outbound.httpclient4.HttpOutboundHandler;
 import io.github.kimmking.gateway.outbound.okhttp.HutoolhttpOutboundHandler;
+import io.github.kimmking.gateway.router.HttpEndpointRouter;
 import io.netty.channel.ChannelHandlerContext;
 import io.netty.channel.ChannelInboundHandlerAdapter;
 import io.netty.handler.codec.http.FullHttpRequest;
@@ -9,6 +10,9 @@
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
+import java.util.Arrays;
+import java.util.List;
+
 public class HttpInboundHandler extends ChannelInboundHandlerAdapter {
 
     private static Logger logger = LoggerFactory.getLogger(HttpInboundHandler.class);
@@ -28,14 +32,23 @@ public void channelReadComplete(ChannelHandlerContext ctx) {
     @Override
     public void channelRead(ChannelHandlerContext ctx, Object msg) {
         try {
-            //logger.info("channelRead流量接口请求开始，时间为{}", startTime);
+
             FullHttpRequest fullRequest = (FullHttpRequest) msg;
-//            String uri = fullRequest.uri();
-//            //logger.info("接收到的请求url为{}", uri);
-//            if (uri.contains("/test")) {
-//                handlerTest(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;
+                }
+            };
+            String proxy = router.route(Arrays.asList(url), Arrays.asList(this.proxyServer.split(",").clone()));
+            HutoolhttpOutboundHandler handler = new HutoolhttpOutboundHandler(proxy);
             handler.handle(fullRequest, ctx);
     
         } catch(Exception e) {
diff --git a/02nio/nio02/src/main/java/io/github/kimmking/gateway/router/HttpEndpointRouter.java b/02nio/nio02/src/main/java/io/github/kimmking/gateway/router/HttpEndpointRouter.java
index f921330a..7369fc45 100644
--- a/02nio/nio02/src/main/java/io/github/kimmking/gateway/router/HttpEndpointRouter.java
+++ b/02nio/nio02/src/main/java/io/github/kimmking/gateway/router/HttpEndpointRouter.java
@@ -2,8 +2,7 @@
 
 import java.util.List;
 
+@FunctionalInterface
 public interface HttpEndpointRouter {
-    
-    String route(List<String> endpoints);
-    
+    String route(List<String> endpoints,List<String> proxyServers);
 }

From b639c46c36b331d42c372a7ed34fda08142ed940 Mon Sep 17 00:00:00 2001
From: tangrong <solike_club@sina.com>
Date: Thu, 5 Nov 2020 01:18:25 +0800
Subject: [PATCH 3/3] =?UTF-8?q?=E8=BF=87=E6=BB=A4?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../kimmking/gateway/filter/TokenFilter.java  | 33 +++++++++++++++++++
 .../gateway/inbound/HttpInboundHandler.java   | 20 ++++++-----
 2 files changed, 44 insertions(+), 9 deletions(-)
 create mode 100644 02nio/nio02/src/main/java/io/github/kimmking/gateway/filter/TokenFilter.java

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
new file mode 100644
index 00000000..54af6980
--- /dev/null
+++ 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
index 81b58433..5dad07e4 100644
--- 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);