1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122
|
//===----------------------------------------------------------------------===//
//
// This source file is part of the Swift open source project
//
// Copyright (c) 2021-2023 Apple Inc. and the Swift project authors
// Licensed under Apache License v2.0 with Runtime Library Exception
//
// See http://swift.org/LICENSE.txt for license information
// See http://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
//
//===----------------------------------------------------------------------===//
import Basics
import Dispatch
import class Foundation.NSLock
import PackageFingerprint
import PackageModel
import struct TSCUtility.Version
public class MockPackageFingerprintStorage: PackageFingerprintStorage {
private var packageFingerprints: [PackageIdentity: [Version: [Fingerprint
.Kind: [Fingerprint.ContentType: Fingerprint]]]]
private let lock = NSLock()
public init(_ packageFingerprints: [PackageIdentity: [Version: [Fingerprint
.Kind: [Fingerprint.ContentType: Fingerprint]]]] = [:])
{
self.packageFingerprints = packageFingerprints
}
public func get(
package: PackageIdentity,
version: Version,
observabilityScope: ObservabilityScope,
callbackQueue: DispatchQueue,
callback: @escaping (Result<[Fingerprint.Kind: [Fingerprint.ContentType: Fingerprint]], Error>) -> Void
) {
if let fingerprints = self.lock.withLock({ self.packageFingerprints[package]?[version] }) {
callbackQueue.async {
callback(.success(fingerprints))
}
} else {
callbackQueue.async {
callback(.failure(PackageFingerprintStorageError.notFound))
}
}
}
public func put(
package: PackageIdentity,
version: Version,
fingerprint: Fingerprint,
observabilityScope: ObservabilityScope,
callbackQueue: DispatchQueue,
callback: @escaping (Result<Void, Error>) -> Void
) {
do {
try self.lock.withLock {
var versionFingerprints = self.packageFingerprints.removeValue(forKey: package) ?? [:]
var fingerprintsForVersion = versionFingerprints.removeValue(forKey: version) ?? [:]
var fingerprintsForKind = fingerprintsForVersion.removeValue(forKey: fingerprint.origin.kind) ?? [:]
if let existing = fingerprintsForKind[fingerprint.contentType] {
// Error if we try to write a different fingerprint
guard fingerprint == existing else {
throw PackageFingerprintStorageError.conflict(given: fingerprint, existing: existing)
}
// Don't need to do anything if fingerprints are the same
return
}
fingerprintsForKind[fingerprint.contentType] = fingerprint
fingerprintsForVersion[fingerprint.origin.kind] = fingerprintsForKind
versionFingerprints[version] = fingerprintsForVersion
self.packageFingerprints[package] = versionFingerprints
}
callbackQueue.async {
callback(.success(()))
}
} catch {
callbackQueue.async {
callback(.failure(error))
}
}
}
public func get(
package: PackageReference,
version: Version,
observabilityScope: ObservabilityScope,
callbackQueue: DispatchQueue,
callback: @escaping (Result<[Fingerprint.Kind: [Fingerprint.ContentType: Fingerprint]], Error>) -> Void
) {
self.get(
package: package.identity,
version: version,
observabilityScope: observabilityScope,
callbackQueue: callbackQueue,
callback: callback
)
}
public func put(
package: PackageReference,
version: Version,
fingerprint: Fingerprint,
observabilityScope: ObservabilityScope,
callbackQueue: DispatchQueue,
callback: @escaping (Result<Void, Error>) -> Void
) {
self.put(
package: package.identity,
version: version,
fingerprint: fingerprint,
observabilityScope: observabilityScope,
callbackQueue: callbackQueue,
callback: callback
)
}
}
|