Serde处理不可预测JSON后，如何迭代HashMap中的Vec类型值？

我之前在处理JSON反序列化时，遇到字段可能是HashMap<String, String>或HashMap<String, Vec<String>>的情况，于是用了如下代码定义枚举来兼容两种类型：

#[derive(Debug, Serialize, Deserialize)]
#[serde(untagged)]
enum StringOrStringVec {
    String(String),
    Vec(Vec<String>)
}

反序列化代码如下：

let deserialized: HashMap<String, StringOrStringVec> = serde_json::from_str(&msg.as_str().unwrap()).unwrap();

现在我想迭代其中值为Vec<String>类型的条目，但尝试调用to_iter时发现没有现成可用的trait，请问我真的需要自行实现这些trait吗？

解决方案 #

不需要自行实现额外trait，直接通过模式匹配就能处理枚举变体的迭代需求，以下是几种常用的处理方式：

1. 直接遍历并匹配变体 #

遍历HashMap的同时，对每个值的枚举类型做匹配，分别处理单字符串和字符串数组的情况：

for (key, value) in deserialized.iter() {
    match value {
        StringOrStringVec::String(s) => {
            // 处理单个字符串，也可以转成单元素Vec统一逻辑
            println!("{}: 单个字符串 - {}", key, s);
        }
        StringOrStringVec::Vec(v) => {
            // 直接迭代字符串数组
            println!("{}: 字符串数组元素:", key);
            for item in v {
                println!("  - {}", item);
            }
        }
    }
}

2. 给枚举添加统一迭代方法 #

如果想让单字符串和数组的迭代逻辑统一，可以给枚举实现一个iter方法，返回统一的迭代器：

impl StringOrStringVec {
    fn iter(&self) -> impl Iterator<Item = &str> {
        match self {
            // 单个字符串返回只包含自身的迭代器
            StringOrStringVec::String(s) => std::iter::once(s.as_str()),
            // 数组直接返回其迭代器并转成&str
            StringOrStringVec::Vec(v) => v.iter().map(|s| s.as_str()),
        }
    }
}

之后就能统一调用iter()遍历所有字符串内容：

for (key, value) in deserialized.iter() {
    println!("{} 的所有内容:", key);
    for content in value.iter() {
        println!("  - {}", content);
    }
}

3. 筛选出所有数组类型的条目 #

如果只需要处理值为Vec<String>的条目，可以用filter_map过滤并提取：

// 收集所有值为数组的条目
let array_entries: Vec<(&String, &Vec<String>)> = deserialized
    .iter()
    .filter_map(|(key, value)| {
        match value {
            StringOrStringVec::Vec(vec) => Some((key, vec)),
            _ => None,
        }
    })
    .collect();

// 遍历筛选后的数组条目
for (key, vec) in array_entries {
    println!("Key: {}, 数组元素: {:?}", key, vec);
}

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