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

本文来自:Segmentfault

感谢作者:达闻西

查看原文:浅谈Golang中创建一个简单的服务器

• 编辑
• 预览