diff --git a/midware_x.go b/midware_x.go index 645fb1d..770273a 100644 --- a/midware_x.go +++ b/midware_x.go @@ -1,89 +1,72 @@ package touka -// UseChainIf 是一个条件中间件包装器。 -// 如果 `condition` 为 true,它将执行提供的 `middlewares` 链。 -// 如果 `condition` 为 false,它会直接跳过这些中间件,调用下一个处理器。 -func (engine *Engine) UseChainIf(condition bool, middlewares ...HandlerFunc) HandlerFunc { - if !condition { +type MiddlewareXFunc func() HandlerFunc + +// UseChainIf 是一个条件中间件包装器,用于一组中间件 +// 如果 `condition` 为 true,它将按顺序创建并执行提供的 `factories` 生成的中间件链 +// 否则,它会直接跳过整个链 +func (engine *Engine) UseChainIf(condition bool, factories ...MiddlewareXFunc) HandlerFunc { + // 如果条件不满足或没有提供任何工厂函数,返回一个“穿透”中间件 + if !condition || len(factories) == 0 { return func(c *Context) { c.Next() } } - // 如果条件为真,返回一个处理器,该处理器负责执行子链。 - // 我们通过修改 Context 的状态来做到这一点。 - // 这比递归闭包或替换 c.Next 要清晰。 + // 在配置路由时就创建好所有中间件实例 + middlewares := make(HandlersChain, 0, len(factories)) + for _, factory := range factories { + if factory != nil { + middlewares = append(middlewares, factory()) + } + } + + // 返回一个处理器,该处理器负责执行这个子链 + // 这个实现通过临时替换 Context 的处理器链来注入子链,是健壮的 return func(c *Context) { - // 1. 将当前的处理链和索引位置保存下来。 + // 将当前的处理链和索引位置保存下来 originalHandlers := c.handlers originalIndex := c.index - // 2. 创建一个新的处理链,它由传入的中间件和最后的 "恢复并继续" 处理器组成。 + // 创建一个新的临时处理链 + // 它由我们预先创建的 `middlewares` 和一个特殊的“恢复”处理器组成 subChain := make(HandlersChain, len(middlewares)+1) copy(subChain, middlewares) - // 3. 在子链的末尾添加一个特殊的 handler。 - // 它的作用是恢复原始的处理链状态,并继续执行原始链中 `If` 之后的处理器。 + // 在子链的末尾添加“恢复”处理器 + // 当所有 `middlewares` 都执行完毕并调用了 Next() 后,这个函数会被执行 subChain[len(middlewares)] = func(ctx *Context) { + // 恢复原始的处理链状态 ctx.handlers = originalHandlers ctx.index = originalIndex + // 继续执行原始处理链中 `UseChainIf` 之后的下一个处理器 ctx.Next() } - // 4. 将 Context 的处理器链临时替换为我们的子链,并重置索引。 + // 将 Context 的处理器链临时替换新的的子链,并重置索引以从头开始 c.handlers = subChain c.index = -1 - // 5. 调用 c.Next() 来启动子链的执行。 c.Next() } } // UseIf 是一个条件中间件包装器 -func (engine *Engine) UseIf(condition bool, middlewares HandlerFunc) HandlerFunc { +func (engine *Engine) UseIf(condition bool, middlewareX MiddlewareXFunc) HandlerFunc { if !condition { return func(c *Context) { c.Next() } } - return middlewares -} + // 只有当条件为 true 时,才调用工厂函数创建中间件实例 + // 注意:这会导致每次请求都创建一个新的中间件实例(如果中间件本身有状态) + // 如果中间件是无状态的,可以进行优化 -func (engine *Engine) UseNewChainIf(condition bool, middlewares ...HandlerFunc) HandlerFunc { - // 如果条件不满足或没有提供中间件,返回一个“穿透”中间件。 - if !condition || len(middlewares) == 0 { - return func(c *Context) { - c.Next() - } - } + // 优化:只创建一次 + middleware := middlewareX() - // 如果条件满足,返回一个处理器,该处理器负责按顺序执行这个子链。 return func(c *Context) { - // 保存原始的处理链和当前执行索引,以便在子链结束后恢复。 - originalHandlers := c.handlers - originalIndex := c.index - - // 创建一个新的临时处理链。 - // 这个链由传入的 `middlewares` 和一个特殊的“恢复”处理器组成。 - subChain := make(HandlersChain, len(middlewares)+1) - copy(subChain, middlewares) - - // 在子链的末尾添加“恢复”处理器。 - // 当所有 `middlewares` 都执行完毕并调用了 Next() 后,这个函数会被执行。 - subChain[len(middlewares)] = func(ctx *Context) { - // 恢复原始的处理链状态。 - ctx.handlers = originalHandlers - ctx.index = originalIndex - // 继续执行原始处理链中 `ChainIf` 之后的下一个处理器。 - ctx.Next() - } - - // 将 Context 的处理器链临时替换为我们的子链,并重置索引以从头开始。 - c.handlers = subChain - c.index = -1 - - // 调用 c.Next() 来启动这个子链的执行。 - c.Next() + middleware(c) } }