mirror of
https://github.com/bitcoinresearchkit/brk.git
synced 2026-07-16 05:28:12 -07:00
nym21
71 lines
2.3 KiB
Rust
71 lines
2.3 KiB
Rust
use brk_error::Result;
|
|
use brk_indexer::Lengths;
|
|
use brk_types::{Cents, Height, Sats, StoredU64, Version};
|
|
use vecdb::{Exit, ReadableVec};
|
|
|
|
use crate::price;
|
|
|
|
/// Dynamic dispatch trait for cohort vectors.
|
|
///
|
|
/// This trait enables heterogeneous cohort processing via trait objects.
|
|
pub trait DynCohortVecs: Send + Sync {
|
|
/// Get minimum length across height-indexed vectors written in block loop.
|
|
fn min_stateful_len(&self) -> usize;
|
|
|
|
/// Reset the starting height for state tracking.
|
|
fn reset_state_starting_height(&mut self);
|
|
|
|
/// Import state from checkpoint at or before the given height.
|
|
fn import_state(&mut self, starting_height: Height) -> Result<Height>;
|
|
|
|
/// Validate that computed vectors have correct versions.
|
|
fn validate_computed_versions(&mut self, base_version: Version) -> Result<()>;
|
|
|
|
/// Push state to height-indexed vectors.
|
|
/// Height is used for the state_starting_height guard check.
|
|
fn push_state(&mut self, height: Height);
|
|
|
|
/// Compute and push unrealized profit/loss states.
|
|
fn push_unrealized_state(&mut self, height_price: Cents);
|
|
|
|
/// First phase of post-processing computations.
|
|
fn compute_rest_part1(
|
|
&mut self,
|
|
prices: &price::Vecs,
|
|
starting_lengths: &Lengths,
|
|
exit: &Exit,
|
|
) -> Result<()>;
|
|
|
|
/// Write state checkpoint to disk.
|
|
fn write_state(&mut self, height: Height, cleanup: bool) -> Result<()>;
|
|
|
|
/// Reset cost basis data (called during fresh start).
|
|
fn reset_cost_basis_data_if_needed(&mut self) -> Result<()>;
|
|
|
|
/// Reset per-block iteration values.
|
|
fn reset_single_iteration_values(&mut self);
|
|
}
|
|
|
|
/// Static dispatch trait for cohort vectors with additional methods.
|
|
///
|
|
/// Used by address cohorts where all cohorts share the same concrete type.
|
|
pub trait CohortVecs: DynCohortVecs {
|
|
/// Compute aggregate cohort from component cohorts.
|
|
fn compute_from_stateful(
|
|
&mut self,
|
|
starting_lengths: &Lengths,
|
|
others: &[&Self],
|
|
exit: &Exit,
|
|
) -> Result<()>;
|
|
|
|
/// Second phase of post-processing computations.
|
|
fn compute_rest_part2(
|
|
&mut self,
|
|
prices: &price::Vecs,
|
|
starting_lengths: &Lengths,
|
|
all_supply_sats: &impl ReadableVec<Height, Sats>,
|
|
all_utxo_count: &impl ReadableVec<Height, StoredU64>,
|
|
exit: &Exit,
|
|
) -> Result<()>;
|
|
}
|