use std::{
collections::{HashMap, VecDeque},
time::{Duration, Instant},
};
use bytes::Bytes;
use nautilus_common::{
cache::database::CacheDatabaseAdapter, custom::CustomData, runtime::get_runtime, signal::Signal,
};
use nautilus_core::nanos::UnixNanos;
use nautilus_model::{
accounts::any::AccountAny,
data::{bar::Bar, quote::QuoteTick, trade::TradeTick, DataType},
events::{order::OrderEventAny, position::snapshot::PositionSnapshot},
identifiers::{
AccountId, ClientId, ClientOrderId, ComponentId, InstrumentId, PositionId, StrategyId,
VenueOrderId,
},
instruments::{any::InstrumentAny, synthetic::SyntheticInstrument},
orderbook::book::OrderBook,
orders::any::OrderAny,
position::Position,
types::currency::Currency,
};
use sqlx::{postgres::PgConnectOptions, PgPool};
use ustr::Ustr;
use crate::sql::{
pg::{connect_pg, get_postgres_connect_options},
queries::DatabaseQueries,
};
#[derive(Debug)]
#[cfg_attr(
feature = "python",
pyo3::pyclass(module = "nautilus_trader.core.nautilus_pyo3.infrastructure")
)]
pub struct PostgresCacheDatabase {
pub pool: PgPool,
tx: tokio::sync::mpsc::UnboundedSender<DatabaseQuery>,
handle: tokio::task::JoinHandle<()>,
}
#[allow(clippy::large_enum_variant)]
#[derive(Debug, Clone)]
pub enum DatabaseQuery {
Close,
Add(String, Vec<u8>),
AddCurrency(Currency),
AddInstrument(InstrumentAny),
AddOrder(OrderAny, Option<ClientId>, bool),
AddPositionSnapshot(PositionSnapshot),
AddAccount(AccountAny, bool),
AddSignal(Signal),
AddCustom(CustomData),
AddQuote(QuoteTick),
AddTrade(TradeTick),
AddBar(Bar),
UpdateOrder(OrderEventAny),
}
impl PostgresCacheDatabase {
pub async fn connect(
host: Option<String>,
port: Option<u16>,
username: Option<String>,
password: Option<String>,
database: Option<String>,
) -> Result<Self, sqlx::Error> {
let pg_connect_options =
get_postgres_connect_options(host, port, username, password, database);
let pool = connect_pg(pg_connect_options.clone().into()).await.unwrap();
let (tx, rx) = tokio::sync::mpsc::unbounded_channel::<DatabaseQuery>();
let handle = tokio::spawn(async move {
PostgresCacheDatabase::process_commands(rx, pg_connect_options.clone().into()).await;
});
Ok(PostgresCacheDatabase { pool, tx, handle })
}
async fn process_commands(
mut rx: tokio::sync::mpsc::UnboundedReceiver<DatabaseQuery>,
pg_connect_options: PgConnectOptions,
) {
tracing::debug!("Starting cache processing");
let pool = connect_pg(pg_connect_options).await.unwrap();
let mut buffer: VecDeque<DatabaseQuery> = VecDeque::new();
let mut last_drain = Instant::now();
let buffer_interval = Duration::from_millis(0);
loop {
if last_drain.elapsed() >= buffer_interval && !buffer.is_empty() {
drain_buffer(&pool, &mut buffer).await;
last_drain = Instant::now();
} else {
match rx.recv().await {
Some(msg) => match msg {
DatabaseQuery::Close => break,
_ => buffer.push_back(msg),
},
None => break, }
}
}
if !buffer.is_empty() {
drain_buffer(&pool, &mut buffer).await;
}
tracing::debug!("Stopped cache processing");
}
}
pub async fn get_pg_cache_database() -> anyhow::Result<PostgresCacheDatabase> {
let connect_options = get_postgres_connect_options(None, None, None, None, None);
Ok(PostgresCacheDatabase::connect(
Some(connect_options.host),
Some(connect_options.port),
Some(connect_options.username),
Some(connect_options.password),
Some(connect_options.database),
)
.await?)
}
#[allow(dead_code)]
#[allow(unused)]
impl CacheDatabaseAdapter for PostgresCacheDatabase {
fn close(&mut self) -> anyhow::Result<()> {
let pool = self.pool.clone();
let (tx, rx) = std::sync::mpsc::channel();
log::debug!("Closing connection pool");
tokio::task::block_in_place(|| {
get_runtime().block_on(async {
pool.close().await;
if let Err(e) = tx.send(()) {
log::error!("Error closing pool: {e:?}");
}
});
});
if let Err(e) = self.tx.send(DatabaseQuery::Close) {
log::error!("Error sending close: {e:?}");
}
log::debug!("Awaiting task 'cache-write'"); tokio::task::block_in_place(|| {
if let Err(e) = get_runtime().block_on(&mut self.handle) {
log::error!("Error awaiting task 'cache-write': {e:?}");
}
});
log::debug!("Closed");
Ok(rx.recv()?)
}
fn flush(&mut self) -> anyhow::Result<()> {
let pool = self.pool.clone();
let (tx, rx) = std::sync::mpsc::channel();
tokio::task::block_in_place(|| {
get_runtime().block_on(async {
if let Err(e) = DatabaseQueries::truncate(&pool).await {
log::error!("Error flushing pool: {e:?}");
}
if let Err(e) = tx.send(()) {
log::error!("Error sending flush result: {e:?}");
}
});
});
Ok(rx.recv()?)
}
fn load(&self) -> anyhow::Result<HashMap<String, Bytes>> {
let pool = self.pool.clone();
let (tx, rx) = std::sync::mpsc::channel();
tokio::spawn(async move {
let result = DatabaseQueries::load(&pool).await;
match result {
Ok(items) => {
let mapping = items
.into_iter()
.map(|(k, v)| (k, Bytes::from(v)))
.collect();
if let Err(e) = tx.send(mapping) {
log::error!("Failed to send general items: {e:?}");
}
}
Err(e) => {
log::error!("Failed to load general items: {e:?}");
if let Err(e) = tx.send(HashMap::new()) {
log::error!("Failed to send empty general items: {e:?}");
}
}
}
});
Ok(rx.recv()?)
}
fn load_currencies(&mut self) -> anyhow::Result<HashMap<Ustr, Currency>> {
let pool = self.pool.clone();
let (tx, rx) = std::sync::mpsc::channel();
tokio::spawn(async move {
let result = DatabaseQueries::load_currencies(&pool).await;
match result {
Ok(currencies) => {
let mapping = currencies
.into_iter()
.map(|currency| (currency.code, currency))
.collect();
if let Err(e) = tx.send(mapping) {
log::error!("Failed to send currencies: {e:?}");
}
}
Err(e) => {
log::error!("Failed to load currencies: {e:?}");
if let Err(e) = tx.send(HashMap::new()) {
log::error!("Failed to send empty currencies: {e:?}");
}
}
}
});
Ok(rx.recv()?)
}
fn load_instruments(&mut self) -> anyhow::Result<HashMap<InstrumentId, InstrumentAny>> {
let pool = self.pool.clone();
let (tx, rx) = std::sync::mpsc::channel();
tokio::spawn(async move {
let result = DatabaseQueries::load_instruments(&pool).await;
match result {
Ok(instruments) => {
let mapping = instruments
.into_iter()
.map(|instrument| (instrument.id(), instrument))
.collect();
if let Err(e) = tx.send(mapping) {
log::error!("Failed to send instruments: {e:?}");
}
}
Err(e) => {
log::error!("Failed to load instruments: {e:?}");
if let Err(e) = tx.send(HashMap::new()) {
log::error!("Failed to send empty instruments: {e:?}");
}
}
}
});
Ok(rx.recv()?)
}
fn load_synthetics(&mut self) -> anyhow::Result<HashMap<InstrumentId, SyntheticInstrument>> {
todo!()
}
fn load_accounts(&mut self) -> anyhow::Result<HashMap<AccountId, AccountAny>> {
let pool = self.pool.clone();
let (tx, rx) = std::sync::mpsc::channel();
tokio::spawn(async move {
let result = DatabaseQueries::load_accounts(&pool).await;
match result {
Ok(accounts) => {
let mapping = accounts
.into_iter()
.map(|account| (account.id(), account))
.collect();
if let Err(e) = tx.send(mapping) {
log::error!("Failed to send accounts: {e:?}");
}
}
Err(e) => {
log::error!("Failed to load accounts: {e:?}");
if let Err(e) = tx.send(HashMap::new()) {
log::error!("Failed to send empty accounts: {e:?}");
}
}
}
});
Ok(rx.recv()?)
}
fn load_orders(&mut self) -> anyhow::Result<HashMap<ClientOrderId, OrderAny>> {
let pool = self.pool.clone();
let (tx, rx) = std::sync::mpsc::channel();
tokio::spawn(async move {
let result = DatabaseQueries::load_orders(&pool).await;
match result {
Ok(orders) => {
let mapping = orders
.into_iter()
.map(|order| (order.client_order_id(), order))
.collect();
if let Err(e) = tx.send(mapping) {
log::error!("Failed to send orders: {e:?}");
}
}
Err(e) => {
log::error!("Failed to load orders: {e:?}");
if let Err(e) = tx.send(HashMap::new()) {
log::error!("Failed to send empty orders: {e:?}");
}
}
}
});
Ok(rx.recv()?)
}
fn load_positions(&mut self) -> anyhow::Result<HashMap<PositionId, Position>> {
todo!()
}
fn load_index_order_position(&self) -> anyhow::Result<HashMap<ClientOrderId, Position>> {
todo!()
}
fn load_index_order_client(&self) -> anyhow::Result<HashMap<ClientOrderId, ClientId>> {
let pool = self.pool.clone();
let (tx, rx) = std::sync::mpsc::channel();
tokio::spawn(async move {
let result = DatabaseQueries::load_distinct_order_event_client_ids(&pool).await;
match result {
Ok(currency) => {
if let Err(e) = tx.send(currency) {
log::error!("Failed to send load_index_order_client result: {e:?}");
}
}
Err(e) => {
log::error!("Failed to run query load_distinct_order_event_client_ids: {e:?}");
if let Err(e) = tx.send(HashMap::new()) {
log::error!("Failed to send empty load_index_order_client result: {e:?}");
}
}
}
});
Ok(rx.recv()?)
}
fn load_currency(&self, code: &Ustr) -> anyhow::Result<Option<Currency>> {
let pool = self.pool.clone();
let code = code.to_owned(); let (tx, rx) = std::sync::mpsc::channel();
tokio::spawn(async move {
let result = DatabaseQueries::load_currency(&pool, &code).await;
match result {
Ok(currency) => {
if let Err(e) = tx.send(currency) {
log::error!("Failed to send currency {code}: {e:?}");
}
}
Err(e) => {
log::error!("Failed to load currency {code}: {e:?}");
if let Err(e) = tx.send(None) {
log::error!("Failed to send None for currency {code}: {e:?}");
}
}
}
});
Ok(rx.recv()?)
}
fn load_instrument(
&self,
instrument_id: &InstrumentId,
) -> anyhow::Result<Option<InstrumentAny>> {
let pool = self.pool.clone();
let instrument_id = instrument_id.to_owned(); let (tx, rx) = std::sync::mpsc::channel();
tokio::spawn(async move {
let result = DatabaseQueries::load_instrument(&pool, &instrument_id).await;
match result {
Ok(instrument) => {
if let Err(e) = tx.send(instrument) {
log::error!("Failed to send instrument {instrument_id}: {e:?}");
}
}
Err(e) => {
log::error!("Failed to load instrument {instrument_id}: {e:?}");
if let Err(e) = tx.send(None) {
log::error!("Failed to send None for instrument {instrument_id}: {e:?}");
}
}
}
});
Ok(rx.recv()?)
}
fn load_synthetic(&self, instrument_id: &InstrumentId) -> anyhow::Result<SyntheticInstrument> {
todo!()
}
fn load_account(&self, account_id: &AccountId) -> anyhow::Result<Option<AccountAny>> {
let pool = self.pool.clone();
let account_id = account_id.to_owned();
let (tx, rx) = std::sync::mpsc::channel();
tokio::spawn(async move {
let result = DatabaseQueries::load_account(&pool, &account_id).await;
match result {
Ok(account) => {
if let Err(e) = tx.send(account) {
log::error!("Failed to send account {account_id}: {e:?}");
}
}
Err(e) => {
log::error!("Failed to load account {account_id}: {e:?}");
if let Err(e) = tx.send(None) {
log::error!("Failed to send None for account {account_id}: {e:?}");
}
}
}
});
Ok(rx.recv()?)
}
fn load_order(&self, client_order_id: &ClientOrderId) -> anyhow::Result<Option<OrderAny>> {
let pool = self.pool.clone();
let client_order_id = client_order_id.to_owned();
let (tx, rx) = std::sync::mpsc::channel();
tokio::spawn(async move {
let result = DatabaseQueries::load_order(&pool, &client_order_id).await;
match result {
Ok(order) => {
if let Err(e) = tx.send(order) {
log::error!("Failed to send order {client_order_id}: {e:?}");
}
}
Err(e) => {
log::error!("Failed to load order {client_order_id}: {e:?}");
let _ = tx.send(None);
}
}
});
Ok(rx.recv()?)
}
fn load_position(&self, position_id: &PositionId) -> anyhow::Result<Position> {
todo!()
}
fn load_actor(&self, component_id: &ComponentId) -> anyhow::Result<HashMap<String, Bytes>> {
todo!()
}
fn delete_actor(&self, component_id: &ComponentId) -> anyhow::Result<()> {
todo!()
}
fn load_strategy(&self, strategy_id: &StrategyId) -> anyhow::Result<HashMap<String, Bytes>> {
todo!()
}
fn delete_strategy(&self, component_id: &StrategyId) -> anyhow::Result<()> {
todo!()
}
fn add(&self, key: String, value: Bytes) -> anyhow::Result<()> {
let query = DatabaseQuery::Add(key, value.into());
self.tx
.send(query)
.map_err(|e| anyhow::anyhow!("Failed to send query to database message handler: {e}"))
}
fn add_currency(&self, currency: &Currency) -> anyhow::Result<()> {
let query = DatabaseQuery::AddCurrency(*currency);
self.tx.send(query).map_err(|e| {
anyhow::anyhow!("Failed to query add_currency to database message handler: {e}")
})
}
fn add_instrument(&self, instrument: &InstrumentAny) -> anyhow::Result<()> {
let query = DatabaseQuery::AddInstrument(instrument.clone());
self.tx.send(query).map_err(|e| {
anyhow::anyhow!("Failed to send query add_instrument to database message handler: {e}")
})
}
fn add_synthetic(&self, synthetic: &SyntheticInstrument) -> anyhow::Result<()> {
todo!()
}
fn add_account(&self, account: &AccountAny) -> anyhow::Result<()> {
let query = DatabaseQuery::AddAccount(account.clone(), false);
self.tx.send(query).map_err(|e| {
anyhow::anyhow!("Failed to send query add_account to database message handler: {e}")
})
}
fn add_order(&self, order: &OrderAny, client_id: Option<ClientId>) -> anyhow::Result<()> {
let query = DatabaseQuery::AddOrder(order.clone(), client_id, false);
self.tx.send(query).map_err(|e| {
anyhow::anyhow!("Failed to send query add_order to database message handler: {e}")
})
}
fn add_position(&self, position: &Position) -> anyhow::Result<()> {
todo!()
}
fn add_position_snapshot(&self, snapshot: &PositionSnapshot) -> anyhow::Result<()> {
let query = DatabaseQuery::AddPositionSnapshot(snapshot.to_owned());
self.tx.send(query).map_err(|e| {
anyhow::anyhow!(
"Failed to send query add_position_snapshot to database message handler: {e}"
)
})
}
fn add_order_book(&self, order_book: &OrderBook) -> anyhow::Result<()> {
todo!()
}
fn add_quote(&self, quote: &QuoteTick) -> anyhow::Result<()> {
let query = DatabaseQuery::AddQuote(quote.to_owned());
self.tx.send(query).map_err(|e| {
anyhow::anyhow!("Failed to send query add_quote to database message handler: {e}")
})
}
fn load_quotes(&self, instrument_id: &InstrumentId) -> anyhow::Result<Vec<QuoteTick>> {
let pool = self.pool.clone();
let instrument_id = instrument_id.to_owned();
let (tx, rx) = std::sync::mpsc::channel();
tokio::spawn(async move {
let result = DatabaseQueries::load_quotes(&pool, &instrument_id).await;
match result {
Ok(quotes) => {
if let Err(e) = tx.send(quotes) {
log::error!("Failed to send quotes for instrument {instrument_id}: {e:?}");
}
}
Err(e) => {
log::error!("Failed to load quotes for instrument {instrument_id}: {e:?}");
if let Err(e) = tx.send(Vec::new()) {
log::error!(
"Failed to send empty quotes for instrument {instrument_id}: {e:?}"
);
}
}
}
});
Ok(rx.recv()?)
}
fn add_trade(&self, trade: &TradeTick) -> anyhow::Result<()> {
let query = DatabaseQuery::AddTrade(trade.to_owned());
self.tx.send(query).map_err(|e| {
anyhow::anyhow!("Failed to send query add_trade to database message handler: {e}")
})
}
fn load_trades(&self, instrument_id: &InstrumentId) -> anyhow::Result<Vec<TradeTick>> {
let pool = self.pool.clone();
let instrument_id = instrument_id.to_owned();
let (tx, rx) = std::sync::mpsc::channel();
tokio::spawn(async move {
let result = DatabaseQueries::load_trades(&pool, &instrument_id).await;
match result {
Ok(trades) => {
if let Err(e) = tx.send(trades) {
log::error!("Failed to send trades for instrument {instrument_id}: {e:?}");
}
}
Err(e) => {
log::error!("Failed to load trades for instrument {instrument_id}: {e:?}");
if let Err(e) = tx.send(Vec::new()) {
log::error!(
"Failed to send empty trades for instrument {instrument_id}: {e:?}"
);
}
}
}
});
Ok(rx.recv()?)
}
fn add_bar(&self, bar: &Bar) -> anyhow::Result<()> {
let query = DatabaseQuery::AddBar(bar.to_owned());
self.tx.send(query).map_err(|e| {
anyhow::anyhow!("Failed to send query add_bar to database message handler: {e}")
})
}
fn load_bars(&self, instrument_id: &InstrumentId) -> anyhow::Result<Vec<Bar>> {
let pool = self.pool.clone();
let instrument_id = instrument_id.to_owned();
let (tx, rx) = std::sync::mpsc::channel();
tokio::spawn(async move {
let result = DatabaseQueries::load_bars(&pool, &instrument_id).await;
match result {
Ok(bars) => {
if let Err(e) = tx.send(bars) {
log::error!("Failed to send bars for instrument {instrument_id}: {e:?}");
}
}
Err(e) => {
log::error!("Failed to load bars for instrument {instrument_id}: {e:?}");
if let Err(e) = tx.send(Vec::new()) {
log::error!(
"Failed to send empty bars for instrument {instrument_id}: {e:?}"
);
}
}
}
});
Ok(rx.recv()?)
}
fn add_signal(&self, signal: &Signal) -> anyhow::Result<()> {
let query = DatabaseQuery::AddSignal(signal.to_owned());
self.tx.send(query).map_err(|e| {
anyhow::anyhow!("Failed to send query add_signal to database message handler: {e}")
})
}
fn load_signals(&self, name: &str) -> anyhow::Result<Vec<Signal>> {
let pool = self.pool.clone();
let name = name.to_owned();
let (tx, rx) = std::sync::mpsc::channel();
tokio::spawn(async move {
let result = DatabaseQueries::load_signals(&pool, &name).await;
match result {
Ok(signals) => {
if let Err(e) = tx.send(signals) {
log::error!("Failed to send signals for '{name}': {e:?}");
}
}
Err(e) => {
log::error!("Failed to load signals for '{name}': {e:?}");
if let Err(e) = tx.send(Vec::new()) {
log::error!("Failed to send empty signals for '{name}': {e:?}");
}
}
}
});
Ok(rx.recv()?)
}
fn add_custom_data(&self, data: &CustomData) -> anyhow::Result<()> {
let query = DatabaseQuery::AddCustom(data.to_owned());
self.tx.send(query).map_err(|e| {
anyhow::anyhow!("Failed to send query add_signal to database message handler: {e}")
})
}
fn load_custom_data(&self, data_type: &DataType) -> anyhow::Result<Vec<CustomData>> {
let pool = self.pool.clone();
let data_type = data_type.to_owned();
let (tx, rx) = std::sync::mpsc::channel();
tokio::spawn(async move {
let result = DatabaseQueries::load_custom_data(&pool, &data_type).await;
match result {
Ok(signals) => {
if let Err(e) = tx.send(signals) {
log::error!("Failed to send custom data for '{data_type}': {e:?}");
}
}
Err(e) => {
log::error!("Failed to load custom data for '{data_type}': {e:?}");
if let Err(e) = tx.send(Vec::new()) {
log::error!("Failed to send empty custom data for '{data_type}': {e:?}");
}
}
}
});
Ok(rx.recv()?)
}
fn index_venue_order_id(
&self,
client_order_id: ClientOrderId,
venue_order_id: VenueOrderId,
) -> anyhow::Result<()> {
todo!()
}
fn index_order_position(
&self,
client_order_id: ClientOrderId,
position_id: PositionId,
) -> anyhow::Result<()> {
todo!()
}
fn update_actor(&self) -> anyhow::Result<()> {
todo!()
}
fn update_strategy(&self) -> anyhow::Result<()> {
todo!()
}
fn update_account(&self, account: &AccountAny) -> anyhow::Result<()> {
let query = DatabaseQuery::AddAccount(account.clone(), true);
self.tx.send(query).map_err(|e| {
anyhow::anyhow!("Failed to send query add_account to database message handler: {e}")
})
}
fn update_order(&self, event: &OrderEventAny) -> anyhow::Result<()> {
let query = DatabaseQuery::UpdateOrder(event.clone());
self.tx.send(query).map_err(|e| {
anyhow::anyhow!("Failed to send query update_order to database message handler: {e}")
})
}
fn update_position(&self, position: &Position) -> anyhow::Result<()> {
todo!()
}
fn snapshot_order_state(&self, order: &OrderAny) -> anyhow::Result<()> {
todo!()
}
fn snapshot_position_state(&self, position: &Position) -> anyhow::Result<()> {
todo!()
}
fn heartbeat(&self, timestamp: UnixNanos) -> anyhow::Result<()> {
todo!()
}
}
async fn drain_buffer(pool: &PgPool, buffer: &mut VecDeque<DatabaseQuery>) {
for cmd in buffer.drain(..) {
let result: anyhow::Result<()> = match cmd {
DatabaseQuery::Close => Ok(()),
DatabaseQuery::Add(key, value) => DatabaseQueries::add(pool, key, value)
.await
.map_err(anyhow::Error::from),
DatabaseQuery::AddCurrency(currency) => DatabaseQueries::add_currency(pool, currency)
.await
.map_err(anyhow::Error::from),
DatabaseQuery::AddInstrument(instrument_any) => match instrument_any {
InstrumentAny::Betting(instrument) => {
DatabaseQueries::add_instrument(pool, "BETTING", Box::new(instrument)).await
}
InstrumentAny::BinaryOption(instrument) => {
DatabaseQueries::add_instrument(pool, "BINARY_OPTION", Box::new(instrument))
.await
}
InstrumentAny::CryptoFuture(instrument) => {
DatabaseQueries::add_instrument(pool, "CRYPTO_FUTURE", Box::new(instrument))
.await
}
InstrumentAny::CryptoPerpetual(instrument) => {
DatabaseQueries::add_instrument(pool, "CRYPTO_PERPETUAL", Box::new(instrument))
.await
}
InstrumentAny::CurrencyPair(instrument) => {
DatabaseQueries::add_instrument(pool, "CURRENCY_PAIR", Box::new(instrument))
.await
}
InstrumentAny::Equity(equity) => {
DatabaseQueries::add_instrument(pool, "EQUITY", Box::new(equity)).await
}
InstrumentAny::FuturesContract(instrument) => {
DatabaseQueries::add_instrument(pool, "FUTURES_CONTRACT", Box::new(instrument))
.await
}
InstrumentAny::FuturesSpread(instrument) => {
DatabaseQueries::add_instrument(pool, "FUTURES_SPREAD", Box::new(instrument))
.await
}
InstrumentAny::OptionsContract(instrument) => {
DatabaseQueries::add_instrument(pool, "OPTIONS_CONTRACT", Box::new(instrument))
.await
}
InstrumentAny::OptionsSpread(instrument) => {
DatabaseQueries::add_instrument(pool, "OPTIONS_SPREAD", Box::new(instrument))
.await
}
},
DatabaseQuery::AddOrder(order_any, client_id, updated) => match order_any {
OrderAny::Limit(order) => {
DatabaseQueries::add_order(pool, "LIMIT", updated, Box::new(order), client_id)
.await
}
OrderAny::LimitIfTouched(order) => {
DatabaseQueries::add_order(
pool,
"LIMIT_IF_TOUCHED",
updated,
Box::new(order),
client_id,
)
.await
}
OrderAny::Market(order) => {
DatabaseQueries::add_order(pool, "MARKET", updated, Box::new(order), client_id)
.await
}
OrderAny::MarketIfTouched(order) => {
DatabaseQueries::add_order(
pool,
"MARKET_IF_TOUCHED",
updated,
Box::new(order),
client_id,
)
.await
}
OrderAny::MarketToLimit(order) => {
DatabaseQueries::add_order(
pool,
"MARKET_TO_LIMIT",
updated,
Box::new(order),
client_id,
)
.await
}
OrderAny::StopLimit(order) => {
DatabaseQueries::add_order(
pool,
"STOP_LIMIT",
updated,
Box::new(order),
client_id,
)
.await
}
OrderAny::StopMarket(order) => {
DatabaseQueries::add_order(
pool,
"STOP_MARKET",
updated,
Box::new(order),
client_id,
)
.await
}
OrderAny::TrailingStopLimit(order) => {
DatabaseQueries::add_order(
pool,
"TRAILING_STOP_LIMIT",
updated,
Box::new(order),
client_id,
)
.await
}
OrderAny::TrailingStopMarket(order) => {
DatabaseQueries::add_order(
pool,
"TRAILING_STOP_MARKET",
updated,
Box::new(order),
client_id,
)
.await
}
},
DatabaseQuery::AddPositionSnapshot(snapshot) => {
DatabaseQueries::add_position_snapshot(pool, snapshot).await
}
DatabaseQuery::AddAccount(account_any, updated) => match account_any {
AccountAny::Cash(account) => {
DatabaseQueries::add_account(pool, "CASH", updated, Box::new(account)).await
}
AccountAny::Margin(account) => {
DatabaseQueries::add_account(pool, "MARGIN", updated, Box::new(account)).await
}
},
DatabaseQuery::AddSignal(signal) => DatabaseQueries::add_signal(pool, &signal).await,
DatabaseQuery::AddCustom(data) => DatabaseQueries::add_custom_data(pool, &data).await,
DatabaseQuery::AddQuote(quote) => DatabaseQueries::add_quote(pool, "e).await,
DatabaseQuery::AddTrade(trade) => DatabaseQueries::add_trade(pool, &trade).await,
DatabaseQuery::AddBar(bar) => DatabaseQueries::add_bar(pool, &bar).await,
DatabaseQuery::UpdateOrder(event) => {
DatabaseQueries::add_order_event(pool, event.into_boxed(), None).await
}
};
if let Err(e) = result {
log::error!("Error processing database query: {e:?}");
}
}
}