red/cdb
red/cdb
1
0
Fork 0
Code Issues 3 Pull requests Projects Releases Packages 1 Wiki Activity

add flows

Browse source
This commit is contained in:
JMARyA 2024-09-12 16:55:36 +02:00
parent 94459d5481
commit 7b6d58be33
Signed by: jmarya
GPG key ID: 901B2ADDF27C2263
4 changed files with 290 additions and 26 deletions
Download patch file Download diff file Expand all files Collapse all files

58
src/variant.rs
View file

@ -196,25 +196,29 @@ impl Variant {
}
pub async fn get_unique_origins(&self) -> Vec<String> {
Transaction::unique(
doc! {
"item": &self.item,
"variant": &self.variant
},
"origin",
unique_flows(
&Transaction::unique(
doc! {
"item": &self.item,
"variant": &self.variant
},
"origin",
)
.await,
)
.await
}
pub async fn get_unique_destinations(&self) -> Vec<String> {
Transaction::unique(
doc! {
"item": &self.item,
"variant": &self.variant
},
"consumed.destination",
unique_flows(
&Transaction::unique(
doc! {
"item": &self.item,
"variant": &self.variant
},
"consumed.destination",
)
.await,
)
.await
}
pub async fn price_history_by_origin(&self, origin: &str) -> Vec<Price> {
@ -287,3 +291,29 @@ impl Variant {
})
}
}
pub fn unique_flows(i: &[String]) -> Vec<String> {
let mut unique_vec: Vec<String> = Vec::new();
for s in i {
// Check if the string starts with "flow::"
if let Some(suffix) = s.strip_prefix("flow::") {
// Extract the part after "flow::" and split on "::" to get the kind (ignoring id)
let parts: Vec<&str> = suffix.split("::").collect();
if let Some(kind) = parts.get(0) {
// Build the common prefix "flow::kind"
let common_prefix = format!("flow::{}", kind);
if !unique_vec.contains(&common_prefix) {
unique_vec.push(common_prefix);
}
}
} else {
// If the string doesn't start with "flow::", retain it
// Assumption: Except "flow::" values, everything should be already unique
unique_vec.push(s.to_string());
}
}
unique_vec
}