Golang Middleware Part 1 · To Be A Better Man

如何在Golang中实现中间件-Part 1

当使用net/http包实现服务的时候，一般使用的是如下的两中处理方式:

http.HandleFunc
http.Handle

http.HandleFunc

分析

当使用这种方式的时候，其接受两个参数，一个是字符串格式的匹配符(pattern),另外一个就是func(ResponWrite, *Request), 因此只要我们的中间件中返回该类型，那么中间件就是可以实现的

func main(){
    http.HandleFunc("/", Hello)
	http.ListenAndServe(":8080", nil)
}

func Hello(w http.ResponseWriter, r *http.Request)  {
	fmt.Print("hello")
}

当我们运行如上的程序的时候，就会打印出hello这个结果，说明我们的写法是没有问题的

实现

接下来，我们需要定义我们的中间件，它需要接收一个｀ http.HandlerFunc｀类型，并且返回一个http.HandlerFunc这样才能被使用

func MyMiddleware(next http.HandlerFunc)http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
        fmt.Println("middleware")
        //doSomethinds
		next.ServeHTTP(w,r)
	}
}

由于调用next.ServeHTTP(w, r)等效于调用next(w, r),处理完后会返回响应w,最终相应会传递到最外层的匿名函数，从而最终会返回到客户端

http.Handle

分析

http.Handle接受两个参数，一个是匹配符，另外一个是http.Handler，当我们查看源码的时候，发现其是一个接口类型

type Handler interface {
	ServeHTTP(ResponseWriter, *Request)
}

因此我们最终得返回一个实现了该接口的类型，假设我们首先定义一个新的结构体,并为其添加一个｀ServiceHTTP｀

解决

type MyResponse struct {
	next http.Handle
	Code int64
	Msg string
	Errors []string
	Data map[string]interface{}
}

func (res *MyResponse)ServeHTTP(w http.ResponseWriter, r *http.Request)  {

	result, err := json.Marshal(res)
	if err 大专栏  Golang Middleware Part 1 · To Be A Better Mancolor:#f92672">!= nil {
		fmt.Println(err.Error())
	}
	w.Write(result)
}

紧接着我们定义两个处理函数

func Hello(w http.ResponseWriter, r *http.Request)  {
	fmt.Println("hello")
}

func World(w http.ResponseWriter, r *http.Request)  {
	fmt.Println("world")
}

然后在主函数里面进行调用

    var res = new(MyResponse)
    res.next = Hello
    http.Handle("/hello", res)
    res.next = World
	http.Handle("/world", res)
	http.ListenAndServe(":8080", nil)

然后你会发现诡异的事情发生了，无论你访问哪一个路由地址，最终打印的都只是world,这是因为Golang是一门静态的预编译语言，编译完成后，｀res｀总的next属性将会永远指向World的地址,

因此我们可以做一个映射，将next变成map类型，但是想一想，如果路由非常多的话，那将是一件多可怕的事情，因此我们需要另辟蹊径，有没有其他的办法可以实现,查看源码我们可以发现之前我们所使用过的http.HandleFunc与之相对应的还有一个类型http.HandlerFunc，该类型实现了ServerHTTP方式

type HandlerFunc func(ResponseWriter, *Request)

// ServeHTTP calls f(w, r).
func (f HandlerFunc) ServeHTTP(w ResponseWriter, r *Request) {
	f(w, r)
}

接下来我们利用Goalng中的强制类型转换，就可以写出如下的中间件

func MySencondMiddleware(next http.HandlerFunc) http.Handler {
	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		//do somethings
		next.ServeHTTP(w, r)
	})
}