如何为枚举中的单个或多个值返回迭代器

Question

考虑以下代码：

struct MediaKind;
struct OtherFields;

enum Media {
    Single(MediaKind),
    Group(Vec<MediaKind>, OtherFields),
}

impl Media {
    fn iter(&self) -> impl Iterator<Item = &MediaKind> {
        match self {
            Self::Single(media) => [media].iter(),
            Self::Group(media_vec, _) => media_vec.iter(),
        }
    }
}

编译错误：

error[E0308]: `match` arms have incompatible types
  --> src/lib.rs:13:42
   |
11 | /         match self {
12 | |             Self::Single(media) => [media].iter(),
   | |                                    -------------- this is found to be of type `std::slice::Iter<'_, &MediaKind>`
13 | |             Self::Group(media_vec, _) => media_vec.iter(),
   | |                                          ^^^^^^^^^^^^^^^^ expected `&MediaKind`, found struct `MediaKind`
14 | |         }
   | |_________- `match` arms have incompatible types
   |
   = note: expected struct `std::slice::Iter<'_, &MediaKind>`
              found struct `std::slice::Iter<'_, MediaKind>`

因为Group arm 可能包含其他字段，所以用Vec替换整个enum不是首选解决方案。

我已尝试either以下方式解决问题：

fn iter(&self) -> impl Iterator<Item = &MediaKind> {
    use either::*;

    match self {
        Self::Single(media) => Left([media].into_iter()),
        Self::Group(media_vec, _) => Right(media_vec.iter()),
    }
}

但我想知道是否有可能在不依赖外部板条箱的情况下修复它。

> 游乐场链接

Answer 1

在这种情况下，因为一种情况是切片迭代器而另一种情况是一个元素，您可以使用一个技巧： std::slice::from_ref() ：

fn iter(&self) -> impl Iterator<Item = &MediaKind> {
    match self {
        Self::Single(media) => std::slice::from_ref(media).iter(),
        Self::Group(media_vec, _) => media_vec.iter(),
    }
}

游乐场。