Rust 获取系统资源监控及自动壁纸设置

主要库对比

推荐使用 sysinfo 库，因为它功能全面、文档完善，并且在所有主要平台上都能良好工作。

实时监控程序

基于上述基础，我们可以构建一个类似任务管理器的实时监控程序。

常见报错处理

新版本中 refresh_cpu_usage() 方法已被移除，正确的做法是使用 refresh_cpu()。同时 uptime() 是实例方法，需要从系统实例调用，即 sys.uptime()。

实时监控代码

use std::{thread, time::Duration};
use sysinfo::{System, SystemExt, CpuExt, ProcessExt};

fn monitor_system() {
    let mut sys = System::new_all();
    loop {
        // 刷新系统信息
        sys.refresh_cpu();
        thread::sleep(Duration::from_secs(1));
        sys.refresh_memory();
        sys.refresh_processes();

        // 清屏并显示信息
        println!("\x1B[2J\x1B[1;1H");
        println!("=== 系统实时监控 ===");
        println!("更新时间：{:?}", sys.uptime());

        // 内存信息
        println!("\n=== 内存信息 ===");
        println!("总内存：{:>6} MB", sys.total_memory() / 1024 / 1024);
        println!("已用内存：{:>6} MB", sys.used_memory() / 1024 / 1024);
        println!("可用内存：{:>6} MB", sys.available_memory() / 1024 / 1024);
        println!(
            "使用率：{:>6.1}%",
            (sys.used_memory() as f64 / sys.total_memory() as f64) * 100.0
        );

        // CPU 信息
        println!("\n=== CPU 信息 ===");
        let global_cpu_usage = if !sys.cpus().is_empty() {
            sys.cpus()
                .iter()
                .map(|cpu| cpu.cpu_usage())
                .sum::<f32>()
                / sys.cpus().len() as f32
        } else {
            0.0
        };
        println!("CPU 核心数：{}", sys.cpus().len());
        println!("全局使用率：{:.1}%", global_cpu_usage);

        // 显示每个 CPU 核心的使用率柱状图
        for (i, cpu) in sys.cpus().iter().enumerate() {
            let usage = cpu.cpu_usage();
            let bars = (usage / 5.0) as usize;
            let bar = "=".repeat(bars.min(20));
            let spaces = " ".repeat(20 - bars.min(20));
            println!(
                "CPU {:2}: [{}{}] {:5.1}%",
                i, bar, spaces, usage
            );
        }

        // 进程信息
        println!("\n=== 内存占用前 5 的进程 ===");
        let mut processes: Vec<_> = sys.processes().values().collect();
        processes.sort_by(|a, b| b.memory().cmp(&a.memory()));
        println!("{:<6} {:<20} {:<10}", "PID", "进程名", "内存 (MB)");
        println!("{}", "-".repeat(40));
        for process in processes.iter().take(5) {
            println!(
                "{:<6} {:<20} {:<10}",
                process.pid(),
                process.name(),
                process.memory() / 1024 / 1024
            );
        }

        // 系统信息
        println!("\n=== 系统信息 ===");
        println!("系统运行时间：{} 秒", sys.uptime());
        println!("进程总数：{}", sys.processes().len());
        println!("\n按 Ctrl+C 退出监控...");

        thread::sleep(Duration::from_secs(2));
    }
}

fn main() {
    println!("启动系统监控...");
    println!("按 Ctrl+C 可以退出程序");
    thread::sleep(Duration::from_secs(2));
    monitor_system();
}

壁纸操作

添加依赖

我们需要引入网络请求、图像处理及壁纸设置相关的库：

actix-web = "4.4"
reqwest = { version = "0.11", features = ["blocking", "json"] }
serde = { version = "1.0", features = ["derive"] }
serde_json = "1.0"
chrono = { version = "0.4", features = ["serde"] }
image = "0.24"
wallpaper = "3"
dirs = "4.0"

注意检查 wallpaper 的版本，确保与当前环境兼容。

完整代码

use std::fs::File;
use std::io::copy;
use std::path::PathBuf;
use dirs::picture_dir;

fn main() -> Result<(), Box<dyn std::error::Error>> {
    println!("开始获取必应每日壁纸...");
    // 1. 获取必应图片 URL
    let image_url = get_bing_image_url()?;
    println!("获取到图片 URL: {}", image_url);
    // 2. 下载图片
    let image_path = download_image(&image_url)?;
    println!("图片已下载到：{:?}", image_path);
    // 3. 设置为壁纸
    set_wallpaper(&image_path)?;
    println!("壁纸设置成功!");
    Ok(())
}

fn get_bing_image_url() -> Result<String, Box<dyn std::error::Error>> {
    // 必应每日图片 API
    let api_url = "https://cn.bing.com/HPImageArchive.aspx?format=js&idx=0&n=1&mkt=zh-CN";
    let response = reqwest::blocking::get(api_url)?;
    let json: serde_json::Value = response.json()?;

    // 解析 JSON 获取图片 URL
    if let Some(images) = json["images"].as_array() {
        if let Some(first_image) = images.get(0) {
            if let Some(url) = first_image["url"].as_str() {
                let full_url = format!("https://cn.bing.com{}", url);
                return Ok(full_url);
            }
        }
    }
    Err("无法解析必应图片 URL".into())
}

fn download_image(url: &str) -> Result<PathBuf, Box<dyn std::error::Error>> {
    // 创建响应
    let response = reqwest::blocking::get(url)?;
    // 获取文件扩展名
    let extension = if url.contains(".jpg") || url.contains(".jpeg") {
        "jpg"
    } else if url.contains(".png") {
        "png"
    } else {
        "jpg"
    };

    // 生成文件名
    let file_name = format!(
        "bing_wallpaper_{}.{}",
        chrono::Local::now().format("%Y%m%d_%H%M%S"),
        extension
    );

    // 保存到图片目录
    let mut picture_dir = picture_dir().ok_or("无法获取图片目录")?;
    picture_dir.push("bing_wallpapers");
    // 创建目录（如果不存在）
    std::fs::create_dir_all(&picture_dir)?;
    let file_path = picture_dir.join(file_name);

    // 下载并保存文件
    let mut dest = File::create(&file_path)?;
    let mut content = std::io::Cursor::new(response.bytes()?);
    copy(&mut content, &mut dest)?;
    Ok(file_path)
}

fn set_wallpaper(image_path: &PathBuf) -> Result<(), Box<dyn std::error::Error>> {
    // 使用 wallpaper 库设置壁纸
    wallpaper::set_from_path(
        image_path
            .to_str()
            .ok_or("无效的图片路径")?,
    )?;
    // 设置壁纸模式（居中、平铺、拉伸等）
    wallpaper::set_mode(wallpaper::Mode::Crop)?;
    Ok(())
}

总结

在使用 Rust 进行系统级开发时，需要注意不同版本库之间的 API 差异，特别是像 sysinfo 这种频繁更新的库。整体而言，只要熟悉基本编程思维，跨语言迁移和调试都会相对顺畅。