GO随笔-搭建一个Web服务器

用 GO 搭建一个Web服务器

有人曾和我说过，一门语言应该能够自己实现一个HTTP服务。PHP做不到，但GO却轻而易举。

只需要几行代码，GO就可以实现一个简单的HTTP服务。

package main

import (
    "fmt"
    "net/http"
)

func IndexHandler(w http.ResponseWriter, r *http.Request) {
    fmt.Fprintln(w, "hello world")
}

func main() {
    http.HandleFunc("/", IndexHandler)
    http.ListenAndServe(":8001", nil)
}

这样，即可监听8001端口，并且当有请求访问的时候完成响应。（这几乎就完成了nginx的主要功能，不过没有做负载平衡）

观察这段代码，我们基本能够从命名大概了解每个函数的作用。

http.HandleFunc("/", IndexHandler)

这相当于绑定url的函数，当请求的url是"/"时，那么将这个请求交给IndexHandler来处理。

http.ListenAndServe(":8001", nil)

顾名思义，监听8001端口并且启动服务。

func IndexHandler(w http.ResponseWriter, r *http.Request) {
    fmt.Fprintln(w, "hello world")
}

处理函数，当有请求的时候输出hello world。

通过以上分析，GO搭建一个HTTP服务只需用到一个net/http包即可。同时结合上篇文章，要完成http服务需要以下几点

• Request 用户请求的信息，用来解析用户的请求信息，包括post、get、cookie、url等信息
• Response 服务端反馈给客户端的信息
• Conn 用户每次请求的连接
• Handler 处理请求和生产返回信息的处理逻辑

前3点是一个HTTP服务必须要的结构，第4点是每一个HTTP服务的核心所在。

我们只需要了解3个问题，就知道GO是如何将Web服务运作起来。

• 如何监听接口
• 如何接受客户端请求
• 如何分配handle

如何监听接口

GO是通过一个函数 ListenAndServe 做到的。

首先初始化一个server对象，然后调用 net.Listen("tcp", addr) 底层建立TCP连接，监听我们设置的端口。

如何接收客户端请求

这块的源码

func (srv *Server) Serve(l net.Listener) error {
    defer l.Close()
    var tempDelay time.Duration // how long to sleep on accept failure
    for {
        rw, e := l.Accept()
        if e != nil {
            if ne, ok := e.(net.Error); ok && ne.Temporary() {
                if tempDelay == 0 {
                    tempDelay = 5 * time.Millisecond
                } else {
                    tempDelay *= 2
                }
                if max := 1 * time.Second; tempDelay > max {
                    tempDelay = max
                }
                log.Printf("http: Accept error: %v; retrying in %v", e, tempDelay)
                time.Sleep(tempDelay)
                continue
            }
            return e
        }
        tempDelay = 0
        c, err := srv.newConn(rw)
        if err != nil {
            continue
        }
        go c.serve()
    }
}

这段代码首先起了一个 for() 。接收请求 Accept() ，若有请求，则建立一个Conn连接，并用协程启动服务。这时再次进入下一个循环，若有请求则新建立一个连接并启动一个协程服务。若没有请求，则休息一段时间后再次循环。正是利用了go协程特性，用户的每一个请求都有一个新的goroutine去服务，互不影响，达到了天然支持高并发特性。

如何分配handle

在建立conn之后，conn会首先解析request， c.readRequest() ，然后获得相应的handle： handler := c.server.Handler ，也就是我们刚才在调用函数 ListenAndServe 时候的第二个参数，我们前面例子传递的是nil，也就是为空，那么默认获取 handler = DefaultServeMux ,那么这个变量用来做什么的呢？对，这个变量就是一个路由器，它用来匹配url跳转到其相应的handle函数，那么这个我们有设置过吗?有，我们调用的代码里面第一句不是调用了 http.HandleFunc("/", IndexHandler) 嘛。这个作用就是注册了请求/的路由规则，当请求uri为"/"，路由就会转到函数 IndexHandler ， DefaultServeMux 会调用 ServeHTTP 方法，这个方法内部其实就是调用 IndexHandler 本身，最后通过写入 response 的信息反馈到客户端。