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);
65 fn purge_account_events(&mut self, ts_now: UnixNanos, lookback_secs: u64);
66
67 fn calculate_balance_locked(
73 &mut self,
74 instrument: InstrumentAny,
75 side: OrderSide,
76 quantity: Quantity,
77 price: Price,
78 use_quote_for_inverse: Option<bool>,
79 ) -> anyhow::Result<Money>;
80
81 fn calculate_pnls(
87 &self,
88 instrument: InstrumentAny,
89 fill: OrderFilled,
90 position: Option<Position>,
91 ) -> anyhow::Result<Vec<Money>>;
92
93 fn calculate_commission(
99 &self,
100 instrument: InstrumentAny,
101 last_qty: Quantity,
102 last_px: Price,
103 liquidity_side: LiquiditySide,
104 use_quote_for_inverse: Option<bool>,
105 ) -> anyhow::Result<Money>;
106}