diff options
| -rw-r--r-- | src/app/store/mod.rs | 26 |
1 files changed, 14 insertions, 12 deletions
diff --git a/src/app/store/mod.rs b/src/app/store/mod.rs index 870e402..fb6e90f 100644 --- a/src/app/store/mod.rs +++ b/src/app/store/mod.rs @@ -1,12 +1,14 @@ pub mod records; mod sqlite; -use miette::{Diagnostic, Result}; +use miette::Diagnostic; use sqlite::SQLiteStore; use thiserror::Error; use records::{EquityChangeEvent, EquitySymbol, FiduciaryAccount}; +type Result<T> = miette::Result<T, ContextError>; + pub enum Context { SQLite(SQLiteStore), } @@ -65,13 +67,13 @@ impl Default for AppContext { } pub trait Store { - fn init(&mut self) -> Result<(), ContextError>; - fn get_accounts(&self) -> Result<Vec<FiduciaryAccount>, ContextError>; - fn get_symbols(&self) -> Result<Vec<EquitySymbol>, ContextError>; - fn get_events(&self) -> Result<Vec<EquityChangeEvent>, ContextError>; - fn add_account(&mut self, account: &FiduciaryAccount) -> Result<(), ContextError>; - fn add_symbol(&mut self, symbol: &EquitySymbol) -> Result<(), ContextError>; - fn add_event(&mut self, event: &EquityChangeEvent) -> Result<(), ContextError>; + fn init(&mut self) -> Result<()>; + fn get_accounts(&self) -> Result<Vec<FiduciaryAccount>>; + fn get_symbols(&self) -> Result<Vec<EquitySymbol>>; + fn get_events(&self) -> Result<Vec<EquityChangeEvent>>; + fn add_account(&mut self, account: &FiduciaryAccount) -> Result<()>; + fn add_symbol(&mut self, symbol: &EquitySymbol) -> Result<()>; + fn add_event(&mut self, event: &EquityChangeEvent) -> Result<()>; } impl AppContext { @@ -103,7 +105,7 @@ impl AppContext { } } - pub fn init_context(&mut self) -> Result<(), ContextError> { + pub fn init_context(&mut self) -> Result<()> { self.unwrap_store().0.init()?; tracing::event!( @@ -131,7 +133,7 @@ impl AppContext { &self.events } - pub fn add_account(&mut self, account: FiduciaryAccount) -> Result<(), ContextError> { + pub fn add_account(&mut self, account: FiduciaryAccount) -> Result<()> { let (store, accounts, ..) = self.unwrap_store(); let acc = accounts.push_mut(account); tracing::event!( @@ -144,7 +146,7 @@ impl AppContext { Ok(()) } - pub fn add_symbol(&mut self, symbol: EquitySymbol) -> Result<(), ContextError> { + pub fn add_symbol(&mut self, symbol: EquitySymbol) -> Result<()> { let (store, _, symbols, ..) = self.unwrap_store(); let sym = symbols.push_mut(symbol); tracing::event!( @@ -157,7 +159,7 @@ impl AppContext { Ok(()) } - pub fn add_event(&mut self, event: EquityChangeEvent) -> Result<(), ContextError> { + pub fn add_event(&mut self, event: EquityChangeEvent) -> Result<()> { let (store, .., events) = self.unwrap_store(); let ev = events.push_mut(event); tracing::event!( |