mirror of
https://github.com/apple/swift-foundation.git
synced 2025-05-18 11:19:30 +08:00
a7692238a8
This adds the necessary guards and includes for the Android modules. While the module does not compile currently due to nullability differences (and in some cases missing declarations), this at least brings the module to a point where we can start working on the errors and differences to create a maintainable codebase for Android.
88 lines
2.5 KiB
Swift
88 lines
2.5 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(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)
|
|
fileprivate typealias PlatformKey = pthread_key_t
|
|
#elseif USE_TSS
|
|
fileprivate typealias PlatformKey = tss_t
|
|
#elseif canImport(WinSDK)
|
|
fileprivate typealias PlatformKey = DWORD
|
|
#endif
|
|
|
|
struct Key<Value> {
|
|
fileprivate let key: PlatformKey
|
|
|
|
init() {
|
|
#if canImport(Darwin) || os(Android) || canImport(Glibc)
|
|
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)
|
|
#endif
|
|
}
|
|
}
|
|
|
|
private static subscript(_ key: PlatformKey) -> UnsafeMutableRawPointer? {
|
|
get {
|
|
#if canImport(Darwin) || os(Android) || canImport(Glibc)
|
|
pthread_getspecific(key)
|
|
#elseif USE_TSS
|
|
tss_get(key)
|
|
#elseif canImport(WinSDK)
|
|
FlsGetValue(key)
|
|
#endif
|
|
}
|
|
|
|
set {
|
|
#if canImport(Darwin) || os(Android) || canImport(Glibc)
|
|
pthread_setspecific(key, newValue)
|
|
#elseif USE_TSS
|
|
tss_set(key, newValue)
|
|
#elseif canImport(WinSDK)
|
|
FlsSetValue(key, 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()
|
|
}
|
|
}
|
|
}
|