gin中间件原理

metabit

已于 2022-11-17 12:05:06 修改

阅读量241

点赞数 1

CC 4.0 BY-SA版权

分类专栏： Go 文章标签： gin 中间件

于 2022-11-16 23:55:11 首次发布

本文链接：https://blog.csdn.net/dawnto/article/details/127895920

Go 专栏收录该内容

167 篇文章

订阅专栏

这篇博客介绍了洋葱模型在服务拆分和中间件管理中的应用。通过定义`Context`结构体，实现`Abort`、`Use`、`Get`和`Next`等方法，展示了如何组织和执行中间件。在示例中，创建了多个中间件并依次执行，最后调用`Run`方法启动流程。中间件的执行顺序类似于洋葱层，从外向内层层推进。

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

洋葱模型

在这里插入图片描述

package main

import (
	"fmt"
	"math"
)

// 中间件索引长度  微服务，handler过多应拆分服务
const abortIndex int8 = math.MaxInt8 >> 1

type Context struct {
	handlers []func(ctx *Context)
	index    int8
}

func (c *Context) Abort() {
	c.index = abortIndex
}

func (c *Context) Use(f func(ctx *Context)) {
	c.handlers = append(c.handlers, f)
}

func (c *Context) Get(path string, f func(ctx *Context)) {
	fmt.Println("Get Function")
	c.handlers = append(c.handlers, f)
}

// Next 进入下一个中间件
func (c *Context) Next() {
	c.index++
	for c.index < int8(len(c.handlers)) {
		c.handlers[c.index](c)
		c.index++
	}
}

func (c *Context) Run() { //
	if len(c.handlers) > 0 {
		c.handlers[0](c)
		c.Next()
	}
}

func main() {
	c := Context{}
	c.Use(func(ctx *Context) {
		fmt.Println("中间件1 开始执行")
		c.Next()
		fmt.Println("中间件1 执行完毕")
	})

	c.Use(func(ctx *Context) {
		fmt.Println("中间件2 开始执行")
		ctx.Next()
		fmt.Println("中间件2 执行完毕")
	})

	c.Use(func(ctx *Context) {
		fmt.Println("中间件3 开始执行")
		ctx.Next()
		fmt.Println("中间件3 执行完毕")
	})

	c.Get("/", func(ctx *Context) {
		fmt.Println("do get")
	})

	fmt.Println(c.handlers)
	c.Run()
}