From a9608a449d9c4d53f076a99329853cb88f27a99b Mon Sep 17 00:00:00 2001
From: aoshiguchen <1052045476@qq.com>
Date: Tue, 17 Oct 2023 15:06:43 +0800
Subject: [PATCH] =?UTF-8?q?=E8=A7=A3=E5=86=B3=E4=B8=8A=E4=BC=A0=E6=96=87?=
 =?UTF-8?q?=E4=BB=B6=E5=8D=A1=E4=BD=8F=E7=9A=84=E9=97=AE=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../client/handler/ProxyMessageTransferHandler.java       | 2 +-
 .../server/proxy/core/ProxyTunnelChannelHandler.java      | 8 ++++----
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/neutrino-proxy-client/src/main/java/org/dromara/neutrinoproxy/client/handler/ProxyMessageTransferHandler.java b/neutrino-proxy-client/src/main/java/org/dromara/neutrinoproxy/client/handler/ProxyMessageTransferHandler.java
index 6e8f2122..60d51a95 100644
--- a/neutrino-proxy-client/src/main/java/org/dromara/neutrinoproxy/client/handler/ProxyMessageTransferHandler.java
+++ b/neutrino-proxy-client/src/main/java/org/dromara/neutrinoproxy/client/handler/ProxyMessageTransferHandler.java
@@ -25,7 +25,7 @@ public class ProxyMessageTransferHandler implements ProxyMessageHandler {
 		if (realServerChannel != null) {
 
 			// 自己可写，则设置来源可读。自己不可写，则设置来源不可读
-			realServerChannel.config().setAutoRead(ctx.channel().isWritable());
+			ctx.channel().config().setAutoRead(realServerChannel.isWritable());
 
 			ByteBuf buf = ctx.alloc().buffer(proxyMessage.getData().length);
 			buf.writeBytes(proxyMessage.getData());
diff --git a/neutrino-proxy-server/src/main/java/org/dromara/neutrinoproxy/server/proxy/core/ProxyTunnelChannelHandler.java b/neutrino-proxy-server/src/main/java/org/dromara/neutrinoproxy/server/proxy/core/ProxyTunnelChannelHandler.java
index 18412de4..12b75031 100644
--- a/neutrino-proxy-server/src/main/java/org/dromara/neutrinoproxy/server/proxy/core/ProxyTunnelChannelHandler.java
+++ b/neutrino-proxy-server/src/main/java/org/dromara/neutrinoproxy/server/proxy/core/ProxyTunnelChannelHandler.java
@@ -70,10 +70,10 @@ public class ProxyTunnelChannelHandler extends SimpleChannelInboundHandler<Proxy
 
     @Override
     public void channelWritabilityChanged(ChannelHandlerContext ctx) throws Exception {
-//        Channel userChannel = ctx.channel().attr(Constants.NEXT_CHANNEL).get();
-//        if (userChannel != null) {
-//            userChannel.config().setOption(ChannelOption.AUTO_READ, ctx.channel().isWritable());
-//        }
+        Channel visitorChannel = ctx.channel().attr(Constants.NEXT_CHANNEL).get();
+        if (visitorChannel != null) {
+            visitorChannel.config().setOption(ChannelOption.AUTO_READ, ctx.channel().isWritable());
+        }
 
         super.channelWritabilityChanged(ctx);
     }