File: async_task_priority_current.swift

package info (click to toggle)
swiftlang 6.0.3-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 2,519,992 kB
  • sloc: cpp: 9,107,863; ansic: 2,040,022; asm: 1,135,751; python: 296,500; objc: 82,456; f90: 60,502; lisp: 34,951; pascal: 19,946; sh: 18,133; perl: 7,482; ml: 4,937; javascript: 4,117; makefile: 3,840; awk: 3,535; xml: 914; fortran: 619; cs: 573; ruby: 573
file content (71 lines) | stat: -rw-r--r-- 2,106 bytes parent folder | download
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
// RUN: %target-run-simple-swift( -Xfrontend -disable-availability-checking %import-libdispatch -parse-as-library) | %FileCheck --dump-input=always %s

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

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

import Dispatch

@available(SwiftStdlib 5.1, *)
@main struct Main {
  static func main() async {
    print("main priority: \(Task.currentPriority)") // CHECK: main priority: TaskPriority.medium
    await test_detach()
    await test_multiple_lo_indirectly_escalated()
  }
}

@available(SwiftStdlib 5.1, *)
func test_detach() async {
  let a1 = Task.currentPriority
  print("a1: \(a1)") // CHECK: a1: TaskPriority.medium

  // Note: remember to detach using a higher priority, otherwise a lower one
  // might be escalated by the get() and we could see `default` in the detached
  // task.
  await detach(priority: .userInitiated) {
    let a2 = Task.currentPriority
    print("a2: \(a2)") // CHECK: a2: TaskPriority.high
  }.get()

  let a3 = Task.currentPriority
  print("a3: \(a3)") // CHECK: a3: TaskPriority.medium
}

@available(SwiftStdlib 5.1, *)
func test_multiple_lo_indirectly_escalated() async {
  @Sendable
  func loopUntil(priority: TaskPriority) async {
    while (Task.currentPriority != priority) {
      await Task.sleep(1_000_000_000)
    }
  }

  let z = detach(priority: .background) {
    await loopUntil(priority: .userInitiated)
  }
  let x = detach(priority: .background) {
    _ = await z // waiting on `z`, but it won't complete since we're also background
    await loopUntil(priority: .userInitiated)
  }

  // detach, don't wait
  detach(priority: .userInitiated) {
    await x // escalates x, which waits on z, so z also escalates
  }

  // since `_` awaited from userInitiated on `x` we:
  // - boost `x` to `userInitiated`
  // and then since `x` waits on `z`
  // - `z` also gets boosted to `userInitiated`
  // which "unlocks" it, allowing the 'default' `await z` to complete:
  await x
  await z
  print("default done") // CHECK: default done
}