Rust中前置条件/参数校验的简易错误处理方案问询

核心问题拆解 #

expect触发的panic会输出开发者视角的调试信息（线程、行号等），不适合展示给普通用户；而直接用unwrap_or_else+std::process::exit会跳过析构函数执行，可能引发资源泄漏。我们需要一种既简洁、又能输出用户友好信息、还能保证析构正常执行的方案。

推荐方案：利用main返回Result+?操作符 #

Rust允许main函数返回Result类型，结合?操作符可以实现类似expect的简洁写法，同时确保所有局部变量的析构函数正常执行，错误信息会自动使用Display特性输出友好内容。

优化后的完整代码 #

use clap::Parser;
use std::collections::HashMap;
use std::error::Error;
use std::process::Command;
use url::Url;

#[derive(Parser, Debug)]
struct Args {
    /// 包含日志信息的URL
    #[arg(short, long)]
    url: String,
}

fn main() {
    if let Err(e) = run() {
        eprintln!("错误：{}", e);
        std::process::exit(1);
    }
}

fn run() -> Result<(), Box<dyn Error>> {
    let args = Args::parse();

    // 解析URL，错误自动传播，用Display输出友好信息
    let url = Url::parse(&args.url)?;

    let query: HashMap<String, String> = url.query_pairs().into_owned().collect();

    // 检查必填参数，自定义用户友好错误
    let server_url = query.get("server_url")
        .ok_or_else(|| "URL中缺少必填参数：server_url")?;

    let revision = query.get("revision")
        .ok_or_else(|| "URL中缺少必填参数：revision")?;

    // 执行命令，错误自动传播
    Command::new("TortoiseProc.exe")
        .arg("/command:log")
        .arg(format!("/startrev:{revision}"))
        .arg(format!("/path:{server_url}"))
        .output()?;

    Ok(())
}

方案优势 #

1. 简洁性：用?替代expect，代码简洁度和expect相当
2. 用户友好：错误信息使用Display特性输出，无调试冗余内容
3. 安全可靠：run函数返回错误时，所有局部变量会正常执行析构函数，避免资源泄漏
4. 统一处理：在main中集中捕获错误，统一输出格式

进阶：自定义错误类型（可选） #

如果需要更精细化的错误分类和统一格式，可以自定义错误类型：

#[derive(Debug)]
enum AppError {
    UrlParse(url::ParseError),
    MissingParam(String),
    CommandFailed(std::io::Error),
}

impl std::fmt::Display for AppError {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match self {
            AppError::UrlParse(e) => write!(f, "URL解析失败：{}", e),
            AppError::MissingParam(param) => write!(f, "缺少必填参数：{}", param),
            AppError::CommandFailed(e) => write!(f, "命令执行失败：{}", e),
        }
    }
}

impl std::error::Error for AppError {}

// 实现From转换，让?自动转换错误类型
impl From<url::ParseError> for AppError {
    fn from(e: url::ParseError) -> Self {
        AppError::UrlParse(e)
    }
}

impl From<std::io::Error> for AppError {
    fn from(e: std::io::Error) -> Self {
        AppError::CommandFailed(e)
    }
}

随后将run函数的返回类型改为Result<(), AppError>即可，错误信息会更规范统一。

内容的提问来源于stack exchange，提问作者idbrii