Go程序如何安全退出(CTRL+C)
http://m.oschina.net/blog/125853
如果是命令行程序需要退出, CTRL+C
是最直接的方法.
C语言如何处理CTRL+C
CTRL+C
会向命令行进程发送中断信号, 在C语言的<signal.h>
中的signal
函数可以注册信号的处理函数.
signal
函数的签名如下:
void (*signal(int sig, void (*func)(int)))(int);
比如, 我们要处理CTRL+C
对应的SIGINT
信号:
#include <stdio.h> #include <stdlib.h> #include <signal.h> void sigHandle(int sig) { switch(sig) { case SIGINT: printf("sigHandle: %d, SIGINT\n", sig); break; default: printf("sigHandle: %d, OTHER\n", sig); break; } exit(1); } int main() { signal(SIGINT, sigHandle); for(;;) {} return 0; }
编译并运行程序后会进入死循环, 按CTRL+C
强制退出会看到以下的输出:
sigHandle: 2, SIGINT
当然, 直接从进程管理杀死程序就没办法收到信号的.
<signal.h>
中除了signal
函数, 还有一个raise
函数用于生成信号:
int raise(int sig);
我们在sigHandle
截获信号之后如果想重新恢复信号, 可以使用raise
函数. 但是, 要注意不要导致无穷递归signal/raise
调用.
Go语言如何处理CTRL+C
Go语言也有类似的函数signal.Notify
(在os/signal
包中), 可以过滤信号.
这是signal.Notify
自带的例子:
// Set up channel on which to send signal notifications. // We must use a buffered channel or risk missing the signal // if we're not ready to receive when the signal is sent. c := make(chan os.Signal, 1) signal.Notify(c, os.Interrupt, os.Kill) // Block until a signal is received. s := <-c fmt.Println("Got signal:", s)
signal.Notify
会将用户关注的信号转发到信道c
, 信道c
不能是阻塞的. 如果信道是缓冲不足的话, 可能会丢失信号. 如果我们不再次转发信号, 设置为1个缓冲大小就可以了.
signal.Notify
从第二个参数起是可变参数的, 用于指定要过滤的信号.
如果不指定第二个参数, 则默认是过滤全部的信号.
信号的定义一般在syscall
. syscall
包是系统相关的,
不同的操作系统信号可能有差异. 不过syscall.SIGINT
和syscall.SIGKILL
各个系统是一致的, 分别对应os.Interrupt
和os.Kill
.
下面是Go语言版完整的例子:
package main import ( "fmt" "os" "os/signal" ) func main() { c := make(chan os.Signal, 1) signal.Notify(c, os.Interrupt, os.Kill) s := <-c fmt.Println("Got signal:", s) }
go run signal.go
运行后会进入死循环, 按CTRL+C
强制退出会看到以下的输出:
Got signal: interrupt
当然, 直接从进程管理杀死程序就没办法收到信号的.
如果要恢复信号, 调用s.Signal()
. 如果要停止信号的过滤, 调用signal.Stop(c)
.
郑重声明:本站内容如果来自互联网及其他传播媒体,其版权均属原媒体及文章作者所有。转载目的在于传递更多信息及用于网络分享,并不代表本站赞同其观点和对其真实性负责,也不构成任何其他建议。