将向量转换为数组并返回

Question

我试图找出从矢量转换为数组并返回的最类似Rust的方式.这些宏可以工作,甚至可以通过一些不安全的块来制作通用,但它们都觉得非常不像Rust.

我会很感激任何输入并且没有任何打击,我认为这段代码远非好或不优.我现在只玩Rust几周了,追逐发布和文档,所以非常感谢帮助.

macro_rules! convert_u8vec_to_array {
    ($container:ident, $size:expr) => {{
    if $container.len() != $size {
            None
    } else {
        use std::mem;
        let mut arr : [_; $size] = unsafe { mem::uninitialized() };
        for element in $container.into_iter().enumerate() {
            let old_val = mem::replace(&mut arr[element.0],element.1);
            unsafe { mem::forget(old_val) };
        }
        Some(arr)
        }
    }};
}

fn array_to_vec(arr: &[u8]) -> Vec<u8> {
     let mut vector = Vec::new();
     for i in arr.iter() {
         vector.push(*i);
     }
     vector
}

fn vector_as_u8_4_array(vector: Vec<u8>) -> [u8;4] {
    let mut arr = [0u8;4];
    for i in (0..4) {
        arr[i] = vector[i];
    }
    arr
}

Answer 1

代码对我来说似乎很好,虽然有一个非常重要的安全事项要注意:arr没有完全初始化时可能没有恐慌.运行在未初始化的内存析构函数很容易导致未定义行为,尤其,这意味着into_iter与next它的方法不应该恐慌(我相信这对是不可能的enumerate和mem::*迭代的部分恐慌给予代码的约束) .

也就是说,人们可以用一个函数表达replace/ forget习语:std::ptr::write.

for (idx, element) in $container.into_iter().enumerate() {
    ptr::write(&mut arr[idx], element);
}

虽然,我会把它写成:

for (place, element) in arr.iter_mut().zip($container.into_iter()) {
    ptr::write(place, element);
}

类似地,可以将一些迭代器优点应用于u8专用版本:

fn array_to_vec(arr: &[u8]) -> Vec<u8> {
     arr.iter().cloned().collect()
}
fn vector_as_u8_4_array(vector: Vec<u8>) -> [u8;4] {
    let mut arr = [0u8;4];
    for (place, element) in arr.iter_mut().zip(vector.iter()) {
        *place = *element;
    }
    arr
}

虽然第一个可能写得更好arr.to_vec(),第二个作为

let mut arr = [0u8; 4];
std::slice::bytes::copy_memory(&vector, &mut arr);
arr

虽然这个功能目前不稳定,因此只能在夜间使用.