blob: 817251667cbafd7ae7c87de3fe15182245cfeb8e (
plain)
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
|
use std::sync::Arc;
use log::trace;
use tokio::sync::RwLock;
use anyhow::Result;
use getset::Getters;
use crate::filestore::Artifact;
use crate::filestore::ReleaseStore;
use crate::filestore::StagingStore;
use crate::package::PackageName;
use crate::package::PackageVersionConstraint;
/// A type that merges the release store and the staging store
///
/// The stores are not actually merged (on disk or in memory), but the querying mechanism works in
/// a way where it _always_ preferes the staging store over the release store.
///
#[derive(Getters)]
pub struct MergedStores {
release: Arc<RwLock<ReleaseStore>>,
#[getset(get = "pub")]
staging: Arc<RwLock<StagingStore>>,
}
impl MergedStores {
pub fn new(release: Arc<RwLock<ReleaseStore>>, staging: Arc<RwLock<StagingStore>>) -> Self {
MergedStores { release, staging }
}
pub async fn get_artifact_by_name_and_version(&self, name: &PackageName, version: &PackageVersionConstraint) -> Result<Vec<Artifact>> {
let v = self.staging
.read()
.await
.0
.values()
.filter(|a| {
trace!("Checking {:?} == {:?} && {:?} == {:?}", a.name(), name, version, a.version());
a.name() == name && version.matches(a.version())
})
.cloned()
.collect::<Vec<_>>();
if v.is_empty() {
Ok({
self.release
.read()
.await
.0
.values()
.filter(|a| a.name() == name && version.matches(a.version()))
.cloned()
.collect()
})
} else {
Ok(v)
}
}
}
|