next() can get stuck if task is already cancelled

sideeffect-io · Jan 3, 2023 · 2f596e1 · 2f596e1
1 parent 1286bef
commit 2f596e1
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/Sources/Operators/AsyncSwitchToLatestSequence.swift b/Sources/Operators/AsyncSwitchToLatestSequence.swift
@@ -218,6 +218,7 @@ where Base.Element: AsyncSequence, Base: Sendable, Base.Element.Element: Sendabl
     }
 
     public mutating func next() async rethrows -> Element? {
+      guard !Task.isCancelled else { return nil }
       self.startBase()
 
       return try await withTaskCancellationHandler { [baseTask] in