Rust中如何结合iter()、iter::once()解决迭代器类型不兼容问题

问题场景 #

下面的代码可以正常运行：

use std::iter;

let a = [1, 2, 3].iter();
let b = iter::once(&456);

for i in a.chain(b) {
    println!("{}", i);
}

输出：

1
2
3
456

但当尝试根据条件切换b的迭代器类型时，会触发类型不兼容错误：

use std::iter;

let a = [1, 2, 3].iter();
let b = if some_condition {
    [4, 5, 6].iter()
} else {
    iter::once(&456)
};

for i in a.chain(b) {
    println!("{}", i);
}

错误信息：

error[E0308]: `if` and `else` have incompatible types
   |
   |       let b = if some_condition {
   |  _____________- 
   | |         [4, 5, 6].iter()
   | |         ---------------- expected because of this
   | |     } else {
   | |         iter::once(&456)
   | |         ^^^^^^^^^^^^^^^^ expected `Iter<'_, {integer}>`, found `Once<&{integer}>`
   | |     };
   | |_____- `if` and `else` have incompatible types
   |
   = note: expected struct `std::slice::Iter<'_, {integer}>`
              found struct `std::iter::Once<&{integer}>`

解决方案 #

方法1：使用 trait 对象装箱（Box） #

通过将不同迭代器装箱为 trait 对象，统一类型：

use std::iter;

fn main() {
    let some_condition = true; // 可修改此值测试不同分支
    let a = [1, 2, 3].iter();
    let b: Box<dyn Iterator<Item = &i32>> = if some_condition {
        Box::new([4, 5, 6].iter())
    } else {
        Box::new(iter::once(&456))
    };

    for i in a.chain(b) {
        println!("{}", i);
    }
}

说明：这种方法通过动态分发统一类型，实现简单，但会带来轻微的运行时开销。

方法2：使用 Either 枚举（静态分发，无运行时开销） #

借助either crate提供的Either枚举，静态处理两种类型分支：

1. 先在Cargo.toml中添加依赖：

[dependencies]
either = "1.9"

2. 编写代码：

use either::Either;
use std::iter;

fn main() {
    let some_condition = true;
    let a = [1, 2, 3].iter();
    let b = if some_condition {
        Either::Left([4, 5, 6].iter())
    } else {
        Either::Right(iter::once(&456))
    };

    for i in a.chain(b) {
        println!("{}", i);
    }
}

说明：Either是专门处理二选一类型的枚举，实现了Iterator trait，属于静态分发，性能更优。

方法3：重构代码，拆分遍历逻辑 #

如果业务逻辑允许，直接拆分两个分支的遍历过程，完全规避类型兼容问题：

use std::iter;

fn main() {
    let some_condition = true;
    let a = [1, 2, 3].iter();
    
    // 先遍历a
    for i in a {
        println!("{}", i);
    }

    // 根据条件遍历不同内容
    if some_condition {
        for i in [4,5,6].iter() {
            println!("{}", i);
        }
    } else {
        println!("{}", 456);
    }
}

说明：这种方法最直接，没有任何额外开销，适合逻辑简单的场景。

内容的提问来源于stack exchange，提问作者Ron