在 Golang 中实现一个简单的Http中间件过程详解
作者:SpringLeee 时间:2024-05-09 14:55:35
本文主要针对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)
猜你喜欢
web标准常见问题集合
Git的使用规范流程总结
![](https://img.aspxhome.com/file/2023/6/132246_0s.jpg)
微信小程序request请求后台接口php的实例详解
sql 查询慢的原因分析
python中的print()输出
oracle查看被锁的表和被锁的进程以及杀掉这个进程
解决Pytorch内存溢出,Ubuntu进程killed的问题
C#从数据库读取图片并保存的两种方法
安装pytorch报错torch.cuda.is_available()=false问题的解决过程
![](https://img.aspxhome.com/file/2023/8/124308_0s.jpg)
mysql 误删除ibdata1之后的恢复方法
vue实现拖拽效果
![](https://img.aspxhome.com/file/2023/1/130081_0s.jpg)
JS、CSS和HTML实现注册页面
![](https://img.aspxhome.com/file/2023/7/136417_0s.jpg)
用Python识别人脸,人种等各种信息
![](https://img.aspxhome.com/file/2023/9/72889_0s.jpg)
Python使用百度API上传文件到百度网盘代码分享
python计算牛顿迭代多项式实例分析
mysql命令行爱好者必备工具mycli
![](https://img.aspxhome.com/file/2023/1/114951_0s.gif)
Windows下ORACLE 10g完全卸载的方法分析
对numpy Array [: ,] 的取值方法详解
Opencv实现计算两条直线或线段角度方法详解
![](https://img.aspxhome.com/file/2023/5/92495_0s.png)