|
|
@@ -1,6 +1,16 @@
|
|
|
+use log::{
|
|
|
+ debug,
|
|
|
+ warn,
|
|
|
+};
|
|
|
+use solana_client::rpc_client::RpcClient;
|
|
|
use solana_sdk::signature::Signature;
|
|
|
|
|
|
-use std::time::Instant;
|
|
|
+use pyth_sdk_solana::state::PriceAccount;
|
|
|
+
|
|
|
+use std::time::{
|
|
|
+ Duration,
|
|
|
+ Instant,
|
|
|
+};
|
|
|
|
|
|
use crate::{
|
|
|
AttestationConditions,
|
|
|
@@ -12,6 +22,7 @@ use crate::{
|
|
|
pub struct BatchState<'a> {
|
|
|
pub group_name: String,
|
|
|
pub symbols: &'a [P2WSymbol],
|
|
|
+ pub last_known_symbol_states: Vec<Option<PriceAccount>>,
|
|
|
pub conditions: AttestationConditions,
|
|
|
status: BatchTxStatus,
|
|
|
status_changed_at: Instant,
|
|
|
@@ -27,6 +38,7 @@ impl<'a> BatchState<'a> {
|
|
|
group_name,
|
|
|
symbols,
|
|
|
conditions,
|
|
|
+ last_known_symbol_states: vec![None; symbols.len()],
|
|
|
status: BatchTxStatus::Sending { attempt_no: 1 },
|
|
|
status_changed_at: Instant::now(),
|
|
|
}
|
|
|
@@ -38,11 +50,108 @@ impl<'a> BatchState<'a> {
|
|
|
pub fn get_status(&self) -> &BatchTxStatus {
|
|
|
&self.status
|
|
|
}
|
|
|
+
|
|
|
/// Ensure that status changes are accompanied by a timestamp bump
|
|
|
pub fn set_status(&mut self, s: BatchTxStatus) {
|
|
|
self.status_changed_at = Instant::now();
|
|
|
self.status = s;
|
|
|
}
|
|
|
+
|
|
|
+ /// Evaluate the configured attestation conditions for this
|
|
|
+ /// batch. Returns Some("<reason>") if any trigger condition was
|
|
|
+ /// met. Only the first encountered condition is mentioned.
|
|
|
+ pub fn should_resend(&mut self, c: &RpcClient) -> Option<String> {
|
|
|
+ let sym_count = self.symbols.len();
|
|
|
+ let mut new_symbol_states: Vec<Option<PriceAccount>> = Vec::with_capacity(sym_count);
|
|
|
+ for (idx, sym) in self.symbols.iter().enumerate() {
|
|
|
+ let new_state = match c
|
|
|
+ .get_account_data(&sym.price_addr)
|
|
|
+ .map_err(|e| e.to_string())
|
|
|
+ .and_then(|bytes| {
|
|
|
+ pyth_sdk_solana::state::load_price_account(&bytes)
|
|
|
+ .map(|state| state.clone())
|
|
|
+ .map_err(|e| e.to_string())
|
|
|
+ }) {
|
|
|
+ Ok(state) => Some(state),
|
|
|
+ Err(e) => {
|
|
|
+ warn!(
|
|
|
+ "Symbol {} ({}/{}): Could not look up state: {}",
|
|
|
+ sym.name
|
|
|
+ .as_ref()
|
|
|
+ .unwrap_or(&format!("Unnamed product {}", sym.product_addr)),
|
|
|
+ idx + 1,
|
|
|
+ sym_count,
|
|
|
+ e.to_string()
|
|
|
+ );
|
|
|
+ None
|
|
|
+ }
|
|
|
+ };
|
|
|
+
|
|
|
+ new_symbol_states.push(new_state);
|
|
|
+ }
|
|
|
+
|
|
|
+ // min interval
|
|
|
+ if let Some(i) = self.conditions.min_interval_secs.as_ref() {
|
|
|
+ if self.get_status_changed_at().elapsed() > Duration::from_secs(*i) {
|
|
|
+ return Some(format!(
|
|
|
+ "minimum interval of {}s elapsed since last state change",
|
|
|
+ i
|
|
|
+ ));
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ let mut ret = None;
|
|
|
+ for (idx, old_new_tup) in self
|
|
|
+ .last_known_symbol_states
|
|
|
+ .iter_mut() // Borrow mutably to make the update easier
|
|
|
+ .zip(new_symbol_states.iter())
|
|
|
+ .enumerate()
|
|
|
+ {
|
|
|
+ // Only evaluate if a triggering condition is not already met
|
|
|
+ if ret.is_none() {
|
|
|
+ match old_new_tup {
|
|
|
+ (Some(old), Some(new)) => {
|
|
|
+ // publish_time_changed
|
|
|
+ if self.conditions.publish_time_changed && new.timestamp > old.timestamp {
|
|
|
+ ret = Some(format!(
|
|
|
+ "publish_time advanced for {:?}",
|
|
|
+ self.symbols[idx].to_string(),
|
|
|
+ ))
|
|
|
+
|
|
|
+ // price_changed_pct
|
|
|
+ } else if let Some(pct) = self.conditions.price_changed_pct {
|
|
|
+ let price_pct_diff = (old.agg.price as f32 - new.agg.price as f32)
|
|
|
+ / old.agg.price as f32
|
|
|
+ * 100.0;
|
|
|
+
|
|
|
+ if price_pct_diff > pct {
|
|
|
+ ret = Some(format!(
|
|
|
+ "price moved by at least {}% for {:?}",
|
|
|
+ pct,
|
|
|
+ self.symbols[idx].to_string()
|
|
|
+ ))
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+ _ => {
|
|
|
+ debug!(
|
|
|
+ "Symbol {:?} {}/{}, old or new state value is None, skipping...",
|
|
|
+ self.symbols[idx].to_string(),
|
|
|
+ idx + 1,
|
|
|
+ sym_count
|
|
|
+ );
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ // Update with newer state if available
|
|
|
+ if old_new_tup.1.is_some() {
|
|
|
+ *old_new_tup.0 = *old_new_tup.1;
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ return ret;
|
|
|
+ }
|
|
|
}
|
|
|
|
|
|
#[derive(Debug)]
|
Stan Drozd