Rust TcpStream.set_nonblocking用法及代码示例

本文简要介绍rust语言中 std::net::TcpStream.set_nonblocking 的用法。

用法

pub fn set_nonblocking(&self, nonblocking: bool) -> Result<()>

将此 TCP 流移入或移出非阻塞模式。

这将导致 read 、 write 、 recv 和 send 操作变为非阻塞，即立即从它们的调用中返回。如果 IO 操作成功，则返回Ok，无需进一步操作。如果 IO 操作无法完成并需要重试，则返回类型为 io::ErrorKind::WouldBlock 的错误。

在 Unix 平台上，调用此方法对应于调用 fcntl FIONBIO 。在 Windows 上调用此方法对应于调用 ioctlsocket FIONBIO 。

例子

以非阻塞模式从 TCP 流中读取字节：

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

let mut stream = TcpStream::connect("127.0.0.1:7878")
    .expect("Couldn't connect to the server...");
stream.set_nonblocking(true).expect("set_nonblocking call failed");

let mut buf = vec![];
loop {
    match stream.read_to_end(&mut buf) {
        Ok(_) => break,
        Err(ref e) if e.kind() == io::ErrorKind::WouldBlock => {
            // wait until network socket is ready, typically implemented
            // via platform-specific APIs such as epoll or IOCP
            wait_for_fd();
        }
        Err(e) => panic!("encountered IO error: {}", e),
    };
};
println!("bytes: {:?}", buf);

相关用法

注：本文由纯净天空筛选整理自rust-lang.org大神的英文原创作品 std::net::TcpStream.set_nonblocking。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。