From 9108d1db46aaf3117940571e391b78d60f237ef3 Mon Sep 17 00:00:00 2001
From: slhafzjw <slhafzjw@slhaf.work>
Date: Mon, 20 Apr 2026 14:38:58 +0800
Subject: [PATCH] feat(gateway): support assign hostname on websocket gateway

---
 .../slhaf/partner/runtime/gateway/WebSocketGateway.java     | 6 ++++--
 .../partner/runtime/gateway/WebSocketGatewayRegistration.kt | 3 ++-
 2 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/Partner-Core/src/main/java/work/slhaf/partner/runtime/gateway/WebSocketGateway.java b/Partner-Core/src/main/java/work/slhaf/partner/runtime/gateway/WebSocketGateway.java
index 0aad6dbb..457a3edd 100644
--- a/Partner-Core/src/main/java/work/slhaf/partner/runtime/gateway/WebSocketGateway.java
+++ b/Partner-Core/src/main/java/work/slhaf/partner/runtime/gateway/WebSocketGateway.java
@@ -34,11 +34,13 @@ public class WebSocketGateway extends WebSocketServer implements AgentGateway<In
     // 记录最后一次收到Pong的时间
     private final ConcurrentHashMap<WebSocket, Long> lastPongTimes = new ConcurrentHashMap<>();
 
-    public WebSocketGateway(int port, long heartbeatInterval) {
-        super(new InetSocketAddress(port));
+    public WebSocketGateway(int port, @NotNull String hostname, long heartbeatInterval) {
+        super(new InetSocketAddress(hostname, port));
         this.heartbeatInterval = heartbeatInterval;
         this.setReuseAddr(true);
         this.executor = Executors.newSingleThreadExecutor();
+
+        log.info("WebSocketGateway started on {}: {}", hostname, port);
     }
 
     public void launch() {
diff --git a/Partner-Core/src/main/java/work/slhaf/partner/runtime/gateway/WebSocketGatewayRegistration.kt b/Partner-Core/src/main/java/work/slhaf/partner/runtime/gateway/WebSocketGatewayRegistration.kt
index 3d93131b..9dd849b2 100644
--- a/Partner-Core/src/main/java/work/slhaf/partner/runtime/gateway/WebSocketGatewayRegistration.kt
+++ b/Partner-Core/src/main/java/work/slhaf/partner/runtime/gateway/WebSocketGatewayRegistration.kt
@@ -10,9 +10,10 @@ object WebSocketGatewayRegistration : AgentGatewayRegistration {
     override fun create(params: Map<String, String>): AgentGateway<*, *> {
         val port = params["port"]?.toIntOrNull() ?: 29600
         val heartbeatInterval = params["heartbeat_interval"]?.toLongOrNull() ?: 10_000L
+        val hostname = params["hostname"] ?: "127.0.0.1"
         require(port > 0) { "port must be greater than 0" }
         require(heartbeatInterval > 0) { "heartbeat_interval must be greater than 0" }
-        return WebSocketGateway(port, heartbeatInterval)
+        return WebSocketGateway(port, hostname, heartbeatInterval)
     }
 
     override fun shutdown(instance: AgentGateway<*, *>) {