4c7791ab24
* Fix various small errors when running in the Swift 6 language mode * Make the `TargetConfiguration` run on the main actor. * Fixed a comment * Add a comment as to why we can't make the whole NSE a main actor. * Fix the unit tests * Fix `blankLinesAtStartOfScope` swiftformat error.
45 lines
1.2 KiB
Swift
45 lines
1.2 KiB
Swift
//
|
|
// Copyright 2023, 2024 New Vector Ltd.
|
|
//
|
|
// SPDX-License-Identifier: AGPL-3.0-only OR LicenseRef-Element-Commercial
|
|
// Please see LICENSE files in the repository root for full details.
|
|
//
|
|
|
|
import Foundation
|
|
|
|
enum ExpiringTaskRunnerError: Error {
|
|
case timeout
|
|
}
|
|
|
|
actor ExpiringTaskRunner<T: Sendable> {
|
|
private var continuation: CheckedContinuation<T, Error>?
|
|
|
|
private var task: () async throws -> T
|
|
|
|
init(_ task: @escaping () async throws -> T) {
|
|
self.task = task
|
|
}
|
|
|
|
func run(timeout: Duration) async throws -> T {
|
|
try await withCheckedThrowingContinuation {
|
|
continuation = $0
|
|
|
|
Task {
|
|
try? await Task.sleep(for: timeout)
|
|
continuation?.resume(with: .failure(ExpiringTaskRunnerError.timeout))
|
|
continuation = nil
|
|
}
|
|
|
|
Task {
|
|
do {
|
|
let result = try await task()
|
|
continuation?.resume(with: .success(result))
|
|
} catch {
|
|
continuation?.resume(with: .failure(error))
|
|
}
|
|
continuation = nil
|
|
}
|
|
}
|
|
}
|
|
}
|