实例
package main
import(
"fmt"
"os"
"os/signal"
"sync"
"syscall"
"time"
)
func worker(stop <-chan bool) {
for {
select {
case <-stop:
fmt.Println("exit")
return
default:
fmt.Println("running...")
time.Sleep(3)
}
}
}
func main() {
stop := make(chan bool)
var wg sync.WaitGroup
for i := 0; i < 3; i++ {
wg.Add(1)
go func(stop <-chan bool) {
defer wg.Done()
worker(stop)
}(stop)
}
waitForSignal()
close(stop)
fmt.Println("stopping all job")
wg.Wait()
}
func waitForSignal() {
sigs := make(chan os.Signal)
signal.Notify(sigs, os.Interrupt)
signal.Notify(sigs, syscall.SIGTERM)
fmt.Println(<-sigs)
}
运行结果
pkill信号syscall.SIGTERM
...
running...
running...
running...
running...
terminated
running...
running...
stopping all job
exit
exit
exit
或者ctrl+c信号os.Interrupt
running...
interrupt
stopping all job
running...
exit
exit
exit
有疑问加站长微信联系(非本文作者)