fix: add thread safety for concurrent state modifications

- Add stateLock to synchronize all getState/setState read-modify-write
  operations across UI thread and AgentLoop background threads
- Synchronize onInput, onPaste, addMessage, appendToStreamingMessage,
  finishStreamingMessage, completeLastToolCall, setThinking, runAgent
- Extract addMessageInternal() to avoid double-locking
- Harden extractToolSummary with explicit indexOf checks

Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
pull/1/head
abel533 1 month ago
parent 75ecaeca58
commit b72fcfea79
  1. 35
      src/main/java/com/claudecode/tui/ClaudeCodeComponent.java

@ -71,6 +71,7 @@ public class ClaudeCodeComponent extends Component<ClaudeCodeComponent.TuiState>
private final Runnable onExit;
// --- 内部状态 ---
private final Object stateLock = new Object(); // 保护 getState/setState 的读-改-写操作
private final List<String> inputHistory = new ArrayList<>();
private int historyIndex = -1;
private String savedInput = "";
@ -441,6 +442,7 @@ public class ClaudeCodeComponent extends Component<ClaudeCodeComponent.TuiState>
@Override
public void onInput(String input, Key key) {
synchronized (stateLock) {
TuiState s = getState();
// Ctrl+D: 退出
@ -453,7 +455,7 @@ public class ClaudeCodeComponent extends Component<ClaudeCodeComponent.TuiState>
if (key.ctrl() && "c".equals(input)) {
if (agentRunning.get()) {
// TODO: 中断 Agent 运行
addMessage(new SystemMsg("^C (interrupt)", Color.BRIGHT_YELLOW));
addMessageInternal(new SystemMsg("^C (interrupt)", Color.BRIGHT_YELLOW), s);
} else {
setState(new TuiState("", s.messages, s.scrollOffset, false, ""));
}
@ -506,14 +508,17 @@ public class ClaudeCodeComponent extends Component<ClaudeCodeComponent.TuiState>
setState(new TuiState(s.inputText + input, s.messages, s.scrollOffset, false, ""));
}
}
}
@Override
public void onPaste(String text) {
synchronized (stateLock) {
if (agentRunning.get() || text == null || text.isEmpty()) return;
TuiState s = getState();
abandonHistoryPreview();
setState(new TuiState(s.inputText + text, s.messages, s.scrollOffset, false, ""));
}
}
/** 处理权限确认输入 */
private void handlePermissionInput(String input, Key key, TuiState s) {
@ -605,9 +610,11 @@ public class ClaudeCodeComponent extends Component<ClaudeCodeComponent.TuiState>
addMessage(new SystemMsg("Error: " + e.getMessage(), Color.BRIGHT_RED));
} finally {
agentRunning.set(false);
synchronized (stateLock) {
TuiState cs = getState();
setState(new TuiState(cs.inputText, cs.messages, 0, false, ""));
}
}
});
}
@ -615,7 +622,13 @@ public class ClaudeCodeComponent extends Component<ClaudeCodeComponent.TuiState>
/** 添加一条消息 */
public void addMessage(UIMessage msg) {
TuiState s = getState();
synchronized (stateLock) {
addMessageInternal(msg, getState());
}
}
/** 内部添加消息(调用方需持有 stateLock) */
private void addMessageInternal(UIMessage msg, TuiState s) {
List<UIMessage> newMsgs = new ArrayList<>(s.messages);
newMsgs.add(msg);
setState(new TuiState(s.inputText, Collections.unmodifiableList(newMsgs),
@ -624,6 +637,7 @@ public class ClaudeCodeComponent extends Component<ClaudeCodeComponent.TuiState>
/** 追加 token 到当前流式助手消息 */
private void appendToStreamingMessage(String token) {
synchronized (stateLock) {
TuiState s = getState();
List<UIMessage> msgs = new ArrayList<>(s.messages);
@ -637,9 +651,11 @@ public class ClaudeCodeComponent extends Component<ClaudeCodeComponent.TuiState>
setState(new TuiState(s.inputText, Collections.unmodifiableList(msgs),
0, s.thinking, s.thinkingText));
}
}
/** 完成当前流式消息(公开给 JinkReplSession 使用) */
public void finishStreamingMessage() {
synchronized (stateLock) {
TuiState s = getState();
List<UIMessage> msgs = new ArrayList<>(s.messages);
@ -649,9 +665,11 @@ public class ClaudeCodeComponent extends Component<ClaudeCodeComponent.TuiState>
0, s.thinking, s.thinkingText));
}
}
}
/** 更新最后一个工具调用消息的结果 */
public void completeLastToolCall(String result) {
synchronized (stateLock) {
TuiState s = getState();
List<UIMessage> msgs = new ArrayList<>(s.messages);
@ -665,6 +683,7 @@ public class ClaudeCodeComponent extends Component<ClaudeCodeComponent.TuiState>
setState(new TuiState(s.inputText, Collections.unmodifiableList(msgs),
s.scrollOffset, s.thinking, s.thinkingText));
}
}
/** 设置权限确认回调 */
public void requestPermission(Consumer<String> callback) {
@ -673,9 +692,11 @@ public class ClaudeCodeComponent extends Component<ClaudeCodeComponent.TuiState>
/** 设置 thinking 状态 */
public void setThinking(boolean thinking, String text) {
synchronized (stateLock) {
TuiState s = getState();
setState(new TuiState(s.inputText, s.messages, s.scrollOffset, thinking, text));
}
}
/** 设置首次用户输入回调 */
public void setOnFirstUserInput(Consumer<String> callback) {
@ -747,11 +768,13 @@ public class ClaudeCodeComponent extends Component<ClaudeCodeComponent.TuiState>
String[] keys = {"command", "file_path", "pattern", "query", "url"};
for (String key : keys) {
String search = "\"" + key + "\"";
if (args.contains(search)) {
int start = args.indexOf(search);
int valStart = args.indexOf("\"", start + search.length()) + 1;
if (start < 0) continue;
int colonPos = args.indexOf("\"", start + search.length());
if (colonPos < 0) continue;
int valStart = colonPos + 1;
int valEnd = args.indexOf("\"", valStart);
if (valStart > 0 && valEnd > valStart) {
if (valEnd < 0 || valEnd <= valStart) continue;
String val = args.substring(valStart, Math.min(valEnd, valStart + 60));
return switch (key) {
case "command" -> "$ " + val;
@ -760,8 +783,6 @@ public class ClaudeCodeComponent extends Component<ClaudeCodeComponent.TuiState>
default -> val;
};
}
}
}
} catch (Exception ignored) {}
return null;
}

Loading…
Cancel
Save