如何在多线程场景下从RwLock嵌套结构中安全返回指定元素引用？

首先明确：你无法直接返回&Human，因为该引用的有效性依赖于两层RwLockReadGuard（外层Vec的锁和内层Human的锁），一旦函数返回时外层锁被释放，Vec的引用失效，内层Human的引用会变成悬垂引用。必须返回一个能同时持有两层锁守卫的自定义类型，才能保证引用安全。

具体实现步骤 #

1. 修正State结构体定义（原代码存在语法错误，补充闭合符号）
2. 定义自定义锁守卫结构体，持有两层RwLockReadGuard
3. 为自定义守卫实现Deref trait，使其可以像&Human一样被使用
4. 实现get_some_element函数，获取两层锁并返回自定义守卫

完整代码示例 #

use tokio::sync::{RwLock, RwLockReadGuard};
use std::ops::Deref;

// 假设Human和Animal已定义
pub struct Human;
pub struct Animal;

pub struct State {
    humans: RwLock<Vec<RwLock<Human>>>,
    animals: RwLock<Vec<RwLock<Animal>>>,
}

// 自定义守卫，同时持有外层Vec的锁和内层Human的锁
pub struct HumanGuard<'a> {
    outer_guard: RwLockReadGuard<'a, Vec<RwLock<Human>>>,
    inner_guard: RwLockReadGuard<'a, Human>,
}

// 实现Deref，让HumanGuard可以直接当作&Human使用
impl<'a> Deref for HumanGuard<'a> {
    type Target = Human;

    fn deref(&self) -> &Self::Target {
        &self.inner_guard
    }
}

impl State {
    pub async fn get_some_element(&self, n: usize) -> Option<HumanGuard<'_>> {
        let outer_guard = self.humans.read().await;
        // 注意：Rust中^是异或运算符，若需平方请用n * n，保持原意则用n ^ 2
        let index = n * n;
        // 获取指定位置的Human的RwLock
        let human_rwlock = outer_guard.get(index)?;
        // 获取内层Human的读锁
        let inner_guard = human_rwlock.read().await;
        // 返回持有两层锁的自定义守卫
        Some(HumanGuard {
            outer_guard,
            inner_guard,
        })
    }
}

关键说明 #

• 自定义HumanGuard会在自身被销毁时自动释放两层锁，无需手动处理
• 通过Deref trait，使用时可以直接对HumanGuard解引用操作，例如：

  let state = State { /* 初始化 */ };
  if let Some(guard) = state.get_some_element(2).await {
      // 直接使用*guard访问Human的字段或方法
      println!("{:?}", *guard);
  }
  

• 之前尝试的RwLockReadGuard::map无法解决问题，因为它只能转换外层锁守卫的目标类型，无法同时持有内层锁守卫，导致内层引用在闭包结束后失效。

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