async_task_locals_basic.swift

// RUN: %target-run-simple-swift( -plugin-path %swift-plugin-dir -parse-as-library %import-libdispatch) | %FileCheck %s

// REQUIRES: executable_test
// REQUIRES: concurrency
// REQUIRES: libdispatch

// rdar://76038845
// REQUIRES: concurrency_runtime
// UNSUPPORTED: back_deployment_runtime

finalclassStringLike:Sendable,ExpressibleByStringLiteral,CustomStringConvertible{
letvalue:String
init(_ value:String){
self.value = value
}
init(stringLiteral value:StringLiteralType){
self.value = value
}

vardescription:String{ value }
}

@available(SwiftStdlib 5.1,*)
enumTL{

@TaskLocal
staticvarstring:String="<undefined>"

@TaskLocal
staticvarnumber:Int=0

@TaskLocal
staticvarnever:StringLike=StringLike("<never>")

@TaskLocal
staticvarclazz:ClassTaskLocal?

@TaskLocal
staticvarforce:ForceUnwrapMe!
}

@TaskLocal
@available(SwiftStdlib 5.1,*)
varglobalTaskLocal:StringLike=StringLike("<not-set>")

@available(SwiftStdlib 5.10,*)
structLessAvailable{}

structForceUnwrapMe{}

@TaskLocal
@available(SwiftStdlib 5.10,*)
varglobalLessAvailable:LessAvailable?

