Kai/spring_ai_app
Template
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

消息记录存储添加排序

Browse Source
This commit is contained in:
liukai
2025-09-06 20:53:32 +08:00
parent 9d5c2aaf04
commit 4e84b88326
3 changed files with 14 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
src/main/java/com/ai/app/config/MysqlChatMemoryRepository.java
View File

@@ -13,6 +13,7 @@ import org.springframework.transaction.annotation.Propagation;
import org.springframework.transaction.annotation.Transactional; import org.springframework.transaction.annotation.Transactional;
import java.time.LocalDateTime; import java.time.LocalDateTime;
import java.util.ArrayList;
import java.util.List; import java.util.List;
import java.util.UUID; import java.util.UUID;
import java.util.stream.Collectors; import java.util.stream.Collectors;
@@ -43,8 +44,13 @@ public class MysqlChatMemoryRepository implements ChatMemoryRepository {
@Transactional(propagation = Propagation.REQUIRED, rollbackFor = Exception.class) @Transactional(propagation = Propagation.REQUIRED, rollbackFor = Exception.class)
public void saveAll(String conversationId, List<Message> messages) { public void saveAll(String conversationId, List<Message> messages) {
chatMessageHistoryMapper.deleteByConversationId(conversationId); chatMessageHistoryMapper.deleteByConversationId(conversationId);
List<ChatMessageHistory> messageHistoryList = messages.stream().map(message -> {
ChatMessageHistory chatMessageHistory = new ChatMessageHistory(); ArrayList<ChatMessageHistory> messageHistoryList = new ArrayList<>();
ChatMessageHistory chatMessageHistory;
Message message;
for (int i = 0; i < messages.size(); i++) {
chatMessageHistory = new ChatMessageHistory();
message = messages.get(i);
chatMessageHistory.setId(UUID.randomUUID().toString()); chatMessageHistory.setId(UUID.randomUUID().toString());
chatMessageHistory.setConversationId(conversationId); chatMessageHistory.setConversationId(conversationId);
chatMessageHistory.setMessageType(message.getMessageType().getValue()); chatMessageHistory.setMessageType(message.getMessageType().getValue());
@@ -53,9 +59,10 @@ public class MysqlChatMemoryRepository implements ChatMemoryRepository {
if (message instanceof AssistantMessage am) { if (message instanceof AssistantMessage am) {
chatMessageHistory.setToolCalls(JSON.toJSONString(am.getToolCalls())); chatMessageHistory.setToolCalls(JSON.toJSONString(am.getToolCalls()));
} }
chatMessageHistory.setSort(i);
chatMessageHistory.setCreateTime(LocalDateTime.now()); chatMessageHistory.setCreateTime(LocalDateTime.now());
return chatMessageHistory; messageHistoryList.add(chatMessageHistory);
}).collect(Collectors.toList()); }
chatMessageHistoryMapper.insert(messageHistoryList); chatMessageHistoryMapper.insert(messageHistoryList);
} }

2
src/main/java/com/ai/app/entity/ChatMessageHistory.java
View File

@@ -28,5 +28,7 @@ public class ChatMessageHistory {
private String toolCalls; private String toolCalls;
private int sort;
private LocalDateTime createTime; private LocalDateTime createTime;
} }

2
src/main/java/com/ai/app/mapper/ChatMessageHistoryMapper.java
View File

@@ -14,7 +14,7 @@ import java.util.List;
*/ */
public interface ChatMessageHistoryMapper extends BaseMapper<ChatMessageHistory> { public interface ChatMessageHistoryMapper extends BaseMapper<ChatMessageHistory> {
@Select("select * from chat_message_history where conversation_id = #{conversationId}") @Select("select * from chat_message_history where conversation_id = #{conversationId} order by sort asc")
List<ChatMessageHistory> selectByConversationId(String conversationId); List<ChatMessageHistory> selectByConversationId(String conversationId);
@Delete("delete from chat_message_history where conversation_id = #{conversationId}") @Delete("delete from chat_message_history where conversation_id = #{conversationId}")