Golang中实现简单的Http Middleware
作者:SpringLeee 时间:2024-05-22 10:30:26
本文主要针对Golang的内置库 net/http 做了简单的扩展,通过添加中间件的形式实现了管道(Pipeline)模式,这样的好处是各模块之间是低耦合的,符合单一职责原则,可以很灵活的通过中间件的形式添加一些功能到管道中,一次请求和响应在管道中的执行过程如下
首先, 我定义了三个测试的中间件 Middleware1,2,3 如下
func Middleware1(next http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
fmt.Println("M1 in")
next.ServeHTTP(w, r)
fmt.Println("M1 out")
})
}
func Middleware2(next http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
fmt.Println("M2 in")
next.ServeHTTP(w, r)
fmt.Println("M2 out")
})
}
func Middleware3(next http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
fmt.Println("M3 in")
next.ServeHTTP(w, r)
fmt.Println("M3 out")
})
}
这里中间件的入参和出参的类型都是 http.Handler, 然后在 next.ServeHTTP() 的前后分别输出了 In 和 Out.
接下来,定义一个 Pipeline 的方法,里面使用嵌套的形式, 使用了上面定义的三个测试的中间件.
func Pipeline(next http.Handler) http.Handler {
return Middleware1(Middleware2(Middleware3(next)))
}
然后还需要业务代码,这里我定义了 LoginHandler 和 RegisterHandler 两个方法
func LoginHandler(w http.ResponseWriter, r *http.Request) {
fmt.Println("Login...")
w.Write([]byte("Login..."))
}
func RegisterHandler(w http.ResponseWriter, r *http.Request) {
fmt.Println("Register...")
w.Write([]byte("Register..."))
}
最后修改程序的 main 函数, 在 Login 接口上使用上面添加过中间件的 Pipeline
func main() {
http.Handle("/Login", Pipeline(http.HandlerFunc(LoginHandler)))
http.Handle("/Register", http.HandlerFunc(RegisterHandler))
http.ListenAndServe(":8080", nil)
}
启动程序后,访问 http://localhost:8080/Login, 程序的输出如下,这和本文最上面的管道的流程图是一致的,然后访问 Register 接口, 控制台没有输出信息,当然也不会执行任何中间件。
现在已经实现了中间件的机制,但是,上面添加中间件是用嵌套的方法,这种方式不能说不太优雅,只能说非常的Low,接下来我们需要对管道进行优化
type Chain struct {
middlewares []func(handler http.Handler) http.Handler
}
func Pipeline(next http.Handler) http.Handler {
//return Middleware1(Middleware2(Middleware3(next)))
return AddMiddlewares(Middleware1,Middleware2,Middleware3).Then(next)
}
func AddMiddlewares(m ...func(handlerFunc http.Handler) http.Handler) Chain {
c := Chain{}
c.middlewares = append(c.middlewares,m...)
return c
}
func (c Chain) Then(next http.Handler) http.Handler {
for i := range c.middlewares {
prev := c.middlewares[len(c.middlewares)-1-i]
next = prev(next)
}
return next
}
首先定义了一个Chain 的struct,用来接收添加到管道中的中间件,在 AddMiddlewares() 函数中,接收了多个Handle, 然后组装到 Chain 对象并返回, 接下来调用 Then() 函数, 把管道中的中间件和业务的Handler 关联起来。在中间件的使用方式上, 这两种方法都是一样的,只需要调用 Pipeline() 方法就行了。
本文在go web中简单的实现了中间件的机制,这样带来的好处也是显而易见的,当然社区也有一些成熟的 middleware 组件,包括 Gin 一些Web框架中也包含了 middleware 相关的功能, 希望对您有用.
来源:https://www.cnblogs.com/myshowtime/p/15071212.html
![](/images/zang.png)
![](/images/jiucuo.png)
猜你喜欢
使用Fabric自动化部署Django项目的实现
python中终止协程和异常处理方式
sqlserver 增删改查一些不常用的小技巧
![](https://img.aspxhome.com/file/2023/5/128395_0s.png)
Go语言之自定义集合Set
Python常见库matplotlib学习笔记之多个子图绘图
![](https://img.aspxhome.com/file/2023/6/128506_0s.png)
2020版Python学习路线图(附学习资料)
![](https://img.aspxhome.com/file/2023/1/130681_0s.png)
Go语言实现二进制与十进制互转的示例代码
python使用多线程+socket实现端口扫描
![](https://img.aspxhome.com/file/2023/2/121372_0s.jpg)
pandas 时间偏移的实现
Maui Blazor 使用摄像头实现代码
写了个监控nginx进程的Python脚本
apache部署python程序出现503错误的解决方法
Python os模块介绍
Numpy中转置transpose、T和swapaxes的实例讲解
使用python将图片按标签分入不同文件夹的方法
![](https://img.aspxhome.com/file/2023/6/118716_0s.jpg)
MySQL 实现lastInfdexOf的功能案例
Python深入学习之内存管理
![](https://img.aspxhome.com/file/2023/9/80719_0s.jpg)