Add cancellation for Array

0xLeif · Aug 18, 2022 · 6fe18c3 · 6fe18c3
1 parent 9c15c66
commit 6fe18c3
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/Sources/Fork/ForkedArray.swift b/Sources/Fork/ForkedArray.swift
@@ -104,13 +104,17 @@ extension ForkedArray {
         case let .single(value):
             guard try await filter(value) else { return [] }
 
+            try Task.checkCancellation()
+
             return [try await using(value)]
         case let .fork(fork):
             return try await fork.merged(
                 using: { leftType, rightType in
                     async let leftOutput = try output(for: leftType, filter: filter, using: using)
                     async let rightOutput = try output(for: rightType, filter: filter, using: using)
 
+                    try Task.checkCancellation()
+
                     return try await leftOutput + rightOutput
                 }
             )