🐛 fix: treat case for possible timer memory leak (#2488)

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