Fix Cancellation

sideeffect-io · Jan 3, 2023 · a5bf1be · a5bf1be
1 parent 2f596e1
commit a5bf1be
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/Sources/Operators/AsyncSwitchToLatestSequence.swift b/Sources/Operators/AsyncSwitchToLatestSequence.swift
@@ -221,8 +221,11 @@ where Base.Element: AsyncSequence, Base: Sendable, Base.Element.Element: Sendabl
       guard !Task.isCancelled else { return nil }
       self.startBase()
 
-      return try await withTaskCancellationHandler { [baseTask] in
+      return try await withTaskCancellationHandler { [baseTask, state] in
         baseTask?.cancel()
+        state.withCriticalRegion {
+          $0.childTask?.cancel()
+        }
       } operation: {
         while true {
           let childTask = await withUnsafeContinuation { [state] (continuation: UnsafeContinuation<Task<ChildValue?, Never>?, Never>) in