mirror of
https://github.com/apple/swift-foundation.git
synced 2025-05-18 11:19:30 +08:00
aecc1b158c
* Get FoundationEssentials building Adding the missing musl imports to get FoundationEssentials building for the Swift static SDKs again. Also providing an option to disable building the macros. The macros aren't necessary for building the library and will not be run as part of the static SDK. No need to bloat the SDK or build times further. For Swift 6, the macros should be provided by the toolchain since the toolchain and SDK are current revlocked due to swiftmodules. * Get FoundationInternationalization building Adding the missing Musl imports to get FoundationInternationalization building for the static SDK.
98 lines
2.9 KiB
Swift
98 lines
2.9 KiB
Swift
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This source file is part of the Swift.org open source project
|
|
//
|
|
// Copyright (c) 2023 Apple Inc. and the Swift project authors
|
|
// Licensed under Apache License v2.0 with Runtime Library Exception
|
|
//
|
|
// See https://swift.org/LICENSE.txt for license information
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#if canImport(Darwin)
|
|
import Darwin
|
|
#elseif os(Android)
|
|
import Bionic
|
|
#elseif canImport(Glibc)
|
|
import Glibc
|
|
#elseif canImport(Musl)
|
|
import Musl
|
|
#elseif canImport(WinSDK)
|
|
import WinSDK
|
|
#elseif canImport(threads_h)
|
|
internal import threads_h
|
|
#elseif canImport(threads)
|
|
internal import threads
|
|
#endif
|
|
|
|
struct _ThreadLocal {
|
|
#if canImport(Darwin) || os(Android) || canImport(Glibc) || canImport(Musl)
|
|
fileprivate typealias PlatformKey = pthread_key_t
|
|
#elseif USE_TSS
|
|
fileprivate typealias PlatformKey = tss_t
|
|
#elseif canImport(WinSDK)
|
|
fileprivate typealias PlatformKey = DWORD
|
|
#elseif os(WASI)
|
|
fileprivate typealias PlatformKey = UnsafeMutablePointer<UnsafeMutableRawPointer?>
|
|
#endif
|
|
|
|
struct Key<Value> {
|
|
fileprivate let key: PlatformKey
|
|
|
|
init() {
|
|
#if canImport(Darwin) || os(Android) || canImport(Glibc) || canImport(Musl)
|
|
var key = PlatformKey()
|
|
pthread_key_create(&key, nil)
|
|
self.key = key
|
|
#elseif USE_TSS
|
|
var key = PlatformKey()
|
|
tss_create(&key, nil)
|
|
self.key = key
|
|
#elseif canImport(WinSDK)
|
|
key = FlsAlloc(nil)
|
|
#elseif os(WASI)
|
|
key = UnsafeMutablePointer<UnsafeMutableRawPointer?>.allocate(capacity: 1)
|
|
#endif
|
|
}
|
|
}
|
|
|
|
private static subscript(_ key: PlatformKey) -> UnsafeMutableRawPointer? {
|
|
get {
|
|
#if canImport(Darwin) || os(Android) || canImport(Glibc) || canImport(Musl)
|
|
pthread_getspecific(key)
|
|
#elseif USE_TSS
|
|
tss_get(key)
|
|
#elseif canImport(WinSDK)
|
|
FlsGetValue(key)
|
|
#elseif os(WASI)
|
|
key.pointee
|
|
#endif
|
|
}
|
|
|
|
set {
|
|
#if canImport(Darwin) || os(Android) || canImport(Glibc) || canImport(Musl)
|
|
pthread_setspecific(key, newValue)
|
|
#elseif USE_TSS
|
|
tss_set(key, newValue)
|
|
#elseif canImport(WinSDK)
|
|
FlsSetValue(key, newValue)
|
|
#elseif os(WASI)
|
|
key.pointee = newValue
|
|
#endif
|
|
}
|
|
}
|
|
|
|
static subscript<Value>(_ key: Key<Value>) -> Value? {
|
|
self[key.key]?.load(as: Value.self)
|
|
}
|
|
|
|
static func withValue<Value, R>(_ value: inout Value, for key: Key<Value>, _ block: () throws -> R) rethrows -> R {
|
|
precondition(Self[key.key] == nil, "Not allowed to set the value for a key within the subscope of that key")
|
|
return try withUnsafeMutablePointer(to: &value) {
|
|
Self[key.key] = UnsafeMutableRawPointer($0)
|
|
defer { Self[key.key] = nil }
|
|
return try block()
|
|
}
|
|
}
|
|
}
|