dht-server add port

zpqsunny · Jan 28, 2021 · d162456 · d162456
1 parent 67096cf
commit d162456
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/dht-server/src/main/java/me/zpq/server/ServerApplication.java b/dht-server/src/main/java/me/zpq/server/ServerApplication.java
@@ -137,6 +137,7 @@ private static void readConfig() throws IOException {
     private static void readEnv() {
         // docker
         String host = System.getenv("HOST");
+        String port = System.getenv("PORT");
         String minNodes = System.getenv("MIN_NODES");
         String maxNodes = System.getenv("MAX_NODES");
         String redisHost = System.getenv("REDIS_HOST");
@@ -146,6 +147,10 @@ private static void readEnv() {
             log.info("=> env HOST: {}", host);
             HOST = host;
         }
+        if (port != null && !port.isEmpty()) {
+            log.info("=> env PORT: {}", port);
+            PORT = Integer.parseInt(port);
+        }
         if (minNodes != null && !minNodes.isEmpty()) {
             log.info("=> env MIN_NODES: {}", minNodes);
             MIN_NODES = Integer.parseInt(minNodes);