如何以分块方式drain Rust的Vector并移动块至函数？

你想要实现的是每次从Vec<T>中移除大小为2的块，将块的所有权转移给do_something函数。现有实现不仅冗长，还存在潜在问题——v.chunks(2)返回的是原向量的切片引用，当你调用drain移除元素后，后续的chunk会变成悬垂引用，可能引发未定义行为。

这里提供两种更简洁且安全的方案：

方案一：循环从头部移除块 #

利用Vec::drain会直接返回被移除元素的迭代器这一特性，循环处理直到向量为空：

fn drain_in_chunks<T>(mut v: Vec<T>) {
    while !v.is_empty() {
        // 每次取前2个（或剩余的所有）元素，收集成块后传递
        let chunk = v.drain(0..v.len().min(2)).collect::<Vec<_>>();
        do_something(chunk);
    }
}

这个方式无需计算索引偏移，每次drain都会修改原向量的长度，确保后续操作基于最新的向量状态，同时直接获取元素所有权，完美契合你“移动元素”的需求。

方案二：自定义迭代器（进阶） #

如果你偏好迭代器风格，可以实现一个自定义迭代器来封装按块移除的逻辑，让代码更符合Rust的迭代器习惯：

struct ChunkDrain<'a, T> {
    vec: &'a mut Vec<T>,
    chunk_size: usize,
}

impl<'a, T> Iterator for ChunkDrain<'a, T> {
    type Item = Vec<T>;

    fn next(&mut self) -> Option<Self::Item> {
        if self.vec.is_empty() {
            None
        } else {
            let end = self.vec.len().min(self.chunk_size);
            Some(self.vec.drain(0..end).collect())
        }
    }
}

fn drain_in_chunks<T>(mut v: Vec<T>) {
    for chunk in ChunkDrain { vec: &mut v, chunk_size: 2 } {
        do_something(chunk);
    }
}

这个方案把按块移除的逻辑封装成迭代器，调用时更简洁，适合需要重复使用该逻辑的场景。

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