读 "优雅关闭的 Go Web 服务器"

读 "优雅关闭的 Go Web 服务器"

GitHub 仓库:https://github.com/stayfoo/stayfoo-hub

文章《优雅的关闭 Go Web 服务器》( Go语言中文网 )写到:可以通过开启一个单独的 goroutine 拦截关闭信号,这样,当服务器真正关闭之前,可以做一些任务,做完任务再发出执行完毕的关闭号。

一些任务比如:清理某些资源;完成数据库事务;一些其他长时间的操作;退出服务的那一刻,刚好收到一个响应,为了保证所有请求完成,就可以在这里,在最大超时时间内去处理这个响应;dump进程当前状态;记录日志的动作。

启动应用,Ctrl + C 中断程序,中断信号被拦截,do something.....

go run main.go -listen-addr=5001
http: 2019/08/31 11:34:30 Server is ready to handle requests at :5001
^Chttp: 2019/08/31 11:34:32 Server is shutting down...
do something start .....  2019-08-31 11:34:32.594668 +0800 CST m=+2.337451148
do something end .....  2019-08-31 11:34:37.598248 +0800 CST m=+7.340881516
http: 2019/08/31 11:34:37 Server stopped

对文中代码做了改造,代码如下:

var listenAddr string

func init() {
    //接收端口号,默认端口号:5000
    flag.StringVar(&listenAddr, "listen-addr", ":5000", "server listen address")
}
func main() {
    flag.Parse() //外部参数解析
    listenAddr = fmt.Sprintf(":%s",listenAddr)

    logger := log.New(os.Stdout, "http: ", log.LstdFlags)

    //创建 server:
    server := newWebServer(logger)

    done := make(chan struct{}, 1)
    quit := make(chan  os.Signal, 1)

    //os.Interrupt: syscall.SIGINT
    signal.Notify(quit, os.Interrupt)
    //启动另一个 goroutine ,监听将要关闭信号:
    go shutdown(server, logger, quit, done)

    //启动 server:
    logger.Println("Server is ready to handle requests at",listenAddr)
    err := server.ListenAndServe()
    if err != nil && err != http.ErrServerClosed {
        logger.Fatalf("Could not listen on %s: %v \n", listenAddr, err)
    }

    //等待已经关闭的信号:
    <-done
    logger.Println("Server stopped")
}

//初始化 server
func newWebServer(logger *log.Logger) *http.Server {
    //路由:
    router := http.NewServeMux()
    router.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
        w.WriteHeader(http.StatusOK)
    })

    //http 服务配置:
    server := &http.Server{
        Addr: listenAddr,
        Handler: router,
        ErrorLog: logger,
        ReadTimeout: 5 * time.Second,
        WriteTimeout: 10 * time.Second,
        IdleTimeout: 15 * time.Second,
    }

    return server
}

//关闭 server
//quit: 接收关闭信号
//done: 发出已经关闭信号
func shutdown(server *http.Server, logger *log.Logger, quit <-chan os.Signal, done chan<- struct{}) {
    //等待接收到退出信号:
    <-quit
    logger.Println("Server is shutting down...")

    ctx, cancel := context.WithTimeout(context.Background(), 30 * time.Second)
    defer cancel()

    server.SetKeepAlivesEnabled(false)
    err := server.Shutdown(ctx)
    if err != nil {
        logger.Fatalf("Could not gracefully shutdown the server: %v \n", err)
    }

    //do Something :
    fmt.Println("do something start ..... ", time.Now())
    time.Sleep(5 * time.Second)
    fmt.Println("do something end ..... ", time.Now())

    close(done)
}

改造的地方

原文代码:

done := make(chan bool, 1)

改造后代码:

done := make(chan struct{}, 1)

可以看到,把关闭信号从 chan bool 改成了 chan struct{}

好处:空 struct{} 不占用空间,chan bool 占1字节。空 struct{} 更节省资源。

