From 2d0b9132e983566f90a9bf7d014438c4ac719ad2 Mon Sep 17 00:00:00 2001
From: aoshiguchen <1052045476@qq.com>
Date: Wed, 31 May 2023 22:42:16 +0800
Subject: [PATCH] =?UTF-8?q?=E5=8D=8F=E8=AE=AE=E6=9E=9A=E4=B8=BE=E8=B0=83?=
 =?UTF-8?q?=E6=95=B4=EF=BC=8C=E7=BB=B4=E6=8A=A4=E5=8D=8F=E8=AE=AE=E5=85=B3?=
 =?UTF-8?q?=E8=81=94=E7=9A=84=E4=B8=8B=E5=B1=82=E5=9F=BA=E7=A1=80=E5=8D=8F?=
 =?UTF-8?q?=E8=AE=AE?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../neutrinoproxy/server/constant/NetworkProtocolEnum.java | 7 ++++---
 .../server/service/VisitorChannelService.java              | 1 +
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/neutrino-proxy-server/src/main/java/org/dromara/neutrinoproxy/server/constant/NetworkProtocolEnum.java b/neutrino-proxy-server/src/main/java/org/dromara/neutrinoproxy/server/constant/NetworkProtocolEnum.java
index be5d23c9..7a053552 100644
--- a/neutrino-proxy-server/src/main/java/org/dromara/neutrinoproxy/server/constant/NetworkProtocolEnum.java
+++ b/neutrino-proxy-server/src/main/java/org/dromara/neutrinoproxy/server/constant/NetworkProtocolEnum.java
@@ -16,11 +16,12 @@ import java.util.stream.Stream;
 @Getter
 @AllArgsConstructor
 public enum NetworkProtocolEnum {
-    TCP("TCP"),
-    UDP("UDP"),
-    HTTP("HTTP"),
+    TCP("TCP", "TCP"),
+    UDP("UDP", "UDP"),
+    HTTP("HTTP", "TCP"),
     ;
     private String desc;
+    private String baseProtocol;
     private static final Map<String, NetworkProtocolEnum> map = Stream.of(NetworkProtocolEnum.values()).collect(Collectors.toMap(NetworkProtocolEnum::getDesc, Function.identity()));
 
     public static NetworkProtocolEnum of(String desc) {
diff --git a/neutrino-proxy-server/src/main/java/org/dromara/neutrinoproxy/server/service/VisitorChannelService.java b/neutrino-proxy-server/src/main/java/org/dromara/neutrinoproxy/server/service/VisitorChannelService.java
index 1db3808e..bd409d69 100644
--- a/neutrino-proxy-server/src/main/java/org/dromara/neutrinoproxy/server/service/VisitorChannelService.java
+++ b/neutrino-proxy-server/src/main/java/org/dromara/neutrinoproxy/server/service/VisitorChannelService.java
@@ -218,6 +218,7 @@ public class VisitorChannelService {
         }
 
         for (PortMappingDO portMapping : portMappingList) {
+            // TODO 此处切入，TCP/UDP代理
             try {
                 proxyMutualService.bindServerPort(cmdChannelAttachInfo, portMapping.getServerPort());
                 tcpServerBootstrap.bind(portMapping.getServerPort()).get();