寻求Rust中双向键值映射（DoubleMap）的标准实现方案

嘿，我之前在Rust里找双向映射的时候也遇到过这个问题，给你分享两个实用的方案，一个是用现成的第三方库，另一个是自己手搓简易版，看你需求选～

1. 用成熟的第三方crate：bimap #

Rust标准库本身没有内置双向键值映射的数据结构，但社区有个很流行的bimap crate专门干这个事，它能保证双向的键唯一性，API也完全贴合你的需求。

首先在Cargo.toml里添加依赖：

[dependencies]
bimap = "0.6" # 可以根据最新版本调整

然后就能像你想要的那样使用了：

use bimap::BiMap;

fn main() {
    let mut map = BiMap::new();
    map.insert(1, "one");
    map.insert(2, "two");

    // 通过数字id找对应的字符串
    assert_eq!(map.get_by_left(&2), Some(&"two"));
    // 通过字符串找对应的数字id
    assert_eq!(map.get_by_right(&"one"), Some(&1));
}

这个crate还提供了很多实用方法，比如remove_by_left、remove_by_right用来删除元素，iter_left、iter_right用来迭代某一侧的所有键值对，甚至支持把双向映射转换成普通的HashMap，日常使用完全够用。

2. 自己实现简易版DoubleMap #

如果你的项目不想引入第三方依赖，也可以用两个HashMap封装成自己的DoubleMap，逻辑很直观：一个存A到B的映射，另一个存B到A的映射，对外暴露统一的操作接口。

示例代码如下：

use std::collections::HashMap;
use std::hash::Hash;

#[derive(Debug, Default)]
struct DoubleMap<A, B> {
    a_to_b: HashMap<A, B>,
    b_to_a: HashMap<B, A>,
}

impl<A: Eq + Hash, B: Eq + Hash> DoubleMap<A, B> {
    // 创建空的DoubleMap
    fn new() -> Self {
        DoubleMap {
            a_to_b: HashMap::new(),
            b_to_a: HashMap::new(),
        }
    }

    // 插入键值对，返回之前存在的旧值（如果有的话）
    fn insert(&mut self, a: A, b: B) -> Option<(B, A)> {
        let old_b = self.a_to_b.insert(a.clone(), b.clone());
        let old_a = self.b_to_a.insert(b, a);
        match (old_b, old_a) {
            (Some(b_val), Some(a_val)) => Some((b_val, a_val)),
            _ => None,
        }
    }

    // 通过A类型的键获取B类型的值
    fn get_a(&self, a: &A) -> Option<&B> {
        self.a_to_b.get(a)
    }

    // 通过B类型的键获取A类型的值
    fn get_b(&self, b: &B) -> Option<&A> {
        self.b_to_a.get(b)
    }
}

// 测试一下功能是否正常
fn main() {
    let mut map = DoubleMap::new();
    map.insert(1, "one");
    map.insert(2, "two");

    assert_eq!(map.get_a(&2), Some(&"two"));
    assert_eq!(map.get_b(&"one"), Some(&1));
}

需要注意的是，自己实现的时候要保证两个HashMap的一致性，比如插入、删除操作要同时更新两个映射，不然会出现数据不匹配的情况。另外，泛型参数A和B都需要实现Eq和Hash trait，这是HashMap的要求。

总结 #

如果项目允许引入第三方依赖，优先用bimap这种成熟的库，它处理了很多你可能没考虑到的边界情况；如果不想加依赖，自己封装两个HashMap的简易版也完全能满足基础需求。

内容来源于stack exchange