go实现程序的优雅退出思路详解

  go实现程序的优雅退出思路详解

===============

首先，何为优雅退出？

我们最常见的杀死一个应用程序是通过kill -9 或者在应用程序上执行ctrl+c。

但是程序内部如何结束的我们不得而知， 如果是暴力立刻中断，显然不是我们想要的优雅退出。

说到这，我们就大致清楚了优雅退出需要考虑哪些问题。

1. 收到请求时，当前正在处理的任务如何处理？ 让其正常执行完毕后再优雅退出。
2. 收到请求时，后来的任务和请求如何处理？ 不处理这些请求任务，等待1种的请求任务处理完毕后退出程序。

下面用一个流程图来说明此问题。

这里主要涉及到2个信号的传递

1. 程序接受到终止信号后，告知任务程序停止接受新的请求任务。
2. 任务程序运行完毕后，告知主程序优雅退出。

下面通过代码来进行演示

package main

import "fmt"
import "os"
import "time"
import "syscall"
import  "os/signal"

var stop bool = false
var finish = make(chan bool)

func main() {
    go start()

    quitSignal := make(chan os.Signal)
    signal.Notify(quitSignal, []os.Signal{syscall.SIGINT, syscall.SIGTERM}...)

    // 收到终止信号后， shutdown
    <- quitSignal
    shutdown()  
    fmt.Println("stop") 

    <- finish
    fmt.Println("exit")
}

func start() {
    fmt.Println("start")
    // 如果收到终止信号，则停止执行任务
    for !stop {
        fmt.Println("do")
        time.Sleep(time.Second)
    }
    // 任务执行完毕后，通知主协程退出程序
    finish <- true
}

func shutdown() {
    stop = true
}

原文链接: https://juejin.cn/post/7367607757208977442