golang实现http中间件 gin框架中间件实现原理分析

suoniao 2021-01-21
需要:0索币

gin-gonic的中间件使用非常简单,这里就不多说了,本文主要讲讲中间件的原理以及DEMO实现。

中间件是一个洋葱模型,中心为最终处理请求的 handler,称为 main handler,其他为 middleware handler 。每一个 middleware handle 可以分为两部分,随着 request 的流动,左边是入,右边为出,而分割点就是 next 。

next 的作用就是明确在这个地方进入到下一个 handler ,如果没有指定,默认是当前 handler 结束后进入下一个 handler。
在这里插入图片描述

由此看出各 middleware handle 都是先进后出的。

deml1:

package main

import (
	"fmt"
)


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

func (c *Context) Next() {
	c.index++
	c.Handlers[c.index](c)
}

func main() {
	c := &Context{}
	c.Handlers = make([]func(*Context), 0)

	// 注册中间件
	c.Handlers = append(c.Handlers, m1)
	c.Handlers = append(c.Handlers, m2)
	c.Handlers = append(c.Handlers, m3)

	// 控制器函数
	c.Handlers = append(c.Handlers, action)

	// 启动
	c.Handlers[0](c)
}

func action(c *Context) {
	fmt.Println("main handler")
}

func m1(c *Context) {
	fmt.Println("m1 start")
	c.Next()
}

func m2(c *Context) {
	fmt.Println("m2 start")
	c.Next()
	fmt.Println("m2 end")
}

func m3(c *Context) {
	c.Next()
	fmt.Println("m3 end")
}

m1, m2, m3 为 middleware handler,action 为 main handler。通过 Next 方法实现链路,以上程序输出:

m1 start
m2 start
main handler
m3 end
m2 end

符合预期,但是如果将 m2 的 c.Next() 去掉了,整个链路就断了,所以还要实现自动链路,思路就是只要链路被启动了,就自动往下运行,而不依靠 Next 方法。

修改 Next 方法:

func (c *Context) Next() {
	c.index++
	for c.index < int8(len(c.Handlers)) {
		c.Handlers[c.index](c)
		c.index++
	}
}

这样,除了 m1 之外,其他几个去掉了 Next 方法,也能按预期运行了,现在需要将 m1 中的 Next 也去掉,则需要修改启动方式了。

// 启动
c.Handlers[0](c)
c.Next()

这样,所有的 middleware handler 都不用强依赖于 Next 也能按预期运行了。

有时候我们需要人为的中断请求,需要实现链路中断,参考gin中的实现方式,它定义了一个最大的 handler 索引最为判断依据,并且在注册 handler 的时候也做同样的判断。

于是,最终代码为:

package main

import (
	"fmt"
	"math"
)

const abortIndex int8 = math.MaxInt8 / 2

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

func (c *Context) Next() {
	c.index++
	for c.index < int8(len(c.Handlers)) {
		c.Handlers[c.index](c)
		c.index++
	}
}

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

func main() {
	c := &Context{}
	c.Handlers = make([]func(*Context), 0)

	// 注册中间件
	c.Handlers = append(c.Handlers, m1)
	c.Handlers = append(c.Handlers, m2)
	c.Handlers = append(c.Handlers, m3)

	// 控制器函数
	c.Handlers = append(c.Handlers, action)

	// 限制handler个数
	if len(c.Handlers) >= int(abortIndex) {
		panic("too many handlers")
	}

	// 启动
	c.Handlers[0](c)
	c.Next()
}

func action(c *Context) {
	fmt.Println("main handler")
}

func m1(c *Context) {
	fmt.Println("m1 start")
}

func m2(c *Context) {
	fmt.Println("m2 start")
	c.Abort()
	c.Next()
	fmt.Println("m2 end")
}

func m3(c *Context) {
	c.Next()
	fmt.Println("m3 end")
}

输出为:

m1 start
m2 start
m2 end

Abort方法实现了禁止向下传播,但是不会影响当前中间件的运行。

回帖
  • 消灭零回复
广告位招租 100元/月 广告位招租 100元/月
相关主题
golang网络数据读取timeout的处理 – 以SetReadDeadline为例 0
golang实现http中间件 gin框架中间件实现原理分析 0
围绕Handler接口的方法ServeHTTP,可以轻松的写出go中的中间件 0
自己动手利用go语言实现一个简易版RPC服务器 0
Go语言数组定义和初始化的三种方式 0
通过go env命令查询go语言环境变量信息 0
GOPATH目录下面创建bin目录、pkg目录和src目录功能说明 0
windows配置golang的环境变量 gopath配置学习go语言第一天 0
何用Go语言开发一个高负荷的WebSocket服务 0
go语言获取windows磁盘列表的代码golang获取磁盘使用信息容量剩余量 0
Go语言实战文本大数据挖掘 0
go语言开发博客系统实战 0
Go语言从基础入门到商品秒杀系统实战18天培训课程 0
海量数据库架构设计分布式数据库和中间件 0
用go语言替代python实现自动化运维实战 0
JAVA高并发编程之中间件大全自研Java内存缓存 0
用go语言实现一个高并发的抽奖系统 0
golang从基础入门到分布式爬虫项目实战 0
go语言与区块链精品入门课程视频 0
go商品秒杀后台教程golang聊天室redis实现 0
相关主题
原来EUC-CN是gb2312编码的一种表示方法 0
索鸟快传2.1.0版本发布 0
golang网络数据读取timeout的处理 – 以SetReadDeadline为例 0
索鸟网广告位招租 0
golang利用通道chan实现一个通用的TCP连接池 0
frpc和frps的交互流程详解了解了frp是如何进行TCP代理的 0
golang实现内存池 go语言字节池byte pool实现代码 0
golang网络编程之基于TCP协议实现长连接 golang心跳检测 0
国产光刻机产业链最新核心材料ArF 193nm光刻胶通过用户认证 0
javascript数组相关函数length属性、delete关键字、pop()栈方法、shift() 0
Qt为我们提供了几个可以用于线程Sleep的函数msleep和usleep 0
QT程序闪退错误捕获教程利用DbgHelp 错误调试技术 0
简单快速修改Qtcreator项目工程的名称 0
golang 的interface接口类型断言 0
golang类型断言type的使用 0
golang依赖包管理 mod使用教程 0
golang表单验证库validator 0
如何使用Go语言实现一个简单的异步任务框架呢?生产者消费者模型 0
内网穿透反向代理工具frp实现TCP协议代理源码分析 0
内网穿透反向代理库frp的实现原理分析一 0