Rust中向HashSet插入对象后如何获取其引用？

你遇到的核心问题是：将id插入HashSet后，id的所有权被转移到集合中，原函数栈内的id变量失效，无法返回它的引用。以下是几种常规解决方案：

方案一：利用HashSet的get方法结合临时字符串引用 #

这种方法无需额外克隆，通过提前获取字符串的引用，在插入后用该引用查找集合中的元素：

use std::collections::HashSet;

pub struct UniqueUuid {
    items: HashSet<String>,
}

impl UniqueUuid {
    // 生成、保存并返回一个UUID
    pub fn next(&mut self) -> &String {
        loop {
            let id = random_string(10);
            let id_str = id.as_str();
            // 检查是否已存在
            if !self.items.contains(id_str) {
                // 转移所有权到集合
                self.items.insert(id);
                // 通过字符串引用获取集合中的元素引用，unwrap安全因为刚插入
                return self.items.get(id_str).unwrap();
            }
        }
    }
}

// 模拟随机字符串生成函数
fn random_string(_len: usize) -> String {
    "abc123".to_string()
}

解释：提前通过id.as_str()获取字符串的引用id_str，该引用指向的内容和后续插入集合的字符串完全一致。插入成功后，用id_str调用get就能拿到集合中拥有所有权的字符串的引用，避免了原变量失效的问题。

方案二：替换为HashMap使用Entry API #

HashSet的API相对有限，HashMap的Entry机制可以更优雅地处理“插入并返回引用”的场景，我们可以用HashMap<String, ()>模拟HashSet的行为：

use std::collections::HashMap;

pub struct UniqueUuid {
    items: HashMap<String, ()>,
}

impl UniqueUuid {
    pub fn next(&mut self) -> &String {
        loop {
            let id = random_string(10);
            match self.items.entry(id) {
                // 元素不存在时插入，直接返回键的引用
                std::collections::hash_map::Entry::Vacant(entry) => {
                    return entry.insert(());
                }
                // 元素已存在则继续循环
                std::collections::hash_map::Entry::Occupied(_) => continue,
            }
        }
    }
}

// 模拟随机字符串生成函数
fn random_string(_len: usize) -> String {
    "abc123".to_string()
}

解释：Entry::Vacant的insert方法会将键的所有权转移到HashMap中，同时返回该键的不可变引用，完美匹配需求，且无需额外的查找操作，效率更高。

方案三：克隆字符串（简单但有性能损耗） #

如果对性能要求不高，也可以通过克隆字符串来绕过所有权问题：

use std::collections::HashSet;

pub struct UniqueUuid {
    items: HashSet<String>,
}

impl UniqueUuid {
    pub fn next(&mut self) -> &String {
        loop {
            let id = random_string(10);
            if !self.items.contains(&id) {
                // 克隆一份插入集合，原id保留用于获取引用
                self.items.insert(id.clone());
                return self.items.get(&id).unwrap();
            }
        }
    }
}

// 模拟随机字符串生成函数
fn random_string(_len: usize) -> String {
    "abc123".to_string()
}

解释：通过id.clone()将副本插入集合，原id变量仍有效，后续用它查找集合中的元素并返回引用。这种方法简单直观，但会产生额外的字符串克隆开销，适合小字符串或低频率调用场景。

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