Compare commits
9 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 6d6beb207d | |||
| 7a68da01f5 | |||
| 5be67fed64 | |||
| cac836601b | |||
| bd22ce265b | |||
| bbc8b7b1cb | |||
| 5dd8cadef3 | |||
| fefe30ec51 | |||
| fb40cce748 |
6
Cargo.lock
generated
6
Cargo.lock
generated
@@ -279,7 +279,7 @@ checksum = "a1d728cc89cf3aee9ff92b05e62b19ee65a02b5702cff7d5a377e32c6ae29d8d"
|
|||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "cm-dashboard"
|
name = "cm-dashboard"
|
||||||
version = "0.1.175"
|
version = "0.1.184"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"anyhow",
|
"anyhow",
|
||||||
"chrono",
|
"chrono",
|
||||||
@@ -301,7 +301,7 @@ dependencies = [
|
|||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "cm-dashboard-agent"
|
name = "cm-dashboard-agent"
|
||||||
version = "0.1.175"
|
version = "0.1.184"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"anyhow",
|
"anyhow",
|
||||||
"async-trait",
|
"async-trait",
|
||||||
@@ -324,7 +324,7 @@ dependencies = [
|
|||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "cm-dashboard-shared"
|
name = "cm-dashboard-shared"
|
||||||
version = "0.1.175"
|
version = "0.1.184"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"chrono",
|
"chrono",
|
||||||
"serde",
|
"serde",
|
||||||
|
|||||||
@@ -1,6 +1,6 @@
|
|||||||
[package]
|
[package]
|
||||||
name = "cm-dashboard-agent"
|
name = "cm-dashboard-agent"
|
||||||
version = "0.1.176"
|
version = "0.1.185"
|
||||||
edition = "2021"
|
edition = "2021"
|
||||||
|
|
||||||
[dependencies]
|
[dependencies]
|
||||||
|
|||||||
@@ -386,7 +386,7 @@ impl DiskCollector {
|
|||||||
/// Get SMART data for drives
|
/// Get SMART data for drives
|
||||||
async fn get_smart_data_for_drives(&self, physical_drives: &[PhysicalDrive], mergerfs_pools: &[MergerfsPool]) -> HashMap<String, SmartData> {
|
async fn get_smart_data_for_drives(&self, physical_drives: &[PhysicalDrive], mergerfs_pools: &[MergerfsPool]) -> HashMap<String, SmartData> {
|
||||||
let mut smart_data = HashMap::new();
|
let mut smart_data = HashMap::new();
|
||||||
|
|
||||||
// Collect all drive names
|
// Collect all drive names
|
||||||
let mut all_drives = std::collections::HashSet::new();
|
let mut all_drives = std::collections::HashSet::new();
|
||||||
for drive in physical_drives {
|
for drive in physical_drives {
|
||||||
@@ -413,23 +413,24 @@ impl DiskCollector {
|
|||||||
|
|
||||||
/// Get SMART data for a single drive
|
/// Get SMART data for a single drive
|
||||||
async fn get_smart_data(&self, drive_name: &str) -> Result<SmartData, CollectorError> {
|
async fn get_smart_data(&self, drive_name: &str) -> Result<SmartData, CollectorError> {
|
||||||
let output = Command::new("sudo")
|
// Use direct smartctl (no sudo) - service has CAP_SYS_RAWIO and CAP_SYS_ADMIN capabilities
|
||||||
.args(&["smartctl", "-a", &format!("/dev/{}", drive_name)])
|
// For NVMe drives, specify device type explicitly
|
||||||
.output()
|
let mut cmd = Command::new("smartctl");
|
||||||
|
if drive_name.starts_with("nvme") {
|
||||||
|
cmd.args(&["-d", "nvme", "-a", &format!("/dev/{}", drive_name)]);
|
||||||
|
} else {
|
||||||
|
cmd.args(&["-a", &format!("/dev/{}", drive_name)]);
|
||||||
|
}
|
||||||
|
|
||||||
|
let output = cmd.output()
|
||||||
.map_err(|e| CollectorError::SystemRead {
|
.map_err(|e| CollectorError::SystemRead {
|
||||||
path: format!("SMART data for {}", drive_name),
|
path: format!("SMART data for {}", drive_name),
|
||||||
error: e.to_string(),
|
error: e.to_string(),
|
||||||
})?;
|
})?;
|
||||||
|
|
||||||
let output_str = String::from_utf8_lossy(&output.stdout);
|
let output_str = String::from_utf8_lossy(&output.stdout);
|
||||||
let error_str = String::from_utf8_lossy(&output.stderr);
|
|
||||||
|
|
||||||
// Debug logging for SMART command results
|
|
||||||
debug!("SMART output for {}: status={}, stdout_len={}, stderr={}",
|
|
||||||
drive_name, output.status, output_str.len(), error_str);
|
|
||||||
|
|
||||||
if !output.status.success() {
|
if !output.status.success() {
|
||||||
debug!("SMART command failed for {}: {}", drive_name, error_str);
|
|
||||||
// Return unknown data rather than failing completely
|
// Return unknown data rather than failing completely
|
||||||
return Ok(SmartData {
|
return Ok(SmartData {
|
||||||
health: "UNKNOWN".to_string(),
|
health: "UNKNOWN".to_string(),
|
||||||
|
|||||||
@@ -4,7 +4,7 @@ use cm_dashboard_shared::{AgentData, ServiceData, SubServiceData, SubServiceMetr
|
|||||||
use std::process::Command;
|
use std::process::Command;
|
||||||
use std::sync::RwLock;
|
use std::sync::RwLock;
|
||||||
use std::time::Instant;
|
use std::time::Instant;
|
||||||
use tracing::{debug, info};
|
use tracing::debug;
|
||||||
|
|
||||||
use super::{Collector, CollectorError};
|
use super::{Collector, CollectorError};
|
||||||
use crate::config::SystemdConfig;
|
use crate::config::SystemdConfig;
|
||||||
@@ -118,10 +118,7 @@ impl SystemdCollector {
|
|||||||
}
|
}
|
||||||
|
|
||||||
if service_name.contains("docker") && active_status == "active" {
|
if service_name.contains("docker") && active_status == "active" {
|
||||||
info!("Collecting Docker sub-services for service: {}", service_name);
|
|
||||||
|
|
||||||
let docker_containers = self.get_docker_containers();
|
let docker_containers = self.get_docker_containers();
|
||||||
info!("Found {} Docker containers", docker_containers.len());
|
|
||||||
for (container_name, container_status) in docker_containers {
|
for (container_name, container_status) in docker_containers {
|
||||||
// For now, docker containers have no additional metrics
|
// For now, docker containers have no additional metrics
|
||||||
// Future: could add memory_mb, cpu_percent, restart_count, etc.
|
// Future: could add memory_mb, cpu_percent, restart_count, etc.
|
||||||
@@ -136,23 +133,20 @@ impl SystemdCollector {
|
|||||||
|
|
||||||
// Add Docker images
|
// Add Docker images
|
||||||
let docker_images = self.get_docker_images();
|
let docker_images = self.get_docker_images();
|
||||||
info!("Found {} Docker images", docker_images.len());
|
for (image_name, image_status, image_size_str, image_size_mb) in docker_images {
|
||||||
for (image_name, image_status, image_size) in docker_images {
|
|
||||||
let mut metrics = Vec::new();
|
let mut metrics = Vec::new();
|
||||||
metrics.push(SubServiceMetric {
|
metrics.push(SubServiceMetric {
|
||||||
label: "size".to_string(),
|
label: "size".to_string(),
|
||||||
value: 0.0, // Size as string in name instead
|
value: image_size_mb,
|
||||||
unit: None,
|
unit: Some("MB".to_string()),
|
||||||
});
|
});
|
||||||
|
|
||||||
sub_services.push(SubServiceData {
|
sub_services.push(SubServiceData {
|
||||||
name: format!("{} ({})", image_name, image_size),
|
name: format!("{} ({})", image_name, image_size_str),
|
||||||
service_status: self.calculate_service_status(&image_name, &image_status),
|
service_status: self.calculate_service_status(&image_name, &image_status),
|
||||||
metrics,
|
metrics,
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
info!("Total Docker sub-services added: {}", sub_services.len());
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Create complete service data
|
// Create complete service data
|
||||||
@@ -175,6 +169,10 @@ impl SystemdCollector {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Sort services alphabetically by name
|
||||||
|
agent_data.services.sort_by(|a, b| a.name.cmp(&b.name));
|
||||||
|
complete_service_data.sort_by(|a, b| a.name.cmp(&b.name));
|
||||||
|
|
||||||
// Update cached state
|
// Update cached state
|
||||||
{
|
{
|
||||||
let mut state = self.state.write().unwrap();
|
let mut state = self.state.write().unwrap();
|
||||||
@@ -779,10 +777,10 @@ impl SystemdCollector {
|
|||||||
fn get_docker_containers(&self) -> Vec<(String, String)> {
|
fn get_docker_containers(&self) -> Vec<(String, String)> {
|
||||||
let mut containers = Vec::new();
|
let mut containers = Vec::new();
|
||||||
|
|
||||||
// Check if docker is available (use sudo for permissions)
|
// Check if docker is available (cm-agent user is in docker group)
|
||||||
// Use -a to show ALL containers (running and stopped)
|
// Use -a to show ALL containers (running and stopped)
|
||||||
let output = Command::new("sudo")
|
let output = Command::new("docker")
|
||||||
.args(&["docker", "ps", "-a", "--format", "{{.Names}},{{.Status}}"])
|
.args(&["ps", "-a", "--format", "{{.Names}},{{.Status}}"])
|
||||||
.output();
|
.output();
|
||||||
|
|
||||||
let output = match output {
|
let output = match output {
|
||||||
@@ -821,24 +819,19 @@ impl SystemdCollector {
|
|||||||
}
|
}
|
||||||
|
|
||||||
/// Get docker images as sub-services
|
/// Get docker images as sub-services
|
||||||
fn get_docker_images(&self) -> Vec<(String, String, String)> {
|
fn get_docker_images(&self) -> Vec<(String, String, String, f32)> {
|
||||||
let mut images = Vec::new();
|
let mut images = Vec::new();
|
||||||
|
// Check if docker is available (cm-agent user is in docker group)
|
||||||
info!("Collecting Docker images");
|
let output = Command::new("docker")
|
||||||
|
.args(&["images", "--format", "{{.Repository}}:{{.Tag}},{{.Size}}"])
|
||||||
// Check if docker is available (use sudo for permissions)
|
|
||||||
let output = Command::new("sudo")
|
|
||||||
.args(&["docker", "images", "--format", "{{.Repository}}:{{.Tag}},{{.Size}}"])
|
|
||||||
.output();
|
.output();
|
||||||
|
|
||||||
let output = match output {
|
let output = match output {
|
||||||
Ok(out) if out.status.success() => out,
|
Ok(out) if out.status.success() => out,
|
||||||
Ok(out) => {
|
Ok(_) => {
|
||||||
info!("Docker images command failed with status: {}", out.status);
|
|
||||||
return images;
|
return images;
|
||||||
}
|
}
|
||||||
Err(e) => {
|
Err(_) => {
|
||||||
info!("Docker images command error: {}", e);
|
|
||||||
return images;
|
return images;
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
@@ -856,23 +849,55 @@ impl SystemdCollector {
|
|||||||
let parts: Vec<&str> = line.split(',').collect();
|
let parts: Vec<&str> = line.split(',').collect();
|
||||||
if parts.len() >= 2 {
|
if parts.len() >= 2 {
|
||||||
let image_name = parts[0].trim();
|
let image_name = parts[0].trim();
|
||||||
let size = parts[1].trim();
|
let size_str = parts[1].trim();
|
||||||
|
|
||||||
// Skip <none>:<none> images (dangling images)
|
// Skip <none>:<none> images (dangling images)
|
||||||
if image_name.contains("<none>") {
|
if image_name.contains("<none>") {
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Parse size to MB (sizes come as "142MB", "1.5GB", "512kB", etc.)
|
||||||
|
let size_mb = self.parse_docker_size(size_str);
|
||||||
|
|
||||||
images.push((
|
images.push((
|
||||||
format!("image_{}", image_name),
|
format!("image_{}", image_name),
|
||||||
"active".to_string(), // Images are always "active" (present)
|
"active".to_string(), // Images are always "active" (present)
|
||||||
size.to_string()
|
size_str.to_string(),
|
||||||
|
size_mb
|
||||||
));
|
));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
images
|
images
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Parse Docker size string to MB
|
||||||
|
fn parse_docker_size(&self, size_str: &str) -> f32 {
|
||||||
|
let size_upper = size_str.to_uppercase();
|
||||||
|
|
||||||
|
// Extract numeric part and unit
|
||||||
|
let mut num_str = String::new();
|
||||||
|
let mut unit = String::new();
|
||||||
|
|
||||||
|
for ch in size_upper.chars() {
|
||||||
|
if ch.is_ascii_digit() || ch == '.' {
|
||||||
|
num_str.push(ch);
|
||||||
|
} else if ch.is_alphabetic() {
|
||||||
|
unit.push(ch);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
let value: f32 = num_str.parse().unwrap_or(0.0);
|
||||||
|
|
||||||
|
// Convert to MB
|
||||||
|
match unit.as_str() {
|
||||||
|
"KB" | "K" => value / 1024.0,
|
||||||
|
"MB" | "M" => value,
|
||||||
|
"GB" | "G" => value * 1024.0,
|
||||||
|
"TB" | "T" => value * 1024.0 * 1024.0,
|
||||||
|
_ => value, // Assume bytes if no unit
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[async_trait]
|
#[async_trait]
|
||||||
|
|||||||
@@ -1,6 +1,6 @@
|
|||||||
[package]
|
[package]
|
||||||
name = "cm-dashboard"
|
name = "cm-dashboard"
|
||||||
version = "0.1.176"
|
version = "0.1.185"
|
||||||
edition = "2021"
|
edition = "2021"
|
||||||
|
|
||||||
[dependencies]
|
[dependencies]
|
||||||
|
|||||||
@@ -1,6 +1,6 @@
|
|||||||
[package]
|
[package]
|
||||||
name = "cm-dashboard-shared"
|
name = "cm-dashboard-shared"
|
||||||
version = "0.1.176"
|
version = "0.1.185"
|
||||||
edition = "2021"
|
edition = "2021"
|
||||||
|
|
||||||
[dependencies]
|
[dependencies]
|
||||||
|
|||||||
Reference in New Issue
Block a user