聊聊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的优雅关闭方式的详细内容,更多请关注其它相关文章!