chore: support graceful shutdown (#4)

appleboy · web-flow · commit 83451988a27e · 2021-09-19T13:26:45.000+08:00
diff --git a/nsq.go b/nsq.go
@@ -240,7 +240,9 @@ func (w *Worker) Shutdown() error {
 
 	w.stopOnce.Do(func() {
 		if atomic.LoadInt32(&w.startFlag) == 1 {
+			w.q.ChangeMaxInFlight(0)
 			w.q.Stop()
+			<-w.q.StopChan
 			w.p.Stop()
 		}
 

Original file line number	Diff line number	Diff line change
`@@ -240,7 +240,9 @@ func (w *Worker) Shutdown() error {`
`240`	`240`
`241`	`241`	`w.stopOnce.Do(func() {`
`242`	`242`	`if atomic.LoadInt32(&w.startFlag) == 1 {`
	`243`	`+ w.q.ChangeMaxInFlight(0)`
`243`	`244`	`w.q.Stop()`
	`245`	`+ <-w.q.StopChan`
`244`	`246`	`w.p.Stop()`
`245`	`247`	`}`
`246`	`248`