xservice/xservice-ai-center
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat:数据库持久化

Browse Source
This commit is contained in:
Ttt
2026-03-13 22:56:19 +08:00
parent d00e3ce11e
commit fa93d4ffe0
13 changed files with 215 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
xservice-service/src/main/java/com/xiang/xservice/ai/agent/SimpleChatAgent.java
View File

@@ -9,7 +9,9 @@ import com.xiang.xservice.ai.core.enums.ModelStrategyEnum;
import com.xiang.xservice.ai.core.enums.ModelTypeEnum;
import com.xiang.xservice.ai.core.handler.MyStreamingHandler;
import com.xiang.xservice.ai.core.route.TaskRouter;
import com.xiang.xservice.ai.core.storage.DbPersistentStore;
import com.xiang.xservice.ai.core.storage.MemoryPersistentStore;
import com.xiang.xservice.ai.repository.manage.IAiSimpleChatMessageManage;
import dev.langchain4j.data.message.SystemMessage;
import dev.langchain4j.data.message.UserMessage;
import dev.langchain4j.memory.chat.ChatMemoryProvider;
@@ -26,12 +28,13 @@ public class SimpleChatAgent implements BaseAgent {
private final TaskRouter router;
private final OpenAIBaseConfig openAIBaseConfig;
private final IAiSimpleChatMessageManage aiSimpleChatMessageManage;
@Override
public void chat(ModelTypeEnum modelType, Long id, String message) {
MemoryPersistentStore store = new MemoryPersistentStore();
DbPersistentStore store = new DbPersistentStore(aiSimpleChatMessageManage);
ChatMemoryProvider chatMemoryProvider = memoryId -> MessageWindowChatMemory.builder()
.id(memoryId)

1
xservice-service/src/main/java/com/xiang/xservice/ai/core/handler/MyStreamingHandler.java
View File

@@ -3,7 +3,6 @@ package com.xiang.xservice.ai.core.handler;
import dev.langchain4j.model.StreamingResponseHandler;
import dev.langchain4j.model.chat.response.ChatResponse;
import dev.langchain4j.model.chat.response.StreamingChatResponseHandler;
import org.checkerframework.checker.units.qual.C;
import org.springframework.stereotype.Component;
@Component

3
xservice-service/src/main/java/com/xiang/xservice/ai/core/route/TaskRouter.java
View File

@@ -3,15 +3,12 @@ package com.xiang.xservice.ai.core.route;
import com.xiang.xservice.ai.core.entity.ModelConfig;
import com.xiang.xservice.ai.core.enums.ModelStrategyEnum;
import com.xiang.xservice.ai.core.provider.BaseProvider;
import dev.langchain4j.model.chat.ChatModel;
import dev.langchain4j.model.chat.StreamingChatModel;
import org.checkerframework.checker.units.qual.C;
import org.springframework.stereotype.Component;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;
@Component
public class TaskRouter {

80
xservice-service/src/main/java/com/xiang/xservice/ai/core/storage/DbPersistentStore.java
View File

@@ -1,23 +1,97 @@
package com.xiang.xservice.ai.core.storage;
import dev.langchain4j.data.message.ChatMessage;
import com.alibaba.fastjson2.JSON;
import com.google.common.collect.Lists;
import com.xiang.xservice.ai.pojo.entity.AiSimpleChatMessageDO;
import com.xiang.xservice.ai.repository.manage.IAiSimpleChatMessageManage;
import dev.langchain4j.data.message.*;
import dev.langchain4j.store.memory.chat.ChatMemoryStore;
import lombok.RequiredArgsConstructor;
import org.apache.commons.collections4.CollectionUtils;
import org.springframework.stereotype.Component;
import java.time.LocalDateTime;
import java.util.List;
/**
* 数据库持久化
*/
@RequiredArgsConstructor
public class DbPersistentStore implements ChatMemoryStore {
private final IAiSimpleChatMessageManage aiSimpleChatMessageManage;
@Override
public List<ChatMessage> getMessages(Object memoryId) {
return List.of();
List<AiSimpleChatMessageDO> msg = aiSimpleChatMessageManage.getMsgByMemoryId((Long) memoryId);
if (CollectionUtils.isEmpty(msg)) {
return Lists.newArrayList();
}
List<ChatMessage> result = Lists.newArrayList();
for (AiSimpleChatMessageDO chatMessageDO : msg) {
String role = chatMessageDO.getRole();
switch (role) {
case "SYSTEM":
result.add(SystemMessage.from(chatMessageDO.getMessage()));
break;
case "USER":
result.add(UserMessage.from(chatMessageDO.getMessage()));
break;
case "AI":
result.add(AiMessage.from(chatMessageDO.getMessage()));
break;
default:
break;
}
}
return result;
}
@Override
public void updateMessages(Object memoryId, List<ChatMessage> list) {
List<AiSimpleChatMessageDO> result = Lists.newArrayList();
for (ChatMessage chatMessage : list) {
StringBuilder str = new StringBuilder();
String type = "";
if (chatMessage instanceof SystemMessage systemMessage) {
str = new StringBuilder(systemMessage.text());
type = "SYSTEM";
}
if (chatMessage instanceof UserMessage userMessage) {
for (Content content : userMessage.contents()) {
if (content instanceof TextContent textContent) {
str.append(textContent.text());
type = "USER";
}
}
}
if (chatMessage instanceof AiMessage aiMessage) {
str = new StringBuilder(aiMessage.text());
type = "AI";
}
AiSimpleChatMessageDO message = AiSimpleChatMessageDO.builder()
.userId(1L)
.memoryId((Long) memoryId)
.message(str.toString())
.createTime(LocalDateTime.now())
.delFlag(0)
.role(type)
.build();
result.add(message);
}
aiSimpleChatMessageManage.saveBatch(result);
}
@Override
public void deleteMessages(Object memoryId) {
aiSimpleChatMessageManage.deleteByMemoryId((Long) memoryId);
}
}

5
xservice-service/src/main/java/com/xiang/xservice/ai/core/storage/MemoryPersistentStore.java
View File

@@ -8,9 +8,12 @@ import dev.langchain4j.store.memory.chat.ChatMemoryStore;
import java.util.List;
import java.util.Map;
/**
* 内存持久化
*/
public class MemoryPersistentStore implements ChatMemoryStore {
private static Map<Long, List<ChatMessage>> MAP = Maps.newHashMap();
private final static Map<Long, List<ChatMessage>> MAP = Maps.newHashMap();
@Override

24
xservice-service/src/main/java/com/xiang/xservice/ai/pojo/entity/AiSimpleChatMessageDO.java Normal file
View File

@@ -0,0 +1,24 @@
package com.xiang.xservice.ai.pojo.entity;
import com.baomidou.mybatisplus.annotation.TableName;
import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;
import java.time.LocalDateTime;
@Data
@AllArgsConstructor
@NoArgsConstructor
@TableName("ai_simple_chat_message")
@Builder
public class AiSimpleChatMessageDO {
private Long id;
private Long userId;
private Long memoryId;
private String role;
private String message;
private LocalDateTime createTime;
private Integer delFlag;
}

26
xservice-service/src/main/java/com/xiang/xservice/ai/repository/manage/AiSimpleChatMessageManageImpl.java Normal file
View File

@@ -0,0 +1,26 @@
package com.xiang.xservice.ai.repository.manage;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.xiang.xservice.ai.pojo.entity.AiSimpleChatMessageDO;
import com.xiang.xservice.ai.repository.mapper.IAiSimpleChatMessageMapper;
import org.springframework.stereotype.Service;
import java.util.List;
@Service
public class AiSimpleChatMessageManageImpl extends ServiceImpl<IAiSimpleChatMessageMapper, AiSimpleChatMessageDO> implements IAiSimpleChatMessageManage {
@Override
public List<AiSimpleChatMessageDO> getMsgByMemoryId(Long memoryId) {
LambdaQueryWrapper<AiSimpleChatMessageDO> lambdaQueryWrapper = Wrappers.lambdaQuery();
lambdaQueryWrapper.eq(AiSimpleChatMessageDO::getMemoryId, memoryId);
lambdaQueryWrapper.eq(AiSimpleChatMessageDO::getDelFlag, 0);
return baseMapper.selectList(lambdaQueryWrapper);
}
@Override
public boolean deleteByMemoryId(Long memoryId) {
return baseMapper.deleteByMemoryId(memoryId) > 0;
}
}

11
xservice-service/src/main/java/com/xiang/xservice/ai/repository/manage/IAiSimpleChatMessageManage.java Normal file
View File

@@ -0,0 +1,11 @@
package com.xiang.xservice.ai.repository.manage;
import com.baomidou.mybatisplus.extension.service.IService;
import com.xiang.xservice.ai.pojo.entity.AiSimpleChatMessageDO;
import java.util.List;
public interface IAiSimpleChatMessageManage extends IService<AiSimpleChatMessageDO> {
List<AiSimpleChatMessageDO> getMsgByMemoryId(Long memoryId);
boolean deleteByMemoryId(Long memoryId);
}

14
xservice-service/src/main/java/com/xiang/xservice/ai/repository/mapper/IAiSimpleChatMessageMapper.java Normal file
View File

@@ -0,0 +1,14 @@
package com.xiang.xservice.ai.repository.mapper;
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.xiang.xservice.ai.pojo.entity.AiSimpleChatMessageDO;
import org.apache.ibatis.annotations.Mapper;
import org.apache.ibatis.annotations.Param;
import org.springframework.stereotype.Repository;
@Mapper
@Repository
public interface IAiSimpleChatMessageMapper extends BaseMapper<AiSimpleChatMessageDO> {
int deleteByMemoryId(@Param("id") Long memoryId);
}

30
xservice-service/src/main/resources/mapper/AiSimpleChatMessageMapper.xml Normal file
View File

@@ -0,0 +1,30 @@
<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
"http://mybatis.org/dtd/mybatis-3-mapper.dtd">
<mapper namespace="com.xiang.dao.AiSimpleChatMessageMapper">
<resultMap id="BaseResultMap" type="com.xiang.xservice.ai.pojo.entity.AiSimpleChatMessageDO">
<result column="id" property="id"/>
<result column="memory_id" property="memoryId"/>
<result column="role" property="role"/>
<result column="message" property="message"/>
<result column="create_time" property="createTime"/>
<result column="user_id" property="userId"/>
<result column="del_flag" property="delFlag"/>
</resultMap>
<sql id="Base_Column_List">
id,
memory_id,
role,
message,
create_time,
user_id,
del_flag
</sql>
<update id="deleteByMemoryId">
update ai_simple_chat_message set del_flag = 1 where memory_id = #{id}
</update>
</mapper>