如何将Arc<tokio::sync::RwLock<MyTrait>>向下转换为Arc<tokio::sync::RwLock<MyStruct>>

问题背景 #

现有代码可实现Arc<MyTrait>向下转换为Arc<MyStruct>：

use std::any::Any;
use std::sync::Arc;

trait MyTrait {
    fn as_any(self: Arc<Self>) -> Arc<dyn Any + Send + Sync>;
}

struct MyStruct {}

impl MyTrait for MyStruct {
    fn as_any(self: Arc<Self>) -> Arc<dyn Any + Send + Sync> {
        self
    }
}

现在需要将Arc<tokio::sync::RwLock<MyTrait>>向下转换为Arc<tokio::sync::RwLock<MyStruct>>，且保证原对象存活（其他模块仍需访问）。尝试在trait中添加如下方法时触发编译器报错：

fn as_any_rw(self: Arc<tokio::sync::RwLock<Self>>) -> Arc<tokio::sync::RwLock<dyn Any + Send + Sync>>;

报错信息：

invalid `self` parameter type: std::sync::Arc<tokio::sync::RwLock<Self>>
type of `self` must be `Self` or a type that dereferences to it
consider changing to `self`, `&self`, `&mut self`, `self: Box<Self>`, `self: Rc<Self>`, `self: Arc<Self>`, or `self: Pin<P>` (where P is one of the previous types except `Self`)rustcE0307

错误原因 #

Rust对trait方法的self参数类型有严格限制：仅允许Self、&Self、&mut Self，或包裹在Box/Rc/Arc/Pin中的Self。Arc<RwLock<Self>>不符合规则，因为RwLock<Self>无法直接解引用为Self，编译器不允许将其作为trait方法的self参数。

解决方案 #

保留原有结构的前提下，通过类型检查+unsafe类型转换实现需求：

1. 扩展原Trait #

添加as_any_ref方法用于类型校验：

use std::any::Any;
use std::sync::Arc;
use tokio::sync::RwLock;

trait MyTrait {
    // 保留原有方法
    fn as_any(self: Arc<Self>) -> Arc<dyn Any + Send + Sync>;
    // 新增方法：获取&dyn Any用于类型判断
    fn as_any_ref(&self) -> &(dyn Any + Send + Sync);
}

struct MyStruct {}

impl MyTrait for MyStruct {
    fn as_any(self: Arc<Self>) -> Arc<dyn Any + Send + Sync> {
        self
    }

    fn as_any_ref(&self) -> &(dyn Any + Send + Sync) {
        self
    }
}

2. 实现转换函数 #

通过Arc的原始指针操作完成类型转换，提前用as_any_ref确保类型匹配：

async fn downcast_rw_lock(
    arc_rw: Arc<RwLock<dyn MyTrait>>,
) -> Option<Arc<RwLock<MyStruct>>> {
    // 先获取读锁，验证内部类型
    let lock = arc_rw.read().await;
    if !lock.as_any_ref().is::<MyStruct>() {
        return None;
    }

    // 类型确认后，unsafe转换Arc的类型
    unsafe {
        // 将Arc转为原始指针，再重新构造为目标类型的Arc
        let raw_ptr = Arc::into_raw(arc_rw) as *mut RwLock<MyStruct>;
        Some(Arc::from_raw(raw_ptr))
    }
}

方案说明 #

• 类型校验：通过as_any_ref().is::<MyStruct>()确保内部对象类型正确，避免unsafe操作的风险。
• 引用计数安全：Arc::into_raw仅提取原始指针，不会减少引用计数；Arc::from_raw重新构造Arc，引用计数保持不变，原对象仍可被其他模块访问。
• 无所有权消耗：转换过程中不会消耗原Arc<RwLock<MyTrait>>，原对象的生命周期不受影响。

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