From 785f4d6183b61c5e451acf90de20f6c1a9dcb262 Mon Sep 17 00:00:00 2001 From: aoshiguchen <1052045476@qq.com> Date: Tue, 30 Aug 2022 16:10:43 +0800 Subject: [PATCH] =?UTF-8?q?=E6=9C=8D=E5=8A=A1=E7=AB=AF=E8=BF=9E=E6=8E=A5?= =?UTF-8?q?=E9=AA=8C=E8=AF=81=E9=80=BB=E8=BE=91=E8=B0=83=E6=95=B4=EF=BC=8C?= =?UTF-8?q?=E5=9C=A8=E6=B2=A1=E6=9C=89=E7=AB=AF=E5=8F=A3=E6=98=A0=E5=B0=84?= =?UTF-8?q?=E6=97=B6=E4=BB=8D=E7=84=B6=E4=BF=9D=E6=8C=81=E4=B8=8E=E5=AE=A2?= =?UTF-8?q?=E6=88=B7=E7=AB=AF=E7=9A=84=E8=BF=9E=E6=8E=A5?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../handler/ProxyMessageAuthHandler.java | 20 ++++++------------- 1 file changed, 6 insertions(+), 14 deletions(-) diff --git a/neutrino-proxy-server/src/main/java/fun/asgc/neutrino/proxy/server/proxy/handler/ProxyMessageAuthHandler.java b/neutrino-proxy-server/src/main/java/fun/asgc/neutrino/proxy/server/proxy/handler/ProxyMessageAuthHandler.java index 94b1a50b..50e609ca 100644 --- a/neutrino-proxy-server/src/main/java/fun/asgc/neutrino/proxy/server/proxy/handler/ProxyMessageAuthHandler.java +++ b/neutrino-proxy-server/src/main/java/fun/asgc/neutrino/proxy/server/proxy/handler/ProxyMessageAuthHandler.java @@ -103,28 +103,20 @@ public class ProxyMessageAuthHandler implements ProxyMessageHandler { return; } List portMappingList = portMappingService.findEnableListByLicenseId(licenseDO.getId()); + // 没有端口映射仍然保持连接 if (CollectionUtil.isEmpty(portMappingList)) { - ctx.channel().writeAndFlush(ProxyMessage.buildErrMessage(ExceptionEnum.AUTH_FAILED, "当前license没有可用的端口映射!")); + return; + } + Channel cmdChannel = ProxyUtil.getCmdChannelByLicenseKey(licenseKey); + if (null != cmdChannel) { + ctx.channel().writeAndFlush(ProxyMessage.buildErrMessage(ExceptionEnum.AUTH_FAILED, "当前license已被另一节点使用!")); ctx.channel().close(); return; } ProxyUtil.initProxyInfo(licenseKey, ProxyMapping.buildList(portMappingList)); Set ports = ProxyUtil.getServerPortsByLicenseKey(licenseKey); - if (ports == null) { - ctx.channel().close(); - return; - } -// Channel channel = ProxyChannelManager.getCmdChannel(licenseKey); - Channel channel = ProxyUtil.getCmdChannelByLicenseKey(licenseKey); - - if (channel != null) { - ctx.channel().close(); - return; - } - -// ProxyChannelManager.addCmdChannel(ports, licenseKey, ctx.channel()); ProxyUtil.addCmdChannel(licenseKey, ctx.channel(), ports); startUserPortServer(ports);