async_task_locals_copy_to_async.swift

// REQUIRES: rdar80824152
// RUN: %target-run-simple-swift( -plugin-path %swift-plugin-dir -target %target-swift-5.1-abi-triple -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,*)
enumTL{
@TaskLocal
staticvarnumber:Int=0
@TaskLocal
staticvarother:Int=0
}

@available(SwiftStdlib 5.1,*)
@discardableResult
func printTaskLocal<V>(
 _ key:TaskLocal<V>,
 _ expected:V?=nil,
 file:String= #file, line:UInt= #line
)->V?{
letvalue= key.get()
print("\(key) (\(value)) at \(file):\(line)")
iflet expected = expected {
assert("\(expected)"=="\(value)",
"Expected [\(expected)] but found: \(value), at \(file):\(line)")
}
return expected
}

// ==== ------------------------------------------------------------------------

@available(SwiftStdlib 5.1,*)
func copyTo_async()async{
awaitTL.$number.withValue(1111){
printTaskLocal(TL.$number) // CHECK: TaskLocal<Int>(defaultValue: 0) (1111)

awaitTL.$number.withValue(2222){
awaitTL.$other.withValue(9999){
printTaskLocal(TL.$number) // CHECK: TaskLocal<Int>(defaultValue: 0) (2222)
printTaskLocal(TL.$other) // CHECK: TaskLocal<Int>(defaultValue: 0) (9999)
lethandle=Task{
printTaskLocal(TL.$number) // CHECK: TaskLocal<Int>(defaultValue: 0) (2222)
printTaskLocal(TL.$other) // CHECK: TaskLocal<Int>(defaultValue: 0) (9999)
TL.$number.withValue(3333){
printTaskLocal(TL.$number) // CHECK: TaskLocal<Int>(defaultValue: 0) (3333)
printTaskLocal(TL.$other) // CHECK: TaskLocal<Int>(defaultValue: 0) (9999)
}
}

 _ =await handle.value
}
}
}
}

@available(SwiftStdlib 5.1,*)
func copyTo_async_noWait()async{
print(#function)
TL.$number.withValue(1111){
TL.$number.withValue(2222){
TL.$other.withValue(9999){
Task{
printTaskLocal(TL.$number) // CHECK: TaskLocal<Int>(defaultValue: 0) (2222)
printTaskLocal(TL.$other) // CHECK: TaskLocal<Int>(defaultValue: 0) (9999)
TL.$number.withValue(3333){
printTaskLocal(TL.$number) // CHECK: TaskLocal<Int>(defaultValue: 0) (3333)
printTaskLocal(TL.$other) // CHECK: TaskLocal<Int>(defaultValue: 0) (9999)
}
}
}
}
}

letsecond=UInt64(100_000_000) // ns
awaitTask.sleep(2* second)
}

@available(SwiftStdlib 5.1,*)
classCustomClass{
@TaskLocal
staticvarcurrent:CustomClass?

init(){
print("init \(ObjectIdentifier(self))")
}

deinit{
print("deinit \(ObjectIdentifier(self))")
}
}

@available(SwiftStdlib 5.1,*)
func test_unstructured_retains()async{
letinstance=CustomClass()
CustomClass.$current.withValue(instance){
print("BEFORE send: \(String(reflecting:CustomClass.current))")
 // don't await on the un-structured tasks on purpose, we want to see that the tasks
 // themselves keep the object alive even if we don't hold onto them
Task{
print("in async task: \(String(reflecting:CustomClass.current))")
}
Task{
print("in async task: \(String(reflecting:CustomClass.current))")
}
print("AFTER send: \(String(reflecting:CustomClass.current))")
}

 // CHECK: init
 // CHECK: BEFORE send: Optional(main.CustomClass)
 // CHECK: in async task: Optional(main.CustomClass)
 // CHECK: in async task: Optional(main.CustomClass)
 // the deinit MUST NOT happen before the async tasks runs
 // CHECK: deinit
awaitTask.sleep(2*1_000_000_000)
}

@available(SwiftStdlib 5.1,*)
func test_unstructured_noValues()async{
awaitTask{
 // no values to copy
}.value
}

@available(SwiftStdlib 5.1,*)
func downloadImage(from url:String)asyncthrows->String{
awaitTask.sleep(10_000)
return""
}

@available(SwiftStdlib 5.1,*)
func test_unstructured_noValues_childTasks()async{
@Sendablefunc work()asyncthrows{
lethandle=Task{
tryawaitdownloadImage(from:"")
}
}

 // these child tasks have a parent pointer in their task local storage.
 // we must not copy it when performing the copyTo for a new unstructured task.
asyncletone=work()
asynclettwo=work()
asyncletthree=work()

try!await one
try!await two
try!await three

}

@available(SwiftStdlib 5.1,*)
@mainstructMain{
staticfunc main()async{
awaitcopyTo_async()
awaitcopyTo_async_noWait()
awaittest_unstructured_retains()
awaittest_unstructured_noValues()
awaittest_unstructured_noValues_childTasks()
}
}