浅谈Golang中创建一个简单的服务器的方法

我们知道，golang中的net/http包对网络的支持非常好，这样会让我们比较容易的建立起一个相对简单的服务器，我们来看一段代码

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

这里就建立起了一个比较简单的服务器，这个是什么意思呢？现在先大概的说一下，这里监听了一个我们本机的端口，然后接受客户端的请求，再响应给客户端相应的数据。

我们再来看一段代码

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

这两段代码作用是一样的，只是在写法上面有点区别，我们来分析一下这些Handler,HandlerFunc,DefaultServeMux 都是干什么用的。

第一段代码中，HandleFunc这个函数的作用我们看看源码。

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

在源码中，调用这个函数，又会在其中调用一个方法。

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

这个方法是定义在ServeMux结构体上的，那么这个ServeMux具体又是干什么的？现在先不用管，我们跟着源码往下看，这里调用了一个定义在ServeMux结构体上的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)

之前的大段代码就是为这两行代码做的准备，如果是按照这种写法的话，我们需要将我们自己定义的函数sayHi放到serveMux中，然而golang为我们已经做了一个默认的serveMux即DefaultServeMux，那么我们需要访问的url路径/sayHi,对应的处理方法就在DefaultServeMux一一对应了，那么到这里我们有点明白了，这个DefaultServeMux是不是就是存储我们的访问路径和要处理的方法的一个集合呢？

是的，存储的就是pattern所对应的handler，在这里我们称其为路由。

那么这个相对应的是如何工作呢，这里就需要看看http.ListenAndServe("localhost:8080", serveMux)源码了。
我们来看看具体如何实现，这里的源码比较长，我们选取比较重要的几段来看看

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

这里将监听的地址和要处理的handler传入，再往下看

go c.serve(ctx)

这里用启动goroutine来服务每一个客户端的请求，最后的处理都在c.serve这个方法中，在这个方法中，有一个地方值得注意。

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

这里又调用了一个ServeHTTP方法，然后去看看这个方法处理了一些什么。

在这个方法里面我们看到有这么一段代码

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

这里看到如果没有传入handler，那么就使用默认的DefaultServeMux,现在我们明白为什么在开头的第一段代码里面我们要传入一个nil，然后handler.ServeHTTP(rw, req)，就是调用了接口Handler的实现方法，这样就能决定某一个pattern要去对应哪一个handler，看到这里我们大概就能明白这个服务器的大概原理是怎么样的，当然这是很表面的一个分析，还有很多深层次的东西需要研究。