Rust如何定义适配内置数值类型的Vec<T>转Vec<U>泛型函数

问题原因分析 #

1. 为什么第一个版本编译失败 #

你的原始代码尝试直接把U类型的元素收集到Vec<T>中，但没有声明U到T的转换约束，也没有实际做转换操作。v.iter().map(|&t|t)返回的是U类型的迭代器，而标准库Vec<T>的FromIterator实现要求迭代器元素类型必须是T，类型不匹配就触发了FromIterator<U> not implemented的报错，并不是Vec真的没实现这个 trait。

2. 为什么自定义实现会冲突 #

Rust有孤儿规则：不允许开发者为外部定义的类型（比如标准库的Vec）实现外部定义的 trait（比如标准库的FromIterator），这个规则是为了避免依赖版本更新时出现实现冲突。FromIterator for Vec的实现已经在标准库中存在，你自己加的实现必然触发冲突。

正确实现方案 #

你要的数值向量转换不需要自定义任何 trait 实现，只需要加对泛型约束、加上类型转换逻辑即可：

基础版（支持无损数值转换） #

如果只需要u8转u32这类无精度损失的数值转换，用标准库的Into trait即可：

// 参数用&[U]比&Vec<U>更通用，支持传入切片、Vec、数组等多种类型
fn my_convert<T, U>(v: &[U]) -> Vec<T>
where
    // 约束1：U实现Copy，可以直接从引用拷贝出原值
    // 约束2：U可以无损转换为T，原生数值类型的无损转换标准库已经自动实现了Into
    U: Copy + Into<T>,
{
    v.iter()
     .copied() // 等价于.map(|&t| t)，写法更规范
     .map(Into::into) // 把U转换为T
     .collect() // 标准库已经实现了对应的FromIterator，直接用即可
}

调用示例：

fn main() {
    let u8_vec = vec![1u8, 2u8, 3u8];
    let u32_vec: Vec<u32> = my_convert(&u8_vec);
    println!("{:?}", u32_vec); // 输出 [1, 2, 3]
}

限制仅允许内置数值类型 #

如果要严格限制函数只能接收数值类型的输入，引入num_traits crate的Num trait做约束即可：

1. 先在Cargo.toml中添加依赖：num_traits = "0.2"
2. 代码修改如下：

use num_traits::Num;

fn my_convert<T, U>(v: &[U]) -> Vec<T>
where
    T: Num,
    U: Num + Copy + Into<T>,
{
    v.iter().copied().map(Into::into).collect()
}

此时如果传入字符串、自定义结构体等非数值类型，编译会直接报错，符合你的要求。

支持有损转换版本 #

如果需要支持u32转u8这类可能溢出的转换，用TryInto trait返回带错误的结果即可：

fn my_convert<T, U>(v: &[U]) -> Result<Vec<T>, U::Error>
where
    U: Copy + TryInto<T>,
{
    v.iter().copied().map(TryInto::try_into).collect()
}

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