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
|
// RUN: %target-run-simple-swift( -plugin-path %swift-plugin-dir -Xfrontend -disable-availability-checking -parse-as-library %import-libdispatch) | %FileCheck %s
// REQUIRES: executable_test
// REQUIRES: concurrency
// REQUIRES: libdispatch
// rdar://76038845
// REQUIRES: concurrency_runtime
// UNSUPPORTED: back_deployment_runtime
@available(SwiftStdlib 5.1, *)
enum TL {
@TaskLocal
static var number = 0
}
@available(SwiftStdlib 5.1, *)
@discardableResult
func printTaskLocal<V>(
_ key: TaskLocal<V>,
_ expected: V? = nil,
file: String = #file, line: UInt = #line
) -> V? {
let value = key.get()
print("\(key) (\(value)) at \(file):\(line)")
if let expected = expected {
assert("\(expected)" == "\(value)",
"Expected [\(expected)] but found: \(value), at \(file):\(line)")
}
return expected
}
// ==== ------------------------------------------------------------------------
@available(SwiftStdlib 5.1, *)
func groups() async {
// no value
_ = await withTaskGroup(of: Int.self) { group in
printTaskLocal(TL.$number) // CHECK: TaskLocal<Int>(defaultValue: 0) (0)
}
// no value in parent, value in child
let x1: Int = await withTaskGroup(of: Int.self) { group in
group.spawn {
printTaskLocal(TL.$number) // CHECK: TaskLocal<Int>(defaultValue: 0) (0)
// inside the child task, set a value
_ = TL.$number.withValue(1) {
printTaskLocal(TL.$number) // CHECK: TaskLocal<Int>(defaultValue: 0) (1)
}
printTaskLocal(TL.$number) // CHECK: TaskLocal<Int>(defaultValue: 0) (0)
return TL.$number.get() // 0
}
return await group.next()!
}
assert(x1 == 0)
// value in parent and in groups
await TL.$number.withValue(2) {
printTaskLocal(TL.$number) // CHECK: TaskLocal<Int>(defaultValue: 0) (2)
let x2: Int = await withTaskGroup(of: Int.self) { group in
printTaskLocal(TL.$number) // CHECK: TaskLocal<Int>(defaultValue: 0) (2)
group.spawn {
printTaskLocal(TL.$number) // CHECK: TaskLocal<Int>(defaultValue: 0) (2)
async let childInsideGroupChild = printTaskLocal(TL.$number)
_ = await childInsideGroupChild // CHECK: TaskLocal<Int>(defaultValue: 0) (2)
return TL.$number.get()
}
printTaskLocal(TL.$number) // CHECK: TaskLocal<Int>(defaultValue: 0) (2)
return await group.next()!
}
assert(x2 == 2)
}
}
@available(SwiftStdlib 5.1, *)
func taskInsideGroup() async {
Task {
print("outside") // CHECK: outside
_ = await withTaskGroup(of: Int.self) { group -> Int in
print("in group") // CHECK: in group
printTaskLocal(TL.$number) // CHECK: TaskLocal<Int>(defaultValue: 0) (0)
for _ in 0..<5 {
Task {
printTaskLocal(TL.$number)
print("some task")
}
}
return 0
}
}
// CHECK: some task
// CHECK: some task
// CHECK: some task
// CHECK: some task
await Task.sleep(5 * 1_000_000_000)
// await t.value
}
@available(SwiftStdlib 5.1, *)
@main struct Main {
static func main() async {
await groups()
await taskInsideGroup()
}
}
|