From fc3cf8628d46b02a7c86c865889954fa6edb6c9a Mon Sep 17 00:00:00 2001
From: qfrjava <13402782+qfrjava@user.noreply.gitee.com>
Date: 星期五, 18 四月 2025 14:28:06 +0800
Subject: [PATCH] feat(websocket): 实现 websocket广播功能
---
JAVA/SMTAIServer/src/main/java/com/smtaiserver/smtaiserver/core/SMTAIServerApp.java | 11 +++++++++--
1 files changed, 9 insertions(+), 2 deletions(-)
diff --git a/JAVA/SMTAIServer/src/main/java/com/smtaiserver/smtaiserver/core/SMTAIServerApp.java b/JAVA/SMTAIServer/src/main/java/com/smtaiserver/smtaiserver/core/SMTAIServerApp.java
index e9026da..4415788 100644
--- a/JAVA/SMTAIServer/src/main/java/com/smtaiserver/smtaiserver/core/SMTAIServerApp.java
+++ b/JAVA/SMTAIServer/src/main/java/com/smtaiserver/smtaiserver/core/SMTAIServerApp.java
@@ -22,6 +22,7 @@
import com.smtaiserver.smtaiserver.javaai.qwen.SMTQwenAgentManager;
import com.smtaiserver.smtaiserver.javaai.qwen.SMTQwenApp;
import com.smtaiserver.smtaiserver.javaai.sse.SMTSSEBroadcastChat;
+import com.smtaiserver.smtaiserver.web.SMTWebSocketHandler;
import com.smtservlet.core.SMTApp;
import com.smtservlet.core.SMTApp.SMTEhCacheManagerInitialize;
import com.smtservlet.util.Json;
@@ -42,6 +43,7 @@
import java.util.Set;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
+import javax.annotation.Resource;
import javax.servlet.http.HttpSessionEvent;
import net.sf.ehcache.Cache;
import net.sf.ehcache.CacheManager;
@@ -144,7 +146,8 @@
private static Pattern _patGlobalMacro = Pattern.compile("\\{\\{\\{([\\w\\.]+)\\}\\}\\}");
private static Logger _logger = LogManager.getLogger(SMTQwenApp.class);
private static Pattern _patIsNumber = Pattern.compile("^[1-9]\\d*$");
-
+ @Resource
+ private SMTWebSocketHandler webSocketHandler;
private static String[] _encacheIdList = {
"GlobalConfig",
"MetricsDefMap",
@@ -212,7 +215,11 @@
return chat;
}
}
-
+ public void webSocketApp(String jsonString)
+ {
+ webSocketHandler.broadcast(jsonString);
+ }
+
public void setServiceEncache(SMTAIServerEncache serverEncache)
{
_serverEncache = serverEncache;
--
Gitblit v1.9.3