如何处理TOML中混合类型（字符串/整数）值的反序列化？

背景 #

需要解析的TOML文件中，count字段的值支持两种形式：整数或字符串，有效值包括：

count = 3          # 合法整数
count = -1         # 合法整数（代表"all"）
count = "all"      # 字符串，需解析为-1
count = "42"       # 字符串形式的整数

要求反序列化时，将"all"解析为-1，同时兼容整数和字符串形式的合法数值（范围-1..i32::MAX）。

问题描述 #

原自定义反序列化器仅能处理字符串形式的值，遇到纯整数时直接报错。原代码及报错如下：

use serde::{de::Error, Deserialize, Deserializer};

#[derive(Debug, Deserialize)]
struct ConfigFileOpts {
    #[serde(deserialize_with = "ds_i32_or_string")]
    count: i32,
}

fn ds_i32_or_string<'de, D>(deserializer: D) -> Result<i32, D::Error>
where
    D: Deserializer<'de>,
{
    let res = String::deserialize(deserializer);
    match res {
        Ok(s) => match s.parse::<i32>() {
            Ok(n) if n >= -1 => Ok(n),
            _ => match s.trim().to_lowercase().as_str() {
                "all" => Ok(-1),
                _ => Err(D::Error::custom("Value must be >= -1 or 'all'.")),
            },
        },
        Err(e) => Err(e),
    }
}

fn main() {
    let cfg_opts: ConfigFileOpts = toml::from_str("count = 3").unwrap();
    println!("deserialized count: {:?}", cfg_opts.count);
}

报错信息：

TomlError {
  message: "invalid type: integer `3`, expected a string",
  original: Some("count = 3"),
  keys: ["count"],
  span: Some(8..9)
}

原因是反序列化器一开始强制尝试将值转为String，纯整数类型无法通过这一步，直接抛出错误。

解决方案 #

修改自定义反序列化函数，先尝试解析为整数类型，失败后再尝试解析为字符串类型处理。这样既兼容纯整数，也能处理字符串形式的数字和"all"。

修改后的完整代码：

use serde::{de::Error, Deserialize, Deserializer};

#[derive(Debug, Deserialize)]
struct ConfigFileOpts {
    #[serde(deserialize_with = "ds_i32_or_string")]
    count: i32,
}

fn ds_i32_or_string<'de, D>(deserializer: D) -> Result<i32, D::Error>
where
    D: Deserializer<'de>,
{
    // 第一步：尝试直接解析为i32
    match i32::deserialize(deserializer) {
        Ok(n) if n >= -1 => return Ok(n),
        Ok(_) => return Err(D::Error::custom("Integer value must be >= -1")),
        Err(_) => {} // 整数解析失败，进入字符串处理逻辑
    }

    // 第二步：整数解析失败，尝试解析为字符串处理
    let s = String::deserialize(deserializer)?;
    match s.trim().to_lowercase().as_str() {
        "all" => Ok(-1),
        _ => {
            let n = s.parse::<i32>().map_err(|_| {
                D::Error::custom("Value must be a valid integer >= -1 or the string 'all'")
            })?;
            if n >= -1 {
                Ok(n)
            } else {
                Err(D::Error::custom("Integer value must be >= -1"))
            }
        }
    }
}

fn main() {
    // 测试纯整数
    let cfg1: ConfigFileOpts = toml::from_str("count = 3").unwrap();
    println!("deserialized count (int): {:?}", cfg1.count); // 输出3

    // 测试字符串"all"
    let cfg2: ConfigFileOpts = toml::from_str("count = \"all\"").unwrap();
    println!("deserialized count (all): {:?}", cfg2.count); // 输出-1

    // 测试字符串形式的整数
    let cfg3: ConfigFileOpts = toml::from_str("count = \"-1\"").unwrap();
    println!("deserialized count (str -1): {:?}", cfg3.count); // 输出-1

    // 测试字符串形式的正整数
    let cfg4: ConfigFileOpts = toml::from_str("count = \"42\"").unwrap();
    println!("deserialized count (str 42): {:?}", cfg4.count); // 输出42
}

关键修改说明 #

1. 优先解析整数：先调用i32::deserialize尝试直接解析纯整数，符合范围则直接返回，不符合则返回明确错误。
2. 字符串 fallback：整数解析失败时，再解析为字符串，处理"all"和字符串形式的数字，同时做范围校验。
3. 错误信息优化：针对不同错误场景返回更清晰的提示，方便排查问题。

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