From a64b97c25630efb4b874736133193422a8d072cc Mon Sep 17 00:00:00 2001
From: Hang <hang@Hangs-MacBook-Pro.local>
Date: Tue, 26 Apr 2022 16:50:12 -0500
Subject: [PATCH] fast push

---
 sample_test/src/server.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/sample_test/src/server.py b/sample_test/src/server.py
index 623f8ab..446225b 100644
--- a/sample_test/src/server.py
+++ b/sample_test/src/server.py
@@ -429,7 +429,8 @@ def parse_config(config_path: str) -> dict:
 
 def map_hostname_to_ip(servers):
     for server_id, server in servers.items():
-        SERVER_ID_TO_IP_MAPPING[server_id] = socket.gethostbyname(server[SERVER_ADDRESS])
+        SERVER_ID_TO_IP_MAPPING[server_id] = socket.gethostbyname(server[SERVER_ADDRESS]) \
+            if server[SERVER_ADDRESS] != 'localhost' else socket.gethostbyname(socket.gethostname())
         IP_TO_SERVER_ID_MAPPING[SERVER_ID_TO_IP_MAPPING[server_id]] = server_id
 
 
-- 
GitLab