use alloc::borrow::Cow;
use alloc::vec::Vec;
use serde::{Deserialize, Serialize};
use serde_with::skip_serializing_none;
use crate::models::Model;
use super::{CommonFields, Request};
#[derive(Debug, Serialize, Deserialize, PartialEq, Eq, Clone)]
pub struct OracleDescriptor<'a> {
pub account: Cow<'a, str>,
pub oracle_document_id: u32,
}
#[skip_serializing_none]
#[derive(Debug, Serialize, Deserialize, PartialEq, Eq, Clone)]
pub struct GetAggregatePrice<'a> {
#[serde(flatten)]
pub common_fields: CommonFields<'a>,
pub base_asset: Cow<'a, str>,
pub quote_asset: Cow<'a, str>,
pub oracles: Vec<OracleDescriptor<'a>>,
pub trim: Option<u8>,
pub time_threshold: Option<u32>,
}
impl Model for GetAggregatePrice<'_> {}
impl<'a> Request<'a> for GetAggregatePrice<'a> {
fn get_common_fields(&self) -> &CommonFields<'a> {
&self.common_fields
}
fn get_common_fields_mut(&mut self) -> &mut CommonFields<'a> {
&mut self.common_fields
}
}
impl<'a> GetAggregatePrice<'a> {
pub fn new(
id: Option<Cow<'a, str>>,
base_asset: Cow<'a, str>,
quote_asset: Cow<'a, str>,
oracles: Vec<OracleDescriptor<'a>>,
trim: Option<u8>,
time_threshold: Option<u32>,
) -> Self {
Self {
common_fields: CommonFields {
command: super::RequestMethod::GetAggregatePrice,
id,
},
base_asset,
quote_asset,
oracles,
trim,
time_threshold,
}
}
}
#[cfg(test)]
mod tests {
use alloc::vec;
use super::*;
#[test]
fn test_serde_round_trip() {
let req = GetAggregatePrice::new(
None,
"XRP".into(),
"USD".into(),
vec![OracleDescriptor {
account: "rHb9CJAWyB4rj91VRWn96DkukG4bwdtyTh".into(),
oracle_document_id: 1,
}],
None,
None,
);
let serialized = serde_json::to_string(&req).unwrap();
let deserialized: GetAggregatePrice = serde_json::from_str(&serialized).unwrap();
assert_eq!(req, deserialized);
assert!(serialized.contains("\"command\":\"get_aggregate_price\""));
assert!(serialized.contains("\"base_asset\":\"XRP\""));
}
#[test]
fn test_with_trim() {
let req = GetAggregatePrice::new(
Some("test-1".into()),
"BTC".into(),
"USD".into(),
vec![
OracleDescriptor {
account: "rHb9CJAWyB4rj91VRWn96DkukG4bwdtyTh".into(),
oracle_document_id: 1,
},
OracleDescriptor {
account: "rsA2LpzuawewSBQXkiju3YQTMzW13pAAdW".into(),
oracle_document_id: 2,
},
],
Some(20),
Some(60),
);
let serialized = serde_json::to_string(&req).unwrap();
assert!(serialized.contains("\"trim\":20"));
assert!(serialized.contains("\"time_threshold\":60"));
let deserialized: GetAggregatePrice = serde_json::from_str(&serialized).unwrap();
assert_eq!(req, deserialized);
}
}