fix(scheduler): 补齐 M3 与设计稿的两处语义偏离(发现即修)

两处都不是风格差异,而是真的偏离设计语义(其一为回归),
已按“先写判据确认失败、再修”的方式处理,判据保留为回归测试。

1. 空闲时到达的中断永远不会被处理(设计 §5.1 ③ 未落地)
   调度器空闲时只阻塞在 select{InputChan, selfInputCh, ctx.Done},
   而 pendingInterrupts 不是 channel——interceptLoop 把中断入队后
   没有任何东西唤醒调度器,中断要等“下一条输入”才被看到。
   修复:调度器加 wake channel,enqueueInterrupt 非阻塞 signalWake,
   空闲分支增加 wake 分支。

2. 临界区内只“不让位”却仍被“取消”(设计 §4.3/§5.2)
   requestPreempt 不判临界区,interceptLoop 照常 cancelLLM,
   于是正在流式的记忆整理被中断,stepLLM 以 error 提前结束——
   整理任务被砍掉一半,而设计要求的是“请求排队等它结束”。
   修复:scheduler 增加原子 critical 标志(帧仍只由调度器读写),
   runInputTask 在 prepare 后设置、结束(含挂起)时清除;
   requestPreempt 在临界区内不 arm、不取消,中断只入队。

- 新增 scheduler_regression_test.go 2 项(先失败后通过)
- 验收:agent 全量 + -race;全仓 vet 通过
This commit is contained in:
JianFeeeee
2026-09-13 06:10:44 +08:00
parent 32f801a775
commit e999c1d95d
3 changed files with 139 additions and 3 deletions

View File

@ -0,0 +1,101 @@
package core
// 回归判据:M3 实现与设计稿 §5.1/§4.3 的两处偏离。
//
// 这两条是**先写判据、确认失败、再修**的(修完保留为回归测试)。
import (
"context"
"sync"
"sync/atomic"
"testing"
"time"
agentAPI "gitcode.com/JianFeeeee/HomeAgent/internal/agent/api"
agentIO "gitcode.com/JianFeeeee/HomeAgent/internal/agent/io"
)
// cancelAwareProvider 第一次调用进入后阻塞,直到 ctx 取消;记录是否被取消。
type cancelAwareProvider struct {
once sync.Once
entered chan struct{}
canceled atomic.Bool
}
func newCancelAwareProvider() *cancelAwareProvider {
return &cancelAwareProvider{entered: make(chan struct{})}
}
func (p *cancelAwareProvider) Name() string { return "cancel-aware" }
func (p *cancelAwareProvider) Chat(ctx context.Context, req *agentAPI.CompletionRequest) (*agentAPI.CompletionResponse, error) {
p.once.Do(func() { close(p.entered) })
<-ctx.Done()
p.canceled.Store(true)
return nil, ctx.Err()
}
func (p *cancelAwareProvider) ChatStream(ctx context.Context, req *agentAPI.CompletionRequest) (<-chan agentAPI.StreamChunk, error) {
return nil, context.Canceled
}
func (p *cancelAwareProvider) MaxContextTokens() int { return 8192 }
// 设计 §5.1 ③:interruptLoop 定级/决策后必须**唤醒调度器**。
// 现状:中断只被放进 pendingInterrupts,而调度器空闲时阻塞在 select(只看
// InputChan/selfInputCh/ctx.Done)——没有任何东西会把它叫醒。
func TestGap_IdleInterruptIsProcessed(t *testing.T) {
sp := &countingProvider{}
a := New(AgentConfig{
ID: "idle-intr",
Provider: sp,
ProviderManager: agentAPI.NewProviderManager(),
IO: agentIO.NewIOManager(),
StageHost: NewStageHost(),
})
a.Start()
defer a.Stop()
// 完全空闲时投递一条中断(模拟定时器通知/插件提醒)。
a.io.InjectInterruptText("qq", "cli", "空闲时的通知")
deadline := time.Now().Add(3 * time.Second)
for sp.n.Load() == 0 && time.Now().Before(deadline) {
time.Sleep(20 * time.Millisecond)
}
if sp.n.Load() == 0 {
snap := a.DumpScheduler()
t.Fatalf("空闲时到达的中断未被处理:pendingInterrupts=%d(调度器未被唤醒)",
len(snap.PendingInterrupts))
}
}
// 设计 §4.3/§5.2:`_consolidation_` 是整任务临界区——抢占请求必须排队等它结束,
// 而不是取消它。现状:requestPreempt 不判临界区,interceptLoop 照常 cancelLLM,
// 于是正在流式的记忆整理被中断 → stepLLM 直接以 error 结束(整理丢一半)。
func TestGap_ConsolidationMustNotBeCancelled(t *testing.T) {
sp := newCancelAwareProvider()
a := New(AgentConfig{
ID: "consol",
Provider: sp,
ProviderManager: agentAPI.NewProviderManager(),
IO: agentIO.NewIOManager(),
StageHost: NewStageHost(),
})
a.Start()
defer a.Stop()
// 走自循环通道发起一次记忆整理。
a.selfInputCh <- selfInputMsg{text: "合并实体", channel: channelConsolidation}
select {
case <-sp.entered:
case <-time.After(3 * time.Second):
t.Fatal("记忆整理未进入 LLM 调用")
}
// L4 中断到达。
a.io.InjectInterruptText("cli", "cli", "L4 打断")
time.Sleep(500 * time.Millisecond)
if sp.canceled.Load() {
t.Fatal("记忆整理是临界区,其 LLM 不该被取消(设计 §4.3/§5.2)")
}
}