我可以从单个字节(u8)创建一个可变切片&mut [u8]吗?

问题描述 投票:6回答:2

有时我想从std::io::Reader读取一个字节。如果我尝试这样做:

use std::io::{self, Read};

fn main() {
    let mut byte: u8 = 0;
    io::stdin().read(&mut byte).unwrap();
    println!("byte: {}", byte);
}

我得到以下错误(很明显,因为byte不是切片):

error[E0308]: mismatched types
 --> src/main.rs:6:22
  |
6 |     io::stdin().read(&mut byte).unwrap();
  |                      ^^^^^^^^^ expected slice, found u8
  |
  = note: expected type `&mut [u8]`
             found type `&mut u8`

有没有办法我可以保持byte作为一个简单的u8,只需要一片它,然后我可以传递给read()?使这段代码工作的显而易见的方法是使用长度为1的数组:

use std::io::{self, Read};

fn main() {
    let mut byte: [u8; 1] = [0];
    io::stdin().read(&mut byte).unwrap();
    println!("byte: {}", byte[0]);
}

但是在整个代码中,这有点奇怪的感觉,使用单个u8而不是[u8; 1]更自然,我必须将其编入索引。

如果不可能从简单的u8创建一个切片,那是可以的,但我不知道它是否可能并且想知道。

rust
2个回答
13
投票

Rust 1.28+

slice::from_mut回来了,它很稳定!

use std::{
    io::{self, Read},
    slice,
};

fn main() {
    let mut byte = 0;
    let bytes_read = io::stdin().read(slice::from_mut(&mut byte)).unwrap();
    if bytes_read == 1 {
        println!("read byte: {:?}", byte);
    }
}

Rust 1.0+

但是在整个代码中,这有点奇怪的感觉,使用单个u8而不是[u8; 1]更自然,我必须将其编入索引。

创建长度为1的数组将是最自然的方式:

use std::io::{self, Read};

fn main() {
    let mut bytes = [0];
    let bytes_read = io::stdin().read(&mut bytes).unwrap();
    let valid_bytes = &bytes[..bytes_read];
    println!("read bytes: {:?}", valid_bytes);
}

但是,可以从引用到单个值的不安全地创建切片:

use std::io::{self, Read};
use std::slice;

fn mut_ref_slice<T>(x: &mut T) -> &mut [T] {
    // It's important to wrap this in its own function because this is
    // the only way to tell the borrow checker what the resulting slice
    // will refer to. Otherwise you might get mutable aliasing or a
    // dangling pointer which is what Rust is trying to avoid.
    unsafe { slice::from_raw_parts_mut(x, 1) }
}

fn main() {
    let mut byte = 0u8;
    let bytes_read = io::stdin().read(mut_ref_slice(&mut byte)).unwrap();
    if bytes_read != 0 {
        println!("byte: {}", byte);
    }
}

请记住,切片基本上是两件事:指向内存区域和长度的指针。使用一个长度为1的片段,您只需要为可变参考和bam添加长度!你得到了一分。

早期版本的Rust有ref_slice and mut_ref_slice functions。它们被删除了,因为它们的实用程序尚未得到证实(这不是常见问题),但它们可以安全地调用。功能被移动到ref_slice crate,所以如果你想继续使用它们,这是一种可能性。


5
投票

要回答你的实际问题:不,你不能这样做,而且几乎从来没有任何需要。即使你无法获得可读的迭代,你也可以将byte[0]放入另一个变量并使用它。

相反,你可以使用the Bytes iterator

let byte: u8 = io::stdin().bytes().next().unwrap();
© www.soinside.com 2019 - 2024. All rights reserved.