Update app.py

app.py

@@ -34,12 +34,28 @@ CORS(app, resources={r"/*": {"origins": "*"}})
 executor = ThreadPoolExecutor(max_workers=10)
 
 proxy_url = os.getenv('PROXY_URL')
-NOTDIAMOND_IP = os.getenv('NOTDIAMOND_IP')
 NOTDIAMOND_DOMAIN = os.getenv('NOTDIAMOND_DOMAIN')
+if not NOTDIAMOND_DOMAIN:
+    logger.error("NOTDIAMOND_DOMAIN environment variable is not set!")
+    raise ValueError("NOTDIAMOND_DOMAIN must be set")
 
-if not NOTDIAMOND_IP:
-    logger.error("NOTDIAMOND_IP environment variable is not set!")
-    raise ValueError("NOTDIAMOND_IP must be set")
+# 动态IP缓存，设置1小时过期
+ip_cache = TTLCache(maxsize=1, ttl=3600)
+ip_refresh_lock = threading.Lock()
+
+def get_current_ip():
+    """动态获取当前域名对应的IP地址"""
+    try:
+        with ip_refresh_lock:
+            if 'current_ip' not in ip_cache:
+                logger.info(f"Resolving IP for domain: {NOTDIAMOND_DOMAIN}")
+                ip = socket.gethostbyname(NOTDIAMOND_DOMAIN)
+                ip_cache['current_ip'] = ip
+                logger.info(f"Successfully resolved IP: {ip}")
+            return ip_cache['current_ip']
+    except socket.gaierror as e:
+        logger.error(f"DNS resolution failed for {NOTDIAMOND_DOMAIN}: {e}")
+        raise Exception(f"Failed to resolve IP for {NOTDIAMOND_DOMAIN}")
 
 refresh_token_cache = TTLCache(maxsize=1000, ttl=3600)
 headers_cache = TTLCache(maxsize=1, ttl=3600)  # 1小时过期
@@ -49,8 +65,9 @@ token_refresh_lock = threading.Lock()
 def patched_create_connection(address, *args, **kwargs):
     host, port = address
     if host == NOTDIAMOND_DOMAIN:
-        logger.info(f"Connecting to {NOTDIAMOND_DOMAIN} using IP: {NOTDIAMOND_IP}")
-        return create_connection((NOTDIAMOND_IP, port), *args, **kwargs)
+        current_ip = get_current_ip()
+        logger.info(f"Connecting to {NOTDIAMOND_DOMAIN} using IP: {current_ip}")
+        return create_connection((current_ip, port), *args, **kwargs)
     return create_connection(address, *args, **kwargs)
 
 # 替换 urllib3 的默认连接函数
@@ -664,6 +681,15 @@ def health_check():
                 if current_time.tm_hour == 0 and current_time.tm_min == 0:
                     multi_auth_manager.reset_all_model_status()
                     logger.info("Reset model status for all accounts")
+                
+                # 每小时刷新一次IP缓存
+                if current_time.tm_min == 0:
+                    try:
+                        ip_cache.clear()
+                        logger.info("Cleared IP cache for periodic refresh")
+                    except Exception as e:
+                        logger.error(f"Error clearing IP cache: {e}")
+                
         except Exception as e:
             logger.error(f"Health check error: {e}")
         time.sleep(60)  # 每分钟检查一次