我有一个未知大小的数组,我想获得该数组的一部分并将其转换为静态大小的数组:
fn pop(barry: &[u8]) -> [u8; 3] {
barry[0..3] // mismatched types: expected `[u8, ..3]` but found `&[u8]`
}
我该怎么做?
这是一个与您要求的类型签名相匹配的函数。
fn pop(barry: &[u8]) -> [u8; 3] {
[barry[0], barry[1], barry[2]]
}
但由于barry
可能少于三个元素,你可能想要返回一个Option<[u8; 3]>
而不是[u8; 3]
。
fn pop(barry: &[u8]) -> Option<[u8; 3]> {
if barry.len() < 3 {
None
} else {
Some([barry[0], barry[1], barry[2]])
}
}
您可以使用TryInto
特征(在Rust 1.34中稳定)轻松完成此操作:
fn pop(barry: &[u8]) -> [u8; 3] {
barry.try_into()
.map(|s: &[u8; 3]| s.clone())
.expect("slice with incorrect length")
}
但更好的是:没有必要克隆你的阵列!实际上可以从&[u8; 3]
获得&[u8]
:
fn pop(barry: &[u8]) -> &[u8; 3] {
barry.try_into().expect("slice with incorrect length")
}
正如其他答案中所提到的,如果barry
的长度不是3,你可能不想恐慌,所以你应该返回一个Option<[u8; 3]>
或类似的东西来优雅地处理这个错误。
这可以归功于相关特征$N
的这些impls(其中TryFrom
只是1到32之间的整数):
impl<'a, T> TryFrom<&'a [T]> for &'a [T; $N]
type Error = TryFromSliceError;
感谢@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: Default + Clone
。
如果你知道你的类型实现了Copy
,你可以使用这个表单:
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!
。
我推荐使用crate arrayref,它有一个方便的宏来做这件事。
请注意,使用此包,您可以创建对数组&[u8; 3]
的引用,因为它不会克隆任何数据!
如果您确实要克隆数据,那么您仍然可以使用宏,但最后调用clone:
#[macro_use]
extern crate arrayref;
fn pop(barry: &[u8]) -> &[u8; 3] {
array_ref!(barry, 0, 3)
}
要么
#[macro_use]
extern crate arrayref;
fn pop(barry: &[u8]) -> [u8; 3] {
array_ref!(barry, 0, 3).clone()
}
您可以手动创建阵列并将其返回。
如果您想要获得比3个元素更多(或更少)的元素,这个函数可以轻松扩展。
请注意,如果切片太小,则数组的结束项将为0。
fn pop(barry: &[u8]) -> [u8; 3] {
let mut array = [0u8; 3];
for (&x, p) in barry.iter().zip(array.iter_mut()) {
*p = x;
}
array
}
我对其他答案不满意,因为我需要几个函数来返回不同长度的固定u8数组。我写了一个宏来生成特定于该任务的函数。希望它可以帮助某人。
#[macro_export]
macro_rules! vec_arr_func {
($name:ident, $type:ty, $size:expr) => {
pub fn $name(data: std::vec::Vec<$type>) -> [$type; $size] {
let mut arr = [0; $size];
arr.copy_from_slice(&data[0..$size]);
arr
}
};
}
//usage - pass in a name for the fn, type of array, length
vec_arr_func!(v32, u8, 32);
v32(data); //where data is std::vec::Vec<u8>