ziro 0.0.12

跨平台端口管理工具 - 快速查找和终止占用端口的进程
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95

use crate::ui;
use anyhow::Result;
use std::io::{self, Write};
use std::thread;
use std::time::{Duration, Instant};
use sysinfo::{ProcessRefreshKind, ProcessesToUpdate, RefreshKind, System};

/// top 子命令的配置
pub struct TopOptions {
    pub interval: f32,
    pub limit: usize,
    pub show_cpu: bool,
    pub show_cmd: bool,
    pub once: bool,
}

/// 用于展示的进程信息
pub struct ProcessView {
    pub pid: u32,
    pub name: String,
    pub memory_bytes: u64,
    pub cpu: f32,
    pub cmd: String,
}

pub fn run_top(opts: TopOptions) -> Result<()> {
    let process_refresh = ProcessRefreshKind::everything();
    let mut system = System::new_with_specifics(RefreshKind::new().with_processes(process_refresh));

    // 进入备用屏幕，避免污染滚动历史（once 模式不需要）
    let use_alt_screen = !opts.once;
    if use_alt_screen {
        print!("\x1b[?1049h");
        let _ = io::stdout().flush();
    }

    let mut tick: u64 = 0;

    loop {
        tick = tick.wrapping_add(1);
        let start = Instant::now();

        system.refresh_processes_specifics(ProcessesToUpdate::All, process_refresh);

        let mut processes: Vec<ProcessView> = system
            .processes()
            .iter()
            .map(|(pid, process)| {
                let cmd = process
                    .cmd()
                    .iter()
                    .map(|s| s.to_string_lossy())
                    .collect::<Vec<_>>()
                    .join(" ");

                ProcessView {
                    pid: pid.as_u32(),
                    name: process.name().to_string_lossy().into_owned(),
                    // sysinfo 的 memory() 返回字节数
                    memory_bytes: process.memory(),
                    cpu: process.cpu_usage(),
                    cmd,
                }
            })
            .collect();

        processes.sort_by(|a, b| b.memory_bytes.cmp(&a.memory_bytes));
        processes.truncate(opts.limit.max(1));

        ui::display_top(
            &processes,
            tick,
            opts.interval,
            opts.show_cpu,
            opts.show_cmd,
        );

        if opts.once {
            break;
        }

        // 补偿刷新时间，保持接近 interval
        let elapsed = start.elapsed();
        let sleep_ms = ((opts.interval * 1000.0) as i64 - elapsed.as_millis() as i64).max(0) as u64;
        thread::sleep(Duration::from_millis(sleep_ms));
    }

    // 离开备用屏幕，恢复原屏幕内容
    if use_alt_screen {
        print!("\x1b[?1049l");
        let _ = io::stdout().flush();
    }

    Ok(())
}