如何优化Rust中解析单值/范围输入生成数字向量的代码？

先说说原代码存在的几个问题：

• 用unwrap()会在解析失败时直接panic，违背了返回Result的设计目的，应该把错误传递出去让调用者处理
• 遇到多个-的情况直接返回空Vec，没有给出明确的错误提示
• if-else分支可以用更符合Rust风格的写法简化

下面是优化后的实现，同时补上了完整的错误处理和输入校验：

首先定义自定义错误类型，让错误信息更清晰：

#[derive(Debug, Clone, PartialEq, Eq)]
enum ParseError {
    InvalidFormat,
    ParseInt(std::num::ParseIntError),
    RangeInvalid,
}

impl From<std::num::ParseIntError> for ParseError {
    fn from(err: std::num::ParseIntError) -> Self {
        ParseError::ParseInt(err)
    }
}

impl std::fmt::Display for ParseError {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match self {
            ParseError::InvalidFormat => write!(f, "输入格式无效，应为单个数字或start-end格式的范围"),
            ParseError::ParseInt(e) => write!(f, "数字解析失败：{}", e),
            ParseError::RangeInvalid => write!(f, "起始数字不能大于结束数字"),
        }
    }
}

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

然后是核心的解析函数：

fn parse_ports(ports: &str) -> Result<Vec<u32>, ParseError> {
    // 先去除每个部分的空格，兼容用户输入带空格的情况（比如"8080 - 9090"）
    let parts: Vec<&str> = ports.split('-').map(str::trim).collect();

    match parts.as_slice() {
        // 单个数值的情况
        [single] => {
            let num = single.parse()?;
            Ok(vec![num])
        }
        // 范围格式的情况
        [start_str, end_str] => {
            let start = start_str.parse()?;
            let end = end_str.parse()?;
            
            // 校验范围有效性
            if start > end {
                return Err(ParseError::RangeInvalid);
            }
            
            // 用闭区间..=替代start..end+1，避免u32溢出风险
            Ok((start..=end).collect())
        }
        // 其他格式（比如多个'-'）直接返回错误
        _ => Err(ParseError::InvalidFormat),
    }
}

关键改进点： #

• 错误处理：用?替代unwrap()，把解析错误向上传递，同时自定义错误类型提供清晰的错误信息
• 输入兼容性：增加str::trim()处理输入中的空格，提升用户体验
• 避免溢出：用..=闭区间生成范围，避免end+1导致的u32溢出问题（比如end是u32::MAX时）
• 格式校验：严格限制输入格式，对不符合要求的情况返回明确错误，而非空Vec
• 代码风格：用match匹配切片替代if-else分支，更符合Rust的惯用写法，可读性更强

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