淺談Golang中創(chuàng)建一個(gè)簡(jiǎn)單的服務(wù)器的方法

我們知道，golang中的net/http包對(duì)網(wǎng)絡(luò)的支持非常好，這樣會(huì)讓我們比較容易的建立起一個(gè)相對(duì)簡(jiǎn)單的服務(wù)器，我們來(lái)看一段代碼

func sayHi(w http.ResponseWriter, r *http.Request) {
  fmt.Fprint(w,"Hi")
}

func main() {
  http.HandleFunc("/sayHi", sayHi)
  log.Fatal(http.ListenAndServe("localhost:8080", nil))
}

這里就建立起了一個(gè)比較簡(jiǎn)單的服務(wù)器，這個(gè)是什么意思呢？現(xiàn)在先大概的說(shuō)一下，這里監(jiān)聽(tīng)了一個(gè)我們本機(jī)的端口，然后接受客戶端的請(qǐng)求，再響應(yīng)給客戶端相應(yīng)的數(shù)據(jù)。

我們?cè)賮?lái)看一段代碼

func sayHi(w http.ResponseWriter, r *http.Request) {
  fmt.Fprint(w,"Hi")
}

func main() {
  serveMux := http.DefaultServeMux
  serveMux.HandleFunc("/sayHi", sayHi)
  log.Fatal(http.ListenAndServe("localhost:8080", serveMux))
}

這兩段代碼作用是一樣的，只是在寫(xiě)法上面有點(diǎn)區(qū)別，我們來(lái)分析一下這些Handler,HandlerFunc,DefaultServeMux 都是干什么用的。

第一段代碼中，HandleFunc這個(gè)函數(shù)的作用我們看看源碼。

func HandleFunc(pattern string, handler func(ResponseWriter, *Request)) {
  DefaultServeMux.HandleFunc(pattern, handler)
}

在源碼中，調(diào)用這個(gè)函數(shù)，又會(huì)在其中調(diào)用一個(gè)方法。

func (mux *ServeMux) HandleFunc(pattern string, handler func(ResponseWriter, *Request)) {
  mux.Handle(pattern, HandlerFunc(handler))
}

這個(gè)方法是定義在ServeMux結(jié)構(gòu)體上的，那么這個(gè)ServeMux具體又是干什么的？現(xiàn)在先不用管，我們跟著源碼往下看，這里調(diào)用了一個(gè)定義在ServeMux結(jié)構(gòu)體上的Handle方法，我們來(lái)看看這個(gè)方法具體又有什么作用。

func (mux *ServeMux) Handle(pattern string, handler Handler) {
  mux.mu.Lock()
  defer mux.mu.Unlock()

  if pattern == "" {
    panic("http: invalid pattern")
  }
  if handler == nil {
    panic("http: nil handler")
  }
  if _, exist := mux.m[pattern]; exist {
    panic("http: multiple registrations for " + pattern)
  }

  if mux.m == nil {
    mux.m = make(map[string]muxEntry)
  }
  mux.m[pattern] = muxEntry{h: handler, pattern: pattern}

  if pattern[0] != '/' {
    mux.hosts = true
  }
}

這里我們看到好像在向ServeMux里面添加了一些東西，好像有pattern，還有handler，添加這些東西有什么用，我們?cè)倩厝タ纯醋钌厦娴牡诙未a。

serveMux := http.DefaultServeMux
serveMux.HandleFunc("/sayHi", sayHi)

之前的大段代碼就是為這兩行代碼做的準(zhǔn)備，如果是按照這種寫(xiě)法的話，我們需要將我們自己定義的函數(shù)sayHi放到serveMux中，然而golang為我們已經(jīng)做了一個(gè)默認(rèn)的serveMux即DefaultServeMux，那么我們需要訪問(wèn)的url路徑/sayHi,對(duì)應(yīng)的處理方法就在DefaultServeMux一一對(duì)應(yīng)了，那么到這里我們有點(diǎn)明白了，這個(gè)DefaultServeMux是不是就是存儲(chǔ)我們的訪問(wèn)路徑和要處理的方法的一個(gè)集合呢？

是的，存儲(chǔ)的就是pattern所對(duì)應(yīng)的handler，在這里我們稱(chēng)其為路由。

那么這個(gè)相對(duì)應(yīng)的是如何工作呢，這里就需要看看http.ListenAndServe("localhost:8080", serveMux)源碼了。
我們來(lái)看看具體如何實(shí)現(xiàn)，這里的源碼比較長(zhǎng)，我們選取比較重要的幾段來(lái)看看

func ListenAndServe(addr string, handler Handler) error {
  server := &Server{Addr: addr, Handler: handler}
  return server.ListenAndServe()
}

這里將監(jiān)聽(tīng)的地址和要處理的handler傳入，再往下看

go c.serve(ctx)

這里用啟動(dòng)goroutine來(lái)服務(wù)每一個(gè)客戶端的請(qǐng)求，最后的處理都在c.serve這個(gè)方法中，在這個(gè)方法中，有一個(gè)地方值得注意。

serverHandler{c.server}.ServeHTTP(w, w.req)

這里又調(diào)用了一個(gè)ServeHTTP方法，然后去看看這個(gè)方法處理了一些什么。

在這個(gè)方法里面我們看到有這么一段代碼

if handler == nil {
    handler = DefaultServeMux
  }
...
handler.ServeHTTP(rw, req)

這里看到如果沒(méi)有傳入handler，那么就使用默認(rèn)的DefaultServeMux,現(xiàn)在我們明白為什么在開(kāi)頭的第一段代碼里面我們要傳入一個(gè)nil，然后handler.ServeHTTP(rw, req)，就是調(diào)用了接口Handler的實(shí)現(xiàn)方法，這樣就能決定某一個(gè)pattern要去對(duì)應(yīng)哪一個(gè)handler，看到這里我們大概就能明白這個(gè)服務(wù)器的大概原理是怎么樣的，當(dāng)然這是很表面的一個(gè)分析，還有很多深層次的東西需要研究。

以上就是本文的全部?jī)?nèi)容，希望對(duì)大家的學(xué)習(xí)有所幫助，也希望大家多多支持億速云。