port mapping

g4proxy-lib/src/main/java/com/virjar/g4proxy/server/client/ClientManager.java

@@ -2,9 +2,11 @@ package com.virjar.g4proxy.server.client;
 
 import com.google.common.collect.BiMap;
 import com.google.common.collect.HashBiMap;
+import com.google.common.collect.Lists;
 import com.virjar.g4proxy.protocol.Constant;
 import com.virjar.g4proxy.server.AvailablePortResManager;
 
+import java.util.List;
 import java.util.Map;
 import java.util.concurrent.ConcurrentHashMap;
 
@@ -28,6 +30,18 @@ public class ClientManager {
         // this.userMappingBootstrap = userMappingBootstrap;
     }
 
+    public List<String> portMapping() {
+        List<String> ret = Lists.newArrayList();
+        for (NatClientImage natClientImage : natClientImageMap.values()) {
+            if (!natClientImage.getUserMappingServerChannel().isActive()) {
+                continue;
+            }
+            int mappingPort = natClientImage.getMappingPort();
+            ret.add(String.valueOf(mappingPort));
+        }
+        return ret;
+    }
+
     public boolean registerNewClient(String client, Channel natChannel) {
         NatClientImage natClientImage = natClientImageMap.get(client);
         Integer mappingPort = null;

g4proxy-server/src/main/java/com/virjar/g4proxyserver/controller/AvailablePortController.java

@@ -16,8 +16,7 @@ public class AvailablePortController {
 
     @GetMapping("portList")
     public String availablePortMapping() {
-
-        return StringUtils.join(ProxyInstanceHolder.g4ProxyServer.getClientManager().getClientPortBiMap().values(), "\r\n");
+        return StringUtils.join(ProxyInstanceHolder.g4ProxyServer.getClientManager().portMapping(), "\r\n");
     }
 
     @GetMapping("deviceMapping")