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。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。