[common]: fix potential leak on Sleep.

shirou · May 28, 2023 · 3e789a4 · 3e789a4
1 parent 03ae19b
commit 3e789a4
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/internal/common/sleep.go b/internal/common/sleep.go
@@ -11,6 +11,9 @@ func Sleep(ctx context.Context, interval time.Duration) error {
 	timer := time.NewTimer(interval)
 	select {
 	case <-ctx.Done():
+		if !timer.Stop() {
+			<-timer.C
+		}
 		return ctx.Err()
 	case <-timer.C:
 		return nil