如何在Rust结构体中更新BTreeSet集合？

问题场景 #

用户编写的Rust代码尝试在结构体中维护一个BTreeSet，但调用插入方法时编译失败：

原始代码 #

use std::collections::BTreeSet;

pub struct Service {
    blobs: BTreeSet<u16>,
}

impl Service {
    fn new() -> Self {
        let bt: BTreeSet<u16> = BTreeSet::new();
        Self { blobs: bt }
    }

    fn insert(&self, val: u16) -> bool {
        &self.blobs.insert(val);
        true
    }
    fn print(&self) {
        println!("{}", self.blobs.len());
    }
}
fn main() {
    let s: Service = Service::new();
    s.print();
    s.insert(4);
    s.print();
}

编译错误 #

`self` is a `&` reference, so the data it refers to cannot be borrowed as mutable

用户尝试将结构体字段改为&'static mut BTreeSet<u16>，又出现新错误：

creates a temporary value which is freed while still in use

正确解决方案 #

核心是遵循Rust的所有权与借用规则：要修改结构体内部字段，方法需接收&mut self（可变引用），同时确保结构体实例是可变的。

修改后完整代码 #

use std::collections::BTreeSet;

pub struct Service {
    blobs: BTreeSet<u16>,
}

impl Service {
    fn new() -> Self {
        let bt = BTreeSet::new();
        Self { blobs: bt }
    }

    // 将&self改为&mut self，允许修改内部字段
    fn insert(&mut self, val: u16) -> bool {
        self.blobs.insert(val) // 直接返回insert的结果，无需额外返回true
    }

    fn print(&self) {
        println!("{}", self.blobs.len());
    }
}

fn main() {
    // 将s声明为mut，允许调用可变方法insert
    let mut s = Service::new();
    s.print();
    s.insert(4);
    s.print();
}

错误原因说明 #

1. 原始代码的问题：

   • insert方法接收的是&self（不可变引用），但BTreeSet::insert需要对集合进行可变借用，Rust不允许通过不可变引用修改内部数据，因此触发编译错误。

2. 使用&'static mut BTreeSet<u16>的错误：

   • &mut BTreeSet::new()创建的是临时值的可变引用，临时值在new方法执行完毕后就会被销毁，而'static生命周期要求引用必须存活至程序结束，二者矛盾，导致编译器报错。
   • 结构体直接持有BTreeSet的所有权才是合理设计，通过可变方法和可变实例修改数据，完全符合Rust的安全规则。

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