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

(编辑:jimmy 日期: 2024/12/23 浏览:2)

我们知道,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,看到这里我们大概就能明白这个服务器的大概原理是怎么样的,当然这是很表面的一个分析,还有很多深层次的东西需要研究。

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持。