fix worker stop

zc2638 · Jan 12, 2024 · caf0eee · caf0eee
1 parent ccd83fc
commit caf0eee
Showing 1 changed file with 6 additions and 7 deletions.
diff --git a/core/worker/worker.go b/core/worker/worker.go
@@ -97,18 +97,17 @@ func (w *Worker) Run(ctx context.Context) error {
 						"wait", waitTimes,
 					)
 
-					select {
-					case <-wCtx.Done():
-						return err
-					default:
-					}
-
 					waitSec := math.Pow(2, float64(waitTimes))
 					if waitSec > 60 {
 						waitSec = constant.DefaultWaitTime
 					}
 					waitTimes++
-					time.Sleep(time.Second * time.Duration(waitSec))
+
+					select {
+					case <-time.After(time.Second * time.Duration(waitSec)):
+					case <-wCtx.Done():
+						return err
+					}
 					continue
 				}
 				waitTimes = 0