深入理解Go语言中的Dispatcher

2020-01-28 12:13:55王振洲

介绍

Go使用goroutines来处理connection的读写事件,不会阻塞:


c, err := srv.newConn(rw)
  if err != nil {
    continue
  }
  go c.serve()

c即为创建的connection,保存了该次请求的信息,然后再传递到对应的handler,handler就可以读取到请求的header信息,保证了请求之间独立。

Go中的ServeMux

上面代码中提到了c(这个c就是connection).serve()方法。其实内部是调用了http包默认的路由器,通过路由器把本次请求的信息传递到了后端的处理函数。

默认路由器ServeMux,结构如下:


type ServeMux struct {
 mu sync.RWMutex  //锁,由于请求涉及到并发处理,因此这里需要一个锁机制
 m map[string]muxEntry // 路由规则,一个string对应一个mux实体,这里的string就是注册的路由表达式
 hosts bool // 是否在任意的规则中带有host信息
}

下面看一下muxEntry:


type muxEntry struct {
 explicit bool  // 是否精确匹配
 h    Handler // 这个路由表达式对应哪个handler
 pattern string //匹配字符串
}

接着看一下Handler的定义:


type Handler interface {
 ServeHTTP(ResponseWriter, *Request) // 路由实现器
}

Handler是一个接口,但是前一小节中的sayhelloName函数并没有实现ServeHTTP这个接口,仍然能添加到路由表中,原因就是http包里还有一个HandlerFunc,我们定义的函数sayhelloName就是这个HandlerFunc调用的结果,而这个类型默认实现了ServeHTTP这个接口,即我们调用了HandlerFunc(f) ,强制类型转换f成为HandlerFunc类型,这样f就拥有了ServeHTTP方法。


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

我们看一下HandlerFunc的官方注解:

HandlerFunc类型是一个适配器,允许使用普通的函数作为HTTP处理程序。如果f是具有适当签名的函数,HandlerFunc(f)是调用f的Handler。

适当的签名,由于作者水平也不深厚(毕竟我本命语言是java),猜一下指的应该是函数的参数以及返回值,也就是说:如果函数的参数是两个,分别是ResponseWriter和一个指向Request的指针,并且返回值为void类型的函数,可以强转为HandlerFunc,而最终调用的f中的Handler接口的方法也就是ServeHttp。

路由器里面存储好了相应的路由规则之后,那么具体的请求又是怎么分发的呢?请看下面的代码,默认的路由器实现了ServeHTTP:


func (mux *ServeMux) ServeHTTP(w ResponseWriter, r *Request) {
 if r.RequestURI == "*" {
 w.Header().Set("Connection", "close")
 w.WriteHeader(StatusBadRequest)
 return
 }
 h, _ := mux.Handler(r)
 h.ServeHTTP(w, r)
}