我试图了解如何在Rust中实现trait对象。如果以下理解是正确的,请告诉我。
我有一个函数,它采用任何实现Write
特性的类型:
fn some_func(write_to: &mut Write) {}
在我们有一个实现此特征的类型并调用上述函数的任何地方,编译器可能会生成一个“特征对象”,可能是通过添加对TraitObject::new(data, vtable)
的调用。
如果我们有类似的东西:
let input = get_user_input(); // say we are expecting the input to be 1 or 2
let mut file = File::new("blah.txt").unwrap();
let mut vec: Vec<u8> = vec![1, 2, 3];
match input {
1 => some_func(&mut file),
2 => some_func(&mut vec),
}
可能会变成:
match input {
1 => {
let file_write_trait_object: &mut Write =
TraitObject::new(&file, &vtable_for_file_write_trait);
some_func(file_write_trait_object);
}
2 => {
let vec_write_trait_object: &mut Write =
TraitObject::new(&vec, &vtable_for_vec_write_trait);
some_func(vec_write_trait_object);
}
}
在some_func
内部,编译器将只访问基于传递的TraitObject
中的vtable使用的方法。
特质对象是胖指针,所以fn some_func(write_to: &mut Write)
编译成像fn some_func(_: *mut OpaqueStruct, _: *const WriteVtable)
的东西。