替换HttpRequestContext的数据传递方式

Author: wangqifox

src/main/java/love/wangqi/codec/DefaultHttpRequestBuilder.java

@@ -7,11 +7,11 @@
 import io.netty.handler.codec.http.multipart.HttpDataFactory;
 import io.netty.handler.codec.http.multipart.HttpPostRequestEncoder;
 import io.netty.handler.codec.http.multipart.InterfaceHttpData;
-import org.slf4j.Logger;
-import org.slf4j.LoggerFactory;
 import love.wangqi.exception.GatewayNoRouteException;
 import love.wangqi.route.Route;
 import love.wangqi.route.RouteMapper;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
 
 import java.net.URI;
 import java.net.URL;

src/main/java/love/wangqi/context/Attributes.java

@@ -0,0 +1,17 @@
+package love.wangqi.context;
+
+import io.netty.handler.codec.http.FullHttpRequest;
+import io.netty.handler.codec.http.FullHttpResponse;
+import io.netty.util.AttributeKey;
+
+/**
+ * @author: wangqi
+ * @description:
+ * @date: Created in 2018-11-27 11:30
+ */
+public interface Attributes {
+    AttributeKey<FullHttpRequest> REQUEST = AttributeKey.newInstance("httpRequest");
+    AttributeKey<Exception> EXCEPTION = AttributeKey.newInstance("exception");
+    AttributeKey<FullHttpResponse> RESPONSE = AttributeKey.newInstance("response");
+    AttributeKey<Boolean> KEEPALIVE = AttributeKey.newInstance("keepAlive");
+}

src/main/java/love/wangqi/context/ContextUtil.java

@@ -0,0 +1,45 @@
+package love.wangqi.context;
+
+import io.netty.channel.Channel;
+import io.netty.handler.codec.http.FullHttpRequest;
+import io.netty.handler.codec.http.FullHttpResponse;
+
+
+/**
+ * @author: wangqi
+ * @description:
+ * @date: Created in 2018-11-27 15:02
+ */
+public class ContextUtil {
+    public static void setRequest(Channel channel, FullHttpRequest request) {
+        channel.attr(Attributes.REQUEST).set(request);
+    }
+
+    public static FullHttpRequest getRequest(Channel channel) {
+        return channel.attr(Attributes.REQUEST).get();
+    }
+
+    public static void setResponse(Channel channel, FullHttpResponse response) {
+        channel.attr(Attributes.RESPONSE).set(response);
+    }
+
+    public static FullHttpResponse getResponse(Channel channel) {
+        return channel.attr(Attributes.RESPONSE).get();
+    }
+
+    public static void setKeepAlive(Channel channel, Boolean keepAlive) {
+        channel.attr(Attributes.KEEPALIVE).set(keepAlive);
+    }
+
+    public static Boolean getKeepAlive(Channel channel) {
+        return channel.attr(Attributes.KEEPALIVE).get();
+    }
+
+    public static void setException(Channel channel, Exception exception) {
+        channel.attr(Attributes.EXCEPTION).set(exception);
+    }
+
+    public static Exception getException(Channel channel) {
+        return channel.attr(Attributes.EXCEPTION).get();
+    }
+}

src/main/java/love/wangqi/context/HttpRequestContext.java

@@ -2,9 +2,7 @@
 
 import io.netty.channel.Channel;
 import io.netty.channel.ChannelFuture;
