面试题：Rust线程停放机制中的内存管理

所有权转移与线程停放
- 原理：Rust的所有权系统确保每个值都有一个唯一的所有者。当线程停放时，如果线程持有某些资源的所有权，在停放线程前，可以将这些资源的所有权转移到其他地方。例如，如果线程thread_a持有一个Vec<i32>的所有权，在停放thread_a前，可以将Vec<i32>的所有权转移给另一个线程thread_b或者主线程。
- 示例代码：

use std::thread;

fn main() {
    let mut data = Vec::from([1, 2, 3]);
    let handle = thread::spawn(move || {
        // 线程获得data的所有权
        data.push(4);
        data
    });
    // 等待线程结束，获取所有权转移后的结果
    let result = handle.join().unwrap();
    println!("{:?}", result);
}

借用与线程停放
- 共享不可变借用：如果线程需要访问某些数据但不需要修改，可使用不可变借用。Rust的借用规则保证在同一时间可以有多个不可变借用。在停放线程时，只要确保借用的数据在借用期间不会被释放，就不会出现悬空指针问题。例如，多个线程可以同时不可变借用一个只读的配置文件数据。
- 示例代码：

use std::thread;

fn main() {
    let data = Vec::from([1, 2, 3]);
    let handle1 = thread::spawn(|| {
        // 不可变借用data
        println!("{:?}", &data);
    });
    let handle2 = thread::spawn(|| {
        // 不可变借用data
        println!("{:?}", &data);
    });
    handle1.join().unwrap();
    handle2.join().unwrap();
}

可变借用：可变借用在同一时间只能有一个。如果线程需要修改数据，在停放线程前，确保可变借用的生命周期合理结束。比如，线程获取一个可变借用进行数据修改，修改完成后，在停放线程前，释放这个可变借用，避免在停放后借用失效导致悬空指针等问题。
示例代码：

use std::thread;

fn main() {
    let mut data = Vec::from([1, 2, 3]);
    let handle = thread::spawn(move || {
        // 可变借用data
        let mut data_ref = &mut data;
        data_ref.push(4);
    });
    handle.join().unwrap();
    println!("{:?}", data);
}

使用智能指针
- Rc<T>和Weak<T>：Rc<T>用于在多个线程间共享不可变数据，通过引用计数来管理内存。Weak<T>是Rc<T>的弱引用，可用于避免循环引用导致的内存泄漏。当线程停放时，Rc<T>的引用计数会相应调整，若引用计数降为0，数据会被释放。例如，多个线程共享一个只读的配置信息，可以用Rc<Config>，如果其中一个线程可能停放且需要避免循环引用，可以结合Weak<Config>。
- Arc<T>：Arc<T>用于在多线程环境下原子引用计数。和Rc<T>类似，但Arc<T>可安全地在多个线程间共享。当停放持有Arc<T>的线程时，Arc<T>的引用计数会调整，若计数为0则释放内存。比如，多个线程需要共享一个可读写的缓存数据，就可以使用Arc<Mutex<CacheData>>，其中Mutex用于提供线程安全的可变访问，Arc用于管理内存。
线程局部存储（TLS）
- 原理：线程局部存储允许每个线程有自己独立的数据副本。当线程停放时，这些线程局部的数据不会影响其他线程，并且在停放后再次启动时，数据依然存在。这有助于避免内存管理问题，因为每个线程的数据生命周期与线程本身紧密相关。
- 示例代码：

use std::thread;
use std::thread::LocalKey;

static TLS_KEY: LocalKey<String> = LocalKey::new();

fn main() {
    let handle = thread::spawn(|| {
        TLS_KEY.with(|s| {
            s.push_str("Thread - specific data");
            println!("{}", s);
        });
    });
    handle.join().unwrap();
}

通过合理应用Rust的所有权和借用规则，结合智能指针及线程局部存储等机制，可以在Rust线程停放过程中有效地进行内存管理，避免内存泄漏和悬空指针等问题。

星途面试题库

面试题：Rust线程停放机制中的内存管理

知识考点

面试题答案