52 lines
1.7 KiB
Swift
52 lines
1.7 KiB
Swift
|
//
|
||
|
|
// Copyright (c) 2021 Open Whisper Systems. All rights reserved.
|
||
|
|
//
|
||
|
|
|
||
|
|
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()
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|