Gin 框架学习：优雅地重启或停止

Go 1.8+ 版本可以使用 http.Server 内置的 Shutdown() 方法优雅地关闭服务。

server.go：

package main

import (
	"context"
	"log"
	"net/http"
	"os"
	"os/signal"
	"time"

	"github.com/gin-gonic/gin"
)

func main() {
	router := gin.Default()
	router.GET("/", func(c *gin.Context) {
		time.Sleep(5 * time.Second)
		c.String(http.StatusOK, "Welcome Gin Server")
	})

	srv := &http.Server{
		Addr:    ":8080",
		Handler: router,
	}

	go func() {
		// 服务连接
		if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {
			log.Fatalf("listen: %s\n", err)
		}
	}()

	// 等待中断信号以优雅地关闭服务器（设置 5 秒的超时时间）
	quit := make(chan os.Signal)
	signal.Notify(quit, os.Interrupt)
	<-quit
	log.Println("Shutdown Server ...")

	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
	defer cancel()
	if err := srv.Shutdown(ctx); err != nil {
		log.Fatal("Server Shutdown:", err)
	}
	log.Println("Server exiting")
}

启动服务后，使用浏览器访问 http://localhost:8080/ 这个链接。由于设置了 5 秒睡眠，页面会一直处于加载中，直到 5 秒后才显示出内容。在页面响应前，按 Ctrl + C 尝试关闭正在运行的服务（Linux 系统还可以使用 kill -2 pid 指令触发），会先看到打印 Shutdown Server ...，等待几秒后前台页面加载完成，最后才打印 Server exiting，服务关闭。

可以试着把代码第 17 行的 5 秒睡眠改成 10 秒或更大，重复一遍上面的操作，会有不同效果。