Rust中非原始类型向下转型：如何获取可变引用修改字段

要实现对Trait对象的可变引用向下转型并修改字段，你需要在Trait中添加一个返回可变Any引用的方法，再配合downcast_mut来完成。以下是修改后的完整代码：

use std::any::Any;

trait MyTrait {
    fn do_something(&self);
    // 新增可变版本的as_any方法
    fn as_any(&self) -> &dyn Any;
    fn as_any_mut(&mut self) -> &mut dyn Any;
}

struct TypeA{field_a: u16}
impl MyTrait for TypeA {
    fn do_something(&self) {
        println!("TypeA is doing something.");
    }
    fn as_any(&self) -> &dyn Any {
        self
    }
    // 实现可变版本的as_any_mut
    fn as_any_mut(&mut self) -> &mut dyn Any {
        self
    }
}

struct TypeB{field_b: u16}
impl MyTrait for TypeB {
    fn do_something(&self) {
        println!("TypeB is doing something.");
    }
    fn as_any(&self) -> &dyn Any {
        self
    }
    // 实现可变版本的as_any_mut
    fn as_any_mut(&mut self) -> &mut dyn Any {
        self
    }
}

fn main() {
    // 将向量声明为可变，允许内部元素被修改
    let mut vec: Vec<Box<dyn MyTrait>> = vec![
        Box::new(TypeA{field_a:10}),
        Box::new(TypeB{field_b:20}),
    ];

    // 遍历向量的可变引用
    for item in &mut vec {
        // 使用downcast_mut获取可变引用并修改字段
        if let Some(type_a) = item.as_any_mut().downcast_mut::<TypeA>() {
            type_a.do_something();
            println!("原field_a值: {}", type_a.field_a);
            type_a.field_a = 100;
            println!("修改后field_a值: {}", type_a.field_a);
        } else if let Some(type_b) = item.as_any_mut().downcast_mut::<TypeB>() {
            type_b.do_something();
            println!("原field_b值: {}", type_b.field_b);
            type_b.field_b = 200;
            println!("修改后field_b值: {}", type_b.field_b);
        } else {
            println!("Unknown type");
        }
    }
}

关键修改点说明： #

• 新增as_any_mut方法：在MyTrait中添加返回&mut dyn Any的方法，让每个实现该Trait的结构体都能提供自身的可变引用。
• 实现as_any_mut：在TypeA和TypeB的Trait实现中，返回self的可变引用即可。
• 可变遍历与downcast_mut：将向量声明为mut，遍历其可变引用&mut vec，使用downcast_mut方法将&mut dyn Any向下转型为具体结构体的可变引用，之后就能直接修改字段值。

运行这段代码，你会看到控制台输出每个结构体字段修改前后的值，验证修改操作成功。

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