006 Rust 网络编程,mio 库介绍

mio

标准库也缺乏对套接字和连接的细粒度控制,例如设置SO_REUSEADDR。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;
        }
    }
}
  • 3
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值