在 Golang 中实现一个简单的Http中间件过程详解
本文主要针对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 相关的功能, 希望对您有用.
到此这篇关于在 Golang 中实现一个简单的Http中间件的文章就介绍到这了,更多相关Golang Http中间件内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！
相关文章
这篇文章主要介绍了Go中strings的常用方法详解，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2020-03-03
golang常用库之gorilla/mux-http路由库使用详解
这篇文章主要介绍了golang常用库之gorilla/mux-http路由库使用,本文通过实例代码给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2020-10-10
go build命令用于编译我们指定的源码文件或代码包以及它们的依赖包,下面这篇文章主要给大家介绍了关于go build失败报方法undefined的解决过程,文中通过实例代码介绍的非常详细,需要的朋友可以参考下2022-11-11
go tool fix 是 Go 工具链中的一个命令，作用是把指定 Go 程序代码包中的的所有旧版本代码修正为新版本的代码，本文将简单介绍一下go tool fix的使用方法，感兴趣的小伙伴可以参考阅读下2023-07-07
模版方法是一种行为设计模式, 它在基类中定义了一个算法的框架, 允许子类在不修改结构的情况下重写算法的特定步骤,本文将通过代码示例给大家详细的介绍一下Go模板方法模式,需要的朋友可以参考下2023-08-08
Golang中的锁机制主要包含互斥锁和读写锁互斥锁互斥锁是传统并发程序对共享资源进行控制访问的主要手段,这篇文章主要给大家介绍了关于Golang并发操作中常见的读写锁的相关资料,需要的朋友可以参考下2021-08-08
单例是一种创建型设计模式，让你能够保证一个类只有一个实例，并提供一个访问该实例的全局节点，本文就通过代码示例给大家讲讲Go单例模式,需要的朋友可以参考下2023-07-07
Go中Map是一个KV对集合,下面这篇文章主要给大家介绍了关于Golang中map探究的相关资料,文中通过实例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2022-09-09
这篇文章主要介绍了go特性之数组与切片的问题,本文通过实例代码给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2020-11-11
这篇文章主要介绍了利用go语言实现查找二叉树中的最大宽度，文章围绕主题展开详细介绍，具有一定的参考价值，需要的小伙伴可以参考一下2022-05-05
最新评论