rust - 如何让函数返回具有相同项目类型的两个迭代器之一？

Question

following不编译，因为两个迭代器有不同的签名。但是，两个迭代器返回相同的项类型并且在概念上表现相同。

fn f(flag: bool) -> impl Iterator<Item = u8> {
    if flag {
        (0..8).into_iter()
    } else {
        (0..8).into_iter().map(|x| x + 1)
    }
}

我想编写一个函数来返回从算法的不同变体生成的项目，这些项目是根据我传递给函数的值选择的。我怎样才能做到这一点？

最终，我需要在 no_std 环境中使用它。这可能吗？

Answer 1

我不知道有任何内置方法可以做到这一点，但实现一些可行的方法并不过分棘手。

您需要的是一种既可以表示 (0..8).into_iter() 结果的类型——它是 Iterator<Item=u8> 的一种，也是 (0..8).into_iter().map(|x| x+1) 的一种。如果您不想使用 Boxed 特征，典型的选择是创建一个具有两个选项的枚举。

然后，如果您在枚举上实现 Iterator<Item=u8> 就完成了。

这样的枚举可能看起来像:

pub enum EitherIter<AIterType, BIterType> {
    A(AIterType),
    B(BIterType),
}

impl<AIterType, BIterType> Iterator for EitherIter<AIterType, BIterType>
where
    AIterType: Iterator,
    BIterType: Iterator<Item = AIterType::Item>,
{
    type Item = AIterType::Item;
    fn next(&mut self) -> Option<<Self as Iterator>::Item> {
        match self {
            EitherIter::A(it) => it.next(),
            EitherIter::B(it) => it.next(),
        }
    }
}

然后使用它您将每个返回类型包装在其中一种类型中。

pub fn f(flag: bool) -> impl Iterator<Item = u8> {
    if flag {
        EitherIter::A((0..8).into_iter())
    } else {
        EitherIter::B((0..8).into_iter().map(|x| x + 1))
    }
}

你可以在 playground 中试试这个。