我们知道,在使用Linux操作系统运行程序时,我们经常会使用 “Kill -9” 和 “Ctrl + C” 的形式退出程序。
所以,我们可以在程序中监听来自系统的退出信号,当主函数接收到退出信号时,再退出程序。像这样:
func main() {
//接收退出信号的chanel
sig := make(chan os.Signal)
//指定哪些信号可以转发到chanel,如果没有列出,会将所有信号传递到chanel
signal.Notify(sig, syscall.SIGINT, syscall.SIGKILL)
fmt.Printf("接收到的信号: %v \n", <-sig)
fmt.Println("主goroutine结束")
}
现在我们编译并运行程序,并通过按下 “Ctrl + C” 退出程序
这里监听到了 中断信号interrupt,我们成功接收到系统发送来的退出信号。但是现实中我们的程序中往往不只有一个主Goroutine,还有很多其他的子Goroutine运行着,那么我们就应该先通知子Goroutine结束任务,再结束整个程序的运行,像这样:
func main() {
sig := make(chan os.Signal)
signal.Notify(sig, syscall.SIGINT, syscall.SIGKILL)
//通过context通知子goroutine结束
ctx, cancel := context.WithCancel(context.Background())
//使用wg确认所有子goroutine已经结束
wg := sync.WaitGroup{}
wg.Add(10)
for i := 0; i < 10; i++ {
go func(ctx context.Context, wg *sync.WaitGroup, i int) {
defer wg.Done()
//do something
for {
select {
case <-ctx.Done():
fmt.Printf("goroutine:%v 结束\n", i)
return
}
}
}(ctx, &wg, i)
}
fmt.Printf("接收到的信号: %v \n", <-sig)
cancel()
wg.Wait()
fmt.Println("主goroutine结束")
}
现在编译运行,并通过按下 “Ctrl + C” 退出程序