Blizzard
16a6c4b1aa
专用「审批」节点方案，全栈打通。 后端： - contract：TaskWaiting/TaskRejected 状态 + ApprovalSubject/ApprovalDecision。 - bus：PublishApproval + WaitApproval（订阅决定主题，带超时）；消费者 AckWait 30s→15min（阻塞等人审期间消息未 ack，否则重投成重复任务）。 - dispatcher：ApprovalWaiter 接口 + approvalNode——执行到审批节点发 await 事件 + 置 waiting，阻塞等决定。批准→回 running 放行下游；拒绝/超时→errRejected 哨兵→ 剪下游→rejected，优雅收尾不计熔断。超时安全默认拒绝。 taskExecTimeout 3→10min（审批5 < 执行10 < AckWait15）。 - 网关：POST /tasks/:id/approve（仅 waiting 态受理，幂等）。 桌面端： - Studio 新增「人工审批」节点（nodeCatalog，可填标题/说明）。 - run.ts pendingApproval() 从 exec 流派生待审中断 + waiting 节点状态。 - BottomDrawer ApprovalBar：琥珀审批条（摘要 + 批准/拒绝 + 备注，调 api.approveTask）。 - ExecTrace waiting 状态灯。 验证：后端 curl 实测 waiting→批准→running→done；waiting→拒绝→rejected（下游未跑）。 前端 tsc + vitest 36 过（pendingApproval 4 例 + waiting 状态）。全模块 build+vet+test 全绿。 EINO_ADOPTION Phase D 标记 HITL 完成。 未覆盖：compose 路径（EINO_COMPOSE，默认关）的 approval 节点；桌面端审批条未在 GUI 实点。 Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
105 lines
3.9 KiB
Go
105 lines
3.9 KiB
Go
// Package nats 是调度器对共享 bus 的薄封装（消费任务 / 回写 Token）。
|
||
package nats
|
||
|
||
import (
|
||
"context"
|
||
"log"
|
||
"time"
|
||
|
||
sharedbus "github.com/sundynix/sundynix-shared/bus"
|
||
"github.com/sundynix/sundynix-shared/contract"
|
||
)
|
||
|
||
// TaskHandler 处理单个任务。
|
||
type TaskHandler func(ctx context.Context, t *contract.Task) error
|
||
|
||
// Subscriber 包装共享 bus，向调度器暴露消费能力。
|
||
type Subscriber struct {
|
||
inner *sharedbus.Bus
|
||
}
|
||
|
||
// MustConnect 接入 NATS 并确保任务流存在（消费者声明在 Consume 时完成）。
|
||
func MustConnect(url string) *Subscriber {
|
||
inner, err := sharedbus.Connect(url)
|
||
if err != nil {
|
||
log.Fatalf("[dispatcher/nats] connect: %v", err)
|
||
}
|
||
if err := inner.EnsureTaskStream(context.Background()); err != nil {
|
||
log.Fatalf("[dispatcher/nats] ensure stream: %v", err)
|
||
}
|
||
log.Printf("[dispatcher/nats] connected %s", url)
|
||
return &Subscriber{inner: inner}
|
||
}
|
||
|
||
// ConsumeTasks 从 sundynix.tasks.* 持续消费任务（队列组负载均衡），阻塞至 ctx 取消。
|
||
func (s *Subscriber) ConsumeTasks(ctx context.Context, h TaskHandler) error {
|
||
stop, err := s.inner.ConsumeTasks(ctx, func(c context.Context, t *contract.Task) error {
|
||
return h(c, t)
|
||
})
|
||
if err != nil {
|
||
return err
|
||
}
|
||
defer stop()
|
||
<-ctx.Done()
|
||
return ctx.Err()
|
||
}
|
||
|
||
// PublishToken / CompleteStream 让 Subscriber 满足 eino.TokenSink，
|
||
// 把推理 Token 回流到 sundynix.streams.<taskID>。
|
||
func (s *Subscriber) PublishToken(taskID string, token []byte) error {
|
||
return s.inner.PublishToken(taskID, token)
|
||
}
|
||
|
||
func (s *Subscriber) CompleteStream(taskID string) error {
|
||
return s.inner.CompleteStream(taskID)
|
||
}
|
||
|
||
// PublishExec / CompleteExec 让 Subscriber 满足 eino.ExecSink，
|
||
// 把执行轨迹事件回流到 sundynix.exec.<taskID>（与 Token 流分开）。
|
||
func (s *Subscriber) PublishExec(taskID string, data []byte) error {
|
||
return s.inner.PublishExec(taskID, data)
|
||
}
|
||
|
||
func (s *Subscriber) CompleteExec(taskID string) error {
|
||
return s.inner.CompleteExec(taskID)
|
||
}
|
||
|
||
// CallTool 让 Subscriber 满足 eino.ToolCaller，经 NATS request-reply 调起第 5 层 MCP 工具。
|
||
func (s *Subscriber) CallTool(ctx context.Context, subject string, call *contract.ToolCall) (*contract.ToolResult, error) {
|
||
return s.inner.CallTool(ctx, subject, call)
|
||
}
|
||
|
||
// ServeHealth 在 dispatcher 心跳主题上应答探活，让管理端「服务状态」判定其在线。
|
||
func (s *Subscriber) ServeHealth(provide func() []byte) (func() error, error) {
|
||
return s.inner.ServeHealth(contract.SubjectHealthDispatcher, provide)
|
||
}
|
||
|
||
// PublishTaskStatus 让 Subscriber 满足 eino.StatusSink，把任务状态流转回写给网关。
|
||
func (s *Subscriber) PublishTaskStatus(taskID, status, detail string) error {
|
||
return s.inner.PublishTaskStatus(&contract.TaskStatusEvent{
|
||
TaskID: taskID, Status: status, Detail: detail, TS: time.Now().UnixMilli(),
|
||
})
|
||
}
|
||
|
||
// WaitApproval 让 Subscriber 满足 eino.ApprovalWaiter，阻塞等待审批节点的人工决定。
|
||
func (s *Subscriber) WaitApproval(ctx context.Context, taskID string, timeout time.Duration) (*contract.ApprovalDecision, error) {
|
||
return s.inner.WaitApproval(ctx, taskID, timeout)
|
||
}
|
||
|
||
// RequestModelConfig 向控制面（Gateway）取当前激活的对话模型配置。
|
||
func (s *Subscriber) RequestModelConfig(ctx context.Context) (*contract.ModelConfig, error) {
|
||
return s.inner.RequestConfig(ctx, contract.ConfigKindChat)
|
||
}
|
||
|
||
// SubscribeModelConfigUpdated 订阅对话模型配置热更新。
|
||
func (s *Subscriber) SubscribeModelConfigUpdated(onUpdate func(*contract.ModelConfig)) (func() error, error) {
|
||
return s.inner.SubscribeConfigUpdated(contract.ConfigKindChat, onUpdate)
|
||
}
|
||
|
||
// FetchModelConfigWithRetry 后台重试拉取初始对话模型配置（容忍 gateway 晚于 dispatcher 启动）。
|
||
func (s *Subscriber) FetchModelConfigWithRetry(ctx context.Context, apply func(*contract.ModelConfig)) {
|
||
s.inner.RequestConfigWithRetry(ctx, contract.ConfigKindChat, apply)
|
||
}
|
||
|
||
func (s *Subscriber) Close() { s.inner.Close() }
|