add response filter

Author: kimmking

02nio/nio02/src/main/java/io/github/kimmking/gateway/NettyServerApplication.java

@@ -14,15 +14,18 @@ public static void main(String[] args) {
 
         String proxyPort = System.getProperty("proxyPort","8888");
 
+        // 这是之前的单个后端url的例子
 //        String proxyServer = System.getProperty("proxyServer","http://localhost:8088");
 //          //  http://localhost:8888/api/hello  ==> gateway API
 //          //  http://localhost:8088/api/hello  ==> backend service
         // java -Xmx512m gateway-server-0.0.1-SNAPSHOT.jar  #作为后端服务
 
-    
+
+        // 这是多个后端url走随机路由的例子
+        String proxyServers = System.getProperty("proxyServers","http://localhost:8801,http://localhost:8802");
         int port = Integer.parseInt(proxyPort);
         System.out.println(GATEWAY_NAME + " " + GATEWAY_VERSION +" starting...");
-        HttpInboundServer server = new HttpInboundServer(port, Arrays.asList("http://localhost:8801","http://localhost:8802"));
+        HttpInboundServer server = new HttpInboundServer(port, Arrays.asList(proxyServers.split(",")));
         System.out.println(GATEWAY_NAME + " " + GATEWAY_VERSION +" started at http://localhost:" + port + " for server:" + server.toString());
         try {
             server.run();

02nio/nio02/src/main/java/io/github/kimmking/gateway/filter/HeaderHttpResponseFilter.java

@@ -0,0 +1,10 @@
+package io.github.kimmking.gateway.filter;
+
+import io.netty.handler.codec.http.FullHttpResponse;
+
+public class HeaderHttpResponseFilter implements HttpResponseFilter {
+    @Override
+    public void filter(FullHttpResponse response) {
+        response.headers().set("kk", "java-1-nio");
+    }
+}

02nio/nio02/src/main/java/io/github/kimmking/gateway/filter/HttpResponseFilter.java

@@ -0,0 +1,9 @@
+package io.github.kimmking.gateway.filter;
+
+import io.netty.handler.codec.http.FullHttpResponse;
+
+public interface HttpResponseFilter {
+
+    void filter(FullHttpResponse response);
+
+}

02nio/nio02/src/main/java/io/github/kimmking/gateway/outbound/httpclient4/HttpOutboundHandler.java

@@ -1,7 +1,9 @@
 package io.github.kimmking.gateway.outbound.httpclient4;
 
 
+import io.github.kimmking.gateway.filter.HeaderHttpResponseFilter;
 import io.github.kimmking.gateway.filter.HttpRequestFilter;
+import io.github.kimmking.gateway.filter.HttpResponseFilter;
 import io.github.kimmking.gateway.router.HttpEndpointRouter;
 import io.github.kimmking.gateway.router.RandomHttpEndpointRouter;
 import io.netty.buffer.Unpooled;
@@ -36,6 +38,7 @@ public class HttpOutboundHandler {
     private ExecutorService proxyService;
     private List<String> backendUrls;
 
+    HttpResponseFilter filter = new HeaderHttpResponseFilter();
     HttpEndpointRouter router = new RandomHttpEndpointRouter();
 
     public HttpOutboundHandler(List<String> backends) {
@@ -121,9 +124,12 @@ private void handleResponse(final FullHttpRequest fullRequest, final ChannelHand
 //            System.out.println(body.length);
 
             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()));
-    
+
+            filter.filter(response);
+
 //            for (Header e : endpointResponse.getAllHeaders()) {
 //                //response.headers().set(e.getName(),e.getValue());
 //                System.out.println(e.getName() + " => " + e.getValue());