如何修改Rust的zip_map函数使其接受二元函数而非元组参数的一元函数？

问题背景 #

现有一个扩展Iterator的trait Iterx，其中zip_map方法要求传入接收二元组的一元函数，但希望改为支持直接传入接收两个独立参数的二元函数，简化调用语法。

原有代码 #

pub trait Iterx: Iterator {
    fn zip_map<U, T, F>(self, other: U, f: F) -> Map<Zip<Self, U::IntoIter>, F>
    where
        Self: Sized,
        U: IntoIterator,
        F: Fn((<Self as Iterator>::Item, <U as IntoIterator>::Item)) -> T,
    {
        self.zip(other.into_iter()).map(f)
    }
}

当前调用方式（需传二元组参数） #

#[test]
fn test_zip_map() {
    assert_equal((1..5).zip_map(1..5, |(a, b)| a + b), vec![2, 4, 6, 8]);
    assert_equal((1..5).zip_map(1..5, |(a, b)| a * b), vec![1, 4, 9, 16]);
}

期望调用方式（直接传二元函数） #

#[test]
fn test_zip_map() {
    assert_equal((1..5).zip_map(1..5, |a, b| a + b), vec![2, 4, 6, 8]);
    assert_equal((1..5).zip_map(1..5, |a, b| a * b), vec![1, 4, 9, 16]);
}

解决方案 #

修改trait Iterx的zip_map方法，调整泛型约束并在内部包装闭包解构二元组：

use std::iter::{Map, Zip};

pub trait Iterx: Iterator {
    fn zip_map<U, T, F>(self, other: U, f: F) -> Map<Zip<Self, U::IntoIter>, impl FnMut((Self::Item, U::Item)) -> T>
    where
        Self: Sized,
        U: IntoIterator,
        F: FnMut(Self::Item, U::Item) -> T,
    {
        self.zip(other.into_iter()).map(move |(x, y)| f(x, y))
    }
}

// 为所有Iterator类型实现该trait
impl<I: Iterator> Iterx for I {}

关键改动说明 #

• 泛型约束调整：将F的约束从Fn((A, B)) -> T改为FnMut(A, B) -> T，允许传入二元函数；使用FnMut而非Fn，兼容需要修改捕获变量的场景，灵活性更强。
• 返回类型调整：由于内部新增了一层闭包包装，返回类型改为Map<..., impl FnMut((A,B)) -> T>，利用Rust的impl Trait特性返回匿名闭包类型，避免堆分配，保证性能。
• 内部实现优化：在map方法中添加move闭包，解构zip生成的二元组，将两个元素分别传递给二元函数f；move关键字确保闭包捕获f的所有权，避免生命周期问题。
• 补全trait实现：添加impl<I: Iterator> Iterx for I {}，让所有迭代器类型都能使用zip_map方法。

验证 #

修改后的代码可以直接运行期望的测试用例，无需调整测试中的闭包语法，完全符合需求。

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