From 9315dc35f91ecc64b2657d9b5a627d03e8e0cc74 Mon Sep 17 00:00:00 2001
From: LXW <1264174498@qq.com>
Date: Thu, 27 Jun 2024 11:19:21 +0800
Subject: [PATCH] (improvement)(launcher) Optimize demo data loading to avoid
 repeatedly writing data to mysql (#1236)

* (improvement)(launcher) Optimize demo data loading to avoid repeatedly writing data to mysql


Co-authored-by: lxwcodemonkey
---
 .../server/service/impl/ChatServiceImpl.java    | 17 -----------------
 .../tencent/supersonic/demo/SmallTalkDemo.java  |  8 +++++++-
 2 files changed, 7 insertions(+), 18 deletions(-)

diff --git a/chat/server/src/main/java/com/tencent/supersonic/chat/server/service/impl/ChatServiceImpl.java b/chat/server/src/main/java/com/tencent/supersonic/chat/server/service/impl/ChatServiceImpl.java
index 88e49a634..3d194bad8 100644
--- a/chat/server/src/main/java/com/tencent/supersonic/chat/server/service/impl/ChatServiceImpl.java
+++ b/chat/server/src/main/java/com/tencent/supersonic/chat/server/service/impl/ChatServiceImpl.java
@@ -4,11 +4,9 @@ import com.tencent.supersonic.auth.api.authentication.pojo.User;
 import com.tencent.supersonic.chat.api.pojo.request.ChatExecuteReq;
 import com.tencent.supersonic.chat.api.pojo.request.ChatParseReq;
 import com.tencent.supersonic.chat.api.pojo.request.ChatQueryDataReq;
-import com.tencent.supersonic.chat.api.pojo.request.SimilarQueryReq;
 import com.tencent.supersonic.chat.server.agent.Agent;
 import com.tencent.supersonic.chat.server.executor.ChatExecutor;
 import com.tencent.supersonic.chat.server.parser.ChatParser;
-import com.tencent.supersonic.chat.server.persistence.dataobject.ChatQueryDO;
 import com.tencent.supersonic.chat.server.pojo.ChatExecuteContext;
 import com.tencent.supersonic.chat.server.pojo.ChatParseContext;
 import com.tencent.supersonic.chat.server.processor.execute.ExecuteResultProcessor;
@@ -18,7 +16,6 @@ import com.tencent.supersonic.chat.server.service.ChatManageService;
 import com.tencent.supersonic.chat.server.service.ChatService;
 import com.tencent.supersonic.chat.server.util.ComponentFactory;
 import com.tencent.supersonic.chat.server.util.QueryReqConverter;
-import com.tencent.supersonic.chat.server.util.SimilarQueryManager;
 import com.tencent.supersonic.common.util.BeanMapper;
 import com.tencent.supersonic.common.util.ContextUtils;
 import com.tencent.supersonic.headless.api.pojo.SemanticParseInfo;
@@ -48,8 +45,6 @@ public class ChatServiceImpl implements ChatService {
     private ChatQueryService chatQueryService;
     @Autowired
     private RetrieveService retrieveService;
-    @Autowired
-    private SimilarQueryManager similarQueryManager;
     private List<ChatParser> chatParsers = ComponentFactory.getChatParsers();
     private List<ChatExecutor> chatExecutors = ComponentFactory.getChatExecutors();
     private List<ParseResultProcessor> parseResultProcessors = ComponentFactory.getParseProcessors();
@@ -95,7 +90,6 @@ public class ChatServiceImpl implements ChatService {
             for (ExecuteResultProcessor processor : executeResultProcessors) {
                 processor.process(chatExecuteContext, queryResult);
             }
-            saveQueryResult(chatExecuteReq, queryResult);
         }
 
         return queryResult;
@@ -143,15 +137,4 @@ public class ChatServiceImpl implements ChatService {
         return chatQueryService.queryDimensionValue(dimensionValueReq, user);
     }
 
-    public void saveQueryResult(ChatExecuteReq chatExecuteReq, QueryResult queryResult) {
-        //The history record only retains the query result of the first parse
-        if (chatExecuteReq.getParseId() > 1) {
-            return;
-        }
-        ChatQueryDO chatQueryDO = chatManageService.saveQueryResult(chatExecuteReq, queryResult);
-        SimilarQueryReq similarQueryReq = SimilarQueryReq.builder().queryId(chatExecuteReq.getQueryId())
-                .queryText(chatQueryDO.getQueryText()).agentId(chatQueryDO.getAgentId()).build();
-        similarQueryManager.saveSimilarQuery(similarQueryReq);
-    }
-
 }
diff --git a/launchers/standalone/src/main/java/com/tencent/supersonic/demo/SmallTalkDemo.java b/launchers/standalone/src/main/java/com/tencent/supersonic/demo/SmallTalkDemo.java
index 63a428ba5..023a6f75a 100644
--- a/launchers/standalone/src/main/java/com/tencent/supersonic/demo/SmallTalkDemo.java
+++ b/launchers/standalone/src/main/java/com/tencent/supersonic/demo/SmallTalkDemo.java
@@ -11,6 +11,9 @@ import lombok.extern.slf4j.Slf4j;
 import org.springframework.core.annotation.Order;
 import org.springframework.stereotype.Component;
 
+import java.util.List;
+import java.util.stream.Collectors;
+
 @Component
 @Slf4j
 @Order(2)
@@ -35,6 +38,9 @@ public class SmallTalkDemo extends S2BaseDemo {
 
     @Override
     boolean checkNeedToRun() {
-        return true;
+        List<String> agentNames = agentService.getAgents()
+                .stream().map(Agent::getName).collect(Collectors.toList());
+        return !agentNames.contains("来闲聊");
     }
+
 }