006 Rust 網絡程式設計，mio 庫介紹

mio

依賴

[dependencies]
mio = { version = "0.7.0", features = ["os-poll", "tcp"]}      

源碼示例

use mio::net::{TcpListener, TcpStream};
use mio::{Events, Interest, Poll, Token};

const SERVER: Token = Token(0);
const CLIENT: Token = Token(1);

fn main() -> std::io::Result<()> {
    let mut poll = Poll::new()?;
    let mut events = Events::with_capacity(128);

    // 建立server端socket
    let addr = "127.0.0.1:8080".parse().unwrap();
    let mut server = TcpListener::bind(addr)?;
    
    // 開始監聽該連結
    poll.registry()
        .register(&mut server, SERVER, Interest::READABLE)?;

    // 建立用戶端連結到服務端
    let mut client = TcpStream::connect(addr)?;

    // 監聽用戶端
    poll.registry()
        .register(&mut client, CLIENT, Interest::READABLE | Interest::WRITABLE)?;

    // 啟動循環
    loop {
        // 輪訓事件
        poll.poll(&mut events, None)?;

        // 處理事件
        let mut cnt = 0;
        for event in events.iter() {
            match event.token() {
                SERVER => {
                    let connection = server.accept();
                    println!("SERVER recv a connection! cnt = {}", cnt);
                    drop(connection);
                }
                CLIENT => {
                    if event.is_writable() {
                        println!("CLIENT write, cnt = {}", cnt);
                    }

                    if event.is_readable() {
                        println!("CLIENT read, cnt = {}", cnt);
                    }
                    
                    return Ok(());
                }

                _ => unreachable!(),
            }
            cnt += 1;
        }
    }
}