diff --git a/wu-lazy-cloud-heartbeat-common/src/main/java/org/framework/lazy/cloud/network/heartbeat/common/NettyChannelContext.java b/wu-lazy-cloud-heartbeat-common/src/main/java/org/framework/lazy/cloud/network/heartbeat/common/NettyChannelContext.java new file mode 100644 index 0000000..9cb7853 --- /dev/null +++ b/wu-lazy-cloud-heartbeat-common/src/main/java/org/framework/lazy/cloud/network/heartbeat/common/NettyChannelContext.java @@ -0,0 +1,25 @@ +package org.framework.lazy.cloud.network.heartbeat.common; + + +import io.netty.channel.Channel; +import io.netty.channel.ChannelHandlerContext; +import lombok.Data; + + +@Data +public class NettyChannelContext { + + + private ChannelHandlerContext channelHandlerContext; + + + /** + * 获取当前通道上下文中的通道 + * + * @return 上下问通道 + */ + public Channel getChannel() { + return channelHandlerContext.channel(); + } + +} diff --git a/wu-lazy-cloud-heartbeat-server/src/main/java/org/framework/lazy/cloud/network/heartbeat/server/netty/permeate/tcp/advanced/ServerHandleTcpChannelHeartbeatTypeAdvanced.java b/wu-lazy-cloud-heartbeat-server/src/main/java/org/framework/lazy/cloud/network/heartbeat/server/netty/permeate/tcp/advanced/ServerHandleTcpChannelHeartbeatTypeAdvanced.java index 963bf35..31fc796 100644 --- a/wu-lazy-cloud-heartbeat-server/src/main/java/org/framework/lazy/cloud/network/heartbeat/server/netty/permeate/tcp/advanced/ServerHandleTcpChannelHeartbeatTypeAdvanced.java +++ b/wu-lazy-cloud-heartbeat-server/src/main/java/org/framework/lazy/cloud/network/heartbeat/server/netty/permeate/tcp/advanced/ServerHandleTcpChannelHeartbeatTypeAdvanced.java @@ -1,6 +1,7 @@ package org.framework.lazy.cloud.network.heartbeat.server.netty.permeate.tcp.advanced; +import io.netty.channel.Channel; import io.netty.channel.ChannelHandlerContext; import org.framework.lazy.cloud.network.heartbeat.common.advanced.permeate.tcp.AbstractTcpHandleChannelHeartbeatTypeAdvanced; import org.springframework.beans.factory.config.BeanDefinition; @@ -26,9 +27,10 @@ public class ServerHandleTcpChannelHeartbeatTypeAdvanced extends AbstractTcpHand */ @Override public void doHandler(ChannelHandlerContext channelHandlerContext, NettyProxyMsg msg) { + Channel channel = channelHandlerContext.channel(); NettyProxyMsg hb = new NettyProxyMsg(); hb.setType(TcpMessageType.TCP_TYPE_HEARTBEAT); - channelHandlerContext.writeAndFlush(hb); + channel.writeAndFlush(hb); } diff --git a/wu-lazy-cloud-heartbeat-server/src/main/java/org/framework/lazy/cloud/network/heartbeat/server/netty/permeate/tcp/advanced/ServerHandleTcpClientConnectSuccessTypeAdvanced.java b/wu-lazy-cloud-heartbeat-server/src/main/java/org/framework/lazy/cloud/network/heartbeat/server/netty/permeate/tcp/advanced/ServerHandleTcpClientConnectSuccessTypeAdvanced.java index 6c9cdbe..e3491ad 100644 --- a/wu-lazy-cloud-heartbeat-server/src/main/java/org/framework/lazy/cloud/network/heartbeat/server/netty/permeate/tcp/advanced/ServerHandleTcpClientConnectSuccessTypeAdvanced.java +++ b/wu-lazy-cloud-heartbeat-server/src/main/java/org/framework/lazy/cloud/network/heartbeat/server/netty/permeate/tcp/advanced/ServerHandleTcpClientConnectSuccessTypeAdvanced.java @@ -137,7 +137,7 @@ public class ServerHandleTcpClientConnectSuccessTypeAdvanced extends AbstractHan }else { // 关闭通道 log.warn("无法认证客户端:【{}】",clientId); - channelHandlerContext.close(); + newChannel.close(); } @@ -162,7 +162,7 @@ public class ServerHandleTcpClientConnectSuccessTypeAdvanced extends AbstractHan nettyMsg.setClientTargetIp(permeateTargetIp); nettyMsg.setClientTargetPort(permeateTargetPort); nettyMsg.setVisitorPort(visitorPort); - channelHandlerContext.writeAndFlush(nettyMsg); + newChannel.writeAndFlush(nettyMsg); } }); @@ -186,7 +186,7 @@ public class ServerHandleTcpClientConnectSuccessTypeAdvanced extends AbstractHan nettyMsg.setVisitorPort(visitorPort); nettyMsg.setClientId(clientId); nettyMsg.setData(toClientId.getBytes(StandardCharsets.UTF_8)); - channelHandlerContext.writeAndFlush(nettyMsg); + newChannel.writeAndFlush(nettyMsg); } }); // 下发服务端路由 @@ -206,7 +206,7 @@ public class ServerHandleTcpClientConnectSuccessTypeAdvanced extends AbstractHan nettyMsg.setType(ProxyMessageType.HTTP_DISTRIBUTE_CLIENT_PROXY_SERVER_SERVER_ROUTE_); nettyMsg.setData((JSON.toJSONString(serverProxyRoute) .getBytes(StandardCharsets.UTF_8))); - channelHandlerContext.writeAndFlush(nettyMsg); + newChannel.writeAndFlush(nettyMsg); } }); @@ -229,7 +229,7 @@ public class ServerHandleTcpClientConnectSuccessTypeAdvanced extends AbstractHan nettyMsg.setType(ProxyMessageType.HTTP_DISTRIBUTE_CLIENT_PROXY_SERVER_CLIENT_ROUTE_); nettyMsg.setData((JSON.toJSONString(clientProxyRoute) .getBytes(StandardCharsets.UTF_8))); - channelHandlerContext.writeAndFlush(nettyMsg); + newChannel.writeAndFlush(nettyMsg); } }); diff --git a/wu-lazy-cloud-heartbeat-server/src/main/java/org/framework/lazy/cloud/network/heartbeat/server/netty/permeate/udp/advanced/ServerHandleUdpChannelHeartbeatTypeAdvanced.java b/wu-lazy-cloud-heartbeat-server/src/main/java/org/framework/lazy/cloud/network/heartbeat/server/netty/permeate/udp/advanced/ServerHandleUdpChannelHeartbeatTypeAdvanced.java index 9abc10b..0611195 100644 --- a/wu-lazy-cloud-heartbeat-server/src/main/java/org/framework/lazy/cloud/network/heartbeat/server/netty/permeate/udp/advanced/ServerHandleUdpChannelHeartbeatTypeAdvanced.java +++ b/wu-lazy-cloud-heartbeat-server/src/main/java/org/framework/lazy/cloud/network/heartbeat/server/netty/permeate/udp/advanced/ServerHandleUdpChannelHeartbeatTypeAdvanced.java @@ -1,6 +1,7 @@ package org.framework.lazy.cloud.network.heartbeat.server.netty.permeate.udp.advanced; +import io.netty.channel.Channel; import io.netty.channel.ChannelHandlerContext; import org.framework.lazy.cloud.network.heartbeat.common.constant.TcpMessageType; import org.framework.lazy.cloud.network.heartbeat.common.NettyProxyMsg; @@ -26,9 +27,10 @@ public class ServerHandleUdpChannelHeartbeatTypeAdvanced extends AbstractUdpHand */ @Override public void doHandler(ChannelHandlerContext channelHandlerContext, NettyProxyMsg msg) { + Channel channel = channelHandlerContext.channel(); NettyProxyMsg hb = new NettyProxyMsg(); hb.setType(TcpMessageType.TCP_TYPE_HEARTBEAT); - channelHandlerContext.writeAndFlush(hb); + channel.writeAndFlush(hb); } diff --git a/wu-lazy-cloud-heartbeat-server/src/main/java/org/framework/lazy/cloud/network/heartbeat/server/netty/proxy/http/advanced/ServerHandleHttpReportClientProxyClientTransferChannelInitSuccessfulTypeAdvanced.java b/wu-lazy-cloud-heartbeat-server/src/main/java/org/framework/lazy/cloud/network/heartbeat/server/netty/proxy/http/advanced/ServerHandleHttpReportClientProxyClientTransferChannelInitSuccessfulTypeAdvanced.java index ec3478f..9a35e5a 100644 --- a/wu-lazy-cloud-heartbeat-server/src/main/java/org/framework/lazy/cloud/network/heartbeat/server/netty/proxy/http/advanced/ServerHandleHttpReportClientProxyClientTransferChannelInitSuccessfulTypeAdvanced.java +++ b/wu-lazy-cloud-heartbeat-server/src/main/java/org/framework/lazy/cloud/network/heartbeat/server/netty/proxy/http/advanced/ServerHandleHttpReportClientProxyClientTransferChannelInitSuccessfulTypeAdvanced.java @@ -63,7 +63,7 @@ public class ServerHandleHttpReportClientProxyClientTransferChannelInitSuccessfu clientConnectTagetNettyProxyMsg.setType(ProxyMessageType.HTTP_DISTRIBUTE_CLIENT_PROXY_CLIENT_TRANSFER_REQUEST_); - channelHandlerContext.writeAndFlush(clientConnectTagetNettyProxyMsg); + channel.writeAndFlush(clientConnectTagetNettyProxyMsg); }