feat: Phase5B 对话历史持久化
新增 ConversationPersistence: - 退出REPL时自动保存对话到 ~/.claude-code-java/conversations/ - JSON格式存储：系统消息、用户消息、助手消息（含工具调用）、工具响应 - 支持保存、加载最近对话、列出所有对话 - 文件名格式: yyyyMMdd_HHmmss_摘要.json 新增 /history 命令: - 列出最近10条保存的对话记录 - 显示时间、摘要、消息数、工作目录 ReplSession 更新: - 自动记录对话摘要（第一次用户输入前40字） - JLine模式和Scanner模式退出时都保存对话 Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
This commit is contained in:
liuzh
@@ -0,0 +1,246 @@
|
||||
package com.claudecode.core;
|
||||
|
||||
import com.fasterxml.jackson.annotation.JsonTypeInfo;
|
||||
import com.fasterxml.jackson.core.type.TypeReference;
|
||||
import com.fasterxml.jackson.databind.ObjectMapper;
|
||||
import com.fasterxml.jackson.databind.SerializationFeature;
|
||||
import com.fasterxml.jackson.databind.jsontype.BasicPolymorphicTypeValidator;
|
||||
import org.slf4j.Logger;
|
||||
import org.slf4j.LoggerFactory;
|
||||
import org.springframework.ai.chat.messages.*;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.nio.file.Files;
|
||||
import java.nio.file.Path;
|
||||
import java.time.LocalDateTime;
|
||||
import java.time.format.DateTimeFormatter;
|
||||
import java.util.*;
|
||||
import java.util.stream.Stream;
|
||||
|
||||
/**
|
||||
* 对话历史持久化 —— 对应 claude-code 的会话持久化机制。
|
||||
* <p>
|
||||
* 将对话消息序列化为 JSON 存储到 ~/.claude-code-java/conversations/ 目录。
|
||||
* 支持保存、加载和列出历史对话。
|
||||
* <p>
|
||||
* 存储格式：每条消息序列化为简化的 JSON 对象，包含角色、内容和工具调用信息。
|
||||
*/
|
||||
public class ConversationPersistence {
|
||||
|
||||
private static final Logger log = LoggerFactory.getLogger(ConversationPersistence.class);
|
||||
private static final ObjectMapper MAPPER = new ObjectMapper()
|
||||
.enable(SerializationFeature.INDENT_OUTPUT);
|
||||
private static final DateTimeFormatter FILE_DATE_FMT = DateTimeFormatter.ofPattern("yyyyMMdd_HHmmss");
|
||||
|
||||
private final Path conversationsDir;
|
||||
|
||||
public ConversationPersistence() {
|
||||
this(Path.of(System.getProperty("user.home"), ".claude-code-java", "conversations"));
|
||||
}
|
||||
|
||||
public ConversationPersistence(Path conversationsDir) {
|
||||
this.conversationsDir = conversationsDir;
|
||||
try {
|
||||
Files.createDirectories(conversationsDir);
|
||||
} catch (IOException e) {
|
||||
log.warn("无法创建对话存储目录: {}", e.getMessage());
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 保存对话历史到文件。
|
||||
*
|
||||
* @param messages 消息列表
|
||||
* @param summary 对话摘要（用于文件名/列表显示）
|
||||
* @return 保存的文件路径，失败返回 null
|
||||
*/
|
||||
public Path save(List<Message> messages, String summary) {
|
||||
if (messages == null || messages.isEmpty()) return null;
|
||||
|
||||
String timestamp = FILE_DATE_FMT.format(LocalDateTime.now());
|
||||
String safeSummary = sanitizeFilename(summary);
|
||||
String filename = timestamp + "_" + safeSummary + ".json";
|
||||
Path file = conversationsDir.resolve(filename);
|
||||
|
||||
try {
|
||||
List<MessageRecord> records = messages.stream()
|
||||
.map(this::toRecord)
|
||||
.filter(Objects::nonNull)
|
||||
.toList();
|
||||
|
||||
ConversationFile conv = new ConversationFile(
|
||||
LocalDateTime.now().toString(),
|
||||
summary,
|
||||
System.getProperty("user.dir"),
|
||||
records
|
||||
);
|
||||
|
||||
MAPPER.writeValue(file.toFile(), conv);
|
||||
log.info("对话已保存: {}", file.getFileName());
|
||||
return file;
|
||||
} catch (IOException e) {
|
||||
log.error("保存对话失败: {}", e.getMessage());
|
||||
return null;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 加载最近一次对话历史。
|
||||
*
|
||||
* @return 消息列表，无历史时返回空列表
|
||||
*/
|
||||
public List<Message> loadLatest() {
|
||||
Path latest = findLatestFile();
|
||||
if (latest == null) return List.of();
|
||||
return loadFromFile(latest);
|
||||
}
|
||||
|
||||
/**
|
||||
* 从指定文件加载对话历史。
|
||||
*/
|
||||
public List<Message> loadFromFile(Path file) {
|
||||
try {
|
||||
ConversationFile conv = MAPPER.readValue(file.toFile(), ConversationFile.class);
|
||||
return conv.messages().stream()
|
||||
.map(this::fromRecord)
|
||||
.filter(Objects::nonNull)
|
||||
.toList();
|
||||
} catch (IOException e) {
|
||||
log.error("加载对话失败: {}", e.getMessage());
|
||||
return List.of();
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 列出所有保存的对话（按时间倒序）。
|
||||
*/
|
||||
public List<ConversationSummary> listConversations() {
|
||||
List<ConversationSummary> summaries = new ArrayList<>();
|
||||
|
||||
try (Stream<Path> paths = Files.list(conversationsDir)) {
|
||||
paths.filter(p -> p.toString().endsWith(".json"))
|
||||
.sorted(Comparator.reverseOrder())
|
||||
.forEach(file -> {
|
||||
try {
|
||||
ConversationFile conv = MAPPER.readValue(file.toFile(), ConversationFile.class);
|
||||
summaries.add(new ConversationSummary(
|
||||
file.getFileName().toString(),
|
||||
conv.summary(),
|
||||
conv.savedAt(),
|
||||
conv.workingDir(),
|
||||
conv.messages().size()
|
||||
));
|
||||
} catch (IOException e) {
|
||||
log.debug("跳过无效对话文件: {}", file.getFileName());
|
||||
}
|
||||
});
|
||||
} catch (IOException e) {
|
||||
log.warn("列出对话失败: {}", e.getMessage());
|
||||
}
|
||||
|
||||
return summaries;
|
||||
}
|
||||
|
||||
/** 获取对话存储目录 */
|
||||
public Path getConversationsDir() {
|
||||
return conversationsDir;
|
||||
}
|
||||
|
||||
// ==================== 序列化/反序列化 ====================
|
||||
|
||||
private MessageRecord toRecord(Message msg) {
|
||||
return switch (msg) {
|
||||
case SystemMessage sm -> new MessageRecord("system", sm.getText(), null, null);
|
||||
case UserMessage um -> new MessageRecord("user", um.getText(), null, null);
|
||||
case AssistantMessage am -> {
|
||||
List<ToolCallRecord> toolCalls = null;
|
||||
if (am.hasToolCalls()) {
|
||||
toolCalls = am.getToolCalls().stream()
|
||||
.map(tc -> new ToolCallRecord(tc.id(), tc.name(), tc.arguments()))
|
||||
.toList();
|
||||
}
|
||||
yield new MessageRecord("assistant", am.getText(), toolCalls, null);
|
||||
}
|
||||
case ToolResponseMessage trm -> {
|
||||
List<ToolResponseRecord> responses = trm.getResponses().stream()
|
||||
.map(tr -> new ToolResponseRecord(tr.id(), tr.name(), tr.responseData()))
|
||||
.toList();
|
||||
yield new MessageRecord("tool_response", null, null, responses);
|
||||
}
|
||||
default -> null;
|
||||
};
|
||||
}
|
||||
|
||||
private Message fromRecord(MessageRecord record) {
|
||||
return switch (record.role()) {
|
||||
case "system" -> new SystemMessage(record.content() != null ? record.content() : "");
|
||||
case "user" -> new UserMessage(record.content() != null ? record.content() : "");
|
||||
case "assistant" -> {
|
||||
if (record.toolCalls() != null && !record.toolCalls().isEmpty()) {
|
||||
List<AssistantMessage.ToolCall> toolCalls = record.toolCalls().stream()
|
||||
.map(tc -> new AssistantMessage.ToolCall(tc.id(), "function", tc.name(), tc.arguments()))
|
||||
.toList();
|
||||
yield AssistantMessage.builder()
|
||||
.content(record.content() != null ? record.content() : "")
|
||||
.toolCalls(toolCalls)
|
||||
.build();
|
||||
}
|
||||
yield new AssistantMessage(record.content() != null ? record.content() : "");
|
||||
}
|
||||
case "tool_response" -> {
|
||||
if (record.toolResponses() != null) {
|
||||
List<ToolResponseMessage.ToolResponse> responses = record.toolResponses().stream()
|
||||
.map(tr -> new ToolResponseMessage.ToolResponse(tr.id(), tr.name(), tr.data()))
|
||||
.toList();
|
||||
yield ToolResponseMessage.builder().responses(responses).build();
|
||||
}
|
||||
yield null;
|
||||
}
|
||||
default -> null;
|
||||
};
|
||||
}
|
||||
|
||||
private Path findLatestFile() {
|
||||
try (Stream<Path> paths = Files.list(conversationsDir)) {
|
||||
return paths.filter(p -> p.toString().endsWith(".json"))
|
||||
.max(Comparator.naturalOrder())
|
||||
.orElse(null);
|
||||
} catch (IOException e) {
|
||||
return null;
|
||||
}
|
||||
}
|
||||
|
||||
private String sanitizeFilename(String name) {
|
||||
if (name == null || name.isBlank()) return "conversation";
|
||||
return name.replaceAll("[^a-zA-Z0-9\\u4e00-\\u9fa5_-]", "_")
|
||||
.substring(0, Math.min(name.length(), 40));
|
||||
}
|
||||
|
||||
// ==================== JSON 数据结构 ====================
|
||||
|
||||
public record ConversationFile(
|
||||
String savedAt,
|
||||
String summary,
|
||||
String workingDir,
|
||||
List<MessageRecord> messages
|
||||
) {}
|
||||
|
||||
public record MessageRecord(
|
||||
String role,
|
||||
String content,
|
||||
List<ToolCallRecord> toolCalls,
|
||||
List<ToolResponseRecord> toolResponses
|
||||
) {}
|
||||
|
||||
public record ToolCallRecord(String id, String name, String arguments) {}
|
||||
public record ToolResponseRecord(String id, String name, String data) {}
|
||||
|
||||
/** 对话摘要（用于列表显示） */
|
||||
public record ConversationSummary(
|
||||
String filename,
|
||||
String summary,
|
||||
String savedAt,
|
||||
String workingDir,
|
||||
int messageCount
|
||||
) {}
|
||||
}
|
||||
Reference in New Issue
Block a user