use async_trait::async_trait;
use crate::{
clients::common_traits::{
Candlestick, Level3OrderBook, OrderBook, OrderBookTopK, Ticker, Trade, BBO,
},
common::{command_translator::CommandTranslator, ws_client_internal::WSClientInternal},
WSClient,
};
use super::{
utils::{BitgetCommandTranslator, BitgetMessageHandler, UPLINK_LIMIT},
EXCHANGE_NAME,
};
const WEBSOCKET_URL: &str = "wss://ws.bitget.com/mix/v1/stream";
pub struct BitgetSwapWSClient {
client: WSClientInternal<BitgetMessageHandler>,
translator: BitgetCommandTranslator<'M'>,
}
impl BitgetSwapWSClient {
pub async fn new(tx: std::sync::mpsc::Sender<String>, url: Option<&str>) -> Self {
let real_url = match url {
Some(endpoint) => endpoint,
None => WEBSOCKET_URL,
};
BitgetSwapWSClient {
client: WSClientInternal::connect(
EXCHANGE_NAME,
real_url,
BitgetMessageHandler {},
Some(UPLINK_LIMIT),
tx,
)
.await,
translator: BitgetCommandTranslator::<'M'> {},
}
}
}
impl_trait!(Trade, BitgetSwapWSClient, subscribe_trade, "trade");
#[rustfmt::skip]
impl_trait!(OrderBookTopK, BitgetSwapWSClient, subscribe_orderbook_topk, "books15");
impl_trait!(OrderBook, BitgetSwapWSClient, subscribe_orderbook, "books");
impl_trait!(Ticker, BitgetSwapWSClient, subscribe_ticker, "ticker");
impl_candlestick!(BitgetSwapWSClient);
panic_bbo!(BitgetSwapWSClient);
panic_l3_orderbook!(BitgetSwapWSClient);
impl_ws_client_trait!(BitgetSwapWSClient);