在const fn中拷贝泛型小数组至更大的泛型数组

方法一：稳定版Const Fn + 索引遍历 #

Rust 1.59及以上版本支持const环境中的while循环，直接通过逐个索引拷贝元素即可，剩余位置用指定填充值初始化：

const fn copy_into_larger<T: Copy, const CURR: usize, const MAX: usize>(
    src: [T; CURR],
    fill: T,
) -> [T; MAX]
where
    [(); MAX - CURR]:, // 编译期约束：确保MAX >= CURR
{
    let mut dest = [fill; MAX];
    let mut i = 0;
    while i < CURR {
        dest[i] = src[i];
        i += 1;
    }
    dest
}

// 使用示例
const SRC: [u32; 3] = [1, 2, 3];
const DEST: [u32; 5] = copy_into_larger(SRC, 0); // 结果：[1, 2, 3, 0, 0]

方法二：递归宏实现（兼容旧版本Rust） #

如果需要兼容Rust 1.59之前的版本，可以用递归宏展开元素拷贝逻辑，同时加入编译期断言保证数组大小约束：

// 自定义编译期断言宏
macro_rules! const_assert {
    ($cond:expr) => {
        const _: () = assert!($cond);
    };
}

macro_rules! copy_small_to_large {
    // 递归终止条件：拷贝0个元素
    (@expand $dest:ident, $src:expr, 0) => {};
    // 递归展开：拷贝第n-1个元素后，继续拷贝前n-1个元素
    (@expand $dest:ident, $src:expr, $n:expr) => {
        copy_small_to_large!(@expand $dest, $src, $n - 1);
        $dest[$n - 1] = $src[$n - 1];
    };

    // 对外接口：源数组、填充值、目标数组长度
    ($src:expr, $fill:expr, $max:expr) => {{
        const CURR: usize = $src.len();
        const MAX: usize = $max;
        const_assert!(MAX >= CURR); // 编译期检查大小关系
        let mut dest = [$fill; MAX];
        copy_small_to_large!(@expand dest, $src, CURR);
        dest
    }};
}

// 使用示例
const SRC: [u32; 3] = [1, 2, 3];
const DEST: [u32; 5] = copy_small_to_large!(SRC, 0, 5);

方法三：Const环境下的数组Map（简洁写法） #

Rust 1.63及以上版本支持const环境中的std::array::from_fn，可以通过索引判断直接生成目标数组，代码更简洁：

const fn copy_into_larger_map<T: Copy, const CURR: usize, const MAX: usize>(
    src: [T; CURR],
    fill: T,
) -> [T; MAX]
where
    [(); MAX - CURR]:,
{
    std::array::from_fn(|i| if i < CURR { src[i] } else { fill })
}

// 使用示例
const SRC: [u32; 3] = [1, 2, 3];
const DEST: [u32; 5] = copy_into_larger_map(SRC, 0);

注意事项 #

• 必须保证元素类型T实现Copy trait，因为const环境下不允许值移动，只能拷贝。
• 编译期约束[(); MAX - CURR]:会强制检查MAX >= CURR，不满足则编译报错。
• 如果需要用元素的默认值填充剩余位置，可以将fill参数替换为T::default()，但要求T实现Default且Default::default()是const函数（Rust 1.61+多数标准类型已支持）。

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