use self::{
channel::OkxChannel, market::OkxMarket, subscription::OkxSubResponse, trade::OkxTrades,
};
use crate::{
ExchangeWsStream, NoInitialSnapshots,
exchange::{Connector, ExchangeSub, PingInterval, StreamSelector},
instrument::InstrumentData,
subscriber::{WebSocketSubscriber, validator::WebSocketSubValidator},
subscription::trade::PublicTrades,
transformer::stateless::StatelessTransformer,
};
use barter_instrument::exchange::ExchangeId;
use barter_integration::protocol::websocket::{WebSocketSerdeParser, WsMessage};
use barter_macro::{DeExchange, SerExchange};
use derive_more::Display;
use serde_json::json;
use std::time::Duration;
use url::Url;
pub mod channel;
pub mod market;
pub mod subscription;
pub mod trade;
pub const BASE_URL_OKX: &str = "wss://ws.okx.com:8443/ws/v5/public";
pub const PING_INTERVAL_OKX: Duration = Duration::from_secs(29);
pub type OkxWsStream<Transformer> = ExchangeWsStream<WebSocketSerdeParser, Transformer>;
#[derive(
Copy,
Clone,
Eq,
PartialEq,
Ord,
PartialOrd,
Hash,
Debug,
Default,
Display,
DeExchange,
SerExchange,
)]
pub struct Okx;
impl Connector for Okx {
const ID: ExchangeId = ExchangeId::Okx;
type Channel = OkxChannel;
type Market = OkxMarket;
type Subscriber = WebSocketSubscriber;
type SubValidator = WebSocketSubValidator;
type SubResponse = OkxSubResponse;
fn url() -> Result<Url, url::ParseError> {
Url::parse(BASE_URL_OKX)
}
fn ping_interval() -> Option<PingInterval> {
Some(PingInterval {
interval: tokio::time::interval(PING_INTERVAL_OKX),
ping: || WsMessage::text("ping"),
})
}
fn requests(exchange_subs: Vec<ExchangeSub<Self::Channel, Self::Market>>) -> Vec<WsMessage> {
vec![WsMessage::text(
json!({
"op": "subscribe",
"args": &exchange_subs,
})
.to_string(),
)]
}
}
impl<Instrument> StreamSelector<Instrument, PublicTrades> for Okx
where
Instrument: InstrumentData,
{
type SnapFetcher = NoInitialSnapshots;
type Stream = OkxWsStream<StatelessTransformer<Self, Instrument::Key, PublicTrades, OkxTrades>>;
}