tangrong · tangrong · commit 1e78fd4f7898 · 2020-11-05T01:01:41.000+08:00
&#36335;&#30001;
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
@@ -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
@@ -2,13 +2,17 @@
 
 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;
 import io.netty.util.ReferenceCountUtil;
 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&#27969;&#37327;&#25509;&#21475;&#35831;&#27714;&#24320;&#22987;&#65292;&#26102;&#38388;&#20026;{}", startTime);
+
             FullHttpRequest fullRequest = (FullHttpRequest) msg;
-//            String uri = fullRequest.uri();
-//            //logger.info("&#25509;&#25910;&#21040;&#30340;&#35831;&#27714;url&#20026;{}", 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
@@ -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);
 }

Original file line number	Diff line number	Diff line change
`@@ -2,8 +2,7 @@`
`2`	`2`
`3`	`3`	`import java.util.List;`
`4`	`4`
	`5`	`+@FunctionalInterface`
`5`	`6`	`public interface HttpEndpointRouter {`
`6`		`-`
`7`		`- String route(List<String> endpoints);`
`8`		`-`
	`7`	`+ String route(List<String> endpoints,List<String> proxyServers);`
`9`	`8`	`}`