由下面测试代码可以知道:

s := struct {}{}
b := true
fmt.Println("struct: ", unsafe.Sizeof(s), " bool: ", unsafe.Sizeof(b)) 
//打印结果:struct:  0  bool:  1

一些其他

Go中的信号

os/signal 包对信号处理。

  • 监听收到的信号:
func Notify(c chan<- os.Signal, sig ...os.Signal)
  • 取消监听:
func Stop(c chan<- os.Signal)
  • 监听 Interrupt 信号,用户 Ctrl+C 触发:
quit := make(chan  os.Signal, 1)
//os.Interrupt: syscall.SIGINT
signal.Notify(quit, os.Interrupt)   
  • 监听所有信号:
c := make(chan os.Signal)
signal.Notify(c)
  • 监听多个指定信号:
c := make(chan os.Signal)
signal.Notify(c,os.Interrupt, os.Kill, syscall.SIGQUIT)
  • 一些信号:
//操作系统收到信号后的动作:
//Term: 表明默认动作为终止进程
//Ign: 表明默认动作为忽略该信号
//Core: 表明默认动作为终止进程同时输出core dump
//Stop: 表明默认动作为停止进程

// Signals
const (
    SIGABRT   = Signal(0x6) //调用abort函数触发,十进制值:6, Core
    SIGALRM   = Signal(0xe) //时钟定时信号,十进制值:14, Term
    SIGBUS    = Signal(0xa) //非法地址(内存地址对齐错误),十进制值:10 Core
    SIGCHLD   = Signal(0x14)//子进程结束(由父进程接收),十进制值:20  Ign
    SIGCONT   = Signal(0x13)//继续执行已经停止的进程(不能被阻塞),十进制:19 Cont
    SIGEMT    = Signal(0x7)
    SIGFPE    = Signal(0x8)//算术运行错误(浮点运算错误、除数为零等),十进制:8  Core
    SIGHUP    = Signal(0x1)//终端控制进程结束(终端连接断开),十进制:1  Term
    SIGILL    = Signal(0x4)//非法指令(程序错误、试图执行数据段、栈溢出等)  Core
    SIGINFO   = Signal(0x1d)
    SIGINT    = Signal(0x2)//用户发送INTR字符(Ctrl+C)触发,十进制值:2
    SIGIO     = Signal(0x17)
    SIGIOT    = Signal(0x6)
    SIGKILL   = Signal(0x9)//无条件结束程序(不能被捕获、阻塞或忽略)十进制:9
    SIGPIPE   = Signal(0xd)//消息管道损坏(FIFO/Socket通信时,管道未打开而进行写操作)
    SIGPROF   = Signal(0x1b)
    SIGQUIT   = Signal(0x3)//用户发送QUIT字符(Ctrl+/)触发,十进制值:3
    SIGSEGV   = Signal(0xb)//无效内存引用(试图访问不属于自己的内存空间、对只读内存空间进行写操作)
    SIGSTOP   = Signal(0x11)//停止进程(不能被捕获、阻塞或忽略)
    SIGSYS    = Signal(0xc)
    SIGTERM   = Signal(0xf)
    SIGTRAP   = Signal(0x5)
    SIGTSTP   = Signal(0x12)//停止进程(可以被捕获、阻塞或忽略)
    SIGTTIN   = Signal(0x15)//后台程序从终端中读取数据时触发
    SIGTTOU   = Signal(0x16)//后台程序向终端中写数据时触发
    SIGURG    = Signal(0x10)
    SIGUSR1   = Signal(0x1e)
    SIGUSR2   = Signal(0x1f)
    SIGVTALRM = Signal(0x1a)
    SIGWINCH  = Signal(0x1c)
    SIGXCPU   = Signal(0x18)//超过CPU时间资源限制(4.2BSD)
    SIGXFSZ   = Signal(0x19)//超过文件大小资源限制(4.2BSD)
)

0 个评论

要回复文章请先登录注册