kiteticker_async_manager/manager/
config.rs

1use crate::models::Mode;
2use std::time::Duration;
3
4/// Configuration for the KiteTicker multi-connection manager
5#[derive(Debug, Clone)]
6pub struct KiteManagerConfig {
7    /// Maximum symbols per WebSocket connection (Kite limit: 3000)
8    pub max_symbols_per_connection: usize,
9    
10    /// Number of WebSocket connections to maintain (Kite limit: 3)
11    pub max_connections: usize,
12    
13    /// Buffer size for each connection's message channel
14    pub connection_buffer_size: usize,
15    
16    /// Buffer size for each parser's output channel
17    pub parser_buffer_size: usize,
18    
19    /// Connection timeout for establishing WebSocket connections
20    pub connection_timeout: Duration,
21    
22    /// Health check interval for monitoring connections
23    pub health_check_interval: Duration,
24    
25    /// Maximum reconnection attempts per connection
26    pub max_reconnect_attempts: usize,
27    
28    /// Delay between reconnection attempts
29    pub reconnect_delay: Duration,
30    
31    /// Enable dedicated parser tasks for each connection
32    pub enable_dedicated_parsers: bool,
33    
34    /// Default subscription mode for new symbols
35    pub default_mode: Mode,
36}
37
38impl Default for KiteManagerConfig {
39    fn default() -> Self {
40        Self {
41            max_symbols_per_connection: 3000,
42            max_connections: 3,
43            connection_buffer_size: 5000,    // High buffer for performance
44            parser_buffer_size: 10000,       // Even higher for parsed messages
45            connection_timeout: Duration::from_secs(30),
46            health_check_interval: Duration::from_secs(10),
47            max_reconnect_attempts: 5,
48            reconnect_delay: Duration::from_secs(2),
49            enable_dedicated_parsers: true,
50            default_mode: Mode::Quote,
51        }
52    }
53}
54
55/// Connection statistics for monitoring
56#[derive(Debug, Clone, Default)]
57pub struct ConnectionStats {
58    pub connection_id: usize,
59    pub is_connected: bool,
60    pub symbol_count: usize,
61    pub messages_received: u64,
62    pub messages_parsed: u64,
63    pub errors_count: u64,
64    pub last_message_time: Option<std::time::Instant>,
65    pub average_latency: Duration,
66    pub connection_uptime: Duration,
67}
68
69/// Manager-wide statistics
70#[derive(Debug, Clone, Default)]
71pub struct ManagerStats {
72    pub total_symbols: usize,
73    pub active_connections: usize,
74    pub total_messages_received: u64,
75    pub total_messages_parsed: u64,
76    pub total_errors: u64,
77    pub uptime: Duration,
78    pub connection_stats: Vec<ConnectionStats>,
79}
80
81/// Channel identifier for output channels
82#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
83pub enum ChannelId {
84    Connection1 = 0,
85    Connection2 = 1,
86    Connection3 = 2,
87}
88
89impl ChannelId {
90    pub fn from_index(index: usize) -> Option<Self> {
91        match index {
92            0 => Some(Self::Connection1),
93            1 => Some(Self::Connection2),
94            2 => Some(Self::Connection3),
95            _ => None,
96        }
97    }
98    
99    pub fn to_index(self) -> usize {
100        self as usize
101    }
102    
103    pub fn all() -> Vec<Self> {
104        vec![Self::Connection1, Self::Connection2, Self::Connection3]
105    }
106}
kiteticker_async_manager/manager/config.rs

kiteticker_async_manager/manager/
config.rs
