From 48ecc1fee2f7a325ea7e324bc4000761310ec85e Mon Sep 17 00:00:00 2001
From: lengmuyun <971434648@qq.com>
Date: Wed, 4 Nov 2020 21:52:04 +0800
Subject: [PATCH 1/3] =?UTF-8?q?okhttp3=E5=8C=85=E5=AF=BC=E5=85=A5?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 02nio/nio02/pom.xml | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/02nio/nio02/pom.xml b/02nio/nio02/pom.xml
index 6cbbeffd..eb37c8b5 100644
--- a/02nio/nio02/pom.xml
+++ b/02nio/nio02/pom.xml
@@ -52,6 +52,17 @@
             <artifactId>httpasyncclient</artifactId>
             <version>4.1.4</version>
         </dependency>
+
+        <dependency>
+            <groupId>com.squareup.okhttp3</groupId>
+            <artifactId>okhttp</artifactId>
+            <version>4.9.0</version>
+        </dependency>
+        <dependency>
+            <groupId>org.jetbrains.kotlin</groupId>
+            <artifactId>kotlin-stdlib</artifactId>
+            <version>1.3.70</version>
+        </dependency>
     
         <!--
         <dependency>

From d79c035e57712fb55340118e35005aed69cdf141 Mon Sep 17 00:00:00 2001
From: lengmuyun <971434648@qq.com>
Date: Wed, 4 Nov 2020 21:54:50 +0800
Subject: [PATCH 2/3] =?UTF-8?q?=E7=AC=AC=E4=B8=80=E9=A2=98=EF=BC=8C?=
 =?UTF-8?q?=E4=BD=BF=E7=94=A8okhttp3=E8=B0=83=E7=94=A8=E5=90=8E=E7=AB=AF?=
 =?UTF-8?q?=E6=9C=8D=E5=8A=A1?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../gateway/inbound/HttpInboundHandler.java   |  8 ++-
 .../okhttp/OkhttpOutboundHandler.java         | 69 +++++++++++++++++++
 2 files changed, 74 insertions(+), 3 deletions(-)

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..8397b821 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,6 @@
 package io.github.kimmking.gateway.inbound;
 
-import io.github.kimmking.gateway.outbound.httpclient4.HttpOutboundHandler;
+import io.github.kimmking.gateway.outbound.okhttp.OkhttpOutboundHandler;
 import io.netty.channel.ChannelHandlerContext;
 import io.netty.channel.ChannelInboundHandlerAdapter;
 import io.netty.handler.codec.http.FullHttpRequest;
