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 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154
|
/*
This source file is part of the Swift.org open source project
Copyright (c) 2020 Apple Inc. and the Swift project authors
Licensed under Apache License v2.0 with Runtime Library Exception
See http://swift.org/LICENSE.txt for license information
See http://swift.org/CONTRIBUTORS.txt for Swift project authors
*/
import Foundation
import _Concurrency
public struct Context {
private var backing: [ObjectIdentifier: Any] = [:]
#if compiler(>=5.5.2)
@available(*, deprecated, message: "Values should be Sendable")
@_disfavoredOverload
public init(dictionaryLiteral keyValuePairs: (ObjectIdentifier, Any)...) {
self.backing = Dictionary(uniqueKeysWithValues: keyValuePairs)
}
public init(dictionaryLiteral keyValuePairs: (ObjectIdentifier, Sendable)...) {
self.backing = Dictionary(uniqueKeysWithValues: keyValuePairs)
}
@available(*, deprecated, message: "Values should be Sendable")
@_disfavoredOverload
public subscript(key: ObjectIdentifier) -> Any? {
get {
return self.backing[key]
}
set {
self.backing[key] = newValue
}
}
public subscript<Value>(key: ObjectIdentifier, as type: Value.Type = Value.self) -> Value? where Value: Sendable {
get {
return self.backing[key] as? Value
}
set {
self.backing[key] = newValue
}
}
#else
public init(dictionaryLiteral keyValuePairs: (ObjectIdentifier, Any)...) {
self.backing = Dictionary(uniqueKeysWithValues: keyValuePairs)
}
@_disfavoredOverload
public subscript(key: ObjectIdentifier) -> Any? {
get {
return self.backing[key]
}
set {
self.backing[key] = newValue
}
}
public subscript<Value>(key: ObjectIdentifier, as type: Value.Type = Value.self) -> Value? {
get {
return self.backing[key] as? Value
}
set {
self.backing[key] = newValue
}
}
#endif
}
#if compiler(>=5.7)
extension Context: /* until we can remove the support for 'Any' values */ @unchecked Sendable {}
#else
#if compiler(>=5.5.2)
extension Context: UnsafeSendable {}
#endif
#endif
@available(*, deprecated, renamed: "init()")
extension Context: ExpressibleByDictionaryLiteral {
public typealias Key = ObjectIdentifier
public typealias Value = Any
}
extension Context {
#if compiler(>=5.5.2)
/// Get the value for the given type.
@available(*, deprecated, message: "Values should be Sendable")
@_disfavoredOverload
public func get<T>(_ type: T.Type = T.self) -> T {
guard let value = getOptional(type) else {
fatalError("no type \(T.self) in context")
}
return value
}
public func get<T: Sendable>(_ type: T.Type = T.self) -> T {
guard let value = self.getOptional(type) else {
fatalError("no type \(T.self) in context")
}
return value
}
/// Get the value for the given type, if present.
@available(*, deprecated, message: "Values should be Sendable")
@_disfavoredOverload
public func getOptional<T>(_ type: T.Type = T.self) -> T? {
guard let value = self[ObjectIdentifier(T.self)] else {
return nil
}
return value as? T
}
/// Get the value for the given type, if present.
public func getOptional<T: Sendable>(_ type: T.Type = T.self) -> T? {
return self[ObjectIdentifier(T.self)]
}
/// Set a context value for a type.
@available(*, deprecated, message: "Values should be Sendable")
@_disfavoredOverload
public mutating func set<T>(_ value: T) {
self[ObjectIdentifier(T.self)] = value
}
public mutating func set<Value: Sendable>(_ value: Value) {
self[ObjectIdentifier(Value.self)] = value
}
#else
/// Get the value for the given type.
public func get<T>(_ type: T.Type = T.self) -> T {
guard let value = getOptional(type) else {
fatalError("no type \(T.self) in context")
}
return value
}
/// Get the value for the given type, if present.
public func getOptional<T>(_ type: T.Type = T.self) -> T? {
guard let value = self[ObjectIdentifier(T.self)] else {
return nil
}
return value as? T
}
/// Set a context value for a type.
public mutating func set<T>(_ value: T) {
self[ObjectIdentifier(T.self)] = value
}
#endif
}
|