digdigdig3-station 0.3.3

Consumer-facing builder over digdigdig3 ExchangeHub. Persistence, cache, replay, cure, orderbook tracker, multiplex, reconnect.
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

//! Persistence configuration for the Station builder.
//!
//! Actual disk I/O lives in [`crate::series::DiskStore`], generic over
//! [`crate::series::DataPoint`]. This module only carries on/off toggles
//! threaded through the builder.

use serde::{Deserialize, Serialize};

/// Per-kind persistence toggles. Master `enabled` gates everything.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct PersistenceConfig {
    pub enabled: bool,
    pub trades: bool,
    pub agg_trades: bool,
    pub klines: bool,
    pub tickers: bool,
    pub orderbook_snapshots: bool,
    pub mark_price: bool,
    pub funding_rate: bool,
    pub open_interest: bool,
    pub liquidations: bool,
}

impl Default for PersistenceConfig {
    fn default() -> Self {
        Self {
            enabled: false,
            trades: false,
            agg_trades: false,
            klines: false,
            tickers: false,
            orderbook_snapshots: false,
            mark_price: false,
            funding_rate: false,
            open_interest: false,
            liquidations: false,
        }
    }
}

impl PersistenceConfig {
    /// Enable persistence for every supported kind.
    pub fn on() -> Self {
        Self {
            enabled: true,
            trades: true,
            agg_trades: true,
            klines: true,
            tickers: true,
            orderbook_snapshots: true,
            mark_price: true,
            funding_rate: true,
            open_interest: true,
            liquidations: true,
        }
    }

    pub fn trades(mut self, on: bool) -> Self { self.trades = on; self }
    pub fn agg_trades(mut self, on: bool) -> Self { self.agg_trades = on; self }
    pub fn klines(mut self, on: bool) -> Self { self.klines = on; self }
    pub fn tickers(mut self, on: bool) -> Self { self.tickers = on; self }
    pub fn orderbook_snapshots(mut self, on: bool) -> Self { self.orderbook_snapshots = on; self }
    pub fn mark_price(mut self, on: bool) -> Self { self.mark_price = on; self }
    pub fn funding_rate(mut self, on: bool) -> Self { self.funding_rate = on; self }
    pub fn open_interest(mut self, on: bool) -> Self { self.open_interest = on; self }
    pub fn liquidations(mut self, on: bool) -> Self { self.liquidations = on; self }

    /// Should `kind` be persisted given the current config?
    pub fn is_enabled_for(&self, kind: &crate::series::Kind) -> bool {
        if !self.enabled {
            return false;
        }
        use crate::series::Kind::*;
        match kind {
            Trade => self.trades,
            AggTrade => self.agg_trades,
            Kline(_) => self.klines,
            Ticker => self.tickers,
            Orderbook => self.orderbook_snapshots,
            MarkPrice => self.mark_price,
            FundingRate => self.funding_rate,
            OpenInterest => self.open_interest,
            Liquidation => self.liquidations,
        }
    }
}