如何优雅地关闭http服务器？

Question

使用net/http包搭建服务器

func SayHello(w http.ResponseWriter, req *http.Request) {
    w.Write([]byte("Hello"))
}

func main() {
    http.HandleFunc("/", SayHello)
    http.ListenAndServe(":3000", nil)
    // TODO
}

问题是如何正常关闭这个服务器？因为没有界面，服务器关闭后还有后续工作，所以不考虑结束进程的方式（ctrl+c）。

Answer 1

package main

import (
    "net"
    "net/http"
    "time"
)

type HTTP struct {
    listener net.Listener
}

func (h *HTTP) ServeHTTP(w http.ResponseWriter, r *http.Request) {
    w.Write([]byte("Hello"))
}

func main() {
    l, _ := net.Listen("tcp", ":3000")
    h := new(HTTP)
    h.listener = l
    go func() {
        http.Serve(h.listener, h)
    }()
    time.Sleep(time.Second * 10)
    h.listener.Close()
    select{}
}

这里有一种方案供你参考。

Answer 2

增加一个close的路由地址,
http.HandleFunc("/close", closeServer)
然后自己写方法