@@ -1,8 +1,8 @@ |
| 1 | 1 | //! Daemon state management |
| 2 | 2 | |
| 3 | | -use crate::collector::{CpuCollector, DiskCollector, History, MemoryCollector, NetworkCollector, ProcessCollector, TempCollector}; |
| 3 | +use crate::collector::{CpuCollector, DiskCollector, GpuCollector, History, MemoryCollector, NetworkCollector, ProcessCollector, TempCollector}; |
| 4 | 4 | use crate::error::Result; |
| 5 | | -use gartop_ipc::{CpuStats, DiskStats, MemoryStats, NetworkStats, ProcessInfo, SortField, TempStats}; |
| 5 | +use gartop_ipc::{CpuStats, DiskStats, GpuStats, MemoryStats, NetworkStats, ProcessInfo, SortField, TempStats}; |
| 6 | 6 | use procfs::Current; |
| 7 | 7 | use std::time::Instant; |
| 8 | 8 | |
@@ -14,12 +14,14 @@ pub struct DaemonState { |
| 14 | 14 | pub disk_collector: DiskCollector, |
| 15 | 15 | pub process_collector: ProcessCollector, |
| 16 | 16 | pub temp_collector: TempCollector, |
| 17 | + pub gpu_collector: GpuCollector, |
| 17 | 18 | pub cpu_history: History<CpuStats>, |
| 18 | 19 | pub memory_history: History<MemoryStats>, |
| 19 | 20 | pub network_history: History<Vec<NetworkStats>>, |
| 20 | 21 | pub disk_history: History<Vec<DiskStats>>, |
| 21 | 22 | pub processes: Vec<ProcessInfo>, |
| 22 | 23 | pub temp_stats: Option<TempStats>, |
| 24 | + pub gpu_stats: Option<GpuStats>, |
| 23 | 25 | pub started: Instant, |
| 24 | 26 | pub sample_interval_ms: u64, |
| 25 | 27 | } |
@@ -34,12 +36,14 @@ impl DaemonState { |
| 34 | 36 | disk_collector: DiskCollector::new(), |
| 35 | 37 | process_collector: ProcessCollector::new()?, |
| 36 | 38 | temp_collector: TempCollector::new()?, |
| 39 | + gpu_collector: GpuCollector::new()?, |
| 37 | 40 | cpu_history: History::new(history_size), |
| 38 | 41 | memory_history: History::new(history_size), |
| 39 | 42 | network_history: History::new(history_size), |
| 40 | 43 | disk_history: History::new(history_size), |
| 41 | 44 | processes: Vec::new(), |
| 42 | 45 | temp_stats: None, |
| 46 | + gpu_stats: None, |
| 43 | 47 | started: Instant::now(), |
| 44 | 48 | sample_interval_ms, |
| 45 | 49 | }) |
@@ -80,6 +84,13 @@ impl DaemonState { |
| 80 | 84 | Ok(stats) |
| 81 | 85 | } |
| 82 | 86 | |
| 87 | + /// Collect GPU stats (no history, just current values). |
| 88 | + pub fn collect_gpu(&mut self) -> Result<GpuStats> { |
| 89 | + let stats = self.gpu_collector.collect()?; |
| 90 | + self.gpu_stats = Some(stats.clone()); |
| 91 | + Ok(stats) |
| 92 | + } |
| 93 | + |
| 83 | 94 | /// Collect process list. |
| 84 | 95 | pub fn collect_processes( |
| 85 | 96 | &mut self, |
