From 2c23c2f574cacf81a67e5a4f8f796a1e38742f5b Mon Sep 17 00:00:00 2001
From: lexluo09 <39718951+lexluo09@users.noreply.github.com>
Date: Mon, 20 Nov 2023 13:24:38 +0800
Subject: [PATCH] [improvement](chat) Fix the shutdown issue with the Python
 service. (#405)

---
 .../model/domain/listener/MetaEmbeddingListener.java     | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/semantic/model/src/main/java/com/tencent/supersonic/semantic/model/domain/listener/MetaEmbeddingListener.java b/semantic/model/src/main/java/com/tencent/supersonic/semantic/model/domain/listener/MetaEmbeddingListener.java
index 0404af90c..44737aa45 100644
--- a/semantic/model/src/main/java/com/tencent/supersonic/semantic/model/domain/listener/MetaEmbeddingListener.java
+++ b/semantic/model/src/main/java/com/tencent/supersonic/semantic/model/domain/listener/MetaEmbeddingListener.java
@@ -11,6 +11,7 @@ import java.util.Map;
 import java.util.stream.Collectors;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.annotation.Value;
 import org.springframework.context.ApplicationListener;
 import org.springframework.scheduling.annotation.Async;
 import org.springframework.stereotype.Component;
@@ -25,6 +26,9 @@ public class MetaEmbeddingListener implements ApplicationListener<DataEvent> {
     @Autowired
     private EmbeddingUtils embeddingUtils;
 
+    @Value("${embedding.operation.sleep.time:3000}")
+    private Integer embeddingOperationSleepTime;
+
     @Async
     @Override
     public void onApplicationEvent(DataEvent event) {
@@ -46,6 +50,11 @@ public class MetaEmbeddingListener implements ApplicationListener<DataEvent> {
         if (CollectionUtils.isEmpty(embeddingQueries)) {
             return;
         }
+        try {
+            Thread.sleep(embeddingOperationSleepTime);
+        } catch (InterruptedException e) {
+            log.error("", e);
+        }
         embeddingUtils.addCollection(COLLECTION_NAME);
         if (event.getEventType().equals(EventType.ADD)) {
             embeddingUtils.addQuery(COLLECTION_NAME, embeddingQueries);