Rust中带const泛型的二维数组转一维slice的非unsafe方法

一、你的unsafe实现的类型限制 #

你的as_flat_slice函数本质上是安全的，但存在以下隐含约束：

• T必须是Sized类型：数组元素本身要求是Sized，因此该约束会自动满足，无需额外标注。
• 内存布局兼容性：Rust中[[T; N]; N]的内存是连续存储的——整个二维方阵在内存中是一块连续的N*N个T的存储空间，因此指针转换合法。但如果T是带有自定义内存布局的类型（比如用#[repr(packed)]修饰），需要额外确认对齐要求是否匹配，这种场景非常少见。
• const上下文限制：由于函数是const fn，T必须支持在const上下文中进行指针转换，大多数基本类型（如u32）都满足这一点。

二、非unsafe的转换方法 #

如果可以接受数据复制，有两种简单的非unsafe方案：

1. 使用concat方法 #

利用数组的concat方法直接将二维数组拼接为一维数组，再转成slice：

fn generate_array<const N: usize>() -> [[u32; N]; N] {
    [[1; N]; N]
}

fn library_function(width: usize, height: usize, data: &[u32]) {
    println!("width: {}, height: {}, data: {:?}", width, height, data);
}

fn main() {
    const N: usize = 3;
    let arr = generate_array::<N>();
    let flat = arr.concat();
    library_function(N, N, &flat);
}

缺点：会创建新的一维数组，产生内存复制开销，不适用于大尺寸数组。

2. 使用迭代器收集为Vec #

通过flat_map展开二维数组的迭代器，再收集为Vec<u32>：

fn main() {
    const N: usize = 3;
    let arr = generate_array::<N>();
    let flat: Vec<u32> = arr.iter().flat_map(|row| row.iter().copied()).collect();
    library_function(N, N, &flat);
}

缺点：同样会产生内存分配和复制，性能不如零成本的unsafe转换。

三、零成本的安全封装 #

如果需要零成本转换，目前标准库没有直接的非unsafe API，但你可以给unsafe函数添加安全断言，明确标注安全前提，将其封装为安全函数：

/// 将二维方阵转换为一维slice，零成本且安全
/// 安全前提：Rust中`[[T; N]; N]`的内存布局是连续的`N*N`个`T`
fn as_flat_slice<const N: usize, T>(arr: &[[T; N]; N]) -> &[T] {
    // 编译期静态断言：验证二维数组内存长度等于N*N个T的长度
    const _: () = assert!(std::mem::size_of::<[[T; N]; N]>() == N * N * std::mem::size_of::<T>());
    unsafe {
        std::slice::from_raw_parts(arr.as_ptr() as *const T, N * N)
    }
}

添加编译期断言后，能在编译阶段验证内存布局的正确性，进一步保证函数的安全性。

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