From 97e1bb49a5540064d627a25d9169d803dfd11243 Mon Sep 17 00:00:00 2001
From: sangge <2251250136@qq.com>
Date: Tue, 24 Oct 2023 19:20:03 +0800
Subject: [PATCH] feat: read host ip from env

---
 src/node.py | 17 +++++------------
 1 file changed, 5 insertions(+), 12 deletions(-)

diff --git a/src/node.py b/src/node.py
index 21aad78..6d71ff2 100644
--- a/src/node.py
+++ b/src/node.py
@@ -5,6 +5,7 @@ import socket
 import asyncio
 from pydantic import BaseModel
 from tpre import *
+import os
 
 
 @asynccontextmanager
@@ -17,7 +18,7 @@ async def lifespan(app: FastAPI):
 app = FastAPI(lifespan=lifespan)
 server_address = "http://10.20.14.232:8000/server"
 id = 0
-ip = "10.16.21.163"
+ip = ""
 client_ip_src = ""  # 发送信息用户的ip
 client_ip_des = ""  # 接收信息用户的ip
 processed_message = ()  # 重加密后的数据
@@ -35,22 +36,14 @@ def send_ip():
     id = requests.get(url)
 
 
-# 用socket获取本机ip
+# 用环境变量获取本机ip
 def get_local_ip():
-    # 创建一个套接字对象
-    s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
-    # 连接到一个外部的服务器，这将自动绑定到本地IP地址
-    s.connect(("8.8.8.8", 80))
-    # 获取本地IP地址
-    local_ip = s.getsockname()[0]
-    s.close()
     global ip
-    ip = local_ip
+    ip = os.environ.get("HOST_IP", "IP not set")
 
 
 def init():
-    # get_local_ip()
-    global id
+    get_local_ip()
     send_ip()
     task = asyncio.create_task(send_heartbeat_internal())
     print("Finish init")