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
69
70
71
72
73
74
75
76
77
78
79
80
81
|
// RUN: %target-run-simple-swift( -Xfrontend -disable-availability-checking %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 asyncEcho(_ value: Int) async -> Int {
value
}
// FIXME: this is a workaround since (A, B) today isn't inferred to be Sendable
// and causes an error, but should be a warning (this year at least)
@available(SwiftStdlib 5.1, *)
struct SendableTuple2<A: Sendable, B: Sendable>: Sendable {
let first: A
let second: B
init(_ first: A, _ second: B) {
self.first = first
self.second = second
}
}
@available(SwiftStdlib 5.1, *)
func test_taskGroup_cancel_then_completions() async {
// CHECK: test_taskGroup_cancel_then_completions
print("before \(#function)")
let result: Int = await withTaskGroup(of: SendableTuple2<Int, Bool>.self) { group in
print("group cancelled: \(group.isCancelled)") // CHECK: group cancelled: false
let spawnedFirst = group.spawnUnlessCancelled {
print("start first")
await Task.sleep(1_000_000_000)
print("done first")
return SendableTuple2(1, Task.isCancelled)
}
print("spawned first: \(spawnedFirst)") // CHECK: spawned first: true
assert(spawnedFirst)
let spawnedSecond = group.spawnUnlessCancelled {
print("start second")
await Task.sleep(3_000_000_000)
print("done second")
return SendableTuple2(2, Task.isCancelled)
}
print("spawned second: \(spawnedSecond)") // CHECK: spawned second: true
assert(spawnedSecond)
group.cancelAll()
print("cancelAll") // CHECK: cancelAll
// let outerCancelled = await outer // should not be cancelled
// print("outer cancelled: \(outerCancelled)") // COM: CHECK: outer cancelled: false
// print("group cancelled: \(group.isCancelled)") // COM: CHECK: outer cancelled: false
let one = await group.next()
print("first: \(one)") // CHECK: first: Optional(main.SendableTuple2<Swift.Int, Swift.Bool>(first: 1,
let two = await group.next()
print("second: \(two)") // CHECK: second: Optional(main.SendableTuple2<Swift.Int, Swift.Bool>(first: 2,
let none = await group.next()
print("none: \(none)") // CHECK: none: nil
return (one?.first ?? 0) + (two?.first ?? 0) + (none?.first ?? 0)
}
print("result: \(result)") // CHECK: result: 3
}
@available(SwiftStdlib 5.1, *)
@main struct Main {
static func main() async {
await test_taskGroup_cancel_then_completions()
}
}
|
