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
|
// RUN: %target-run-simple-swift( -Xfrontend -disable-availability-checking %import-libdispatch -parse-as-library ) | %FileCheck %s
// REQUIRES: executable_test
// REQUIRES: concurrency
// REQUIRES: libdispatch
// REQUIRES: concurrency_runtime
// UNSUPPORTED: back_deployment_runtime
import Dispatch
import StdlibUnittest
import _Concurrency
// For sleep
#if canImport(Darwin)
import Darwin
#elseif canImport(Glibc)
import Glibc
#elseif canImport(Android)
import Android
#endif
final class NaiveQueueExecutor: TaskExecutor {
let queue: DispatchQueue
let sem: DispatchSemaphore
init(_ sem: DispatchSemaphore, _ queue: DispatchQueue) {
print("init \(Self.self)")
self.sem = sem
self.queue = queue
}
deinit {
print("deinit \(Self.self)")
sem.signal()
}
public func enqueue(_ _job: consuming ExecutorJob) {
print("Enqueue on \(Self.self)!")
let job = UnownedJob(_job)
queue.async {
job.runSynchronously(on: self.asUnownedTaskExecutor())
}
}
@inlinable
public func asUnownedTaskExecutor() -> UnownedTaskExecutor {
print("\(Self.self).\(#function)")
return UnownedTaskExecutor(ordinary: self)
}
}
nonisolated func nonisolatedFunc(expectedQueue queue: DispatchQueue) async {
dispatchPrecondition(condition: .onQueue(queue))
print("Invoked: \(#function)")
}
@main struct Main {
static func main() async {
let queue = DispatchQueue(label: "example-queue")
let deinitSem = DispatchSemaphore(value: 0)
var executor: NaiveQueueExecutor? = NaiveQueueExecutor(deinitSem, queue)
// Task retains the executor, so it should never deinit before the task completes
// CHECK: init NaiveQueueExecutor
// The concurrency runtime invokes the...
// CHECK: NaiveQueueExecutor.asUnownedTaskExecutor
// And we enqueue on the task executor...
// CHECK: Enqueue on NaiveQueueExecutor
// CHECK: Task start
let task = Task(executorPreference: executor!) {
print("Task start")
// CHECK: Invoked: nonisolatedFunc
await nonisolatedFunc(expectedQueue: queue)
print("Task done")
}
executor = nil
print("In main: executor = nil")
await task.value
// The executor is ONLY released after the task has completed,
// regardless when the reference in main() was released.
// CHECK: Task done
// CHECK-NEXT: deinit NaiveQueueExecutor
deinitSem.wait()
print("Done")
// CHECK: Done
}
}
|
