Go的http包中默认路由匹配规则

# 一、执行流程 首先我们构建一个简单http server： ```go package main import ( "log" "net/http" ) func main() { http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) { w.Write([]byte("hello world")) }) log.Fatal(http.ListenAndServe(":8080", nil)) } ``` 使用`http://127.0.0.1:8080/` 就可以看到输出了 通过跟踪http.go包代码，可以发现执行流程基本如下： - 1.创建一个`Listener`监听`8080`端口 - 2.进入`for`循环并Accept请求，没有请求则处于阻塞状态 - 3.接收到请求，并创建一个conn对象，放入goroutine处理（实现高并发关键） - 4.解析请求来源信息获得请求路径等重要信息 - 5.请求ServerHTTP方法，已经通过上一步获得了ResponseWriter和Request对象 ```go func (sh serverHandler) ServeHTTP(rw ResponseWriter, req *Request) { //此handler即为http.ListenAndServe 中的第二个参数 handler := sh.srv.Handler if handler == nil { //如果handler为空则使用内部的DefaultServeMux 进行处理 handler = DefaultServeMux } if req.RequestURI == "*" && req.Method == "OPTIONS" { handler = globalOptionsHandler{} } //这里就开始处理http请求 //如果需要使用自定义的mux，就需要实现ServeHTTP方法，即实现Handler接口。 handler.ServeHTTP(rw, req) } ``` - 6.进入DefaultServeMux中的逻辑就是根据请求path在map中匹配查找handler，并交由handler处理 >http请求处理流程更多信息可以参考[《Go Web 编程 》3.3 Go如何使得Web工作](https://github.com/astaxie/build-web-application-with-golang/blob/master/zh/03.3.md) # 二、DefaultServeMux 路由匹配规则 先看几个路由规则： ```go package main import ( "log" "net/http" ) func main() { //规则1 http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) { w.Write([]byte("hello world")) }) //规则2 http.HandleFunc("/path/", func(w http.ResponseWriter, r *http.Request) { w.Write([]byte("pattern path: /path/ ")) }) //规则3 http.HandleFunc("/path/subpath", func(w http.ResponseWriter, r *http.Request) { w.Write([]byte("pattern path: /path/subpath")) }) log.Fatal(http.ListenAndServe(":8080", nil)) } ``` 情景一： 访问：`http://127.0.0.1:8080/` 返回：`hello world` 情景二： 访问：`http://127.0.0.1:8080/path` 返回：`pattern path: /path/ ` 情景三： 访问：`http://127.0.0.1:8080/path/subpath/` 返回：`pattern path: /path/ ` 情景四： 访问：`http://127.0.0.1:8080/hahaha/` 返回：`hello world` 先说明一些规则吧，再看代码是怎么实现的： 1.如果匹配路径中后带有`/`，则会自动增加一个匹配规则不带`/`后缀的，并跳转转到`path/`,解释了情景二的场景，为什么匹配到的`/path/` 2.我设置了这么多规则为什么规则一可以通用匹配未设置的路由信息，而且又不影响已经存在路由， 内部是怎么实现的？ ## 2.1 添加路由规则 先看两个struct，这是存放默认路由规则的： ```go type ServeMux struct { mu sync.RWMutex //处理并发，增加读写锁 m map[string]muxEntry //存放规则map，key即为设置的path hosts bool // whether any patterns contain hostnames（是否包含host） } type muxEntry struct { explicit bool //是否完全匹配 h Handler//相应匹配规则的handler pattern string//匹配路径 } ``` 通过跟踪`http.HandleFunc`定位到如下代码，正是往上面两个`struct`中增加规则： ```go func (mux *ServeMux) Handle(pattern string, handler Handler) { mux.mu.Lock() defer mux.mu.Unlock() if pattern == "" { panic("http: invalid pattern " + pattern) } if handler == nil { panic("http: nil handler") } //如果已经匹配到了则panic if mux.m[pattern].explicit { panic("http: multiple registrations for " + pattern) } //增加一个新的匹配规则 mux.m[pattern] = muxEntry{explicit: true, h: handler, pattern: pattern} //根据path的第一个字母判断是否有host if pattern[0] != '/' { mux.hosts = true } //！！这里看清楚 就是实现了情景二的情况 ，看判断条件 n := len(pattern) if n > 0 && pattern[n-1] == '/' && !mux.m[pattern[0:n-1]].explicit{ // If pattern contains a host name, strip it and use remaining // path for redirect. path := pattern if pattern[0] != '/' { // In pattern, at least the last character is a '/', so // strings.Index can't be -1. path = pattern[strings.Index(pattern, "/"):] } url := &url.URL{Path: path} mux.m[pattern[0:n-1]] = muxEntry{h: RedirectHandler(url.String(), StatusMovedPermanently), pattern: pattern} } } ``` 上面有个`Helpful behavior`的注释行为，就是实现了情景二的情况，他是判断如果匹配的路径中最后含有`/`，并且之前也不存在添加去除反斜杠的规则的话，就自动给他增加一个301的跳转指向`/path/` ## 2.2 查找路由规则 路由规则的查找就是从`ServeMux `中的map去匹配查找的,的到这个handler并执行，只是会有一些处理机制，比如怎么样确保访问`/path/subpath`的时候是先匹配`/path/subpath`而不是匹配`/path/`呢？ 当一个请求过来的时候，跟踪到了`mux.match`方法： >过程`mux.ServerHTTP`->`mux.Handler`->`mux.handler`->`mux.match` ```go func (mux *ServeMux) match(path string) (h Handler, pattern string) { var n = 0 for k, v := range mux.m { if !pathMatch(k, path) { continue } //如果匹配到了一个规则，并没有马上返回handler，而且继续匹配并且判断path的长度是否是最长的，这是关键！！！ if h == nil || len(k) > n { n = len(k) h = v.h pattern = v.pattern } } return } ``` 1.这里就解释了为什么设置的精确的path是最优匹配到的，因为它是根据path的长度判断。 当然也就解释了为什么`/`可以匹配所有（看`pathMatch`函数就知道了，`/`是匹配所有的，只是这是最后才被匹配成功） 2.得到了处理请求的handler，再调用`h.ServeHTTP(w, r)`，去执行相应的handler方法。 等一下，handler中哪里有`ServeHTTP`这个方法？？ 因为在调用 `http.HandleFunc`的时候已经将自定义的handler处理函数，强制转为`HandlerFunc`类型的，就拥有了`ServeHTTP`方法： ```go type HandlerFunc func(ResponseWriter, *Request) // ServeHTTP calls f(w, r). func (f HandlerFunc) ServeHTTP(w ResponseWriter, r *Request) { f(w, r) } ``` `f(w,r)`就实现了handler的执行。 > 关注"学点程序"公众号，了解更多干货内容