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
|
//===----------------------------------------------------------------------===//
//
// 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
) throws -> [Fingerprint.Kind: [Fingerprint.ContentType: Fingerprint]] {
guard let fingerprints = self.lock.withLock({ self.packageFingerprints[package]?[version] }) else {
throw PackageFingerprintStorageError.notFound
}
return fingerprints
}
public func put(
package: PackageIdentity,
version: Version,
fingerprint: Fingerprint,
observabilityScope: ObservabilityScope
) throws {
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
}
}
public func get(
package: PackageReference,
version: Version,
observabilityScope: ObservabilityScope
) throws -> [Fingerprint.Kind: [Fingerprint.ContentType: Fingerprint]] {
try self.get(
package: package.identity,
version: version,
observabilityScope: observabilityScope
)
}
public func put(
package: PackageReference,
version: Version,
fingerprint: Fingerprint,
observabilityScope: ObservabilityScope
) throws {
try self.put(
package: package.identity,
version: version,
fingerprint: fingerprint,
observabilityScope: observabilityScope
)
}
}
|