加入读写超时控制

Author: wangqifox

src/main/java/love/wangqi/GatewayServerDemo.java

@@ -41,6 +41,8 @@ protected List<Route> locateRouteList(Set<Long> ids) {
                     routeList.add(new Route(8L, HttpMethod.GET, "/path", new URL("http://127.0.0.1:9990/path")));
                     routeList.add(new Route(9L, HttpMethod.GET, "/html", new URL("http://10.100.64.71/html/user.json")));
                     routeList.add(new Route(10L, HttpMethod.GET, "/css", new URL("https://ss0.bdstatic.com/5aV1bjqh_Q23odCf/static/mantpl/css/news/init_7637f86c.css")));
+                    routeList.add(new Route(11L, HttpMethod.GET, "/google", new URL("https://www.google.com")));
+                    routeList.add(new Route(12L, HttpMethod.GET, "/local", new URL("http://127.0.0.1:9999")));
                 } catch (MalformedURLException e) {
                 }
                 return routeList;

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

@@ -57,7 +57,7 @@ public RequestHolder build(FullHttpRequest originRequest) throws Exception {
             throw new GatewayNoRouteException();
         }
         URL url  = route.getMapUrl();
-        logger.debug("proxy_pass {}", url.toString());
+        logger.info("proxy_pass {}", url.toString());
 
         // 请求路径
         QueryStringEncoder queryStringEncoder = new QueryStringEncoder(url.getPath());

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

@@ -5,6 +5,7 @@
 import io.netty.handler.codec.http.FullHttpRequest;
 import io.netty.handler.codec.http.FullHttpResponse;
 import io.netty.util.AttributeKey;
+import love.wangqi.codec.RequestHolder;
 
 /**
  * @author: wangqi
@@ -18,4 +19,5 @@ public interface Attributes {
     AttributeKey<Boolean> KEEPALIVE = AttributeKey.newInstance("keepAlive");
     AttributeKey<Channel> SERVER_CHANNEL = AttributeKey.newInstance("serverChannel");
     AttributeKey<SimpleChannelPool> CLIENT_POOL = AttributeKey.newInstance("clientPool");
+    AttributeKey<RequestHolder> REQUEST_HOLDER = AttributeKey.newInstance("requestHolder");
 }

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

@@ -3,6 +3,7 @@
 import io.netty.channel.Channel;
 import io.netty.handler.codec.http.FullHttpRequest;
 import io.netty.handler.codec.http.FullHttpResponse;
+import love.wangqi.codec.RequestHolder;
 
 
 /**
@@ -35,6 +36,14 @@ public static Boolean getKeepAlive(Channel channel) {
         return channel.attr(Attributes.KEEPALIVE).get();
     }
 
+    public static void setRequestHolder(Channel channel, RequestHolder requestHolder) {
+        channel.attr(Attributes.REQUEST_HOLDER).set(requestHolder);
+    }
+
+    public static RequestHolder getRequestHolder(Channel channel) {
+        return channel.attr(Attributes.REQUEST_HOLDER).get();
+    }
+
     public static void setException(Channel channel, Exception exception) {
         channel.attr(Attributes.EXCEPTION).set(exception);
     }

src/main/java/love/wangqi/filter/SendErrorFilter.java

@@ -1,6 +1,7 @@
 package love.wangqi.filter;
 
 import io.netty.channel.Channel;
+import love.wangqi.codec.RequestHolder;
 import love.wangqi.config.GatewayConfig;
 import love.wangqi.context.ContextUtil;
 import org.slf4j.Logger;
@@ -30,7 +31,8 @@ public int filterOrder() {
     public void filter(Channel channel) throws Exception {
         Exception e = ContextUtil.getException(channel);
         if (e != null) {
-            logger.error(e.getMessage(), e);
+            RequestHolder requestHolder = ContextUtil.getRequestHolder(channel);
+            logger.error("Route Id: " + requestHolder.route.getId() + " Route Url: " + requestHolder.route.getMapUrl() + " " + e.getMessage(), e);
             config.getExceptionHandler().handle(channel, e);
         }
     }

src/main/java/love/wangqi/filter/SendForwardFilter.java

@@ -34,6 +34,7 @@ public synchronized void filter(Channel channel) throws Exception {
         FullHttpRequest httpRequest = ContextUtil.getRequest(channel);
         RequestHolder requestHolder = httpRequestBuilder.build(httpRequest);
         httpRequest.release();
+        ContextUtil.setRequestHolder(channel, requestHolder);
 
         BackClientPool.INSTANCE.request(requestHolder, channel);
     }

src/main/java/love/wangqi/handler/back/BackClientPool.java

@@ -12,13 +12,17 @@
 import io.netty.channel.socket.nio.NioSocketChannel;
 import io.netty.handler.codec.http.HttpRequest;
 import io.netty.handler.codec.http.multipart.HttpPostRequestEncoder;
+import io.netty.handler.timeout.ReadTimeoutHandler;
+import io.netty.handler.timeout.WriteTimeoutHandler;
 import io.netty.util.concurrent.Future;
 import io.netty.util.concurrent.FutureListener;
 import love.wangqi.codec.RequestHolder;
 import love.wangqi.context.Attributes;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
+import java.util.concurrent.TimeUnit;
+
 
 /**
  * @author: wangqi
@@ -59,6 +63,13 @@ public synchronized void request(RequestHolder requestHolder, Channel serverChan
                 HttpPostRequestEncoder bodyRequestEncoder = requestHolder.bodyRequestEncoder;
 
                 Channel clientChannel = future.getNow();
+
+                // 添加读写超时控制器
+                clientChannel.pipeline().addFirst("ReadTimeoutHandler",
+                        new ReadTimeoutHandler(requestHolder.route.getTimeoutInMilliseconds(), TimeUnit.MILLISECONDS));
+                clientChannel.pipeline().addFirst("WriteTimeoutHandler",
+                        new WriteTimeoutHandler(500, TimeUnit.MILLISECONDS));
+
                 clientChannel.attr(Attributes.SERVER_CHANNEL).set(serverChannel);
                 clientChannel.attr(Attributes.CLIENT_POOL).set(pool);
 

src/main/java/love/wangqi/handler/back/BackHandler.java

@@ -38,6 +38,10 @@ public void channelRead0(ChannelHandlerContext ctx, FullHttpResponse response) t
         GatewayRunner.getInstance().postRoutAction(serverChannel);
 
         clientChannel.attr(Attributes.CLIENT_POOL).get().release(clientChannel);
+
+        // 移除读写超时控制器
+        ctx.channel().pipeline().remove("ReadTimeoutHandler");
+        ctx.channel().pipeline().remove("WriteTimeoutHandler");
     }
 
     @Override