1. 1//! Operational telemetry state: CPU/Memory gauges, token burn-rate waveform,
2. 2//! agent swarm radar, and provider latency.
3. 3//!
4. 4//! All values are populated from real server data — `/health` for the
5. 5//! system/provider snapshot and `AgentEvent` stream events for live
6. 6//! token usage, worker counts, and pending task counts. The previous
7. 7//! implementation hardcoded all values; this one derives them from actual
8. 8//! API responses.
9. 9
10. 10use crate::api::HealthReport;
11. 11
12. 12#[derive(Debug, Clone)]
13. 13pub struct TelemetryState {
14. 14 pub cpu_percent: f32,
15. 15 pub rss_mb: f64,
16. 16 pub token_rate_history: Vec<u32>,
17. 17 pub max_rate: u32,
18. 18 pub active_workers: usize,
19. 19 pub pending_tasks: usize,
20. 20 pub radar_angle_deg: f32,
21. 21 pub spend_today_usd: f64,
22. 22 pub spend_budget_cap_usd: f64,
23. 23 pub circuit_breaker_healthy: bool,
24. 24 pub active_services_count: (usize, usize),
25. 25 pub bus_queue_depth: usize,
26. 26 pub bus_dlq_count: usize,
27. 27 pub anthropic_latency_ms: u32,
28. 28 pub ollama_latency_ms: u32,
29. 29 pub tick_count: usize,
30. 30 pub connected: bool,
31. 31 pub last_error: Option<String>,
32. 32}
33. 33
34. 34impl Default for TelemetryState {
35. 35 fn default() -> Self {
36. 36 Self {
37. 37 cpu_percent: 0.0,
38. 38 rss_mb: 0.0,
39. 39 token_rate_history: Vec::new(),
40. 40 max_rate: 0,
41. 41 active_workers: 0,
42. 42 pending_tasks: 0,
43. 43 radar_angle_deg: 0.0,
44. 44 spend_today_usd: 0.0,
45. 45 spend_budget_cap_usd: 0.0,
46. 46 circuit_breaker_healthy: false,
47. 47 active_services_count: (0, 0),
48. 48 bus_queue_depth: 0,
49. 49 bus_dlq_count: 0,
50. 50 anthropic_latency_ms: 0,
51. 51 ollama_latency_ms: 0,
52. 52 tick_count: 0,
53. 53 connected: false,
54. 54 last_error: None,
55. 55 }
56. 56 }
57. 57}
58. 58
59. 59impl TelemetryState {
60. 60 pub fn new() -> Self {
61. 61 Self::default()
62. 62 }
63. 63
64. 64 /// Advance visual animations on tick (50ms).
65. 65 pub fn tick(&mut self) {
66. 66 self.tick_count = self.tick_count.wrapping_add(1);
67. 67 self.radar_angle_deg = (self.radar_angle_deg + 6.0) % 360.0;
68. 68 }
69. 69
70. 70 /// Update telemetry from a real `/health` response.
71. 71 pub fn update_from_health(&mut self, health: &HealthReport) {
72. 72 self.circuit_breaker_healthy = health.circuit_breaker_healthy;
73. 73 self.connected = health.healthy;
74. 74 }
75. 75
76. 76 /// Push a real token-rate sample from an agent TurnEnd event.
77. 77 pub fn push_token_rate(&mut self, rate: u32) {
78. 78 if self.token_rate_history.len() >= 60 {
79. 79 self.token_rate_history.remove(0);
80. 80 }
81. 81 self.token_rate_history.push(rate);
82. 82 if rate > self.max_rate {
83. 83 self.max_rate = rate;
84. 84 }
85. 85 }
86. 86
87. 87 /// Update worker presence from `WorkerStarted` / `WorkerFinished`.
88. 88 pub fn set_active_workers(&mut self, count: usize) {
89. 89 self.active_workers = count;
90. 90 }
91. 91
92. 92 /// Record a connection error so the UI can surface it.
93. 93 pub fn set_connection_error(&mut self, msg: String) {
94. 94 self.connected = false;
95. 95 self.last_error = Some(msg);
96. 96 }
97. 97
98. 98 /// Clear the error indicator on successful data receipt.
99. 99 pub fn clear_error(&mut self) {
100. 100 if self.connected {
101. 101 self.last_error = None;
102. 102 }
103. 103 }
104. 104}
105. 105