Golang开启http服务的三种方式

前言

都说go标准库实用，Api设计简洁。这次就用go 标准库中的net/http包实现一个简洁的http web服务器，包括三种版本。

v1最简单版

直接使用http.HandleFunc(partern,function(http.ResponseWriter,
*http.Request){})
HandleFunc接受两个参数，第一个为路由地址，第二个为处理方法。

//v1
func main() {
    http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
         w.Write([]byte("httpserver v1"))
       })
       http.HandleFunc("/bye", sayBye)
       log.Println("Starting v1 server ...")
       log.Fatal(http.ListenAndServe(":1210", nil))
 }

func sayBye(w http.ResponseWriter, r *http.Request) {
      w.Write([]byte("bye bye ,this is v1 httpServer"))
}

v2自定义Handler

查看标准库源码，v1版本实际上是调用了handle方法,传入的HandlerFunc实现了Handler的ServeHTTP方法，实际上是ServeHTTP在做http请求处理。

HandleFunc调用.png

HandleFunc实现Handler.png

Handler接口定义.png

由此我们可以自定义自己的Handler，v2版本代码如下:

// v2
func main() {
   mux := http.NewServeMux()
   mux.Handle("/", &myHandler{})
   mux.HandleFunc("/bye", sayBye)

   log.Println("Starting v2 httpserver")
   log.Fatal(http.ListenAndServe(":1210", mux))
}
type myHandler struct{}

func (*myHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
      w.Write([]byte("this is version 2"))
}
func sayBye(w http.ResponseWriter, r *http.Request) {
     w.Write([]byte("bye bye ,this is v2 httpServer"))
}

v3自定义server配置

前面对Handler开了一次刀，下面我们看看http.ListenAndServe()中有些什么秘密。

ListenAndServe.png

原来这里可以自定义http服务器配置，都在Server这个结构体中,这个对象能配置监听地址端口，配置读写超时时间，配置handler,配置请求头最大字节数...，所有稍微改造一下v2的程序得到v3版:

// v3
func main() {
    mux := http.NewServeMux()
    mux.Handle("/", &myHandler{})
    mux.HandleFunc("/bye", sayBye)

    server := &http.Server{
        Addr:         ":1210",
        WriteTimeout: time.Second * 3,            //设置3秒的写超时
        Handler:      mux,
    }
    log.Println("Starting v3 httpserver")
    log.Fatal(server.ListenAndServe())
}

type myHandler struct{}

func (*myHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
    w.Write([]byte("this is version 3"))
}

func sayBye(w http.ResponseWriter, r *http.Request) {
       // 睡眠4秒  上面配置了3秒写超时，所以访问 “/bye“路由会出现没有响应的现象
    time.Sleep(4 * time.Second)              
    w.Write([]byte("bye bye ,this is v3 httpServer"))
}

拓展一下(如何平滑关闭http服务)

在go1.8中新增了一个新特性，利用Shutdown(ctx context.Context) 优雅地关闭http服务。
文档中描述:
Shutdown 将无中断的关闭正在活跃的连接，然后平滑的停止服务。处理流程如下：

1. 首先关闭所有的监听;
2. 然后关闭所有的空闲连接;
3. 然后无限期等待连接处理完毕转为空闲，并关闭;
4. 如果提供了 带有超时的Context，将在服务关闭前返回 Context的超时错误;

利用这个特性改造一下v3版本的程序，实现一个关闭http的提示

 // 主动关闭服务器
var server *http.Server
func main() {
    
 // 一个通知退出的chan
quit := make(chan os.Signal)
signal.Notify(quit, os.Interrupt)

mux := http.NewServeMux()
mux.Handle("/", &myHandler{})
mux.HandleFunc("/bye", sayBye)

server = &http.Server{
    Addr:         ":1210",
    WriteTimeout: time.Second * 4,
    Handler:      mux,
}

go func() {
    // 接收退出信号
    <-quit
    if err := server.Close(); err != nil {
        log.Fatal("Close server:", err)
    }
}()

log.Println("Starting v3 httpserver")
err := server.ListenAndServe()
if err != nil {
    // 正常退出
    if err == http.ErrServerClosed {
        log.Fatal("Server closed under request")
    } else {
        log.Fatal("Server closed unexpected", err)
    }
}
  log.Fatal("Server exited")

}

type myHandler struct{}

func (*myHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
  w.Write([]byte("this is version 3"))
}

 // 关闭http
func sayBye(w http.ResponseWriter, r *http.Request) {
      w.Write([]byte("bye bye ,shutdown the server"))     // 没有输出
      err := server.Shutdown(nil)
      if err != nil {
        log.([]byte("shutdown the server err"))
      }
 }

尝试访问http://localhost:1210/bye 在控制台会得到以下提示结果，平滑关闭http服务成功:

成功平滑关闭.png