golang平滑重启(优雅重启)方案

package main
 
import (
    "context"
    "fmt"
    "os"
    "os/signal"
    "syscall"
    "time"
)
 
// 优雅地重启服务的示例函数
func gracefulRestart(ctx context.Context) error {
    // 创建系统信号监听器，监听中断（interrupt）信号
    stop := make(chan os.Signal, 1)
    signal.Notify(stop, syscall.SIGINT) // 当按下Ctrl+C时，会发送SIGINT信号
 
    // 创建一个超时上下文，当服务在2s内没有正常关闭时会强制退出
    var cancel context.CancelFunc
    ctx, cancel = context.WithTimeout(ctx, 2*time.Second)
    defer cancel()
 
    // 阻塞等待接收信号
    <-stop
 
    // 在这里可以添加代码来关闭外部资源（如数据库连接、清理文件等）
    // 这里只是简单打印信息
    fmt.Println("服务正在优雅地关闭...")
 
    // 返回nil表示不需要进行错误处理，因为优雅关闭不应该有错误返回
    return nil
}
 
func main() {
    // 创建一个空的上下文
    ctx := context.Background()
 
    // 调用优雅重启函数
    if err := gracefulRestart(ctx); err != nil {
        fmt.Println("优雅关闭失败:", err)
        os.Exit(1)
    }
 
    fmt.Println("优雅关闭成功")
    os.Exit(0)
}

这段代码演示了如何在Go语言中实现一个简单的优雅重启逻辑。它创建了一个监听系统中断信号（如Ctrl+C）的goroutine，并设置了一个超时上下文，以确定优雅关闭的最后期限。在实际应用中，你可以在gracefulRestart函数中添加关闭数据库连接、清理资源等逻辑。