如何在Rust中获得切片作为阵列？

Question

我有一个大小未知的数组，我想获取该数组的一个切片，并将其转换为静态大小的数组：

fn pop(barry: &[u8]) -> [u8; 3] {
    barry[0..3] // expected array `[u8; 3]`, found slice `[u8]`
}

我该怎么做？

Answer 1

这里有一个与您要求的类型签名匹配的函数。

fn pop(barry: &[u8]) -> [u8; 3] {
    [barry[0], barry[1], barry[2]]
}

但是由于barry的元素可能少于三个，因此您可能需要返回一个选项

fn pop(barry: &[u8]) -> Option<[u8; 3]> {
    if barry.len() < 3 {
        None
    } else {
        Some([barry[0], barry[1], barry[2]])
    }
}

Answer 2

感谢@malbarbo，我们可以使用此帮助功能：

use std::convert::AsMut;

fn clone_into_array<A, T>(slice: &[T]) -> A
where
    A: Default + AsMut<[T]>,
    T: Clone,
{
    let mut a = A::default();
    <A as AsMut<[T]>>::as_mut(&mut a).clone_from_slice(slice);
    a
}

要获得更简洁的语法，请执行以下操作：

fn main() {
    let original = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10];

    let e = Example {
        a: clone_into_array(&original[0..4]),
        b: clone_into_array(&original[4..10]),
    };

    println!("{:?}", e);
}

只要T：默认克隆。

如果您知道您的类型实现了复制，则可以使用以下表单：

use std::convert::AsMut;

fn copy_into_array<A, T>(slice: &[T]) -> A
where
    A: Default + AsMut<[T]>,
    T: Copy,
{
    let mut a = A::default();
    <A as AsMut<[T]>>::as_mut(&mut a).copy_from_slice(slice);
    a
}

两种变体都将panic 如果目标数组和传入的切片的长度不相同。

Answer 3

您可以使用TryIntotrait（稳定在Rust 1.34中）轻松做到这一点：

use std::convert::TryInto;

fn pop(barry: &[u8]) -> [u8; 3] {
    barry.try_into().expect("slice with incorrect length")
}

但更好的是：不需要克隆/复制您的元素！实际上可以得到一个

fn pop(barry: &[u8]) -> &[u8; 3] {
    barry.try_into().expect("slice with incorrect length")
}

正如在其他答案中提到的，如果barry的长度不是3，您可能不想惊慌失措，而是要优雅地处理此错误。

这要归功于来自的相关特性尝试的这些impl（在Rust 1.47之前，这些仅适用于长度不超过32的阵列）：

impl<'_, T, const N: usize> TryFrom<&'_ [T]> for [T; N]
where
    T: Copy, 

impl<'a, T, const N: usize> TryFrom<&'a [T]> for &'a [T; N]

impl<'a, T, const N: usize> TryFrom<&'a mut [T]> for &'a mut [T; N]