Rust中实现u16/u32/u64泛型按大小端转换为Vec<u8>的方法

通过自定义trait抽象整数按端序转字节的行为，仅为u16、u32、u64三个目标类型实现该trait，即可在泛型函数中完成类型约束，全程使用标准库原生方法，无unsafe代码、无额外运行时开销。

首先定义端序枚举和转换trait：

// 端序枚举，指定转换时使用的字节序
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum Endian {
    Big,
    Little,
}

// 定义统一的转换行为trait
trait ToEndianBytes {
    fn to_endian_bytes(self, endian: Endian) -> Vec<u8>;
}

为三个目标无符号整数类型实现该trait，直接复用标准库内置的端序转换方法：

impl ToEndianBytes for u16 {
    fn to_endian_bytes(self, endian: Endian) -> Vec<u8> {
        match endian {
            Endian::Big => self.to_be_bytes().to_vec(),
            Endian::Little => self.to_le_bytes().to_vec(),
        }
    }
}

impl ToEndianBytes for u32 {
    fn to_endian_bytes(self, endian: Endian) -> Vec<u8> {
        match endian {
            Endian::Big => self.to_be_bytes().to_vec(),
            Endian::Little => self.to_le_bytes().to_vec(),
        }
    }
}

impl ToEndianBytes for u64 {
    fn to_endian_bytes(self, endian: Endian) -> Vec<u8> {
        match endian {
            Endian::Big => self.to_be_bytes().to_vec(),
            Endian::Little => self.to_le_bytes().to_vec(),
        }
    }
}

实现通用泛型转换函数，泛型参数仅允许实现了ToEndianBytes的类型（也就是我们限定的三个无符号整数类型）：

fn convert<T: ToEndianBytes>(val: T, endian: Endian) -> Vec<u8> {
    val.to_endian_bytes(endian)
}

用法验证 #

对应给出的0x01234567（u32类型）转换场景，大端模式下返回结果完全符合预期：

fn main() {
    let test_num: u32 = 0x01234567;
    // 大端转换，返回 vec![0x01, 0x23, 0x45, 0x67]
    let big_endian_res = convert(test_num, Endian::Big);
    assert_eq!(big_endian_res, vec![0x01, 0x23, 0x45, 0x67]);

    // 小端转换结果
    let little_endian_res = convert(test_num, Endian::Little);
    assert_eq!(little_endian_res, vec![0x67, 0x45, 0x23, 0x01]);

    // u16、u64类型同样支持
    let u16_num: u16 = 0x1234;
    assert_eq!(convert(u16_num, Endian::Big), vec![0x12, 0x34]);

    let u64_num: u64 = 0x1122334455667788;
    assert_eq!(convert(u64_num, Endian::Big), vec![0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, 0x88]);
}

方案说明 #

• 类型约束严格：除了u16、u32、u64之外的任何类型传入convert函数都会直接触发编译错误，完全符合泛型限定要求。
• 无额外开销：直接调用标准库为整数类型实现的原生端序转换方法，没有手动做字节移位、掩码操作，不存在跨平台端序不一致问题，性能和原生方法完全一致。
• 无unsafe代码：整个实现完全使用安全Rust语法，不存在内存安全风险。
• 如果你需要处理更复杂的字节流读写场景，也可以使用byteorder库提供的接口完成同类需求，针对单值转字节向量的场景，上述标准库实现已经足够轻量简洁。

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