golang http 中间件
源码链接
type Handler interface {
ServeHTTP(ResponseWriter, *Request)
}
type Router struct {
route map[string]Handle
}
func (r *Router) ServeHTTP(w http.ResponseWriter, req *http.Request) {
}
通过函数包裹的方式实现
中间件v1.0
func withMiddle(h Handle) Handle {
return func(writer http.ResponseWriter, request *http.Request) {
t := time.Now()
defer func() {
log.Println("time spend is ", time.Since(t))
}()
h(writer, request)
}
}
func (r *Router) Register1(route string, f Handle) {
r.route[route] = withMiddle(f)
}
func (r *Router) Register2(route string, f Handle) {
r.route[route] = withMiddLog(withMiddTime(f))
}
Register("/bench", func(writer http.ResponseWriter, request *http.Request) {
time.Sleep(time.Second)
fmt.Println("bench sleep 1 second")
})
中间件v1.1
func (r *Router) Register(route string, f HandlerFunc) {
r.route[route] = withMiddLog(withMiddTime(func(writer http.ResponseWriter, request *http.Request) {
f()
}))
}
中间件v2.0
func (r *Server) Register(route string, f HandlerFunc) {
r.route[route] = withMiddLog(withMiddTime(func(writer http.ResponseWriter, request *http.Request) {
f(r.createContext(writer, request))
}))
}
r.Register("/bench", func(c *Context) {
time.Sleep(time.Second)
fmt.Println("bench sleep 1 second")
c.Writer.Write([]byte("hello!\r\n"))
})
golang框架gin中的实现
中间件v3.0
func (c *Context) Next() {
c.index++
//for中的index++是为了退出循环 否则没法退出
for ; c.index < len(c.middle); c.index++ {
c.middle[c.index](c)
}
}
func withMiddTime() HandleContext {
return func(c *Context) {
t := time.Now()
defer func() {
fmt.Println("withMiddTime end time", time.Since(t))
}()
fmt.Println("withMiddTime start ", time.Since(t))
c.Next()
}
}
func (s *Server) Register(path string, f ...HandleContext) {
handleNew := make([]HandleContext, 0, len(s.handle)+len(f))
handleNew = append(handleNew, s.handle...)
handleNew = append(handleNew, f...)
s.routeHandler(path, func(writer http.ResponseWriter, request *http.Request) {
s.createContext(writer, request, handleNew).Next()
})
}