Question

我试图找出从矢量转换为数组并返回的最类似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方法永远不会出现恐慌（我认为{{1}不可能在给定代码约束的情况下迭代器的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

虽然that function目前不稳定，因此只能在夜间使用。

将向量转换为数组并返回

1 个答案: