在最近的项目中，使用了go 搭建了web服务器，所以”优雅的关闭服务器”，经过查资料，发现go在1.8之后，http包已经有Shutdown()方法，但是使用起来还是有一些要注意的地方。
如果对go很熟悉的，对这些包很熟悉的人，肯定不会碰到这个问题。只怪我太菜。。。碰到了坑。下面直接上代码。
查询资料的时候，在网上看到这样的代码
1
2
3
4
5
6
7
8
9
10
11
12
13
| func gracefullShutdown(server *http.Server, logger *log.Logger, quit <-chan os.Signal, done chan<- bool) {
<-quit
logger.Println("Server is shutting down...")
ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
defer cancel()
server.SetKeepAlivesEnabled(false)
if err := server.Shutdown(ctx); err != nil {
logger.Fatalf("Could not gracefully shutdown the server: %v\n", err)
}
close(done)
}
|
像这种写法，是启动了别的goroutin去执行gracefullShoudown，所以需要额外的一个done channel来阻塞主线程。我们的http服务器必然会调用这个ListenAndServer(如下)，当调用了gracefullShutdown中server.Shutdown(ctx)，ListenAndServer会直接返回，而主线程直接结束了的话，别的线程也就会跟着一起结束。也就没有了等待服务器关闭这么一说法了。
1
2
3
4
5
6
7
8
9
10
11
| router := http.NewServeMux()
router.HandleFunc("/healthcheck", healthcheck)
port := fmt.Sprint(":8080")
srv := &http.Server{
Addr: port,
Handler: router,
}
if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {
logger.Error("[HTTP] server listen on", port, ", err:", err)
}
|
** 注意：server.ListenAndServe() 方法在 Shutdown 时会立刻返回，Shutdown 方法会阻塞至所有连接闲置或 context 完成，所以 Shutdown 的方法要写在主 goroutine 中或者。如果像上面一样使用在新的goroutine中，则需要自己写好阻塞主携程的方法。**
最后附上完整使用新goroutine监听关闭服务器的demo：
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
| package main
import (
"context"
"fmt"
"net/http"
"os"
"os/signal"
"syscall"
"time"
)
func gracefullShutdown(server *http.Server, quit <-chan os.Signal, done chan<- bool) {
<-quit
// 30s，让服务器做一些清理操作
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
defer cancel()
server.SetKeepAlivesEnabled(false)
if err := server.Shutdown(ctx); err != nil {
fmt.Println("server.Shutdown err ..... ", err)
}
//do Something ：
fmt.Println("do something start ..... ", time.Now())
time.Sleep(5 * time.Second)
fmt.Println("do something end ..... ", time.Now())
close(done)
}
func main() {
var (
done = make(chan bool, 1)
quit = make(chan os.Signal, 1)
)
router := http.NewServeMux()
port := fmt.Sprint(":36010")
srv := &http.Server{
Addr: port,
Handler: router,
}
// 给信号挖个坑，如果抓到这些信号就退出程序
signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
go gracefullShutdown(srv, quit, done)
fmt.Println("[HTTP] http server listen", port)
if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {
fmt.Println("ListenAndServe err ..... ", err)
os.Exit(1)
}
<-done
fmt.Println("[HTTP] Showdown end")
}
|
