fix: stop iteration on context cancellation

fixes #4330
block · Feb 7, 2025 · 87d847f · 87d847f
1 parent 8c68e45
commit 87d847f
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 11 deletions.
diff --git a/examples/go/echo/types.ftl.go b/examples/go/echo/types.ftl.go
diff --git a/internal/raft/cluster.go b/internal/raft/cluster.go
@@ -239,10 +239,12 @@ func (s *ShardHandle[Q, R, E]) StateIter(ctx context.Context, query Q) (iter.Seq
 
 		for {
 			select {
+			case <-ctx.Done():
+				close(result)
+				return
 			case <-s.cluster.runningCtx.Done():
 				logger.Infof("Changes channel closed")
 				close(result)
-
 				return
 			case <-timer.C:
 				last, err := s.getLastIndex()