barter_data/exchange/coinbase/
mod.rs

1use self::{
2    channel::CoinbaseChannel, market::CoinbaseMarket, subscription::CoinbaseSubResponse,
3    trade::CoinbaseTrade,
4};
5use crate::{
6    ExchangeWsStream, NoInitialSnapshots,
7    exchange::{Connector, ExchangeSub, StreamSelector},
8    instrument::InstrumentData,
9    subscriber::{WebSocketSubscriber, validator::WebSocketSubValidator},
10    subscription::trade::PublicTrades,
11    transformer::stateless::StatelessTransformer,
12};
13use barter_instrument::exchange::ExchangeId;
14use barter_integration::{error::SocketError, protocol::websocket::WsMessage};
15use barter_macro::{DeExchange, SerExchange};
16use derive_more::Display;
17use serde_json::json;
18use url::Url;
19
20/// Defines the type that translates a Barter [`Subscription`](crate::subscription::Subscription)
21/// into an exchange [`Connector`] specific channel used for generating [`Connector::requests`].
22pub mod channel;
23
24/// Defines the type that translates a Barter [`Subscription`](crate::subscription::Subscription)
25/// into an exchange [`Connector`] specific market used for generating [`Connector::requests`].
26pub mod market;
27
28/// [`Subscription`](crate::subscription::Subscription) response type and response
29/// [`Validator`](barter_integration::Validator) for [`Coinbase`].
30pub mod subscription;
31
32/// Public trade types for [`Coinbase`].
33pub mod trade;
34
35/// [`Coinbase`] server base url.
36///
37/// See docs: <https://docs.cloud.coinbase.com/exchange/docs/websocket-overview>
38pub const BASE_URL_COINBASE: &str = "wss://ws-feed.exchange.coinbase.com";
39
40/// [`Coinbase`] exchange.
41///
42/// See docs: <https://docs.cloud.coinbase.com/exchange/docs/websocket-overview>
43#[derive(
44    Copy,
45    Clone,
46    Eq,
47    PartialEq,
48    Ord,
49    PartialOrd,
50    Hash,
51    Debug,
52    Default,
53    Display,
54    DeExchange,
55    SerExchange,
56)]
57pub struct Coinbase;
58
59impl Connector for Coinbase {
60    const ID: ExchangeId = ExchangeId::Coinbase;
61    type Channel = CoinbaseChannel;
62    type Market = CoinbaseMarket;
63    type Subscriber = WebSocketSubscriber;
64    type SubValidator = WebSocketSubValidator;
65    type SubResponse = CoinbaseSubResponse;
66
67    fn url() -> Result<Url, SocketError> {
68        Url::parse(BASE_URL_COINBASE).map_err(SocketError::UrlParse)
69    }
70
71    fn requests(exchange_subs: Vec<ExchangeSub<Self::Channel, Self::Market>>) -> Vec<WsMessage> {
72        exchange_subs
73            .into_iter()
74            .map(|ExchangeSub { channel, market }| {
75                WsMessage::text(
76                    json!({
77                        "type": "subscribe",
78                        "product_ids": [market.as_ref()],
79                        "channels": [channel.as_ref()],
80                    })
81                    .to_string(),
82                )
83            })
84            .collect()
85    }
86}
87
88impl<Instrument> StreamSelector<Instrument, PublicTrades> for Coinbase
89where
90    Instrument: InstrumentData,
91{
92    type SnapFetcher = NoInitialSnapshots;
93    type Stream =
94        ExchangeWsStream<StatelessTransformer<Self, Instrument::Key, PublicTrades, CoinbaseTrade>>;
95}