@@ -12,11 +12,13 @@ public class HttpInboundHandler extends ChannelInboundHandlerAdapter {
 
     private static Logger logger = LoggerFactory.getLogger(HttpInboundHandler.class);
     private final String proxyServer;
-    private HttpOutboundHandler handler;
+//    private HttpOutboundHandler handler;
+    private OkhttpOutboundHandler handler;
     
     public HttpInboundHandler(String proxyServer) {
         this.proxyServer = proxyServer;
-        handler = new HttpOutboundHandler(this.proxyServer);
+//        handler = new HttpOutboundHandler(this.proxyServer);
+        handler = new OkhttpOutboundHandler(this.proxyServer);
     }
     
     @Override
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
index 5f194588..3562ee7d 100644
--- 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
@@ -1,4 +1,73 @@
 package io.github.kimmking.gateway.outbound.okhttp;
 
+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 okhttp3.OkHttpClient;
+import okhttp3.Request;
+import okhttp3.Response;
+import okhttp3.ResponseBody;
+import org.jetbrains.annotations.NotNull;
+
+import java.io.IOException;
+
+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 OkhttpOutboundHandler {
+
+    private String backendUrl;
+    private OkHttpClient client;
+
+    public OkhttpOutboundHandler(String backendUrl) {
+        this.backendUrl = backendUrl.endsWith("/") ? backendUrl.substring(0, backendUrl.length()-1) : backendUrl;
+        this.client = new OkHttpClient();
+    }
+
+    public void handle(FullHttpRequest fullRequest, ChannelHandlerContext ctx) {
+        Request request = buildRequest(fullRequest);
+
+        FullHttpResponse httpResponse = null;
+        try (Response response = client.newCall(request).execute()) {
+            ResponseBody body = response.body();
+
+            httpResponse = new DefaultFullHttpResponse(HTTP_1_1, OK, Unpooled.wrappedBuffer(body.bytes()));
+            httpResponse.headers().set("Content-Type", "application/json");
+            httpResponse.headers().setInt("Content-Length", (int) body.contentLength());
+        } catch (IOException e) {
+            e.printStackTrace();
+            httpResponse = new DefaultFullHttpResponse(HTTP_1_1, NO_CONTENT);
+            ctx.close();
+        } finally {
+            if (!HttpUtil.isKeepAlive(fullRequest)) {
+                ctx.write(httpResponse).addListener(ChannelFutureListener.CLOSE);
+            } else {
+                ctx.write(httpResponse);
+            }
+            ctx.flush();
+        }
+    }
+
+    @NotNull
+    private Request buildRequest(FullHttpRequest fullRequest) {
+        String url = this.backendUrl + fullRequest.uri();
+        Request.Builder builder = new Request.Builder().url(url);
+        enhanceRequest(builder, fullRequest);
+        return builder.build();
+    }
+
+    /**
+     * 请求头处理
+     * @param builder
+     * @param fullHttpRequest
+     */
+    protected void enhanceRequest(Request.Builder builder, FullHttpRequest fullHttpRequest) {
+
+    }
+
 }

From b1e4865c3f8c8a4418d32f647c321c67fc4f66c3 Mon Sep 17 00:00:00 2001
From: lengmuyun <971434648@qq.com>
Date: Wed, 4 Nov 2020 21:58:11 +0800
Subject: [PATCH 3/3] =?UTF-8?q?=E7=AC=AC=E4=B8=89=E9=A2=98=EF=BC=8C?=
 =?UTF-8?q?=E4=BD=BF=E7=94=A8okhttp3=E8=B0=83=E7=94=A8=E5=90=8E=E7=AB=AF?=
 =?UTF-8?q?=E6=9C=8D=E5=8A=A1=E5=89=8D=EF=BC=8C=E7=94=A8filter=E5=81=9A?=
 =?UTF-8?q?=E5=A2=9E=E5=BC=BA=EF=BC=8C=E8=B0=83=E7=94=A8=E6=97=B6=E6=B7=BB?=
 =?UTF-8?q?=E5=8A=A0=E5=A2=9E=E5=BC=BA=E5=90=8E=E7=9A=84=E8=AF=B7=E6=B1=82?=
 =?UTF-8?q?=E5=A4=B4?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../filter/HttpRequestHeaderFilter.java       | 13 ++++++
 .../inbound/HttpInboundFilterHandler.java     | 42 +++++++++++++++++++
 .../inbound/HttpInboundInitializer.java       |  3 +-
 .../okhttp/OkhttpOutboundFilterHandler.java   | 27 ++++++++++++
 4 files changed, 84 insertions(+), 1 deletion(-)
 create mode 100644 02nio/nio02/src/main/java/io/github/kimmking/gateway/filter/HttpRequestHeaderFilter.java
 create mode 100644 02nio/nio02/src/main/java/io/github/kimmking/gateway/inbound/HttpInboundFilterHandler.java
 create mode 100644 02nio/nio02/src/main/java/io/github/kimmking/gateway/outbound/okhttp/OkhttpOutboundFilterHandler.java

diff --git a/02nio/nio02/src/main/java/io/github/kimmking/gateway/filter/HttpRequestHeaderFilter.java b/02nio/nio02/src/main/java/io/github/kimmking/gateway/filter/HttpRequestHeaderFilter.java
new file mode 100644
index 00000000..7ca0e395
--- /dev/null
+++ b/02nio/nio02/src/main/java/io/github/kimmking/gateway/filter/HttpRequestHeaderFilter.java
@@ -0,0 +1,13 @@
+package io.github.kimmking.gateway.filter;
+
+import io.netty.channel.ChannelHandlerContext;
+import io.netty.handler.codec.http.FullHttpRequest;
+
+public class HttpRequestHeaderFilter implements HttpRequestFilter {
+
+    @Override
+    public void filter(FullHttpRequest fullRequest, ChannelHandlerContext ctx) {
+        fullRequest.headers().set("nio", "fangkuangzhang");
+    }
+
+}
diff --git a/02nio/nio02/src/main/java/io/github/kimmking/gateway/inbound/HttpInboundFilterHandler.java b/02nio/nio02/src/main/java/io/github/kimmking/gateway/inbound/HttpInboundFilterHandler.java
new file mode 100644
index 00000000..a74ae11a
--- /dev/null
+++ b/02nio/nio02/src/main/java/io/github/kimmking/gateway/inbound/HttpInboundFilterHandler.java
@@ -0,0 +1,42 @@
+package io.github.kimmking.gateway.inbound;
+
+import io.github.kimmking.gateway.filter.HttpRequestFilter;
+import io.github.kimmking.gateway.filter.HttpRequestHeaderFilter;
+import io.github.kimmking.gateway.outbound.okhttp.OkhttpOutboundFilterHandler;
+import io.github.kimmking.gateway.outbound.okhttp.OkhttpOutboundHandler;
+import io.netty.channel.ChannelHandlerContext;
+import io.netty.channel.ChannelInboundHandlerAdapter;
+import io.netty.handler.codec.http.FullHttpRequest;
+import io.netty.util.ReferenceCountUtil;
+
+public class HttpInboundFilterHandler extends ChannelInboundHandlerAdapter {
+
+    private final String proxyServer;
+    private OkhttpOutboundHandler handler;
+    private HttpRequestFilter filter;
+
+    public HttpInboundFilterHandler(String proxyServer) {
+        this.proxyServer = proxyServer;
+        handler = new OkhttpOutboundFilterHandler(this.proxyServer);
+        filter = new HttpRequestHeaderFilter();
+    }
+
+    @Override
+    public void channelReadComplete(ChannelHandlerContext ctx) {
+        ctx.flush();
+    }
+
+    @Override
+    public void channelRead(ChannelHandlerContext ctx, Object msg) {
+        try {
+            FullHttpRequest fullRequest = (FullHttpRequest) msg;
+            filter.filter(fullRequest, ctx);
+            handler.handle(fullRequest, ctx);
+        } catch(Exception e) {
+            e.printStackTrace();
+        } finally {
+            ReferenceCountUtil.release(msg);
+        }
+    }
+
+}
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
index d902b4c8..89d7c68e 100644
--- 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
@@ -23,6 +23,7 @@ public void initChannel(SocketChannel ch) {
 		p.addLast(new HttpServerCodec());
 		//p.addLast(new HttpServerExpectContinueHandler());
 		p.addLast(new HttpObjectAggregator(1024 * 1024));
-		p.addLast(new HttpInboundHandler(this.proxyServer));
+//		p.addLast(new HttpInboundHandler(this.proxyServer));
+		p.addLast(new HttpInboundFilterHandler(this.proxyServer));
 	}
 }
