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

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

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))
}

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

我們再來看一段代碼

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))
}

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

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

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

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

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

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

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，添加這些東西有什么用，我們再回去看看最上面的第二段代碼。

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

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

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

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

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

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

go c.serve(ctx)

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

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

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

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

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

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

以上就是本文的全部內(nèi)容，希望對大家的學(xué)習(xí)有所幫助，也希望大家多多支持腳本之家。