golang程序设计:Go middleware中间件以及Gin 中间件分析
先从业务开发角度出发,来逐渐引出中间件。
一、刚开始时业务开发
开始业务开发时,业务需求比较少。
- 当我们最开始进行业务开发时,需求不是很多。 第一个需求产是品向大家打声招呼:“hello world”。
接到需求任务,我们就进行代码开发了。
一般都会写下如下的代码,用handlefunc来处理请求的服务
package main
import (
"net/http"
)
func helloHandler(w http.ResponseWriter, r *http.Request) {
w.Write([]byte("hello world"))
}
func main() {
http.HandleFunc("/", helloHandler)
http.ListenAndServe(":8080", nil)
}
- 假如现在业务有变化了,我们要新增一个hello服务的处理耗时,怎么做?
这个需求比较简单,修改代码如下:
package main
import (
"log"
"net/http"
"os"
"time"
)
var logger = log.New(os.Stdout, "", 0)
func helloHandler(w http.ResponseWriter, r *http.Request) {
timeStart := time.Now()
w.Write([]byte("hello world"))
timeElapsed := time.Since(timeStart)
logger.Println(timeElapsed)
}
func main() {
http.HandleFunc("/", helloHandler)
http.ListenAndServe(":8080", nil)
}
这样就可以输出当前hello请求到日志文件了。
- 完成了这个需求后。过了没多久,又有新的需求来了,我们要显示信息,显示Email,
显示好朋友,并且这就一个接口也需要增加耗时记录。
一下子又增加了很多api, 简略示例代码如下:
package main
func helloHandler(wr http.ResponseWriter, r *http.Request) {
// ...
}
func showInfoHandler(wr http.ResponseWriter, r *http.Request) {
// ...
}
func showEmailHandler(wr http.ResponseWriter, r *http.Request) {
// ...
}
func showFriendsHandler(wr http.ResponseWriter, r *http.Request) {
timeStart := time.Now()
wr.Write([]byte("your friends is tom and alex"))
timeElapsed := time.Since(timeStart)
logger.Println(timeElapsed)
}
func main() {
http.HandleFunc("/", helloHandler)
http.HandleFunc("/info/show", showInfoHandler)
http.HandleFunc("/email/show", showEmailHandler)
http.HandleFunc("/friends/show", showFriendsHandler)
// ...
}
每一个handler里面都需要记录运行的时间,每次新加路由都要写同样的代码。都要把业务逻辑代码拷贝过来。
- 业务继续发展,又有了新的需求,增加一个监控系统,需要你上报接口运行时间到监控系统里面,以便监控接口的稳定性。这个监控系统叫metrics。
好了,现在你又要修改代码,通过http post方式把耗时时间发送给metrics系统。
而且你要修改好多个handler,增加metrics上报接口代码。
修改代码:
func helloHandler(wr http.ResponseWriter, r *http.Request) {
timeStart := time.Now()
wr.Write([]byte("hello"))
timeElapsed := time.Since(timeStart)
logger.Println(timeElapsed)
// 新增耗时上报
metrics.Upload("timeHandler", timeElapsed)
}
func showInfoHandler(wr http.ResponseWriter, r *http.Request) {
// ...
// 新增耗时上报
metrics.Upload("timeHandler", timeElapsed)
}
func showEmailHandler(wr http.ResponseWriter, r *http.Request) {
// ...
// 新增耗时上报
metrics.Upload("timeHandler", timeElapsed)
}
func showFriendsHandler(wr http.ResponseWriter, r *http.Request) {
timeStart := time.Now()
wr.Write([]byte("your friends is tom and alex"))
timeElapsed := time.Since(timeStart)
logger.Println(timeElapsed)
// 新增耗时上报
metrics.Upload("timeHandler", timeElapsed)
}
到这里,发现要修改好多的handler函数,才能把接口的耗时时间上报到metrics系统里。
随着新需求越来越多,handler也会越多,那么我们修改的地方也就越多。增加了一个简单的业务统计,就要修改好多个handler函数。
虽然只是增加一个业务统计,我们就要去修改handler,来增加这些和业务无关的代码。
一开始我们并没有做错什么, 但是随着业务的发展,我们逐渐陷入了代码的泥潭。
接下来,我们该怎么办呢?怎么处理这种情况?
二、业务逐渐多了后
随着业务发展,handler越来越多,增加与业务无关的代码所要修改的地方也越来越多。这时候怎么办?有没有办法可以处理这种情况呢?
想一下,java里面有一个Filter的技术,可以拦截请求的处理。我们可不可以利用这个思想来解决我们的问题呢。这种思想当然是可以的。
在go里面就是利用 http.Handler
来把你要处理的函数包起来(实际就是拦截了),然后处理。
在go里面有一个学名叫 middleware(中间件),中间件常见的位置是ServeMux和应用处理程序之间。
http的请求控制流程:
ServeMux => Middleware Handler => Application Handler
针对上面的需求,我们用中间件的方法来修改下:
package main
import (
"log"
"net/http"
"os"
"time"
)
var logger = log.New(os.Stdout, "", 0)
func hello(w http.ResponseWriter, r *http.Request) {
w.Write([]byte("hello world"))
}
func timeMiddleware(next http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
timeStart := time.Now()
//next handler
next.ServeHTTP(w, r)
timeElapsed := time.Since(timeStart)
logger.Println(timeElapsed)
})
}
func main() {
http.HandleFunc("/", timeMiddleWare(hello))
http.ListenAndServe(":8080", nil)
}
这样就实现了中间件。
也是把业务代码和非业务代码进行了剥离。
三、怎么实现中间件
上面的中间件是怎么实现的呢?
- 他要满足
http.Handler
接口
type Handler interface {
ServeHTTP(ResponseWriter, *Request)
}
写一个简单的程序:
func showinfoHandler(info string) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
w.Write([]byte(info)
})
}
上面程序我们把简单的处理程序w.Write放在了一个匿名函数中,然后引用了外部的info构成了一个闭包。接下来我们用 http.HandlerFunc
适配器将此闭包转换为处理程序,然后返回它。
我们可以用相同的方法,将下一个处理程序作为变量来进行传递,然后调用ServeHTTP() 方法将控制转移到下一个处理程序,然后返回它。
func demoMiddleware(next http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
// out logic
next.ServeHTTP(w, r)
})
}
上面的中间件函数有一个 func(http.Handler) http.Handler
的函数签名。它接收一个处理程序作为参数并返回另外一个处理程序。
一个完整的例子
用一个完整的例子来看看中间件的执行流程:
middlewaredemo.go
package main
import (
"log"
"net/http"
)
func middleOne(next http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
log.Println("before middleone")
next.ServeHTTP(w, r)
log.Println("after middlewareOne again")
})
}
func middleTwo(next http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
log.Println("before middletwo")
if r.URL.Path == "/foo" {
return
}
next.ServeHTTP(w, r)
log.Println("after middleTwo again")
})
}
func final(w http.ResponseWriter, r *http.Request) {
log.Println("exec finalHandler")
w.Write([]byte("OK"))
}
func main() {
finalHandler := http.HandlerFunc(final)
http.Handle("/", middleOne(middleTwo(finalHandler)))
http.ListenAndServe(":3030", nil)
}
先执行下面的命令
go run middlewaredemo.go
然后在浏览器上执行:http://localhost:3030/
运行结果:
2020/04/19 23:07:14 before middleone
2020/04/19 23:07:14 before middletwo
2020/04/19 23:07:14 exec finalHandler
2020/04/19 23:07:14 after middleTwo again
2020/04/19 23:07:14 after middlewareOne again
看看执行结果,在 next.ServeHTTP
前,看到处理顺序是依次按照嵌套的顺序出结果, 但是在 next.ServeHTTP
后 的程序,是按照相反的方向出结果。
然后在运行 : http://localhost:3030/foo
运行结果:
2020/04/19 23:12:15 before middleone
2020/04/19 23:12:15 before middletwo
2020/04/19 23:12:15 after middlewareOne again
middleTwo
函数里 return
后面的程序,都没有显示了。
所以,在中间件中,我们可以用 return
来停止在中间件程序的传播。
四、Gin框架的中间件
github.com/gin-gonic/gin,这个web框架使用很广泛,它也有中间件功能。
- 使用方法 一
// 定义中间件
func middlewareOne(c *gin.Context) {
// 中间件逻辑
}
// 使用中间件
r := gin.Default()
r.Use(middlewareOne)
- 使用方法 二
func middlewareTwo() gin.HandlerFunc {
// 自定义逻辑
return func(c *gin.Context) {
// 中间件逻辑
}
}
// 使用中间件
r := gin.Deafult()
r.Use(middlewareTwo()
Gin还有一种像java中的Filter,处理前,处理后的一种方法 Next()
比如:demo1.go
package main
import (
"fmt"
"time"
"github.com/gin-gonic/gin"
)
func middlewareOne() gin.HandlerFunc {
return func(c *gin.Context) {
c.String(200, "before middlewareOne: handler "+time.Now().String()+"\n")
c.Next()
c.String(200, "after middlewareOne : handler "+time.Now().String()+"\n")
}
}
func middlewareTwo() gin.HandlerFunc {
return func(c *gin.Context) {
c.String(200, "before middlewareTwo: handler \n")
c.Next()
c.String(200, "after middlewareTwo : handler \n")
}
}
func main() {
r := gin.Default()
r.Use(middlewareOne(), middlewareTwo())
r.GET("/", func(c *gin.Context) {
fmt.Println("start!")
c.String(200, "GET method \n")
fmt.Println("end!")
})
r.Run(":8080")
}
先在命令行运行:
go run demo1.go
然后在浏览器上输入:http://localhost:8080/
就可以看到输出结果:
before middlewareOne: handler 2020-04-20 01:03:25.4547842 +0800 CST m=+21.318612501
before middlewareTwo: handler
GET method
after middlewareTwo : handler
after middlewareOne : handler 2020-04-20 01:03:25.4547842 +0800 CST m=+21.318612501
其实我们看到gin框架实现的中间件,它书写形式并不是像上面的那种嵌套模式。如果有很多中间件的话,那么这种嵌套模式写出来就让人感觉非常复杂。
而gin中间件这种书写模式,就很清晰,适合人阅读。是一种优雅的实现方式。
它是怎么实现的呢?
五、Gin中间件的实现
gin v1.6.2 版本
gin.go
// HandlerFunc defines the handler used by gin middleware as return value.
type HandlerFunc func(*Context)
// HandlersChain defines a HandlerFunc array.
//定义了Handlers链
type HandlersChain []HandlerFunc
// Last returns the last handler in the chain. ie. the last handler is the main one.
// 从Handlers里取出最后一个Handler,就是main自己
func (c HandlersChain) Last() HandlerFunc {
if length := len(c); length > 0 {
return c[length-1]
}
return nil
}
// RouteInfo represents a request route's specification which contains method and path and its handler.
type RouteInfo struct {
Method string
Path string
Handler string
HandlerFunc HandlerFunc
}
// RoutesInfo defines a RouteInfo array.
type RoutesInfo []RouteInfo
type Engine struct {
RouterGroup
// Enables automatic redirection if the current route can't be matched but a
// handler for the path with (without) the trailing slash exists.
// For example if /foo/ is requested but a route only exists for /foo, the
// client is redirected to /foo with http status code 301 for GET requests
// and 307 for all other request methods.
RedirectTrailingSlash bool
... ...
}
// Default returns an Engine instance with the Logger and Recovery middleware already attached.
// 初始化Engine,里面就用到了2个中间件函数Logger和Recovery
func Default() *Engine {
debugPrintWARNINGDefault()
engine := New()
engine.Use(Logger(), Recovery())
return engine
}
// Use attaches a global middleware to the router. ie. the middleware attached though Use() will be
// included in the handlers chain for every single request. Even 404, 405, static files...
// For example, this is the right place for a logger or error management middleware.
// 增加 middleware -> 实质是到 RouterGroup.Use()
func (engine *Engine) Use(middleware ...HandlerFunc) IRoutes {
engine.RouterGroup.Use(middleware...) //到了 RouterGroup 里的Use
engine.rebuild404Handlers()
engine.rebuild405Handlers()
return engine
}
context.go
// Context is the most important part of gin. It allows us to pass variables between middleware,
// manage the flow, validate the JSON of a request and render a JSON response for example.
type Context struct {
writermem responseWriter
Request *http.Request
Writer ResponseWriter
Params Params
handlers HandlersChain //这里有一个handlers 链,一个slice
index int8
fullPath string
engine *Engine
// This mutex protect Keys map
KeysMutex *sync.RWMutex
// Keys is a key/value pair exclusively for the context of each request.
Keys map[string]interface{}
... ...
}
// Handler returns the main handler.
// 返回main handler
func (c *Context) Handler() HandlerFunc {
return c.handlers.Last()
}
// Next should be used only inside middleware.
// It executes the pending handlers in the chain inside the calling handler.
// See example in GitHub.
func (c *Context) Next() {
c.index++
for c.index < int8(len(c.handlers)) {
c.handlers[c.index](c)
c.index++
}
}
还有一个routegroup.go里的handler chain
// Use adds middleware to the group, see example code in GitHub.
// 增加middleware
func (group *RouterGroup) Use(middleware ...HandlerFunc) IRoutes {
group.Handlers = append(group.Handlers, middleware...)
return group.returnObj()
}
func (group *RouterGroup) combineHandlers(handlers HandlersChain) HandlersChain {
finalSize := len(group.Handlers) + len(handlers)
if finalSize >= int(abortIndex) {
panic("too many handlers")
}
mergedHandlers := make(HandlersChain, finalSize)
copy(mergedHandlers, group.Handlers)
copy(mergedHandlers[len(group.Handlers):], handlers)
return mergedHandlers
}
六、改进以前框架
写到了这里,想起了以前写的 golang web框架 文章,那只是实现了一个简单的MVC功能,并不具备可扩展性,有了这个中间件技术,就可以把以前的框架进行改进。
改进后的全新 go web 框架 lilac
先实现功能,然后再进行优化改进 - 论开发。
七、参考
- Go高级语言编程-中间件 ,这个写的非常非常非常好,业务部分基本是从这里来的
- gin框架