聊聊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函数通知操作系统程序关闭请求已经处理完毕了。

在主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函数中,完成程序退出工作。

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

程序监听的信号

在Linux系统中,一般有以下几种信号:

信号名称信号描述
SIGINT终端中断字符,通常通过 Control+C 发出
SIGTERM终止信号,通常用来请求程序自行停止运行
SIGHUP挂起信号,通常表示某个终端连接已经中断,需要重新读取配置文件等
SIGUSR1-31自定义信号
SIGKILL杀死进程信号,无法被捕捉或忽略,只能暴力杀死进程
SIGSTOP停止进程信号,无法被捕捉或忽略,只能暴力停止进程
SIGCONT继续运行进程
SIGPIPE管道破裂信号,通常表示读取数据时对方已经关闭了连接
SIGALRM闹钟信号,用来定时触发某个操作
SIGCHLD子进程结束信号

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

优雅关闭的作用

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

总结

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

以上就是聊聊golang的优雅关闭方式的详细内容,更多请关注其它相关文章!