repeater context with done chan

ydb-platform · Feb 3, 2025 · 9c21116 · 9c21116
1 parent 6b25d26
commit 9c21116
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/internal/repeater/repeater.go b/internal/repeater/repeater.go
@@ -96,13 +96,15 @@ func New(
 	task func(ctx context.Context) (err error),
 	opts ...option,
 ) *repeater {
-	ctx, cancel := xcontext.WithCancel(ctx)
+	stopped := make(chan struct{})
+
+	ctx, cancel := xcontext.WithDone(ctx, stopped)
 
 	r := &repeater{
 		interval: interval,
 		task:     task,
 		cancel:   cancel,
-		stopped:  make(chan struct{}),
+		stopped:  stopped,
 		force:    make(chan struct{}, 1),
 		clock:    clockwork.NewRealClock(),
 		trace:    &trace.Driver{},