Function BuildSubAgentMiddlewareStack

pkg/middleware/subagent.go:558–593

BuildSubAgentMiddlewareStack 构建子代理的中间件栈 根据子代理规格决定是否继承父代理中间件

(spec SubAgentSpec, parentMiddlewares []Middleware)

// BuildSubAgentMiddlewareStack 构建子代理的中间件栈
// 根据子代理规格决定是否继承父代理中间件
func BuildSubAgentMiddlewareStack(spec SubAgentSpec, parentMiddlewares []Middleware) []Middleware {
	if !spec.InheritMiddlewares {
		// 不继承,只使用子代理专用中间件
		return spec.MiddlewareOverrides
	}

	// 继承父代理中间件
	stack := make([]Middleware, 0, len(parentMiddlewares)+len(spec.MiddlewareOverrides))

	// 1. 首先添加父代理中间件
	stack = append(stack, parentMiddlewares...)

	// 2. 然后添加子代理专用中间件(覆盖或追加)
	if len(spec.MiddlewareOverrides) > 0 {
		// 创建名称映射用于覆盖检测
		nameMap := make(map[string]int)
		for i, m := range stack {
			nameMap[m.Name()] = i
		}

		// 处理覆盖和追加
		for _, override := range spec.MiddlewareOverrides {
			if idx, exists := nameMap[override.Name()]; exists {
				// 覆盖同名中间件
				stack[idx] = override
				saLog.Debug(context.Background(), "override middleware", map[string]any{"name": override.Name()})
			} else {
				// 追加新中间件
				stack = append(stack, override)
				saLog.Debug(context.Background(), "append middleware", map[string]any{"name": override.Name()})
			}
		}
	}

	return stack
}

// GetMiddlewareForSubAgent 获取子代理应该使用的中间件栈
// 这是一个辅助函数,供 SubAgentFactory 实现使用