@available(SwiftStdlib 5.1,*)
finalclassClassTaskLocal:Sendable{
init(){
print("clazz init \(ObjectIdentifier(self))")
}

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

@available(SwiftStdlib 5.1,*)
@discardableResult
func printTaskLocalAsync<V>(
 _ key:TaskLocal<V>,
 _ expected:V?=nil,
 file:String= #file, line:UInt= #line
)async->V?{
printTaskLocal(key, expected, file: file, line: line)
}

@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 simple()async{
printTaskLocal(TL.$number) // CHECK: TaskLocal<Int>(defaultValue: 0) (0)
TL.$number.withValue(1){
printTaskLocal(TL.$number) // CHECK-NEXT: TaskLocal<Int>(defaultValue: 0) (1)
}

 // async body
awaitTL.$number.withValue(1){
awaitTask.yield()
print("OK: \(TL.number)")
}

 // sync body
TL.$number.withValue(1){
print("OK: \(TL.number)")
}
}

@available(SwiftStdlib 5.1,*)
func simple_deinit()async{
TL.$clazz.withValue(ClassTaskLocal()){
 // CHECK: clazz init [[C:.*]]
printTaskLocal(TL.$clazz) // CHECK: TaskLocal<Optional<ClassTaskLocal>>(defaultValue: nil) (Optional(main.ClassTaskLocal))
}
 // CHECK: clazz deinit [[C]]
printTaskLocal(TL.$clazz) // CHECK: TaskLocal<Optional<ClassTaskLocal>>(defaultValue: nil) (nil)
}

structBoom:Error{
letvalue:String
}
@available(SwiftStdlib 5.1,*)
func simple_throw()async{
do{
tryTL.$clazz.withValue(ClassTaskLocal()){
throwBoom(value:"oh no!")
}
}catch{
 //CHECK: error: Boom(value: "oh no!")
print("error: \(error)")
}
}

@available(SwiftStdlib 5.1,*)
func nested()async{
printTaskLocal(TL.$string) // CHECK: TaskLocal<String>(defaultValue: <undefined>) (<undefined>)
TL.$string.withValue("hello"){
printTaskLocal(TL.$number) // CHECK-NEXT: TaskLocal<Int>(defaultValue: 0) (0)
printTaskLocal(TL.$string)// CHECK-NEXT: TaskLocal<String>(defaultValue: <undefined>) (hello)
TL.$number.withValue(2){
printTaskLocal(TL.$number) // CHECK-NEXT: TaskLocal<Int>(defaultValue: 0) (2)
printTaskLocal(TL.$string,"hello") // CHECK: TaskLocal<String>(defaultValue: <undefined>) (hello)
}
printTaskLocal(TL.$number) // CHECK-NEXT: TaskLocal<Int>(defaultValue: 0) (0)
printTaskLocal(TL.$string) // CHECK-NEXT: TaskLocal<String>(defaultValue: <undefined>) (hello)
}
printTaskLocal(TL.$number) // CHECK-NEXT: TaskLocal<Int>(defaultValue: 0) (0)
printTaskLocal(TL.$string) // CHECK-NEXT: TaskLocal<String>(defaultValue: <undefined>) (<undefined>)
}

@available(SwiftStdlib 5.1,*)
func nested_allContribute()async{
printTaskLocal(TL.$string) // CHECK: TaskLocal<String>(defaultValue: <undefined>) (<undefined>)
TL.$string.withValue("one"){
printTaskLocal(TL.$string,"one")// CHECK-NEXT: TaskLocal<String>(defaultValue: <undefined>) (one)
TL.$string.withValue("two"){
printTaskLocal(TL.$string,"two") // CHECK-NEXT: TaskLocal<String>(defaultValue: <undefined>) (two)
TL.$string.withValue("three"){
printTaskLocal(TL.$string,"three") // CHECK-NEXT: TaskLocal<String>(defaultValue: <undefined>) (three)
}
printTaskLocal(TL.$string,"two") // CHECK-NEXT: TaskLocal<String>(defaultValue: <undefined>) (two)
}
printTaskLocal(TL.$string,"one")// CHECK-NEXT: TaskLocal<String>(defaultValue: <undefined>) (one)
}
printTaskLocal(TL.$string) // CHECK-NEXT: TaskLocal<String>(defaultValue: <undefined>) (<undefined>)
}

@available(SwiftStdlib 5.1,*)
func nested_3_onlyTopContributes()async{
printTaskLocal(TL.$string) // CHECK: TaskLocal<String>(defaultValue: <undefined>) (<undefined>)
TL.$string.withValue("one"){
printTaskLocal(TL.$string)// CHECK-NEXT: TaskLocal<String>(defaultValue: <undefined>) (one)
TL.$number.withValue(2){
printTaskLocal(TL.$string) // CHECK-NEXT: TaskLocal<String>(defaultValue: <undefined>) (one)
TL.$number.withValue(3){
printTaskLocal(TL.$string) // CHECK-NEXT: TaskLocal<String>(defaultValue: <undefined>) (one)
}
printTaskLocal(TL.$string) // CHECK-NEXT: TaskLocal<String>(defaultValue: <undefined>) (one)
}
printTaskLocal(TL.$string)// CHECK-NEXT: TaskLocal<String>(defaultValue: <undefined>) (one)
}
printTaskLocal(TL.$string) // CHECK-NEXT: TaskLocal<String>(defaultValue: <undefined>) (<undefined>)
}

@available(SwiftStdlib 5.1,*)
func nested_3_onlyTopContributesAsync()async{
awaitprintTaskLocalAsync(TL.$string) // CHECK: TaskLocal<String>(defaultValue: <undefined>) (<undefined>)
awaitTL.$string.withValue("one"){
awaitprintTaskLocalAsync(TL.$string)// CHECK-NEXT: TaskLocal<String>(defaultValue: <undefined>) (one)
awaitTL.$number.withValue(2){
awaitprintTaskLocalAsync(TL.$string) // CHECK-NEXT: TaskLocal<String>(defaultValue: <undefined>) (one)
awaitTL.$number.withValue(3){
awaitprintTaskLocalAsync(TL.$string) // CHECK-NEXT: TaskLocal<String>(defaultValue: <undefined>) (one)
}
awaitprintTaskLocalAsync(TL.$string) // CHECK-NEXT: TaskLocal<String>(defaultValue: <undefined>) (one)
}
awaitprintTaskLocalAsync(TL.$string)// CHECK-NEXT: TaskLocal<String>(defaultValue: <undefined>) (one)
}
awaitprintTaskLocalAsync(TL.$string) // CHECK-NEXT: TaskLocal<String>(defaultValue: <undefined>) (<undefined>)
}

@available(SwiftStdlib 5.1,*)
func nested_3_onlyTopContributesMixed()async{
awaitprintTaskLocalAsync(TL.$string) // CHECK: TaskLocal<String>(defaultValue: <undefined>) (<undefined>)
awaitTL.$string.withValue("one"){
awaitprintTaskLocalAsync(TL.$string)// CHECK-NEXT: TaskLocal<String>(defaultValue: <undefined>) (one)
awaitTL.$number.withValue(2){
printTaskLocal(TL.$string) // CHECK-NEXT: TaskLocal<String>(defaultValue: <undefined>) (one)
TL.$number.withValue(3){
printTaskLocal(TL.$string) // CHECK-NEXT: TaskLocal<String>(defaultValue: <undefined>) (one)
}
awaitprintTaskLocalAsync(TL.$string) // CHECK-NEXT: TaskLocal<String>(defaultValue: <undefined>) (one)
}
awaitprintTaskLocalAsync(TL.$string)// CHECK-NEXT: TaskLocal<String>(defaultValue: <undefined>) (one)
}
awaitprintTaskLocalAsync(TL.$string) // CHECK-NEXT: TaskLocal<String>(defaultValue: <undefined>) (<undefined>)
}

@available(SwiftStdlib 5.1,*)
func withLocal_body_mustNotEscape()async{
varsomething="Nice"
TL.$string.withValue("xxx"){
 something ="very nice"
}
 _ = something // silence not used warning
}

@available(SwiftStdlib 5.1,*)
actorWorker{
@TaskLocal
staticvardeclaredInActor:String="<default-value>"

func setAndRead()async{
print("setAndRead") // CHECK: setAndRead
awaitWorker.$declaredInActor.withValue("value-1"){
awaitprintTaskLocalAsync(Worker.$declaredInActor) // CHECK-NEXT: TaskLocal<String>(defaultValue: <default-value>) (value-1)
}
}
}

@available(SwiftStdlib 5.1,*)
func inside_actor()async{
awaitWorker().setAndRead()
}

@available(SwiftStdlib 5.1,*)
func global_task_local()async{
await $globalTaskLocal.withValue("value-1"){
awaitprintTaskLocalAsync($globalTaskLocal) // CHECK-NEXT: TaskLocal<StringLike>(defaultValue: <not-set>) (value-1)
}
}

@available(SwiftStdlib 5.1,*)
@mainstructMain{
staticfunc main()async{
awaitsimple()
awaitsimple_deinit()
awaitsimple_throw()
awaitnested()
awaitnested_allContribute()
awaitnested_3_onlyTopContributes()
awaitnested_3_onlyTopContributesAsync()
awaitnested_3_onlyTopContributesMixed()
awaitinside_actor()
awaitglobal_task_local()
}
}