在实际项目中我们修改了配置文件后,希望在不重启进程的情况下重新加载配置文件,这时候就需要通过信号传递来进行处理了。golang中对信号的处理主要使用os/signal包中的两个方法:一个是notify方法用来监听收到的信号;一个是 stop方法用来取消监听。下面给一个些示例。
监听信号
notify方法原型
func Notify(c chan<- os.Signal, sig ...os.Signal)
第一个参数表示接收信号的管道
第二个及后面的参数表示设置要监听的信号,如果不设置表示监听所有的信号。
package main
import (
"fmt"
"os"
"os/signal"
//"syscall"
)
func main() {
c := make(chan os.Signal)
signal.Notify(c)
//监听指定信号
//signal.Notify(c, syscall.SIGHUP, syscall.SIGUSR2)
//阻塞直至有信号传入
s := <-c
fmt.Println("get signal:", s)
}
运行该程序后,搜索到该程序的进程编号,运行kill命令就可以看到信号的输出。
终端1:
$ go run main.go
//此时处于阻塞状态,当终端2执行kill时输出如下信息
get signal: user defined signal 2
终端2:
$ ps -ef | grep main
501 839 543 0 11:16PM ttys000 0:00.03 go run main.go
501 842 839 0 11:16PM ttys000 0:00.00 /var/folders/s8/kbqz28xd3wl8q5rw8vsr129c0000gn/T/go-build384577908/command-line-arguments/_obj/exe/main
501 844 709 0 11:16PM ttys001 0:00.00 grep main
$ kill -USR2 842
当然这样的程序在实际中并没有什么用处,改进一下就可以用来接收信号并且处理一些内容了。
package main
import (
"fmt"
"os"
"os/signal"
"syscall"
"time"
)
func main() {
go signalListen()
for {
time.Sleep(10 * time.Second)
}
}
func signalListen() {
c := make(chan os.Signal)
signal.Notify(c, syscall.SIGUSR2)
for {
s := <-c
//收到信号后的处理,这里只是输出信号内容,可以做一些更有意思的事
fmt.Println("get signal:", s)
}
}
主要就是开启一个gorutine单独处理信号的监听。
停止监听
package main
import (
"fmt"
"os"
"os/signal"
"syscall"
)
func main() {
c := make(chan os.Signal)
signal.Notify(c, syscall.SIGUSR2)
//当调用了该方法后,下面的for循环内<-c就收到不到信号,就退出不了。
signal.Stop(c)
for {
s := <-c
fmt.Println("get signal:", s)
}
}
小结
golang中处理信号非常简单,但是关于信号本身需要了解的还有很多,建议可以参考《Unix高级编程》中的信号篇章。