Rust是否提供包装迭代器？支持还原原迭代器的场景问询

标准库现有适配方案 #

如果你仅使用标准库的基础迭代器适配器（如take），在Rust 1.63及以上版本中，Take、Skip、Filter等适配器已内置into_inner方法，可直接取回被包装的原迭代器。你的场景可以简化为：

let data = vec!(1, 2, 3, 4, 5);
let mut it = data.into_iter();

// 用Take适配器处理前2个元素
let mut take_iter = it.take(2);
take_iter.for_each(|e| process_first_2_elems(e));

// 取回原迭代器继续处理剩余元素
it = take_iter.into_inner();
it.for_each(|e| process_remaining(e));

自定义包装迭代器 #

如果需要更复杂的自定义包装逻辑，你可以手动实现一个持有原迭代器的结构体，同时实现Iterator trait和原迭代器回收方法：

// 自定义包装迭代器结构体
struct WrappedIter<I> {
    inner: I,
}

impl<I> WrappedIter<I> {
    // 包装原迭代器的构造函数
    fn new(inner: I) -> Self {
        WrappedIter { inner }
    }

    // 取回原迭代器的方法
    fn into_inner(self) -> I {
        self.inner
    }
}

// 实现Iterator trait，直接代理原迭代器的核心逻辑
impl<I: Iterator> Iterator for WrappedIter<I> {
    type Item = I::Item;

    fn next(&mut self) -> Option<Self::Item> {
        self.inner.next()
    }

    // 可选：代理size_hint方法以优化性能
    fn size_hint(&self) -> (usize, Option<usize>) {
        self.inner.size_hint()
    }
}

// 业务处理函数示例
fn process_first_2_elems(e: i32) {
    println!("处理首批元素: {}", e);
}

fn process_remaining(e: i32) {
    println!("处理剩余元素: {}", e);
}

fn main() {
    let data = vec!(1, 2, 3, 4, 5);
    let mut it = data.into_iter();
    let mut wrapped = WrappedIter::new(it);

    wrapped.take(2).for_each(|e| process_first_2_elems(e));

    // 取回原迭代器
    it = wrapped.into_inner();
    it.for_each(|e| process_remaining(e));
}

关键注意事项 #

• 迭代器是惰性求值的，只有调用next或依赖它的方法（如for_each、collect）时才会消耗元素。
• 自定义包装器时，若需添加元素转换、过滤等逻辑，只需修改next方法的实现，保留into_inner即可回收原迭代器。
• 并非所有标准库适配器都提供into_inner，遇到此类情况时，自定义包装器是更灵活的替代方案。

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