mirror of
https://github.com/apple/swift.git
synced 2025-12-14 20:36:38 +01:00
71 lines
2.2 KiB
Swift
71 lines
2.2 KiB
Swift
// RUN: %target-run-simple-swift( -target %target-swift-5.1-abi-triple %import-libdispatch -parse-as-library) | %FileCheck %s --dump-input=always
|
|
// RUN: %target-run-simple-swift( -target %target-swift-5.1-abi-triple %import-libdispatch -parse-as-library -swift-version 5 -strict-concurrency=complete -enable-upcoming-feature NonisolatedNonsendingByDefault) | %FileCheck %s --dump-input=always
|
|
// REQUIRES: swift_feature_NonisolatedNonsendingByDefault
|
|
|
|
// REQUIRES: executable_test
|
|
// REQUIRES: concurrency
|
|
// REQUIRES: libdispatch
|
|
|
|
// rdar://76038845
|
|
// REQUIRES: concurrency_runtime
|
|
// UNSUPPORTED: back_deployment_runtime
|
|
|
|
import Dispatch
|
|
|
|
@available(SwiftStdlib 5.1, *)
|
|
func test_sum_nextOnCompleted() async {
|
|
let numbers = [1, 2, 3, 4, 5]
|
|
let expected = 15 // FIXME: numbers.reduce(0, +) this hangs?
|
|
|
|
let sum = try! await withTaskGroup(of: Int.self) { group async -> Int in
|
|
for n in numbers {
|
|
group.spawn {
|
|
print(" complete group.spawn { \(n) }")
|
|
return n
|
|
}
|
|
}
|
|
|
|
// We specifically want to await on completed child tasks in this test,
|
|
// so give them some time to complete before we hit group.next()
|
|
try! await Task.sleep(nanoseconds: 2_000_000_000)
|
|
|
|
var sum = 0
|
|
while let r = await group.next() {
|
|
print("next: \(r)")
|
|
sum += r
|
|
print("sum: \(sum)")
|
|
}
|
|
|
|
assert(group.isEmpty, "Group must be empty after we consumed all tasks")
|
|
|
|
print("task group returning: \(sum)")
|
|
return sum
|
|
}
|
|
|
|
// The completions may arrive in any order, we make no strong guarantees about it:
|
|
// CON: CHECK-DAG: complete group.spawn { [[N1:[0-9]+]] }
|
|
// CON: CHECK-DAG: complete group.spawn { [[N2:[0-9]+]] }
|
|
// CON: CHECK-DAG: complete group.spawn { [[N3:[0-9]+]] }
|
|
// CON: CHECK-DAG: complete group.spawn { [[N4:[0-9]+]] }
|
|
// CON: CHECK-DAG: complete group.spawn { [[N5:[0-9]+]] }
|
|
// CON: CHECK-DAG: next: [[N1]]
|
|
// CON: CHECK-DAG: next: [[N2]]
|
|
// CON: CHECK-DAG: next: [[N3]]
|
|
// CON: CHECK-DAG: next: [[N4]]
|
|
// CON: CHECK-DAG: next: [[N5]]
|
|
|
|
// CHECK: sum: 15
|
|
//
|
|
// CHECK: task group returning: 15
|
|
|
|
// CHECK: result: 15
|
|
print("result: \(sum)")
|
|
}
|
|
|
|
@available(SwiftStdlib 5.1, *)
|
|
@main struct Main {
|
|
static func main() async {
|
|
await test_sum_nextOnCompleted()
|
|
}
|
|
}
|