【发布时间】:2016-03-26 08:29:00
【问题描述】:
我正在编写一个简单的基于 TCP 的回显服务器。当我尝试使用BufReader 和BufWriter 读取和写入TcpStream 时,我发现将TcpStream 按值传递给BufReader::new() 会移动其所有权,因此我无法将其传递给BufWriter。然后,我在this thread找到了解决问题的答案:
fn handle_client(stream: TcpStream) {
let mut reader = BufReader::new(&stream);
let mut writer = BufWriter::new(&stream);
// Receive a message
let mut message = String::new();
reader.read_line(&mut message).unwrap();
// ingored
}
这很简单,而且很有效。但是,我不太明白为什么这段代码有效。为什么我可以只传递对BufReader::new() 的不可变引用,而不是可变引用?
整个程序可以在here找到。
更多详情
在上面的代码中,我使用了reader.read_line(&mut message)。于是我打开了Rust标准库中BufRead的源代码,看到了这个:
fn read_line(&mut self, buf: &mut String) -> Result<usize> {
// ignored
append_to_string(buf, |b| read_until(self, b'\n', b))
}
在这里我们可以看到它将自我(在我的情况下可能是&mut BufReader)传递给read_until()。接下来,我在同一个文件中找到了以下代码:
fn read_until<R: BufRead + ?Sized>(r: &mut R, delim: u8, buf: &mut Vec<u8>)
-> Result<usize> {
let mut read = 0;
loop {
let (done, used) = {
let available = match r.fill_buf() {
Ok(n) => n,
Err(ref e) if e.kind() == ErrorKind::Interrupted => continue,
Err(e) => return Err(e)
};
match memchr::memchr(delim, available) {
Some(i) => {
buf.extend_from_slice(&available[..i + 1]);
(true, i + 1)
}
None => {
buf.extend_from_slice(available);
(false, available.len())
}
}
};
r.consume(used);
read += used;
if done || used == 0 {
return Ok(read);
}
}
}
在这一部分中,有两个地方使用了BufReader:r.fill_buf() 和r.consume(used)。我以为r.fill_buf() 是我想看到的。于是,我去Rust标准库中BufReader的代码,发现是这样的:
fn fill_buf(&mut self) -> io::Result<&[u8]> {
// ignored
if self.pos == self.cap {
self.cap = try!(self.inner.read(&mut self.buf));
self.pos = 0;
}
Ok(&self.buf[self.pos..self.cap])
}
它似乎使用self.inner.read(&mut self.buf) 从self.inner 读取数据。然后,我们来看看BufReader和BufReader::new()的结构:
pub struct BufReader<R> {
inner: R,
buf: Vec<u8>,
pos: usize,
cap: usize,
}
// ignored
impl<R: Read> BufReader<R> {
// ignored
#[stable(feature = "rust1", since = "1.0.0")]
pub fn new(inner: R) -> BufReader<R> {
BufReader::with_capacity(DEFAULT_BUF_SIZE, inner)
}
// ignored
#[stable(feature = "rust1", since = "1.0.0")]
pub fn with_capacity(cap: usize, inner: R) -> BufReader<R> {
BufReader {
inner: inner,
buf: vec![0; cap],
pos: 0,
cap: 0,
}
}
// ignored
}
从上面的代码我们可以知道inner是一个实现Read的类型。就我而言,inner 可能是&TcpStream。
我知道Read.read()的签名是:
fn read(&mut self, buf: &mut [u8]) -> Result<usize>
这里需要一个可变引用,但我只借给它一个不可变引用。当程序在 fill_buf() 中到达 self.inner.read() 时,这应该是一个问题吗?
【问题讨论】:
-
@Shepmaster 我最初认为 TcpStream 的工作方式与 File 不同。但是在阅读了 Lukas Kalbertodt 的回答后,我突然意识到它背后的想法是一样的。谢谢你的链接。
标签: reference rust immutability borrowing