diff --git a/02nio/nio02/src/main/java/io/github/kimmking/gateway/outbound/okhttp/OkhttpOutboundFilterHandler.java b/02nio/nio02/src/main/java/io/github/kimmking/gateway/outbound/okhttp/OkhttpOutboundFilterHandler.java
new file mode 100644
index 00000000..42a1daf7
--- /dev/null
+++ b/02nio/nio02/src/main/java/io/github/kimmking/gateway/outbound/okhttp/OkhttpOutboundFilterHandler.java
@@ -0,0 +1,27 @@
+package io.github.kimmking.gateway.outbound.okhttp;
+
+import io.netty.handler.codec.http.FullHttpRequest;
+import io.netty.handler.codec.http.HttpHeaders;
+import okhttp3.Request;
+
+import java.util.Iterator;
+import java.util.Map;
+
+public class OkhttpOutboundFilterHandler extends OkhttpOutboundHandler {
+
+    public OkhttpOutboundFilterHandler(String backendUrl) {
+        super(backendUrl);
+    }
+
+    @Override
+    protected void enhanceRequest(Request.Builder builder, FullHttpRequest fullHttpRequest) {
+        HttpHeaders headers = fullHttpRequest.headers();
+        Iterator<Map.Entry<String, String>> iterator = headers.iteratorAsString();
+        while (iterator.hasNext()) {
+            Map.Entry<String, String> next = iterator.next();
+            builder.header(next.getKey(), next.getValue());
+        }
+    }
+
+}
+