From 1ca5584d7e3200a9af65a099bd26d3593e2ba702 Mon Sep 17 00:00:00 2001
From: liyong <18434998025@163.com>
Date: 星期四, 07 五月 2026 14:36:08 +0800
Subject: [PATCH] 迁移pro
---
src/main/java/com/ruoyi/ai/store/MongoChatMemoryStore.java | 62 +++++++++++++++++++++++++++++++
1 files changed, 62 insertions(+), 0 deletions(-)
diff --git a/src/main/java/com/ruoyi/ai/store/MongoChatMemoryStore.java b/src/main/java/com/ruoyi/ai/store/MongoChatMemoryStore.java
new file mode 100644
index 0000000..e88f0e9
--- /dev/null
+++ b/src/main/java/com/ruoyi/ai/store/MongoChatMemoryStore.java
@@ -0,0 +1,62 @@
+package com.ruoyi.ai.store;
+
+import com.ruoyi.ai.mongodbBean.ChatMessages;
+import dev.langchain4j.data.message.ChatMessage;
+import dev.langchain4j.data.message.ChatMessageDeserializer;
+import dev.langchain4j.data.message.ChatMessageSerializer;
+import dev.langchain4j.store.memory.chat.ChatMemoryStore;
+import lombok.RequiredArgsConstructor;
+import org.springframework.data.mongodb.core.MongoTemplate;
+import org.springframework.data.mongodb.core.query.Criteria;
+import org.springframework.data.mongodb.core.query.Query;
+import org.springframework.data.mongodb.core.query.Update;
+import org.springframework.stereotype.Component;
+
+import java.util.Date;
+import java.util.LinkedList;
+import java.util.List;
+
+@Component
+@RequiredArgsConstructor
+public class MongoChatMemoryStore implements ChatMemoryStore {
+
+ private final MongoTemplate mongoTemplate;
+
+ @Override
+ public List<ChatMessage> getMessages(Object memoryId) {
+ Query query = Query.query(Criteria.where("memoryId").is(memoryIdString(memoryId)));
+ ChatMessages chatMessages = mongoTemplate.findOne(query, ChatMessages.class);
+ if (chatMessages == null || chatMessages.getContent() == null) {
+ return new LinkedList<>();
+ }
+ return ChatMessageDeserializer.messagesFromJson(chatMessages.getContent());
+ }
+
+ @Override
+ public void updateMessages(Object memoryId, List<ChatMessage> messages) {
+ String memoryIdValue = memoryIdString(memoryId);
+ Query query = Query.query(Criteria.where("memoryId").is(memoryIdValue));
+ Update update = new Update();
+ update.set("memoryId", memoryIdValue);
+ update.set("content", ChatMessageSerializer.messagesToJson(messages));
+ update.set("updateTime", new Date());
+ update.setOnInsert("createTime", new Date());
+ mongoTemplate.upsert(query, update, ChatMessages.class);
+ }
+
+ @Override
+ public void deleteMessages(Object memoryId) {
+ Query query = Query.query(Criteria.where("memoryId").is(memoryIdString(memoryId)));
+ mongoTemplate.remove(query, ChatMessages.class);
+ }
+
+ public void appendMessages(Object memoryId, List<ChatMessage> appendList) {
+ List<ChatMessage> messages = new LinkedList<>(getMessages(memoryId));
+ messages.addAll(appendList);
+ updateMessages(memoryId, messages);
+ }
+
+ private String memoryIdString(Object memoryId) {
+ return memoryId == null ? "" : memoryId.toString();
+ }
+}
--
Gitblit v1.9.3