perf: simplify code logic and improve efficiency

This commit is contained in:
wonfen 2025-03-23 02:16:06 +08:00
parent bc39d56b0f
commit 9b04721b3d

View File

@ -795,21 +795,24 @@ impl CoreManager {
.output()?;
let output = String::from_utf8_lossy(&output.stdout);
let mut pids = Vec::new();
for line in output.lines() {
if line.contains(process_name) {
let pids: Vec<u32> = output
.lines()
.filter(|line| line.contains(process_name))
.filter_map(|line| {
println!("[进程检查] 发现匹配行: {}", line);
let parts: Vec<&str> = line.split(',').collect();
if parts.len() >= 2 {
let pid_str = parts[1].trim_matches('"');
if let Ok(pid) = pid_str.parse::<u32>() {
pid_str.parse::<u32>().ok().map(|pid| {
println!("[进程检查] 发现进程 PID: {}", pid);
pids.push(pid);
}
pid
})
} else {
None
}
}
}
})
.collect();
println!("[进程检查] 共发现 {} 个相关进程", pids.len());
Ok(pids)
@ -820,17 +823,22 @@ impl CoreManager {
use std::process::Command;
println!("[进程检查] Linux系统使用pgrep命令");
let output = Command::new("pgrep").arg("-f").arg(process_name).output()?;
let output = Command::new("pgrep")
.arg("-f")
.arg(process_name)
.output()?;
let output = String::from_utf8_lossy(&output.stdout);
let mut pids = Vec::new();
for line in output.lines() {
if let Ok(pid) = line.trim().parse::<u32>() {
println!("[进程检查] 发现进程 PID: {}", pid);
pids.push(pid);
}
}
let pids: Vec<u32> = output
.lines()
.filter_map(|line| {
line.trim().parse::<u32>().ok().map(|pid| {
println!("[进程检查] 发现进程 PID: {}", pid);
pid
})
})
.collect();
println!("[进程检查] 共发现 {} 个相关进程", pids.len());
Ok(pids)
@ -846,20 +854,23 @@ impl CoreManager {
.output()?;
let output = String::from_utf8_lossy(&output.stdout);
let mut pids = Vec::new();
for line in output.lines() {
if line.contains(process_name) {
let pids: Vec<u32> = output
.lines()
.filter(|line| line.contains(process_name))
.filter_map(|line| {
println!("[进程检查] 发现匹配行: {}", line);
let parts: Vec<&str> = line.split_whitespace().collect();
if !parts.is_empty() {
if let Ok(pid) = parts[0].parse::<u32>() {
parts[0].parse::<u32>().ok().map(|pid| {
println!("[进程检查] 发现进程 PID: {}", pid);
pids.push(pid);
}
pid
})
} else {
None
}
}
}
})
.collect();
println!("[进程检查] 共发现 {} 个相关进程", pids.len());
Ok(pids)