Handle 'done' channel in keep-alive logic.

streamingfast · Feb 13, 2025 · 2a2d0d0 · 2a2d0d0
1 parent 147cdb2
commit 2a2d0d0
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/service/tier2.go b/service/tier2.go
@@ -464,6 +464,8 @@ excludable:
 					select {
 					case <-ctx.Done():
 						return
+					case <-done:
+						return
 					case <-time.After(keepAliveDelay):
 						s.logger.Info("keep alive timer expired, calling keep alive", zap.String("worker_id", workerID), zap.Bool("keep", false))
 						_, err := s.remoteWorkerClient.KeepAlive(ctx,