如何在Rust中使用编译期已知大小的泛型作为结构体数组长度

问题场景 #

尝试实现一个泛型结构体，利用泛型类型（已实现Sized）的编译期大小，计算结构体内部数组的固定长度，出现如下编译错误：

error: generic parameters may not be used in const operations
 --> src/lib.rs:4:43
  |
4 |     arr: [Key; 1024 / std::mem::size_of::<Key>()]
  |                                           ^^^ cannot perform const operation using `Key`
  |
  = note: type parameters may not be used in const expressions

对应代码示例：

type SomeKey = u64;

// 无法编译的泛型结构体
struct NotWorking<Key: Sized> {
    arr: [Key; 1024 / std::mem::size_of::<Key>()]
}

// 可正常编译的非泛型版本
struct Working {
    arr: [SomeKey; 1024 / std::mem::size_of::<SomeKey>()]
}

解决方案 #

方法一：通过where子句确认编译期常量有效性 #

在Rust 1.64及以上版本，可通过where子句让编译器验证数组长度是合法的编译期常量：

type SomeKey = u64;

struct FixedSizeArray<Key: Sized>
where
    [(); 1024 / std::mem::size_of::<Key>()]:,
{
    arr: [Key; 1024 / std::mem::size_of::<Key>()],
}

// 使用示例
fn main() {
    let _: FixedSizeArray<u64> = FixedSizeArray { arr: [0; 128] };
    let _: FixedSizeArray<u32> = FixedSizeArray { arr: [0; 256] };
}

方法二：显式使用const泛型传递数组长度 #

如果需要更灵活的定义，可将数组长度作为const泛型参数传入，在使用时计算具体值：

struct FixedSizeArray<Key: Sized, const N: usize> {
    arr: [Key; N],
}

// 使用示例
fn main() {
    let _: FixedSizeArray<u64, {1024 / std::mem::size_of::<u64>()}> = FixedSizeArray { arr: [0; 128] };
}

原因说明 #

早期Rust不允许在const表达式中直接使用泛型类型参数，编译器无法确定std::mem::size_of::<Key>()是否为编译期已知的常量。通过where子句的约束，或显式const泛型参数，可让编译器确认数组长度的有效性，从而通过编译。

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