mirror of
https://gitcode.com/JianFeeeee/HomeAgent.git
synced 2026-09-22 18:08:04 +00:00
ec13eb391a
线上验证「内核开、两个插件接」时发现的真实缺陷：`echo`、`ls` 这类在首个 200ms ticker 之前就结束的短命令，readLoop 走到 `!terminalRunning(t)` 分支 直接 return，残留在 stream 里的输出从未 flush。 症状：输出只留在 session.buf 里——agent 用 terminal_read 能看到，但 terminal_output 事件永远发不出去，于是内核权威视图（以及 WebUI/CLI 的 /terminals）的 output 恒为空。实测 term_2（echo HELLO_KERNEL_REGISTRY） 在 /terminals 里 output="" 而 agent 同期 terminal_read 拿到了正文。 修法： - 抽出 flushTermStream(s, t)，ticker 与所有退出路径共用同一条推送路径 （避免以后再出现「某条退出路径忘了 flush」）。 - readLoop 顶部加 defer：defer flushTermStream 后于 defer emitTermState 声明 → LIFO 下先 flush 再报停止，保证「最后一段输出」先于 running=false 到达订阅者。 测试：plugin_flush_test.go 新增 TestReadLoopFlushesOutputOnExit——走 EventBus 捕获事件，推入输出后立即让进程退出（远早于 ticker），断言输出 已补推且末态 running=false。已验证去掉修复即 FAIL、加回即 PASS。 注：handleRead(clear=true) 会主动 Reset stream（避免与读取结果重复）， 属既有设计；本修复针对的是「未被读取就退出」的路径。
96 lines
2.4 KiB
Go
96 lines
2.4 KiB
Go
//go:build linux || windows
|
||
|
||
package agentcli
|
||
|
||
import (
|
||
"sync"
|
||
"testing"
|
||
"time"
|
||
|
||
"gitcode.com/JianFeeeee/HomeAgent/internal/events"
|
||
sdk "gitcode.com/JianFeeeee/HomeAgent/internal/sdk"
|
||
)
|
||
|
||
// eventCapture 订阅事件总线，收下 terminal_output 事件供断言。
|
||
type eventCapture struct {
|
||
mu sync.Mutex
|
||
outs []map[string]interface{}
|
||
}
|
||
|
||
func (c *eventCapture) add(ev *events.Event) {
|
||
c.mu.Lock()
|
||
c.outs = append(c.outs, ev.Payload)
|
||
c.mu.Unlock()
|
||
}
|
||
|
||
// outputFor 汇总某个终端已发布的全部 output 片段。
|
||
func (c *eventCapture) outputFor(id string) string {
|
||
c.mu.Lock()
|
||
defer c.mu.Unlock()
|
||
var s string
|
||
for _, p := range c.outs {
|
||
if p["terminal_id"] == id {
|
||
if o, _ := p["output"].(string); o != "" {
|
||
s += o
|
||
}
|
||
}
|
||
}
|
||
return s
|
||
}
|
||
|
||
func (c *eventCapture) statesFor(id string) []bool {
|
||
c.mu.Lock()
|
||
defer c.mu.Unlock()
|
||
var out []bool
|
||
for _, p := range c.outs {
|
||
if p["terminal_id"] == id {
|
||
r, _ := p["running"].(bool)
|
||
out = append(out, r)
|
||
}
|
||
}
|
||
return out
|
||
}
|
||
|
||
// 短命令（在首个 200ms ticker 之前就结束）的输出必须在退出时补推，
|
||
// 否则只留在 buf 里、永远发不出 terminal_output 事件，
|
||
// 内核权威视图（以及 WebUI/CLI 的 /terminals）output 恒为空。
|
||
func TestReadLoopFlushesOutputOnExit(t *testing.T) {
|
||
p := New("agentcli")
|
||
bus := events.NewBus()
|
||
capture := &eventCapture{}
|
||
bus.Subscribe(events.EventTerminalOutput, capture.add)
|
||
|
||
sdkInst := sdk.New("agentcli", sdk.SDKConfig{
|
||
RegTool: func(string, sdk.ToolDef, sdk.ToolHandler) error { return nil },
|
||
RegStage: func(sdk.Stage, sdk.StageHandler) {},
|
||
RegAPI: func(string) error { return nil },
|
||
Settings: sdk.NewSettings("agentcli", nil),
|
||
EventBus: bus,
|
||
})
|
||
sdkInst.SetIOInjector(&injectCapture{})
|
||
|
||
term := newMockTerm()
|
||
ts := newTestSession(term)
|
||
startReadLoop(p, sdkInst, ts)
|
||
|
||
// 推入输出后立刻让进程退出——远早于 200ms ticker。
|
||
term.push([]byte("HELLO_KERNEL_REGISTRY\n"))
|
||
time.Sleep(50 * time.Millisecond)
|
||
term.setRunning(false)
|
||
|
||
select {
|
||
case <-ts.done:
|
||
case <-time.After(3 * time.Second):
|
||
t.Fatal("readLoop did not exit")
|
||
}
|
||
|
||
if got := capture.outputFor(ts.id); got != "HELLO_KERNEL_REGISTRY\n" {
|
||
t.Fatalf("output on exit = %q, want %q", got, "HELLO_KERNEL_REGISTRY\n")
|
||
}
|
||
// 停止状态也要报到（readLoop 退出时 emitTermState(false)）。
|
||
states := capture.statesFor(ts.id)
|
||
if len(states) == 0 || states[len(states)-1] {
|
||
t.Fatalf("expected trailing running=false, got %v", states)
|
||
}
|
||
}
|