topdog/supersonic
1
0
Fork 0
mirror of https://github.com/tencentmusic/supersonic.git synced 2025-12-12 04:27:39 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

[improvement](chat) Fix the shutdown issue with the Python service. (#405)

Browse Source
This commit is contained in:
lexluo09
2023-11-20 13:24:38 +08:00
committed by GitHub
parent 80ad75503b
commit 2c23c2f574
1 changed files with 9 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
semantic/model/src/main/java/com/tencent/supersonic/semantic/model/domain/listener/MetaEmbeddingListener.java
View File

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