1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
|
// RUN: %target-run-simple-swift( -target %target-swift-5.1-abi-triple %import-libdispatch -parse-as-library) | %FileCheck %s --dump-input=always
// 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()
}
}
|
