第三周作业

leopard3699 · leopard3699 · commit 99313df71351 · 2022-03-20T22:06:42.000+08:00
diff --git a/02nio/nio02/src/main/java/io/github/kimmking/gateway/filter/HeaderHttpRequestFilter.java b/02nio/nio02/src/main/java/io/github/kimmking/gateway/filter/HeaderHttpRequestFilter.java
@@ -1,5 +1,6 @@
 package io.github.kimmking.gateway.filter;
 
+import io.netty.buffer.Unpooled;
 import io.netty.channel.ChannelHandlerContext;
 import io.netty.handler.codec.http.FullHttpRequest;
 
@@ -8,5 +9,7 @@ public class HeaderHttpRequestFilter implements HttpRequestFilter {
     @Override
     public void filter(FullHttpRequest fullRequest, ChannelHandlerContext ctx) {
         fullRequest.headers().set("mao", "soul");
+
+        fullRequest.headers().set("token","cafe-babe");
     }
 }
diff --git a/02nio/nio02/src/main/java/io/github/kimmking/gateway/filter/HeaderHttpResponseFilter.java b/02nio/nio02/src/main/java/io/github/kimmking/gateway/filter/HeaderHttpResponseFilter.java
@@ -1,10 +1,16 @@
 package io.github.kimmking.gateway.filter;
 
+import io.netty.buffer.Unpooled;
 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");
+
+        response.headers().set("Content-Type", "application/json");
+
+        response.replace(Unpooled.wrappedBuffer("you are hacked".getBytes()));
+
     }
 }
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
@@ -81,9 +81,7 @@ public void handle(final FullHttpRequest fullRequest, final ChannelHandlerContex
     
     private void fetchGet(final FullHttpRequest inbound, final ChannelHandlerContext ctx, final String url) {
         final HttpGet httpGet = new HttpGet(url);
-        //httpGet.setHeader(HTTP.CONN_DIRECTIVE, HTTP.CONN_CLOSE);
         httpGet.setHeader(HTTP.CONN_DIRECTIVE, HTTP.CONN_KEEP_ALIVE);
-        httpGet.setHeader("mao", inbound.headers().get("mao"));
 
         httpclient.execute(httpGet, new FutureCallback<HttpResponse>() {
             @Override
@@ -113,27 +111,17 @@ public void cancelled() {
     private void handleResponse(final FullHttpRequest fullRequest, final ChannelHandlerContext ctx, final HttpResponse endpointResponse) throws Exception {
         FullHttpResponse response = null;
         try {
-//            String value = "hello,kimmking";
-//            response = new DefaultFullHttpResponse(HTTP_1_1, OK, Unpooled.wrappedBuffer(value.getBytes("UTF-8")));
-//            response.headers().set("Content-Type", "application/json");
-//            response.headers().setInt("Content-Length", response.content().readableBytes());
-    
+
     
             byte[] body = EntityUtils.toByteArray(endpointResponse.getEntity());
-//            System.out.println(new String(body));
-//            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());
-//            } 
         
         } catch (Exception e) {
             e.printStackTrace();

Original file line number	Diff line number	Diff line change
`@@ -1,5 +1,6 @@`
`1`	`1`	`package io.github.kimmking.gateway.filter;`
`2`	`2`
	`3`	`+import io.netty.buffer.Unpooled;`
`3`	`4`	`import io.netty.channel.ChannelHandlerContext;`
`4`	`5`	`import io.netty.handler.codec.http.FullHttpRequest;`
`5`	`6`
`@@ -8,5 +9,7 @@ public class HeaderHttpRequestFilter implements HttpRequestFilter {`
`8`	`9`	`@Override`
`9`	`10`	`public void filter(FullHttpRequest fullRequest, ChannelHandlerContext ctx) {`
`10`	`11`	`fullRequest.headers().set("mao", "soul");`
	`12`	`+`
	`13`	`+ fullRequest.headers().set("token","cafe-babe");`
`11`	`14`	`}`
`12`	`15`	`}`