From e6ef4d1546c2b9e205b3f92ac619055ec06c65ba Mon Sep 17 00:00:00 2001
From: Redkale <8730487+redkale@users.noreply.github.com>
Date: Fri, 27 Jul 2018 20:02:08 +0800
Subject: [PATCH]

---
 src/org/redkale/net/http/WebSocketNode.java       | 4 ++--
 src/org/redkale/service/WebSocketNodeService.java | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

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<Integer> broadcastMessage(@RpcTargetAddress InetSocketAddress targetAddress, WebSocketRange wsrange, Object message, boolean last);
 
-    protected abstract CompletableFuture<Void> connect(Serializable userid, InetSocketAddress addr);
+    protected abstract CompletableFuture<Void> connect(Serializable userid, @RpcTargetAddress InetSocketAddress addr);
 
-    protected abstract CompletableFuture<Void> disconnect(Serializable userid, InetSocketAddress addr);
+    protected abstract CompletableFuture<Void> disconnect(Serializable userid, @RpcTargetAddress InetSocketAddress addr);
 
     protected abstract CompletableFuture<Void> 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<Void> connect(Serializable userid, InetSocketAddress sncpAddr) {
+    public CompletableFuture<Void> connect(Serializable userid, @RpcTargetAddress InetSocketAddress sncpAddr) {
         tryAcquireSemaphore();
         CompletableFuture<Void> 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<Void> disconnect(Serializable userid, InetSocketAddress sncpAddr) {
+    public CompletableFuture<Void> disconnect(Serializable userid, @RpcTargetAddress InetSocketAddress sncpAddr) {
         tryAcquireSemaphore();
         CompletableFuture<Void> future = sncpNodeAddresses.removeSetItemAsync(SOURCE_SNCP_USERID_PREFIX + userid, sncpAddr);
         if (semaphore != null) future.whenComplete((r, e) -> releaseSemaphore());