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
|
// RUN: %target-run-simple-swift( -plugin-path %swift-plugin-dir -Xfrontend -disable-availability-checking -parse-as-library) | %FileCheck %s
// REQUIRES: executable_test
// REQUIRES: concurrency
// REQUIRES: reflection
// rdar://76038845
// REQUIRES: concurrency_runtime
// UNSUPPORTED: back_deployment_runtime
// rdar://105496007
// UNSUPPORTED: CPU=arm64e
@available(SwiftStdlib 5.1, *)
enum TL {
@TaskLocal
static var number: Int = 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 async_let_nested() async {
printTaskLocal(TL.$number) // CHECK: TaskLocal<Int>(defaultValue: 0) (0)
async let x1: () = TL.$number.withValue(2) {
async let x2 = printTaskLocal(TL.$number) // CHECK: TaskLocal<Int>(defaultValue: 0) (2)
@Sendable
func test() async {
printTaskLocal(TL.$number) // CHECK: TaskLocal<Int>(defaultValue: 0) (2)
async let x31 = printTaskLocal(TL.$number) // CHECK: TaskLocal<Int>(defaultValue: 0) (2)
_ = await x31
}
async let x3: () = test()
_ = await x2
await x3
}
_ = await x1
printTaskLocal(TL.$number) // CHECK: TaskLocal<Int>(defaultValue: 0) (0)
}
@available(SwiftStdlib 5.1, *)
func async_let_nested_skip_optimization() async {
async let x1: Int? = TL.$number.withValue(2) {
async let x2: Int? = { () async -> Int? in
async let x3: Int? = { () async -> Int? in
async let x4: Int? = { () async -> Int? in
async let x5: Int? = { () async -> Int? in
assert(TL.$number.get() == 2)
async let xx = printTaskLocal(TL.$number) // CHECK: TaskLocal<Int>(defaultValue: 0) (2)
return await xx
}()
return await x5
}()
return await x4
}()
return await x3
}()
return await x2
}
_ = await x1
}
@available(SwiftStdlib 5.1, *)
@main struct Main {
static func main() async {
await async_let_nested()
await async_let_nested_skip_optimization()
}
}
|