个人中心
gpt4 book ai didi

Golang中实现简单的Http Middleware

转载 作者：qq735679552 更新时间：2022-09-29 22:32:09 27 4
gpt4 key购买 nike

CFSDN坚持开源创造价值，我们致力于搭建一个资源共享平台，让每一个IT人在这里找到属于你的精彩世界.

这篇CFSDN的博客文章Golang中实现简单的Http Middleware由作者收集整理，如果你对这篇文章有兴趣，记得点赞哟.

本文主要针对Golang的内置库 net/http 做了简单的扩展，通过添加中间件的形式实现了管道(Pipeline)模式，这样的好处是各模块之间是低耦合的，符合单一职责原则，可以很灵活的通过中间件的形式添加一些功能到管道中，一次请求和响应在管道中的执行过程如下 。

Golang中实现简单的Http Middleware

首先, 我定义了三个测试的中间件 Middleware1,2,3 如下 。

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
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 的方法，里面使用嵌套的形式, 使用了上面定义的三个测试的中间件. 。

?
1
2
3
4
5
func Pipeline(next http.Handler) http.Handler {
 
     return Middleware1(Middleware2(Middleware3(next)))
 
}

然后还需要业务代码，这里我定义了 LoginHandler 和 RegisterHandler 两个方法 。

?
1
2
3
4
5
6
7
8
9
10
11
12
13
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 。

?
1
2
3
4
5
6
7
8
9
func main() {
 
     http.Handle("/Login", Pipeline(http.HandlerFunc(LoginHandler)))
 
     http.Handle("/Register", http.HandlerFunc(RegisterHandler))
 
     http.ListenAndServe(":8080", nil)
 
}

启动程序后，访问 http://localhost:8080/Login, 程序的输出如下，这和本文最上面的管道的流程图是一致的，然后访问 Register 接口, 控制台没有输出信息，当然也不会执行任何中间件.

Golang中实现简单的Http Middleware

现在已经实现了中间件的机制，但是，上面添加中间件是用嵌套的方法，这种方式不能说不太优雅，只能说非常的Low，接下来我们需要对管道进行优化 。

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
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 Middleware的文章就介绍到这了,更多相关Golang实现Http Middleware内容请搜索我以前的文章或继续浏览下面的相关文章希望大家以后多多支持我！ 。

原文链接：https://www.cnblogs.com/myshowtime/p/15071212.html 。

最后此篇关于Golang中实现简单的Http Middleware的文章就讲到这里了,如果你想了解更多关于Golang中实现简单的Http Middleware的内容请搜索CFSDN的文章或继续浏览相关文章，希望大家以后支持我的博客！ 。

27 4 0
文章推荐： 基于charles抓取https请求使用过程解析
文章推荐： Java 程序员必备的 Intellij IDEA 插件
文章推荐： Java11 发布前抓紧掌握这些新特性
文章推荐： 浅谈mysql 树形结构表设计与优化
qq735679552
个人简介

我是一名优秀的程序员,十分优秀！

滴滴打车优惠券免费领取
滴滴打车优惠券
全站热门文章
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com