swift/test/Concurrency/Runtime/async_taskgroup_asynciterator_semantics.swift at main · ian-twilightcoder/swift · GitHub

Name: swift/test/Concurrency/Runtime/async_taskgroup_asynciterator_semantics.swift at main · ian-twilightcoder/swift · GitHub
Rating: 4.7 (1473 reviews)
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
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
// RUN: %target-run-simple-swift( -target %target-swift-5.1-abi-triple -parse-as-library) | %FileCheck %s --dump-input=always

// REQUIRES: executable_test
// REQUIRES: concurrency
// REQUIRES: concurrency_runtime
// UNSUPPORTED: back_deployment_runtime

structBoom:Error{}

func boom()asyncthrows->Int{
throwBoom()
}

@available(SwiftStdlib 5.1,*)
func test_taskGroup_next()async{
letsum=awaitwithThrowingTaskGroup(of:Int.self, returning:Int.self){ group in
fornin1...10{
 group.addTask{
return n.isMultiple(of:3)?tryawaitboom(): n
}
}

varsum=0
varcatches=0
for_in1...5{
do{
whilelet r =tryawait group.next(){
 sum += r
print("add \(r) -> sum: \(sum)")
}
}catch{
 catches +=1
print("catch: \(catches)")
}
}

 // CHECK: catches with group.next(): 3
print("catches with group.next(): \(catches)")

return sum
}

 // CHECK: result with group.next(): 37
print("result with group.next(): \(sum)")
}

@available(SwiftStdlib 5.1,*)
func test_taskGroup_for_in()async{
letsum=awaitwithThrowingTaskGroup(of:Int.self, returning:Int.self){ group in
fornin1...10{
 group.addTask{
return n.isMultiple(of:3)?tryawaitboom(): n
}
}

varsum=0
varcatches=0
for_in1...5{
do{
fortryawaitrin group {
 sum += r
}
}catch{
 catches +=1
}
}

 // CHECK: catches with for-in: 3
print("catches with for-in: \(catches)")

return sum
}

 // CHECK: result with for-in: 37
print("result with for-in: \(sum)")
}

@available(SwiftStdlib 5.1,*)
func test_taskGroup_asyncIterator()async{
letsum=awaitwithThrowingTaskGroup(of:Int.self, returning:Int.self){ group in
fornin1...10{
 group.addTask{
return n.isMultiple(of:3)?tryawaitboom(): n
}
}

varsum=0
varcatches=0
for_in1...5{
variterator= group.makeAsyncIterator()
do{
whilelet r =tryawait iterator.next(){
 sum += r
}
 if try!await iterator.next()!=nil{
fatalError("Element returned from iterator after nil")
}
}catch{
 catches +=1
iftry!await iterator.next()!=nil{
fatalError("Element returned from iterator after throw")
}
}
}

 // CHECK: catches with for-in: 3
print("catches with for-in: \(catches)")

return sum
}

 // CHECK: result with async iterator: 37
print("result with async iterator: \(sum)")
}

@available(SwiftStdlib 5.1,*)
func test_taskGroup_contains()async{
letsum=awaitwithTaskGroup(of:Int.self, returning:Int.self){ group in
fornin1...4{
 group.addTask{
return n
}
}

letthree=await group.contains(3)
print("three = \(three)") // CHECK: three = true

fornin5...7{
 group.addTask{
return n
}
}

letsix=await group.contains(6)
print("six = \(six)") // CHECK: six = true


letsixAgain=await group.contains(6)
print("six again = \(sixAgain)") // CHECK: six again = false

return0
}

 // CHECK: result with async iterator: 0
print("result with async iterator: \(sum)")
}

@available(SwiftStdlib 5.1,*)
@mainstructMain{
staticfunc main()async{
awaittest_taskGroup_next()
awaittest_taskGroup_for_in()
awaittest_taskGroup_asyncIterator()
awaittest_taskGroup_contains()
}
}