perf[proxy]: don't send any packet when the tunnel buffer is full

Author: jaysunxiao

net/src/main/java/com/zfoo/net/core/proxy/TunnelProtocolServer2Client.java

@@ -67,16 +67,12 @@ public static TunnelPacketInfo read(ByteBuf in) {
 
 
     public void write(ByteBuf out) {
-        try {
-            out.ensureWritable(22);
-            out.writerIndex(PacketService.PACKET_HEAD_LENGTH);
-            ByteBufUtils.writeLong(out, sid);
-            ByteBufUtils.writeLong(out, uid);
-            out.writeBytes(retainedByteBuf);
-            NetContext.getPacketService().writeHeaderBefore(out);
-        } finally {
-            ReferenceCountUtil.release(retainedByteBuf);
-        }
+        out.ensureWritable(22);
+        out.writerIndex(PacketService.PACKET_HEAD_LENGTH);
+        ByteBufUtils.writeLong(out, sid);
+        ByteBufUtils.writeLong(out, uid);
+        out.writeBytes(retainedByteBuf);
+        NetContext.getPacketService().writeHeaderBefore(out);
     }
     // -----------------------------------------------------------------------------------------------------------------
 

net/src/main/java/com/zfoo/net/core/proxy/handler/ProxyCodecHandler.java

@@ -26,6 +26,8 @@
 import io.netty.channel.ChannelHandlerContext;
 import io.netty.handler.codec.ByteToMessageCodec;
 import io.netty.util.ReferenceCountUtil;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
 
 import java.util.List;
 
@@ -34,6 +36,8 @@
  */
 public class ProxyCodecHandler extends ByteToMessageCodec<TunnelProtocolServer2Client> {
 
+    private static final Logger logger = LoggerFactory.getLogger(ProxyCodecHandler.class);
+
 
     @Override
     protected void decode(ChannelHandlerContext ctx, ByteBuf in, List<Object> out) {
@@ -57,11 +61,20 @@ protected void decode(ChannelHandlerContext ctx, ByteBuf in, List<Object> out) {
 
         if (CollectionUtils.isEmpty(TunnelServer.tunnels)) {
             in.readSlice(length);
+            logger.warn("Tunnel server has no tunnels");
             return;
         }
+
         var tunnel = RandomUtils.randomEle(TunnelServer.tunnels);
         if (!SessionUtils.isActive(tunnel)) {
             in.readSlice(length);
+            logger.warn("Tunnel server has no active tunnels");
+            return;
+        }
+
+        if (!tunnel.isWritable()) {
+            in.readSlice(length);
+            logger.warn("Tunnel server has no writable tunnels");
             return;
         }
 

net/src/main/java/com/zfoo/net/core/proxy/handler/TunnelServerCodecHandler.java

@@ -60,7 +60,11 @@ protected void decode(ChannelHandlerContext ctx, ByteBuf in, List<Object> out) {
 
     @Override
     protected void encode(ChannelHandlerContext ctx, TunnelProtocolServer2Client tunnelProtocol, ByteBuf out) {
-        tunnelProtocol.write(out);
+        try {
+            tunnelProtocol.write(out);
+        } finally {
+            ReferenceCountUtil.release(tunnelProtocol.getRetainedByteBuf());
+        }
     }
 
 }