aboutsummaryrefslogtreecommitdiff
path: root/src/app/store
diff options
context:
space:
mode:
Diffstat (limited to 'src/app/store')
-rw-r--r--src/app/store/ledger.rs58
-rw-r--r--src/app/store/mod.rs3
-rw-r--r--src/app/store/records.rs8
3 files changed, 6 insertions, 63 deletions
diff --git a/src/app/store/ledger.rs b/src/app/store/ledger.rs
deleted file mode 100644
index df65b58..0000000
--- a/src/app/store/ledger.rs
+++ /dev/null
@@ -1,58 +0,0 @@
-use bigdecimal::{BigDecimal, Zero};
-use crate::app::store::records::{EquityChangeEvent, EquitySymbol, FiduciaryAccount, ROUNDING_MODE};
-
-/// An estimated ledger is the effective sum of all the equity change events
-/// derived from record for a particular fiduciary account and equity symbol.
-pub struct EstimatedLedger<'a> {
- pub fiduciary_account: &'a FiduciaryAccount,
- pub equity_symbol: &'a EquitySymbol,
- pub equity_change_events: Vec<&'a EquityChangeEvent>,
- pub account_symbol_quantity: BigDecimal,
- pub account_symbol_cost_basis_price: BigDecimal,
- pub account_symbol_cost_basis_total: BigDecimal,
- pub account_symbol_last_effective_price: BigDecimal,
- pub account_symbol_last_update_timestamp: chrono::DateTime<chrono::Utc>,
-}
-
-impl EstimatedLedger<'_> {
-
- /// Takes a list of all change events, a fiduciary account, and an equity
- /// symbol for which to create a ledger.
- ///
- /// All values are taken by reference, so they need to be maintained elsewhere
- pub fn compile_change_events<'a>(fiduciary_account: &'a FiduciaryAccount, equity_symbol: &'a EquitySymbol, events: Vec<&'a EquityChangeEvent>) -> EstimatedLedger<'a> {
- let mut acsym_events = vec![];
- let mut quantity = BigDecimal::zero();
- let mut cost_basis_total = BigDecimal::zero();
- let mut last_event: Option<&EquityChangeEvent> = None;
-
- for event in events {
- if event.fiduciary_account_id != fiduciary_account.fiduciary_account_id || event.equity_symbol_id != equity_symbol.equity_symbol_id {
- continue;
- }
-
- if last_event.is_none() || last_event.unwrap().equity_change_timestamp < event.equity_change_timestamp {
- last_event = Some(event);
- }
-
- quantity += event.scored_quantity();
- cost_basis_total += event.scored_cost_basis();
-
- acsym_events.push(event);
- }
-
- let cost_basis_price = (cost_basis_total.clone() / quantity.clone()).with_scale_round(2, ROUNDING_MODE);
-
- EstimatedLedger {
- fiduciary_account,
- equity_symbol,
- equity_change_events: acsym_events,
- account_symbol_quantity: quantity,
- account_symbol_cost_basis_price: cost_basis_price,
- account_symbol_cost_basis_total: cost_basis_total.with_scale_round(2, ROUNDING_MODE),
- account_symbol_last_effective_price: last_event.map_or_else(|| BigDecimal::zero().with_scale(2), |ev| ev.estimated_price()),
- account_symbol_last_update_timestamp: last_event.map_or_else(|| chrono::DateTime::<chrono::Utc>::MIN_UTC, |ev| ev.equity_change_timestamp),
- }
- }
-
-} \ No newline at end of file
diff --git a/src/app/store/mod.rs b/src/app/store/mod.rs
index da01fba..2fa87a4 100644
--- a/src/app/store/mod.rs
+++ b/src/app/store/mod.rs
@@ -1,6 +1,5 @@
mod sqlite;
-mod records;
-mod ledger;
+pub mod records;
use sqlite::{SQLiteStore};
use miette::{Diagnostic, Result};
diff --git a/src/app/store/records.rs b/src/app/store/records.rs
index 024c06e..682f676 100644
--- a/src/app/store/records.rs
+++ b/src/app/store/records.rs
@@ -5,6 +5,8 @@ pub static ROUNDING_MODE: RoundingMode = RoundingMode::HalfEven;
/// Will use default precision (either 100 or ${RUST_BIGDECIMAL_DEFAULT_PRECISION})
pub static DECIMAL_CONTEXT: LazyLock<Context> = std::sync::LazyLock::new(|| Context::default().with_rounding_mode(ROUNDING_MODE));
+type DateTime = chrono::DateTime<chrono::Utc>;
+
/// Represents an account that holds equity. An account may hold equity with
/// multiple equity symbols.
pub struct FiduciaryAccount {
@@ -12,7 +14,7 @@ pub struct FiduciaryAccount {
pub fiduciary_account_number: usize,
pub fiduciary_account_name: String,
pub fiduciary_account_description: String,
- pub fiduciary_account_created_timestamp: chrono::DateTime<chrono::Utc>,
+ pub fiduciary_account_created_timestamp: DateTime,
}
/// An atomic enum to describe for what element of equity a symbol tracks
@@ -41,7 +43,7 @@ pub struct EquitySymbol {
pub equity_symbol_name: String,
pub equity_symbol_type: EquityType,
pub equity_symbol_managing_company: String,
- pub equity_symbol_created_timestamp: chrono::DateTime<chrono::Utc>,
+ pub equity_symbol_created_timestamp: DateTime,
}
/// An atomic enum to describe the basic action of an equity change event
@@ -82,7 +84,7 @@ pub struct EquityChangeEvent {
pub equity_change_quantity: BigDecimal,
pub equity_change_cost_basis: BigDecimal,
pub equity_change_type: ChangeEventType,
- pub equity_change_timestamp: chrono::DateTime<chrono::Utc>,
+ pub equity_change_timestamp: DateTime,
}
impl EquityChangeEvent {