//開啟web服務
func test(){
http.HandleFunc("/", sayHello)
err := http.ListenAndServe(":9090",nil)
if err!=nil {
log.Fatal("ListenAndServer:",err)
}
}
func sayHello(w http.ResponseWriter, r *http.Request){
r.ParseForm()
fmt.Println("path",r.URL.Path)
fmt.Println("scheme",r.URL.Scheme)
fmt.Fprintf(w, "Hello Guest!")
}複製程式碼
在使用ListenAndServe
這個方法時,系統就會給我們指派一個路由器,DefaultServeMux
是系統預設使用的路由器,如果ListenAndServe
這個方法的第2個引數傳入nil,系統就會預設使用DefaultServeMux
。當然,這裡也可以傳入自定義的路由器。
先來看http.HandleFunc("/", sayHello)
,從HandleFunc
方法點進去,如下:
func HandleFunc(pattern string, handler func(ResponseWriter, *Request)) {
DefaultServeMux.HandleFunc(pattern, handler)
}複製程式碼
在這裡呼叫了DefaultServeMux
的HandleFunc
方法,這個方法有兩個引數,pattern
是匹配的路由規則,handler
表示這個路由規則對應的處理方法,並且這個處理方法有兩個引數。
在我們書寫的程式碼示例中,pattern
對應/
,handler
對應sayHello
,當我們在瀏覽器中輸入http://localhost:9090
時,就會觸發sayHello
方法。
我們再順著DefaultServeMux
的HandleFunc
方法繼續點下去,如下:
func (mux *ServeMux) HandleFunc(pattern string, handler func(ResponseWriter, *Request)) {
mux.Handle(pattern, HandlerFunc(handler))
}
複製程式碼
在這個方法中,路由器又呼叫了Handle
方法,注意這個Handle
方法的第2個引數,將之前傳入的handler
這個響應方法強制轉換成了HandlerFunc
型別。
這個HandlerFunc
型別到底是個什麼呢?如下:
type HandlerFunc func(ResponseWriter, *Request)
複製程式碼
看來和我們定義的SayHello
方法的型別都差不多。但是!!!
這個HandlerFunc
預設實現了ServeHTTP
介面!這樣HandlerFunc
物件就有了ServeHTTP
方法!如下:
// ServeHTTP calls f(w, r).
func (f HandlerFunc) ServeHTTP(w ResponseWriter, r *Request) {
f(w, r)
}
複製程式碼
這個細節是十分重要的,因為這一步關乎到當路由規則匹配時,相應的響應方法是否會被呼叫的問題!這個方法的呼叫時機會在下一小節中講到。
接下來,我們返回去繼續看mux
的Handle
方法,也就是這段程式碼mux.Handle(pattern, HandlerFunc(handler))
。這段程式碼做了哪些事呢?原始碼如下:
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")
}
if mux.m[pattern].explicit {
panic("http: multiple registrations for " + pattern)
}
if mux.m == nil {
mux.m = make(map[string]muxEntry)
}
mux.m[pattern] = muxEntry{explicit: true, h: handler, pattern: pattern}
if pattern[0] != '/' {
mux.hosts = true
}
// Helpful behavior:
// If pattern is /tree/, insert an implicit permanent redirect for /tree.
// It can be overridden by an explicit registration.
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}
}
}
複製程式碼
程式碼挺多,其實主要就做了一件事,向DefaultServeMux
的map[string]muxEntry
中增加對應的路由規則和handler
。
map[string]muxEntry
是個什麼鬼?
map
是一個字典物件,它儲存的是key-value
。[string]
表示這個字典的key
是string
型別的,這個key
值會儲存我們的路由規則。muxEntry
是一個例項物件,這個物件內儲存了路由規則對應的處理方法。
找到相應程式碼,如下:
//路由器
type ServeMux struct {
mu sync.RWMutex
m map[string]muxEntry //路由規則,一個string對應一個mux例項物件,map的key就是註冊的路由表示式(string型別的)
hosts bool // whether any patterns contain hostnames
}
//muxEntry
type muxEntry struct {
explicit bool
h Handler //這個路由表示式對應哪個handler
pattern string
}
//路由響應方法
type Handler interface {
ServeHTTP(ResponseWriter, *Request) //handler的路由實現器
}
複製程式碼
ServeMux
就是這個系統預設的路由器。
最後,總結一下這個部分:
1.呼叫http.HandleFunc("/", sayHello)
2.呼叫DefaultServeMux
的HandleFunc()
,把我們定義的sayHello()
包裝成HandlerFunc
型別
3.繼續呼叫DefaultServeMux
的Handle()
,向DefaultServeMux
的map[string]muxEntry
中增加路由規則和對應的handler
OK,這部分程式碼做的事就這麼多,第一部分結束。
第二部分主要就是研究這句程式碼err := http.ListenAndServe(":9090",nil)
,也就是ListenAndServe
這個方法。從這個方法點進去,如下:
func ListenAndServe(addr string, handler Handler) error {
server := &Server{Addr: addr, Handler: handler}
return server.ListenAndServe()
}
複製程式碼
在這個方法中,初始化了一個server
物件,然後呼叫這個server
物件的ListenAndServe
方法,在這個方法中,如下:
func (srv *Server) ListenAndServe() error {
addr := srv.Addr
if addr == "" {
addr = ":http"
}
ln, err := net.Listen("tcp", addr)
if err != nil {
return err
}
return srv.Serve(tcpKeepAliveListener{ln.(*net.TCPListener)})
}
複製程式碼
在這個方法中,呼叫了net.Listen("tcp", addr)
,也就是底層用TCP協議搭建了一個服務,然後監控我們設定的埠。
程式碼的最後,呼叫了srv
的Serve
方法,如下:
func (srv *Server) Serve(l net.Listener) error {
defer l.Close()
if fn := testHookServerServe; fn != nil {
fn(srv, l)
}
var tempDelay time.Duration // how long to sleep on accept failure
if err := srv.setupHTTP2_Serve(); err != nil {
return err
}
srv.trackListener(l, true)
defer srv.trackListener(l, false)
baseCtx := context.Background() // base is always background, per Issue 16220
ctx := context.WithValue(baseCtx, ServerContextKey, srv)
ctx = context.WithValue(ctx, LocalAddrContextKey, l.Addr())
for {
rw, e := l.Accept()
if e != nil {
select {
case <-srv.getDoneChan():
return ErrServerClosed
default:
}
if ne, ok := e.(net.Error); ok && ne.Temporary() {
if tempDelay == 0 {
tempDelay = 5 * time.Millisecond
} else {
tempDelay *= 2
}
if max := 1 * time.Second; tempDelay > max {
tempDelay = max
}
srv.logf("http: Accept error: %v; retrying in %v", e, tempDelay)
time.Sleep(tempDelay)
continue
}
return e
}
tempDelay = 0
c := srv.newConn(rw)
c.setState(c.rwc, StateNew) // before Serve can return
go c.serve(ctx)
}
}
複製程式碼
最後3段程式碼比較重要,也是Go語言支援高併發的體現,如下:
c := srv.newConn(rw)
c.setState(c.rwc, StateNew) // before Serve can return
go c.serve(ctx)
複製程式碼
上面那一大坨程式碼,總體意思是進入方法後,首先開了一個for
迴圈,在for
迴圈內時刻Accept請求,請求來了之後,會為每個請求建立一個Conn
,然後單獨開啟一個goroutine
,把這個請求的資料當做引數扔給這個Conn
去服務:go c.serve()
。使用者的每一次請求都是在一個新的goroutine
去服務,每個請求間相互不影響。
在conn
的serve
方法中,有一句程式碼很重要,如下:
serverHandler{c.server}.ServeHTTP(w, w.req)
複製程式碼
表示serverHandler
實現了ServeHTTP
介面,ServeHTTP
方法實現如下:
func (sh serverHandler) ServeHTTP(rw ResponseWriter, req *Request) {
handler := sh.srv.Handler
if handler == nil {
handler = DefaultServeMux
}
if req.RequestURI == "*" && req.Method == "OPTIONS" {
handler = globalOptionsHandler{}
}
handler.ServeHTTP(rw, req)
}
複製程式碼
在這裡如果handler
為空(這個handler
就可以理解為是我們自定義的路由器),就會使用系統預設的DefaultServeMux
,程式碼的最後呼叫了DefaultServeMux
的ServeHTTP()
func (mux *ServeMux) ServeHTTP(w ResponseWriter, r *Request) {
if r.RequestURI == "*" {
if r.ProtoAtLeast(1, 1) {
w.Header().Set("Connection", "close")
}
w.WriteHeader(StatusBadRequest)
return
}
h, _ := mux.Handler(r) //這裡返回的h是Handler介面物件
h.ServeHTTP(w, r) //呼叫Handler介面物件的ServeHTTP方法實際上就呼叫了我們定義的sayHello方法
}
複製程式碼
路由器接收到請求之後,如果是*
那麼關閉連結,如果不是*
就呼叫mux.Handler(r)
返回該路由對應的處理Handler
,然後執行該handler
的ServeHTTP
方法,也就是這句程式碼h.ServeHTTP(w, r)
,mux.Handler(r)
做了什麼呢?如下:
func (mux *ServeMux) Handler(r *Request) (h Handler, pattern string) {
if r.Method != "CONNECT" {
if p := cleanPath(r.URL.Path); p != r.URL.Path {
_, pattern = mux.handler(r.Host, p)
url := *r.URL
url.Path = p
return RedirectHandler(url.String(), StatusMovedPermanently), pattern
}
}
return mux.handler(r.Host, r.URL.Path)
}
func (mux *ServeMux) handler(host, path string) (h Handler, pattern string) {
mux.mu.RLock()
defer mux.mu.RUnlock()
// Host-specific pattern takes precedence over generic ones
if mux.hosts {
h, pattern = mux.match(host + path)
}
if h == nil {
h, pattern = mux.match(path)
}
if h == nil {
h, pattern = NotFoundHandler(), ""
}
return
}
func (mux *ServeMux) match(path string) (h Handler, pattern string) {
var n = 0
for k, v := range mux.m { //mux.m就是系統預設路由的map
if !pathMatch(k, path) {
continue
}
if h == nil || len(k) > n {
n = len(k)
h = v.h
pattern = v.pattern
}
}
return
}
複製程式碼
它會根據使用者請求的URL
到路由器裡面儲存的map
中匹配,匹配成功就會返回儲存的handler
,呼叫這個handler
的ServeHTTP()
就可以執行到相應的處理方法了,這個處理方法實際上就是我們剛開始定義的sayHello()
,只不過這個sayHello()
被HandlerFunc
又包了一層,因為HandlerFunc
實現了ServeHTTP
介面,所以在呼叫HandlerFunc
物件的ServeHTTP()
時,實際上在ServeHTTP ()
的內部呼叫了我們的sayHello()
。
總結一下:
1.呼叫http.ListenAndServe(":9090",nil)
2.例項化server
3.呼叫server
的ListenAndServe()
4.呼叫server
的Serve
方法,開啟for
迴圈,在迴圈中Accept請求
5.對每一個請求例項化一個Conn
,並且開啟一個goroutine
為這個請求進行服務go c.serve()
6.讀取每個請求的內容c.readRequest()
7.呼叫serverHandler
的ServeHTTP()
,如果handler
為空,就把handler
設定為系統預設的路由器DefaultServeMux
8.呼叫handler
的ServeHTTP()
=>實際上是呼叫了DefaultServeMux
的ServeHTTP()
9.在ServeHTTP()
中會呼叫路由對應處理handler
10.在路由對應處理handler
中會執行sayHello()
有一個需要注意的點:DefaultServeMux
和路由對應的處理方法handler
都實現了ServeHTTP
介面,他們倆都有ServeHTTP
方法,但是方法要達到的目的不同,在DefaultServeMux
的ServeHttp()
裡會執行路由對應的處理handler
的ServeHttp()
。