Merge pull request #100 from PeterPeterWang/main

Fix dialogue memory leak risk

Author: joey-zhou

src/main/java/com/xiaozhi/communication/common/SessionManager.java

@@ -4,6 +4,7 @@
 import com.xiaozhi.entity.SysDevice;
 import com.xiaozhi.entity.SysRole;
 import com.xiaozhi.enums.ListenMode;
+import com.xiaozhi.event.ChatSessionCloseEvent;
 import jakarta.annotation.PostConstruct;
 import jakarta.annotation.PreDestroy;
 import jakarta.annotation.Resource;
@@ -153,6 +154,7 @@ public void closeSession(ChatSession chatSession) {
             // 重置会话状态
             chatSession.setStreamingState(false);
             chatSession.setAudioSinks(null);
+            applicationContext.publishEvent(new ChatSessionCloseEvent(chatSession));
             // 从会话映射中移除
             logger.info("会话已关闭 - SessionId: {} SessionType: {}", chatSession.getSessionId(), chatSession.getClass().getSimpleName());
         } catch (Exception e) {

src/main/java/com/xiaozhi/dialogue/service/DialogueService.java

@@ -10,12 +10,14 @@
 import com.xiaozhi.dialogue.tts.factory.TtsServiceFactory;
 import com.xiaozhi.entity.SysConfig;
 import com.xiaozhi.entity.SysDevice;
+import com.xiaozhi.event.ChatSessionCloseEvent;
 import com.xiaozhi.utils.AudioUtils;
 import com.xiaozhi.utils.EmojiUtils;
 import com.xiaozhi.utils.EmojiUtils.EmoSentence;
 import jakarta.annotation.Resource;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
+import org.springframework.context.ApplicationListener;
 import org.springframework.stereotype.Service;
 import org.springframework.util.ObjectUtils;
 import org.springframework.util.StringUtils;
@@ -33,7 +35,7 @@
  * 负责处理语音识别和对话生成的业务逻辑
  */
 @Service
-public class DialogueService {
+public class DialogueService implements ApplicationListener<ChatSessionCloseEvent> {
     private static final Logger logger = LoggerFactory.getLogger(DialogueService.class);
     private static final DecimalFormat df = new DecimalFormat("0.00");
     private static final long TIMEOUT_MS = 5000;
@@ -84,6 +86,20 @@ public class DialogueService {
     private final Map<String, Semaphore> sessionSemaphores = new ConcurrentHashMap<>();
     private final Map<String, PriorityBlockingQueue<TtsTask>> sessionTaskQueues = new ConcurrentHashMap<>();
 
+    @Override
+    public void onApplicationEvent(ChatSessionCloseEvent event) {
+        ChatSession chatSession = event.getSession();
+        if(chatSession != null) {
+            // clean up dialogue audio paths and responses
+            if (StringUtils.hasText(chatSession.getDialogueId())) {
+                String dialogueId = chatSession.getDialogueId();
+                dialogueAudioPaths.remove(dialogueId);
+                dialogueResponses.remove(dialogueId);
+            }
+            cleanupSession(chatSession.getSessionId());
+        }
+    }
+
     /**
      * 句子对象，用于跟踪每个句子的处理状态
      */

src/main/java/com/xiaozhi/event/ChatSessionCloseEvent.java

@@ -0,0 +1,25 @@
+package com.xiaozhi.event;
+
+import com.xiaozhi.communication.common.ChatSession;
+import org.springframework.context.ApplicationEvent;
+
+import java.time.Clock;
+
+/**
+ * Session 关闭事件
+ */
+public class ChatSessionCloseEvent extends ApplicationEvent {
+
+    public ChatSessionCloseEvent(Object source) {
+        super(source);
+    }
+
+    public ChatSessionCloseEvent(Object source, Clock clock) {
+        super(source, clock);
+    }
+
+    public ChatSession getSession() {
+        return (ChatSession) getSource();
+    }
+
+}