From b6a3adb21cdc90e953558ec166c0c6707a1a5fab Mon Sep 17 00:00:00 2001
From: Redkale <8730487+redkale@users.noreply.github.com>
Date: Sun, 7 Jun 2020 22:11:49 +0800
Subject: [PATCH]

---
 src/org/redkale/net/sncp/Sncp.java | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/org/redkale/net/sncp/Sncp.java b/src/org/redkale/net/sncp/Sncp.java
index c9ddaba9b..b6b39fb5d 100644
--- a/src/org/redkale/net/sncp/Sncp.java
+++ b/src/org/redkale/net/sncp/Sncp.java
@@ -164,6 +164,11 @@ public abstract class Sncp {
             Field ts = service.getClass().getDeclaredField(FIELDPREFIX + "_messageagent");
             ts.setAccessible(true);
             ts.set(service, messageAgent);
+            if (service instanceof WebSocketNode) {
+                Field c = WebSocketNode.class.getDeclaredField("messageAgent");
+                c.setAccessible(true);
+                c.set(service, messageAgent);
+            }
         } catch (Exception e) {
             throw new RuntimeException(service + " not found " + FIELDPREFIX + "_messageagent");
         }