primitives - PricingBounds::to_vec & From<&Channel> for Output

Author: elpiel

primitives/src/channel.rs

@@ -102,6 +102,22 @@ pub struct PricingBounds {
     pub click: Option<Pricing>,
 }
 
+impl PricingBounds {
+    pub fn to_vec(&self) -> Vec<(&str, Pricing)> {
+        let mut vec = Vec::new();
+
+        if let Some(pricing) = self.impression.as_ref() {
+            vec.push(("IMPRESSION", pricing.clone()));
+        }
+
+        if let Some(pricing) = self.click.as_ref() {
+            vec.push(("CLICK", pricing.clone()))
+        }
+
+        vec
+    }
+}
+
 #[derive(Serialize, Deserialize, Debug, Clone)]
 #[serde(rename_all = "camelCase")]
 pub struct ChannelSpec {

primitives/src/targeting.rs

@@ -1,4 +1,4 @@
-use crate::BigNum;
+use crate::{BigNum, Channel};
 use std::collections::HashMap;
 
 pub use eval::*;
@@ -139,6 +139,25 @@ pub struct Output {
     pub price: HashMap<String, BigNum>,
 }
 
+impl From<&Channel> for Output {
+    fn from(channel: &Channel) -> Self {
+        let price = match &channel.spec.pricing_bounds {
+            Some(pricing_bounds) => pricing_bounds
+                .to_vec()
+                .into_iter()
+                .map(|(key, price)| (key.to_string(), price.min))
+                .collect(),
+            _ => Default::default(),
+        };
+
+        Self {
+            show: true,
+            boost: 1.0,
+            price,
+        }
+    }
+}
+
 #[cfg(test)]
 mod test {
     use super::*;