聊聊golang的优雅关闭方式

在开发使用golang的时候，一般都会遇到需要关闭或停止运行的情况。比如，程序运行结束或者出现异常，这时需要优雅地关闭。

优雅关闭可以让程序停止运行的同时，能够让已经启动的任务处理完毕，避免数据丢失和资源泄漏等问题，也不会对客户端造成不必要的损失。本文就讨论一下golang的优雅关闭方式。

优雅关闭方法

在golang中，优雅关闭的实现不难，我们可以通过设置信号量或者channel的方式进行，具体的实现方法如下：

通过信号量实现

package main

import (
    "log"
    "os"
    "os/signal"
    "syscall"
    "time"
)

func main() {
    stop := make(chan os.Signal, 1)
    signal.Notify(stop, syscall.SIGINT, syscall.SIGTERM)

    log.Println("Starting...")

    go func() {
        <-stop
        log.Println("Stopping...")

        // 处理关闭逻辑
        time.Sleep(3 * time.Second)

        log.Println("Graceful shutdown finished")
        os.Exit(0)
    }()

    // 处理业务逻辑
    for {
        log.Println("Running...")
        time.Sleep(1 * time.Second)
    }
}

这段代码中通过make创建了一个类型为os.Signal，长度为1的信号缓冲channel，然后通过signal包中的Notify函数告诉操作系统，我们想要监听的信号是SIGINT和SIGTERM，这两个信号都可以代表程序的关闭请求。

接着，我们启动了一个goroutine，监听stop这个channel，一旦接受到SIGINT或SIGTERM信号，就会触发这个goroutine的执行。在执行过程中，我们处理关闭逻辑，这里用了time.Sleep模拟了一个比较耗时的操作，以便更好的测试我们的程序是否正常结束。最后通过os.Exit函数通知操作系统程序关闭请求已经处理完毕了。

立即学习“go语言免费学习笔记（深入）”；

在主goroutine中，我们处理了业务逻辑，每秒钟打印一次"Running..."，这里只是一个简单的示例；通常情况下，我们在处理业务逻辑时，需要定时保存数据和释放资源等。

通过channel实现

package main

import (
    "log"
    "os"
    "time"
)

func main() {
    done := make(chan bool)

    log.Println("Starting...")

    go func() {
        osSignals := make(chan os.Signal, 1)
        signal.Notify(osSignals, syscall.SIGINT, syscall.SIGTERM)

        select {
        case <-osSignals:
            log.Println("Stopping...")

            // 处理关闭逻辑
            time.Sleep(3 * time.Second)

            log.Println("Graceful shutdown finished")
            done <- true
        }
    }()

    // 处理业务逻辑
    for {
        log.Println("Running...")
        time.Sleep(1 * time.Second)
        select {
        case <-done:
            return // 正常退出
        default:
            continue
        }
    }
}

这段代码中，我们创建了一个bool类型的done变量，表示我们的程序是否已经完成，然后在主goroutine中处理业务逻辑，定时打印"Running..."，并通过select函数监听done。如果done变量被设为true，程序就结束运行，返回到main函数中，完成程序退出工作。

Bandy AI

全球领先的电商设计Agent

下载

在另外一个goroutine中，我们通过make创建了一个类型为os.Signal，长度为1的信号缓冲channel，然后通过signal包中的Notify函数告诉操作系统，我们想要监听的信号是SIGINT和SIGTERM。接下来，通过select函数监听osSignals，如果接收到SIGINT或SIGTERM信号，就会触发特定操作。

程序监听的信号

在Linux系统中，一般有以下几种信号：

通常情况下，我们只需要监听SIGINT和SIGTERM信号即可。

优雅关闭的作用

通过优雅关闭，我们可以避免因未能处理完毕的请求和操作而导致的数据丢失、资源泄漏等问题，对于长时间运行的程序和提供服务的应用程序尤其重要。特别是在分布式系统和微服务架构中，优雅关闭能够有效保证系统稳定性和性能，如果没有针对性的问题处理函数，程序在接到停止信号后，可能会武断的结束或触发崩溃，给客户端造成困惑甚至不必要的损失。

总结

通过本文，我们学习到了如何使用golang的channel和os/signal包进行优雅关闭，以及通过监听信号来完成优雅关闭的操作。在实际开发中，应用程序的正常关闭是一项非常重要的任务，通过优雅关闭能够有效提高程序的稳定性，避免资源浪费和数据丢失等问题。