go 简单中间件实现

package main

import (
	"net/http"
)

func main(){
	//中间件的链式调用
	http.HandleFunc("/",MiddlewareOne(MiddlewareTwo(hello)))
	http.ListenAndServe(":8080",nil)
}

func hello(w http.ResponseWriter,r *http.Request){
	w.Write([]byte("hello!"))
}

//中间件1 传入参数是一个http.HandlerFunc; 返回参数也是一个http.HandlerFunc
func MiddlewareOne(h http.HandlerFunc) http.HandlerFunc {
	return http.HandlerFunc(
		func(w http.ResponseWriter,r * http.Request) {
			w.Write([]byte("middlewareone!\n"))
			h(w,r)
		})
}
//中间件2
func MiddlewareTwo(h http.HandlerFunc) http.HandlerFunc{
	return http.HandlerFunc(
		func(w http.ResponseWriter,r *http.Request) {
			w.Write([]byte("middlewaretwo!\n"))
			h(w,r)
		})
}
发布了6 篇原创文章 · 获赞 0 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/manbudezhu/article/details/104192788