53 lines
1.7 KiB
Swift
53 lines
1.7 KiB
Swift
//
|
|
// Copyright 2024 Signal Messenger, LLC
|
|
// SPDX-License-Identifier: AGPL-3.0-only
|
|
//
|
|
|
|
import Foundation
|
|
|
|
public extension UnfairLock {
|
|
|
|
/// Acquires and releases the lock around the provided closure. Blocks the current thread until the lock can be
|
|
/// acquired.
|
|
@objc
|
|
@available(swift, obsoleted: 1.0)
|
|
final func withLockObjc(_ criticalSection: () -> Void) {
|
|
withLock(criticalSection)
|
|
}
|
|
|
|
/// Acquires and releases the lock around the provided closure. Blocks the current thread until the lock can be
|
|
/// acquired.
|
|
final func withLock<T>(_ criticalSection: () throws -> T) rethrows -> T {
|
|
lock()
|
|
defer { unlock() }
|
|
|
|
return try criticalSection()
|
|
}
|
|
|
|
/// Acquires and releases the lock around the provided closure. Returns without performing the closure if the lock
|
|
/// can not be acquired.
|
|
/// - Returns: `true` if the lock was acquired and the closure was invoked. `false` if the lock could not be
|
|
/// acquired.
|
|
@discardableResult
|
|
final func tryWithLock(_ criticalSection: () throws -> Void) rethrows -> Bool {
|
|
guard tryLock() else { return false }
|
|
defer { unlock() }
|
|
|
|
try criticalSection()
|
|
return true
|
|
}
|
|
|
|
/// Acquires and releases the lock around the provided closure. Returns without performing the closure if the lock
|
|
/// can not be acquired.
|
|
/// - Returns: nil if the lock could not be acquired. Otherwise, returns the returns the result of the provided
|
|
/// closure
|
|
@discardableResult
|
|
final func tryWithLock<T>(_ criticalSection: () throws -> T) rethrows -> T? {
|
|
guard tryLock() else { return nil }
|
|
defer { unlock() }
|
|
|
|
return try criticalSection()
|
|
}
|
|
|
|
}
|