一个参数,它接受一系列 trait 对象并在 Rust 中多次迭代它

问题描述

我有一个函数,它接受一系列特征对象并对其进行多次迭代,例如,

trait Animal {
    fn make_noise(&self);
}

fn make_lots_of_noises(animals: &[&dyn Animal]) {
    animals.iter().for_each(|animal| animal.make_noise());
    animals.iter().for_each(|animal| animal.make_noise());
    animals.iter().for_each(|animal| animal.make_noise());
}

但我希望该函数能够对借用和拥有的数据结构进行操作。以下是我尝试过的选项:

&[&dyn Animal]代码片段所示。问题:如果我拥有 trait 对象,例如 animals: &[Box<dyn Animal>],那么我必须调用 make_lots_of_noises(animals.map(Box::as_ref).collect::<Vec<_>>().as_slice(),这涉及在堆上进行不必要的分配。

&[&R] where R: AsRef<dyn Animal> + ?Sized。问题:T 没有实现 AsRef<T>!它现在适用于各种智能指针,但不适用于 &[&dyn Animal]

解决方法

使用 Borrow 特征,它确实具有您想要的 impl Borrow<T> for T

use std::borrow::Borrow;

trait Animal {
    fn make_noise(&self);
}

struct Mouse;
impl Animal for Mouse {
    fn make_noise(&self) { /* squeak */ }
}

fn make_lots_of_noises<A: Borrow<dyn Animal>>(animals: &[A]) {
    animals.iter().for_each(|animal| animal.borrow().make_noise());
    animals.iter().for_each(|animal| animal.borrow().make_noise());
    animals.iter().for_each(|animal| animal.borrow().make_noise());
}

fn main() {
    make_lots_of_noises(&[&Mouse as &dyn Animal]);
    make_lots_of_noises(&[Box::new(Mouse) as Box<dyn Animal>]);
}