Golang Clean关闭信号中断

Question

我正试图找到一种有效的方法，一旦我得到操作系统中断信号，就可以关闭所有的go例程。在这里，我正在轮询事件(比如从某个队列(并在goroutine中处理它。但是，当我收到操作系统中断时，我想确保正在运行的作业在终止之前已经完成。我还需要一些额外的事情，只有在所有的goroutines完成后才能做。下面的代码对我来说似乎很好，但有什么更好/有效的方法可以做到这一点吗？

package main
import (
"fmt"
"os"
"os/signal"
"sync"
"syscall"
"time" // or "runtime"
)
func something(wg *sync.WaitGroup){
defer wg.Done()
fmt.Println("something is happening here...")
time.Sleep(10 * time.Second)
fmt.Println("job done...")
}
func main() {
c := make(chan os.Signal)
mutex := sync.Mutex{}
stop := make(chan int, 1)
signal.Notify(c, os.Interrupt, syscall.SIGINT, syscall.SIGTERM)
wg := sync.WaitGroup{}
count := 0
go func() {
<-c
currentTime := time.Now()
fmt.Println("Interrupt signal got at: ", currentTime.String())
// do not let the code shutdown without running everything we needed to do
mutex.Lock()
stop <- 1
fmt.Println("Done .. try shutting down")
wg.Wait()
// do cleanup
time.Sleep(3*time.Second)
fmt.Println("All cleanup completed .. shut down")
currentTime = time.Now()
fmt.Println("Kill at : ", currentTime.String())
mutex.Unlock()
}()
// This for loop is for reading messages from queue like sqs, and it has to be infinite loop because there might be scenarios where there are no events for period of time.
for {            
// read off of queue
select {
case stop <- 1:
fmt.Println("Not stopped yet")
wg.Add(1)
go something(&wg)
<- stop
count ++
default:
// try getting the lock before exiting (so that other cleanups are done)
mutex.Lock()
fmt.Println("Done! All jobs completed: Jobs count",count)
return
}
fmt.Println("Processing job -", count)
time.Sleep(1 * time.Second)
}
}

Answer 1

我认为您的解决方案过于复杂。这可能是正确的，但这要简单得多：

func goroutine(wg *sync.WaitGroup,stop chan struct{}) {
defer wg.Done()
for {
select {
case <-stop:
return
default:
}
// do stuff
}
}
func main() {
stop:=make(chan struct{})
// Setup signal handlers
go func() {
<-c
// This will send the stop signal to all goroutines
close(stop)
}()
// Start goroutines
...
// wait for them to finish
wg.Wait()