From 86b54ec52373a7051c44cea6855260d0224aee56 Mon Sep 17 00:00:00 2001
From: kira <yinruoxi666@126.com>
Date: Tue, 12 Mar 2024 22:34:51 +0800
Subject: [PATCH] =?UTF-8?q?yinruoxi=20feature=EF=BC=9A=20=20=20=20=201.?=
 =?UTF-8?q?=E6=96=B0=E5=A2=9E=EF=BC=9A=E8=AF=AD=E6=84=8F=E5=AE=8C=E6=95=B4?=
 =?UTF-8?q?=E6=80=A7=E5=88=86=E6=9E=90?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../server/handler/ChatServerHandler.java     | 40 ++++++++++++++++++-
 1 file changed, 39 insertions(+), 1 deletion(-)

diff --git a/xueyi-modules/xueyi-nlt/src/main/java/com/xueyi/nlt/netty/server/handler/ChatServerHandler.java b/xueyi-modules/xueyi-nlt/src/main/java/com/xueyi/nlt/netty/server/handler/ChatServerHandler.java
index 36d33b16..a204d9cf 100644
--- a/xueyi-modules/xueyi-nlt/src/main/java/com/xueyi/nlt/netty/server/handler/ChatServerHandler.java
+++ b/xueyi-modules/xueyi-nlt/src/main/java/com/xueyi/nlt/netty/server/handler/ChatServerHandler.java
@@ -9,8 +9,10 @@ import com.xueyi.common.core.web.result.R;
 import com.xueyi.nlt.api.nlt.domain.vo.DmIntentVo;
 import com.xueyi.nlt.api.nlt.domain.vo.response.DmIntentResponse;
 import com.xueyi.nlt.netty.server.config.ServerConfig;
+import com.xueyi.nlt.nlt.domain.vo.WordProcessVo;
 import com.xueyi.nlt.nlt.service.IDmIntentService;
 import com.xueyi.nlt.nlt.service.IDmRegularService;
+import com.xueyi.nlt.nlt.service.IDmWordProcessService;
 import com.xueyi.nlt.nlt.service.impl.LogServiceImpl;
 import com.xueyi.nlt.nlt.template.FreeChatTemplate;
 import com.xueyi.nlt.nlt.template.MovieChatTemplate;
@@ -57,6 +59,8 @@ public class ChatServerHandler extends SimpleChannelInboundHandler<TextWebSocket
     @Autowired
     private IDmIntentService intentService;
 
+    @Autowired
+    private IDmWordProcessService wordProcessService;
 
     @Autowired
     private MovieChatTemplate movieChatTemplate;
@@ -75,6 +79,7 @@ public class ChatServerHandler extends SimpleChannelInboundHandler<TextWebSocket
         INSTANCE.manDeviceService = this.manDeviceService;
         INSTANCE.remoteEnterpriseService = this.remoteEnterpriseService;
         INSTANCE.logService = this.logService;
+        INSTANCE.wordProcessService = this.wordProcessService;
     }
     @Override
     protected void channelRead0(ChannelHandlerContext channelHandlerContext, TextWebSocketFrame textWebSocketFrame) throws Exception {
@@ -146,6 +151,32 @@ public class ChatServerHandler extends SimpleChannelInboundHandler<TextWebSocket
                     return;
                 }
                 else {
+                    // 语意判断
+                    WordProcessVo wordProcessVo = new WordProcessVo();
+                    wordProcessVo.setMetadata(msg);
+                    WordProcessVo result = INSTANCE.wordProcessService.semanticIntegrityDetection(wordProcessVo);
+                    if (result != null) {
+                        System.out.println(result.getProcessedResult());
+                        // 判断结果
+                        switch (result.getProcessedResult()){
+
+                            case "0":
+                                // 返回前端，抛弃
+                                JSONObject jo = new JSONObject();
+                                jo.put("action","chat");
+                                jo.put("motion","idle");
+                                jo.put("traceId","");
+                                jo.put("status",2);
+                                jo.put("code", -1);
+                                jo.put("tts","");
+                                reply(channel, msg,jo, enterpriseName,"大模型");
+                                return;
+                            case "1":
+                            case "2":
+                                break;
+                            default:
+                        }
+                    }
                     JSONObject jo = new JSONObject();
                     jo.put("action","chat");
                     jo.put("motion","idle");
@@ -163,6 +194,7 @@ public class ChatServerHandler extends SimpleChannelInboundHandler<TextWebSocket
                 jo.put("traceId","");
                 jo.put("status",2);
                 jo.put("tts","大模型出现异常，请稍后重试。");
+                jo.put("code",500);
                 String str = jo.toJSONString();
                 log.info("发生异常client：{},内容：{}",devId,jo.toJSONString());
                 channel.writeAndFlush(new TextWebSocketFrame(str));
@@ -236,7 +268,9 @@ public class ChatServerHandler extends SimpleChannelInboundHandler<TextWebSocket
     @TLogAspect(str = "testtest")
 //    @TLogAspect({"enterpriseName","type"})
     private void reply(Channel channel, String msg, JSONObject jo,String enterpriseName,String type ) {
-
+        if (!jo.containsKey("code")) {
+            jo.put("code",0);
+        }
         INSTANCE.logService.record(jo,msg,enterpriseName,type);
         channel.writeAndFlush(new TextWebSocketFrame(jo.toJSONString()));
     }
@@ -285,6 +319,10 @@ public class ChatServerHandler extends SimpleChannelInboundHandler<TextWebSocket
             log.info("断开废弃连接,userId：{},channel:{}：", userId, channel.id().asLongText());
             return;
         }
+        // 清楚traceId信息
+        if (ServerConfig.currentTraceMap.containsKey(userId)) {
+            ServerConfig.currentTraceMap.remove(userId);
+        }
         if (ServerConfig.sessionMap.containsKey(userId)) {
             ServerConfig.sessionMap.remove(userId);
             log.info("用户下线,userId：{},channel:{}：", userId, channel.id().asLongText());