martini or gin Next()

最新推荐文章于 2025-02-28 17:10:05 发布

原创最新推荐文章于 2025-02-28 17:10:05 发布 · 227 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#golang

Go 专栏收录该内容

167 篇文章

订阅专栏

本文介绍了Go语言中类似gin和martini框架的`Next()`方法的使用。`Next()`确保在调用它之前执行当前处理器，然后继续执行后续处理器，直到所有处理器完成，最后执行`Next()`之后的代码。文中通过示例展示了如何在自定义的上下文`context`结构体中实现`Next()`方法，并展示了如何注册和执行处理器。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

package main

import (
	"fmt"
	"reflect"
)

type Handler interface{}

type context struct {
	handlers []Handler
	index    int
}

func (c *context) handler() Handler {
	if c.index < len(c.handlers) {
		return c.handlers[c.index]
	}
	if c.index == len(c.handlers) {
		return func() {
			println("do end")
		}
	}
	panic("invalid index for context handler")
}

func (c *context) Next() {
	c.index += 1
	c.run()
}

func (c *context) run() {
	for c.index <= len(c.handlers) {
		val := reflect.ValueOf(c.handler())
		val.Call(nil)
		c.index += 1
	}
}

func (c *context) use(handler Handler) {
	validateHandler(handler)
	c.handlers = append(c.handlers, handler)
}

func validateHandler(handler Handler) {
	if reflect.TypeOf(handler).Kind() != reflect.Func {
		panic("handler 非函数")
	}
}

func main() {

	ctx := new(context)
	ctx.use(func() {
		ctx.Next()
		fmt.Println("函数1")
	})
	ctx.use(func() {
		fmt.Println("函数2")
		ctx.Next()
	})
	ctx.use(func() {
		fmt.Println("函数3")
	})

	ctx.run()
}