任务之间的通道通信

问题描述 投票:0回答:1

我正在尝试在一个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

谢谢 :)

rust rust-tokio hyper
1个回答
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!"))
        })
    });
© www.soinside.com 2019 - 2024. All rights reserved.