向mio注册频道

时间:2019-04-26 06:59:51

标签: rust mio

在mio doc的旧版本中,我找到了mio :: channel,它似乎用于创建实现EventedFd的通道,该通道可以向Poll注册。 我还在reddit上看到这已更改为其他功能,但我不知道注册频道的新方法是什么。

使用mio在频道(std::sync::mpsc或其他东西)上等待的当前日期是什么?我在Google上能找到的就是链接到mio docs的旧版本。 谢谢!

当我遇到相同的问题时,我希望可以提供一些代码使问题更容易理解:

use std::io;
use std::net::*; //{TcpListener,TcpStream,IpAddr,Ipv4Addr,SocketAddr};
use std::thread::*;
use std::sync::mpsc::*; //{Receiver,SyncSender,Sender,channel,sync_channel};

fn poll_quit( rx : &Receiver::<u8> ) -> bool {
    match rx.try_recv() {
        Ok(_) => true,
        Err(TryRecvError::Empty) => false,
        Err(TryRecvError::Disconnected) => true
    }
}

fn guard_external_tcp_port( rx : Receiver::<u8>) -> () {
    let listener = TcpListener::bind("127.0.0.1:8384").expect("tcp guard - bind failed!");
    listener.set_nonblocking(true).expect("cannot set tcp listener to nonblocking!");
    while false == poll_quit(&rx) {
        match listener.accept() {
            Ok((_s,pa)) => {
                println!("tcp client connected: {} - closing it down."
                    , pa);
            }
            Err(ref e) if e.kind() == io::ErrorKind::WouldBlock => {
                //wait_for_fd();
                continue;
            }
            Err(e) => panic!("encountered IO error: {}", e),

        }
    }
}

fn start_tcpguard() -> (JoinHandle<()>,SyncSender::<u8>) {
    let (tx,rx) = sync_channel::<u8>(0);
    let thrd = spawn(move || {
        guard_external_tcp_port(rx);
    });
    (thrd,tx)
}

到目前为止,我尝试不使用mio来执行操作,但是问题很明显:您要阻塞,直到关闭通道消息到达或向侦听器套接字发出信号为止。有足够的动力转向mio。除非mio无法等待Receiver

1 个答案:

答案 0 :(得分:0)

现在还有另一个mio crate,具有与频道相关的功能:

mio-extras

因此,您要做的就是将板条箱添加到cargo.toml [dependencies]中。
然后,在您的main.rs中添加extern crate mio_extras,您应该准备使用它们。

因此,在我添加到问题的代码段中,它最终看起来像这样:

extern crate mio;
extern crate mio_extras;

use std::io;
use mio::*;
use mio::net::*;
use mio_extras::channel::*;
use std::thread::*;

fn guard_external_tcp_port( rx : mio_extras::channel::Receiver::<u8>) -> () {
    let poll = Poll::new().expect("could not create a new Poll instance.");
    const QUIT : Token = Token(0);
    const CONNECT : Token = Token(1);


    let mut events = Events::with_capacity(5);
    poll.register(&rx,QUIT,Ready::readable(),PollOpt::level() ).expect("could not register channel Receiver.");


    let addr : std::net::SocketAddr = "127.0.0.1:8384".parse().unwrap();
    let listener = TcpListener::bind(&addr).expect("tcp guard - bind failed!");

    poll.register(&listener,CONNECT,Ready::all(),PollOpt::edge()).expect("could not register our listening socket.");

    let mut running : bool = true;
    //listener.set_nonblocking(true).expect("cannot set tcp listener to nonblocking!");
    while running {
        let _nevents = poll.poll(&mut events,None).unwrap();
        println!("poll returned!");
        for event in &events {
            match event.token() {
                QUIT => running = false,
                CONNECT => {
                    match listener.accept() {
                        Ok((_s,pa)) => {
                            println!("tcp client connected: {} - closing it down."
                                , pa);
                        }
                        Err(ref e) if e.kind() == io::ErrorKind::WouldBlock => {
                            //wait_for_fd();
                            continue;
                        }
                        Err(e) => panic!("encountered IO error: {}", e),
                    }
                },
                Token(_) => continue
            }
        }
    }
}

fn start_tcpguard() -> (JoinHandle<()>,Sender::<u8>) {
    let (tx,rx) = mio_extras::channel::channel::<u8>();
    let thrd = spawn(move || {
        guard_external_tcp_port(rx);
    });
    (thrd,tx)
}

我还尝试了sync_channel(0)版本,但该版本无法按预期工作。上面代码中的版本,尽管对于接收关闭事件的民意测验是有效的。