diff --git a/manager/src/main/java/org/jiayunet/service/MentorMockChatService.java b/manager/src/main/java/org/jiayunet/service/MentorMockChatService.java index 28c12d0..6e5f030 100644 --- a/manager/src/main/java/org/jiayunet/service/MentorMockChatService.java +++ b/manager/src/main/java/org/jiayunet/service/MentorMockChatService.java @@ -79,12 +79,18 @@ public class MentorMockChatService { throw new RuntimeException("模拟对话生成失败", e); } - // 6. 入库 - MentorMockChat record = new MentorMockChat(); - record.setMentorId(mentorId); - record.setMessages(chatMessages); - record.setCreateTime(Instant.now()); - mentorMockChatMapper.insert(record); + // 6. 入库(存在则更新,不存在则插入) + MentorMockChat existRecord = mentorMockChatMapper.selectOne(new LambdaQueryWrapper().eq(MentorMockChat::getMentorId, mentorId)); + if (existRecord != null) { + existRecord.setMessages(chatMessages); + mentorMockChatMapper.updateById(existRecord); + } else { + MentorMockChat record = new MentorMockChat(); + record.setMentorId(mentorId); + record.setMessages(chatMessages); + record.setCreateTime(Instant.now()); + mentorMockChatMapper.insert(record); + } // 7. 返回 MentorMockChatDto dto = new MentorMockChatDto();