如何在Rust中为固定大小类型实现类似C++重载的泛型写入函数？

嘿，我懂你的需求——想写一个泛型函数，像C++重载那样处理所有固定大小的数值类型（u8、i8、u16这些），不用重复写一堆几乎一样的函数对吧？你之前尝试用fixed库的Fixed trait踩的坑，其实是这个trait的设计和你的需求不匹配，咱们一步步来解决。

先说说你遇到的两个错误原因：

• 第一个错误里，[u8]是动态大小类型（DST），编译期没法确定它的长度，而局部变量必须有固定的编译期大小，所以data.to_ne_bytes()返回的不定长数组没法直接存在栈上。
• 第二个错误更直接：mem::size_of_val(data)是运行时才能拿到的值，Rust要求数组长度必须是编译期常量，所以那行代码肯定过不了编译。

那正确的思路是什么？我们需要让泛型类型能转换为固定长度的字节数组——毕竟所有原生数值类型的to_ne_bytes方法本来就返回固定长度的数组（比如u8返回[u8;1]，u16返回[u8;2]），所以可以自定义一个trait来约束这个行为：

use std::io::Write;

// 定义一个trait，要求类型能转换为固定长度的字节数组
trait ToNeBytes {
    // 编译期常量，标记该类型对应的字节数组长度
    const BYTE_LEN: usize;
    // 返回固定长度的字节数组
    fn to_ne_bytes(&self) -> [u8; Self::BYTE_LEN];
}

接下来给所有需要的数值类型实现这个trait，比如：

// 为u8实现
impl ToNeBytes for u8 {
    const BYTE_LEN: usize = 1;
    fn to_ne_bytes(&self) -> [u8; 1] {
        u8::to_ne_bytes(*self)
    }
}

// 为i8实现
impl ToNeBytes for i8 {
    const BYTE_LEN: usize = 1;
    fn to_ne_bytes(&self) -> [u8; 1] {
        i8::to_ne_bytes(*self)
    }
}

// u16同理
impl ToNeBytes for u16 {
    const BYTE_LEN: usize = 2;
    fn to_ne_bytes(&self) -> [u8; 2] {
        u16::to_ne_bytes(*self)
    }
}

// 你可以继续为i16、u32、i32、u64、i64等类型实现这个trait

然后咱们的泛型write函数就可以正常工作了：

fn write<T: ToNeBytes>(data: T, buf: &mut dyn Write) -> std::io::Result<()> {
    let bytes = data.to_ne_bytes();
    // 用write_all确保所有字节都写入，比write更稳妥
    buf.write_all(&bytes)?;
    Ok(())
}

如果你不想手动写这么多实现，也可以用现成的crates简化工作，比如bytemuck——这个库专门处理字节转换，所有原生数值类型都实现了它的Pod trait（Plain Old Data，简单来说就是可以安全按字节复制的类型）。用它的话代码会更简洁：

use bytemuck::Pod;
use std::io::Write;

fn write<T: Pod>(data: T, buf: &mut dyn Write) -> std::io::Result<()> {
    // bytes_of直接获取类型的字节切片，不用手动转数组
    buf.write_all(bytemuck::bytes_of(&data))?;
    Ok(())
}

这里要注意，Pod trait要求类型是可安全按字节复制的，所有原生数值类型都满足这个条件，所以完全没问题。

回到你之前用的Fixed trait，它的Bytes关联类型是[u8]，这是动态大小的，没法满足我们需要固定长度数组的需求，所以不太适合这个场景。

总结一下：要么自定义一个约束固定长度字节转换的trait，要么用bytemuck这类现成库，就能实现你想要的、类似C++重载效果的泛型函数啦。

备注：内容来源于stack exchange，提问作者Sync