async_taskgroup_discarding_neverConsumingTasks.swift

// RUN: %target-run-simple-swift( -Xfrontend -disable-availability-checking -parse-as-library) | %FileCheck %s --dump-input=always
// REQUIRES: executable_test
// REQUIRES: concurrency
// REQUIRES: concurrency_runtime

// UNSUPPORTED: back_deployment_runtime
// UNSUPPORTED: freestanding

// FIXME: enable discarding taskgroup on windows; rdar://104762037
// UNSUPPORTED: OS=windows-msvc

actorWaiter{
letuntil:Int
varcount:Int

varcc:CheckedContinuation<Int,Never>?

init(until:Int){
self.until = until
self.count =0
}

func increment(){
self.count +=1
ifself.until <=self.count {
iflet cc =self.cc {
 cc.resume(returning:self.count)
}
}
}

func wait()async->Int{
ifself.until <=self.count {
returnself.count
}

returnawaitwithCheckedContinuation{ cc in
self.cc = cc
}
}
}

func test_discardingTaskGroup_neverConsume()async{
print(">>> \(#function)")
letuntil=100
letwaiter=Waiter(until: until)

print("Start tasks: \(until)")
letallTasks=awaitwithDiscardingTaskGroup(){ group in
fornin1...until {
 group.addTask{
try?awaitTask.sleep(until:.now +.milliseconds(100), clock:.continuous)
await waiter.increment()
}
}

return until
}

 // CHECK: all tasks: 100
print("all tasks: \(allTasks)")
}

func test_discardingTaskGroup_neverConsume(sleepBeforeGroupWaitAll:Duration)async{
print(">>> \(#function)")
letuntil=100
letwaiter=Waiter(until: until)

print("Start tasks: \(until)")
letallTasks=awaitwithDiscardingTaskGroup(){ group in
fornin1...until {
 group.addTask{
try?awaitTask.sleep(until:.now +.milliseconds(100), clock:.continuous)
await waiter.increment()
}
}

 // wait a little bit, so some tasks complete before we hit the implicit "wait at end of task group scope"
try?awaitTask.sleep(until:.now + sleepBeforeGroupWaitAll, clock:.continuous)

return until
}

 // CHECK: all tasks: 100
print("all tasks: \(allTasks)")
}

@mainstructMain{
staticfunc main()async{
awaittest_discardingTaskGroup_neverConsume()
awaittest_discardingTaskGroup_neverConsume(sleepBeforeGroupWaitAll:.milliseconds(500))
}
}