任务之间的渠道交流

时间:2019-03-08 11:33:49

标签: rust rust-tokio hyper

我正在尝试在一个hyper服务和一个tokio流之间建立基于信道的通信。问题是编译器发出以下错误消息:

  

闭包为FnOnce,因为它将变量tx_queue移出   它的环境。

在阅读rustc --explain E0525提供的解释之后,看来tokio::sync::mpsc::Sender实现了Clone但没有实现Copy(除非我忽略了某些内容)。

所以我有点卡住了。如何让我的服务通过tokio::sync::mpsc通道将消息发送到tokio流?我确定我想念一些明显的东西,但看不到:/

有问题的代码摘录(经过修改,使其在请求@ E_net4时更短):

    extern crate hyper;
    extern crate tokio;
    extern crate tokio_signal;

    use futures::Stream;
    use hyper::rt::Future;
    use hyper::service::service_fn_ok;
    use hyper::{Body, Request, Response, Server};

    use futures::sink::Sink;
    use futures::sync::{mpsc, oneshot};
    use futures::{future, stream};

    fn main() {
        let mut runtime = tokio::runtime::Runtime::new().unwrap();

        let (tx1, rx1) = oneshot::channel::<()>();

        let (tx_queue, rx_queue) = mpsc::channel(10);

        // ----

        runtime.spawn(start_queue(rx_queue));

        // ----

        let http_server = Server::bind(&([127, 0, 0, 1], 3000).into()).serve(|| {
            service_fn_ok(move |_: Request<Body>| {
                tx_queue.send(1);
                Response::new(Body::from("Hello World!"))
            })
        });

        let graceful = http_server
            .with_graceful_shutdown(rx1)
            .map_err(|err| eprintln!("server error: {}", err))
            .and_then(|_| {
                dbg!("stopped");
                // TODO: stop order queue listener
                Ok(())
            });

        dbg!("HTTP server listening ...");

        runtime.spawn(graceful);

        // ----

        tx1.send(()).unwrap();

        dbg!("exited");
    }

    pub fn start_queue(rx: mpsc::Receiver<usize>) -> impl Future<Item = (), Error = ()> {
        #[derive(Eq, PartialEq)]
        enum Item {
            Value(usize),
            Tick,
            Done,
        }

        let items = rx
            .map(Item::Value)
            .chain(stream::once(Ok(Item::Done)))
            .take_while(|item| future::ok(*item != Item::Done));

        items
            .fold(0, |num, _item| {
                dbg!("x");
                future::ok(num)
            })
            .map(|_| ())
    }

此处提供了完整的代码:https://gist.github.com/jeromer/52aa2da43c5c93584c6ee55be68dd04e

谢谢:)

1 个答案:

答案 0 :(得分:2)

futures::sync::mpsc::Sender::send消耗Sender并产生一个Send对象,这是将来必须运行到完成才能实际发送数据的对象。如果频道已满,它将阻塞直到其他人收到该频道。完成后,它会退还Sender,可用于发送更多数据。

在这种情况下,我认为您不能仅使用Sender的单个实例来构造代码。您需要对其进行克隆,以便对服务功能的每次调用都有新的克隆。注意,两个闭包现在都是move

    let http_server = Server::bind(&([127, 0, 0, 1], 3000).into()).serve(move || {
        // This closure has one instance of tx_queue that was moved-in here.
        // Now we make a copy to be moved into the closure below.
        let tx_queue = tx_queue.clone();
        service_fn_ok(move |_: Request<Body>| {
            // This closure has one instance of tx_queue, but it will be called
            // multiple times, so it can not consume it. It must make a copy
            // before consuming it.
            tx_queue.clone().send(111);
            Response::new(Body::from("Hello World!"))
        })
    });

但是,这会给您以下警告:

warning: unused `futures::sink::send::Send` that must be used

就像我说的,send只是给您一个未来,必须运行它才能实际执行发送。如果忽略返回值,则不会发生任何事情。在这种情况下,最好将其spawn作为一个单独的任务执行(这样就不会阻止对客户端的响应)。要生成它,您需要一个来自运行时的执行程序,还必须为内部闭包克隆该执行程序:

    let executor = runtime.executor();
    let http_server = Server::bind(&([127, 0, 0, 1], 3000).into()).serve(move || {
        let tx_queue = tx_queue.clone();
        let executor = executor.clone();
        service_fn_ok(move |_: Request<Body>| {
            executor.spawn(tx_queue.clone().send(111).map(|_| ()).map_err(|err| {
                // TODO: Handle the error differenty!
                panic!("Error in mpsc {:?}", err);
            }));
            Response::new(Body::from("Hello World!"))
        })
    });