-import io.netty.handler.codec.http.FullHttpRequest;
-import love.wangqi.context.HttpRequestContext;
-import love.wangqi.context.RequestConstant;
+import love.wangqi.context.ContextUtil;
 import love.wangqi.listener.ChannelCloseFutureListener;
 
 /**
@@ -13,14 +11,12 @@
  * @date: Created in 2018/11/22 下午4:21
  */
 public class DefaultChannelCloseFutureListener implements ChannelCloseFutureListener {
-    private HttpRequestContext httpRequestContext = HttpRequestContext.getInstance();
 
     @Override
     public void operationComplete(Channel channel, ChannelFuture future) {
-        Boolean keepAlive = httpRequestContext.get(channel, RequestConstant.KEEPALIVE);
+        Boolean keepAlive = ContextUtil.getKeepAlive(channel);
 
-        ((FullHttpRequest)httpRequestContext.getHttpRequest(channel)).release();
-        httpRequestContext.remove(channel);
+        ContextUtil.getRequest(channel).release();
 
         if (!keepAlive) {
             channel.close();

src/main/java/love/wangqi/context/RequestConstant.java

@@ -7,8 +7,7 @@
 import io.netty.handler.codec.http.HttpHeaderNames;
 import io.netty.handler.codec.http.HttpUtil;
 import love.wangqi.config.GatewayConfig;
-import love.wangqi.context.HttpRequestContext;
-import love.wangqi.context.RequestConstant;
+import love.wangqi.context.ContextUtil;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
@@ -21,11 +20,10 @@ public class ResponseHandler {
     private Logger logger = LoggerFactory.getLogger(ResponseHandler.class);
 
     private GatewayConfig config = GatewayConfig.getInstance();
-    private HttpRequestContext httpRequestContext = HttpRequestContext.getInstance();
 
     public void send(Channel channel, FullHttpResponse response) {
         response.headers().set(HttpHeaderNames.CONTENT_LENGTH, response.content().readableBytes());
-        Boolean keepAlive = httpRequestContext.get(channel, RequestConstant.KEEPALIVE);
+        Boolean keepAlive = ContextUtil.getKeepAlive(channel);
         HttpUtil.setKeepAlive(response, keepAlive == null ? false : keepAlive);
         channel.writeAndFlush(response).addListener(new ChannelFutureListener() {
             @Override

src/main/java/love/wangqi/core/DefaultChannelCloseFutureListener.java

@@ -1,6 +1,6 @@
 package love.wangqi.exception.handler;
 
-import io.netty.channel.ChannelHandlerContext;
+import io.netty.channel.Channel;
 
 /**
  * @author: wangqi
@@ -10,8 +10,8 @@
 public abstract class AbstractExceptionHandler implements ExceptionHandler {
 
     @Override
-    public final void handle(ChannelHandlerContext ctx, Exception exception) {
+    public final void handle(Channel channel, Exception exception) {
         ExceptionResponse exceptionResponse = getExceptionResponse(exception);
-        send(ctx, exceptionResponse);
+        send(channel, exceptionResponse);
     }
 }

src/main/java/love/wangqi/core/ResponseHandler.java

@@ -1,6 +1,6 @@
 package love.wangqi.exception.handler;
 
-import io.netty.channel.ChannelHandlerContext;
+import io.netty.channel.Channel;
 import io.netty.handler.codec.http.*;
 import love.wangqi.config.GatewayConfig;
 import love.wangqi.exception.GatewayException;
@@ -41,13 +41,13 @@ public ExceptionResponse getExceptionResponse(Exception exception) {
     }
 
     @Override
-    public void send(ChannelHandlerContext ctx, ExceptionResponse exceptionResponse) {
+    public void send(Channel channel, ExceptionResponse exceptionResponse) {
         String content = exceptionResponse.getContent();
         FullHttpResponse response = new DefaultFullHttpResponse(HttpVersion.HTTP_1_1, exceptionResponse.getStatus());
         response.headers().set(HttpHeaderNames.CONTENT_LENGTH, response.content().readableBytes());
         if (content != null) {
             response.headers().set("X-Ca-Error-Message", content);
         }
-        config.getResponseHandler().send(ctx.channel(), response);
+        config.getResponseHandler().send(channel, response);
     }
 }

src/main/java/love/wangqi/exception/handler/AbstractExceptionHandler.java

@@ -1,6 +1,6 @@
 package love.wangqi.exception.handler;
 
-import io.netty.channel.ChannelHandlerContext;
+import io.netty.channel.Channel;
 
 /**
  * @author: wangqi
@@ -17,15 +17,15 @@ public interface ExceptionHandler {
 
     /**
      * 发送异常
-     * @param ctx
+     * @param channel
      * @param exceptionResponse
      */
-    void send(ChannelHandlerContext ctx, ExceptionResponse exceptionResponse);
+    void send(Channel channel, ExceptionResponse exceptionResponse);
 
     /**
      * 处理异常
-     * @param ctx
+     * @param channel
      * @param exception
      */
-    void handle(ChannelHandlerContext ctx, Exception exception);
+    void handle(Channel channel, Exception exception);
 }