如何在Rust中实现特征对象?

时间:2017-12-01 20:27:24

标签: rust traits trait-objects

我正在尝试了解如何在Rust中实现特征对象。如果以下理解是正确的,请告诉我。

我有一个函数,它接受任何实现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使用的方法。

1 个答案:

答案 0 :(得分:0)

Trait对象是胖指针,因此fn some_func(write_to: &mut Write)会编译为fn some_func(_: *mut OpaqueStruct, _: *const WriteVtable)