rustrade-data 0.2.0

High performance & normalised WebSocket intergration for leading cryptocurrency exchanges - batteries included.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125

use super::{Alpaca, AlpacaServerCrypto, AlpacaServerIex, AlpacaServerSip};
use crate::{Identifier, instrument::MarketInstrumentData, subscription::Subscription};
use rustrade_instrument::{
    Keyed, asset::name::AssetNameInternal, instrument::market_data::MarketDataInstrument,
};
use serde::{Deserialize, Serialize};
use smol_str::{SmolStr, StrExt, format_smolstr};

/// Alpaca market identifier.
///
/// - Equities (IEX/SIP): uppercase symbol, e.g., `"AAPL"`, `"SPY"`
/// - Crypto: `"BASE/USD"` format, e.g., `"BTC/USD"`, `"ETH/USD"`
#[derive(Clone, Eq, PartialEq, Ord, PartialOrd, Hash, Debug, Deserialize, Serialize)]
pub struct AlpacaMarket(pub SmolStr);

impl AsRef<str> for AlpacaMarket {
    fn as_ref(&self) -> &str {
        &self.0
    }
}

fn equities_market(base: &AssetNameInternal) -> AlpacaMarket {
    AlpacaMarket(base.name().to_uppercase_smolstr())
}

fn crypto_market(base: &AssetNameInternal, quote: &AssetNameInternal) -> AlpacaMarket {
    AlpacaMarket(format_smolstr!(
        "{}/{}",
        base.name().to_uppercase_smolstr(),
        quote.name().to_uppercase_smolstr()
    ))
}

impl<Kind> Identifier<AlpacaMarket>
    for Subscription<Alpaca<AlpacaServerIex>, MarketDataInstrument, Kind>
{
    fn id(&self) -> AlpacaMarket {
        equities_market(&self.instrument.base)
    }
}

impl<Kind> Identifier<AlpacaMarket>
    for Subscription<Alpaca<AlpacaServerSip>, MarketDataInstrument, Kind>
{
    fn id(&self) -> AlpacaMarket {
        equities_market(&self.instrument.base)
    }
}

impl<Kind> Identifier<AlpacaMarket>
    for Subscription<Alpaca<AlpacaServerCrypto>, MarketDataInstrument, Kind>
{
    fn id(&self) -> AlpacaMarket {
        crypto_market(&self.instrument.base, &self.instrument.quote)
    }
}

impl<InstrumentKey, Kind> Identifier<AlpacaMarket>
    for Subscription<Alpaca<AlpacaServerIex>, Keyed<InstrumentKey, MarketDataInstrument>, Kind>
{
    fn id(&self) -> AlpacaMarket {
        equities_market(&self.instrument.value.base)
    }
}

impl<InstrumentKey, Kind> Identifier<AlpacaMarket>
    for Subscription<Alpaca<AlpacaServerSip>, Keyed<InstrumentKey, MarketDataInstrument>, Kind>
{
    fn id(&self) -> AlpacaMarket {
        equities_market(&self.instrument.value.base)
    }
}

impl<InstrumentKey, Kind> Identifier<AlpacaMarket>
    for Subscription<Alpaca<AlpacaServerCrypto>, Keyed<InstrumentKey, MarketDataInstrument>, Kind>
{
    fn id(&self) -> AlpacaMarket {
        crypto_market(&self.instrument.value.base, &self.instrument.value.quote)
    }
}

impl<InstrumentKey, Kind> Identifier<AlpacaMarket>
    for Subscription<Alpaca<AlpacaServerIex>, MarketInstrumentData<InstrumentKey>, Kind>
{
    fn id(&self) -> AlpacaMarket {
        AlpacaMarket(self.instrument.name_exchange.name().clone())
    }
}

impl<InstrumentKey, Kind> Identifier<AlpacaMarket>
    for Subscription<Alpaca<AlpacaServerSip>, MarketInstrumentData<InstrumentKey>, Kind>
{
    fn id(&self) -> AlpacaMarket {
        AlpacaMarket(self.instrument.name_exchange.name().clone())
    }
}

impl<InstrumentKey, Kind> Identifier<AlpacaMarket>
    for Subscription<Alpaca<AlpacaServerCrypto>, MarketInstrumentData<InstrumentKey>, Kind>
{
    fn id(&self) -> AlpacaMarket {
        AlpacaMarket(self.instrument.name_exchange.name().clone())
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use rustrade_instrument::asset::name::AssetNameInternal;

    #[test]
    fn test_equities_market_format() {
        let base = AssetNameInternal::new("aapl");
        let market = equities_market(&base);
        assert_eq!(market.as_ref(), "AAPL");
    }

    #[test]
    fn test_crypto_market_format() {
        let base = AssetNameInternal::new("btc");
        let quote = AssetNameInternal::new("usd");
        let market = crypto_market(&base, &quote);
        assert_eq!(market.as_ref(), "BTC/USD");
    }
}