diff --git a/src/org/redkale/net/http/WebSocketNode.java b/src/org/redkale/net/http/WebSocketNode.java index 21829ac1f..b88293788 100644 --- a/src/org/redkale/net/http/WebSocketNode.java +++ b/src/org/redkale/net/http/WebSocketNode.java @@ -92,9 +92,9 @@ public abstract class WebSocketNode { protected abstract CompletableFuture broadcastMessage(@RpcTargetAddress InetSocketAddress targetAddress, WebSocketRange wsrange, Object message, boolean last); - protected abstract CompletableFuture connect(Serializable userid, InetSocketAddress addr); + protected abstract CompletableFuture connect(Serializable userid, @RpcTargetAddress InetSocketAddress addr); - protected abstract CompletableFuture disconnect(Serializable userid, InetSocketAddress addr); + protected abstract CompletableFuture disconnect(Serializable userid, @RpcTargetAddress InetSocketAddress addr); protected abstract CompletableFuture changeUserid(Serializable fromuserid, Serializable touserid, @RpcTargetAddress InetSocketAddress addr); diff --git a/src/org/redkale/service/WebSocketNodeService.java b/src/org/redkale/service/WebSocketNodeService.java index c9fdcff21..227c1d616 100644 --- a/src/org/redkale/service/WebSocketNodeService.java +++ b/src/org/redkale/service/WebSocketNodeService.java @@ -76,7 +76,7 @@ public class WebSocketNodeService extends WebSocketNode implements Service { * @return 无返回值 */ @Override - public CompletableFuture connect(Serializable userid, InetSocketAddress sncpAddr) { + public CompletableFuture connect(Serializable userid, @RpcTargetAddress InetSocketAddress sncpAddr) { tryAcquireSemaphore(); CompletableFuture future = sncpNodeAddresses.appendSetItemAsync(SOURCE_SNCP_USERID_PREFIX + userid, sncpAddr); future = future.thenAccept((a) -> sncpNodeAddresses.appendSetItemAsync(SOURCE_SNCP_ADDRS_KEY, sncpAddr)); @@ -94,7 +94,7 @@ public class WebSocketNodeService extends WebSocketNode implements Service { * @return 无返回值 */ @Override - public CompletableFuture disconnect(Serializable userid, InetSocketAddress sncpAddr) { + public CompletableFuture disconnect(Serializable userid, @RpcTargetAddress InetSocketAddress sncpAddr) { tryAcquireSemaphore(); CompletableFuture future = sncpNodeAddresses.removeSetItemAsync(SOURCE_SNCP_USERID_PREFIX + userid, sncpAddr); if (semaphore != null) future.whenComplete((r, e) -> releaseSemaphore());