nautilus_model/accounts/
mod.rs1pub mod any;
19pub mod base;
20pub mod cash;
21pub mod margin;
22
23#[cfg(any(test, feature = "stubs"))]
24pub mod stubs;
25
26use ahash::AHashMap;
27use enum_dispatch::enum_dispatch;
28use nautilus_core::UnixNanos;
29
30pub use crate::accounts::{
32 any::AccountAny, base::BaseAccount, cash::CashAccount, margin::MarginAccount,
33};
34use crate::{
35 enums::{AccountType, LiquiditySide, OrderSide},
36 events::{AccountState, OrderFilled},
37 identifiers::AccountId,
38 instruments::InstrumentAny,
39 position::Position,
40 types::{AccountBalance, Currency, Money, Price, Quantity},
41};
42
43#[enum_dispatch]
44pub trait Account: 'static + Send {
45 fn id(&self) -> AccountId;
46 fn account_type(&self) -> AccountType;
47 fn base_currency(&self) -> Option<Currency>;
48 fn is_cash_account(&self) -> bool;
49 fn is_margin_account(&self) -> bool;
50 fn calculated_account_state(&self) -> bool;
51 fn balance_total(&self, currency: Option<Currency>) -> Option<Money>;
52 fn balances_total(&self) -> AHashMap<Currency, Money>;
53 fn balance_free(&self, currency: Option<Currency>) -> Option<Money>;
54 fn balances_free(&self) -> AHashMap<Currency, Money>;
55 fn balance_locked(&self, currency: Option<Currency>) -> Option<Money>;
56 fn balances_locked(&self) -> AHashMap<Currency, Money>;
57 fn balance(&self, currency: Option<Currency>) -> Option<&AccountBalance>;
58 fn last_event(&self) -> Option<AccountState>;
59 fn events(&self) -> Vec<AccountState>;
60 fn event_count(&self) -> usize;
61 fn currencies(&self) -> Vec<Currency>;
62 fn starting_balances(&self) -> AHashMap<Currency, Money>;
63 fn balances(&self) -> AHashMap<Currency, AccountBalance>;
64 fn apply(&mut self, event: AccountState) -> anyhow::Result<()>;
71 fn purge_account_events(&mut self, ts_now: UnixNanos, lookback_secs: u64);
72
73 fn calculate_balance_locked(
79 &mut self,
80 instrument: InstrumentAny,
81 side: OrderSide,
82 quantity: Quantity,
83 price: Price,
84 use_quote_for_inverse: Option<bool>,
85 ) -> anyhow::Result<Money>;
86
87 fn calculate_pnls(
93 &self,
94 instrument: InstrumentAny,
95 fill: OrderFilled,
96 position: Option<Position>,
97 ) -> anyhow::Result<Vec<Money>>;
98
99 fn calculate_commission(
105 &self,
106 instrument: InstrumentAny,
107 last_qty: Quantity,
108 last_px: Price,
109 liquidity_side: LiquiditySide,
110 use_quote_for_inverse: Option<bool>,
111 ) -> anyhow::Result<Money>;
112}