rust - 类型不匹配解决从 future channel 转发消息到 WebSocket Sink 时的错误类型

标签 rust future rust-tokio

我正在努力研究 Rust 中的 future ,但我对这段代码感到困惑,这段代码应该将到达 rx 的消息发送到 sink:

extern crate futures;
extern crate tokio_core;
extern crate websocket;

use websocket::message::OwnedMessage;
use websocket::server::InvalidConnection;
use websocket::async::Server;

use tokio_core::reactor::Core;
use futures::{Future, Sink, Stream};
use futures::sync::mpsc;
use std::{thread, time};
use futures::sync::mpsc::Receiver;

fn main() {
    let mut core = Core::new().unwrap();
    let (mut tx, rx) = mpsc::channel(5);
    thread::spawn(|| worker(rx));
    let mut i = 0;
    loop {
        let res = tx.clone().send(OwnedMessage::Text(format!("Test {}", i)));
        core.run(res);
        i += 1;
        let period = time::Duration::from_millis(200);
        thread::sleep(period);
    }
}

fn worker(rx: Receiver<OwnedMessage>) {
    let mut core = Core::new().unwrap();
    let handle = core.handle();
    // bind to the server
    let server = Server::bind("127.0.0.1:9000", &handle).unwrap();
    let f = server.incoming()
            // we don't wanna save the stream if it drops
            .map_err(|InvalidConnection { error, .. }| error)
            .for_each(|(upgrade, addr)| {
                // accept the request to be a ws connection if it does
                let f = upgrade
                    .use_protocol("rust-websocket")
                    .accept()
                    .and_then(|(s, _)| {
                        let (sink, stream) = s.split();
                        rx // using stream (echoing back) works
                            .forward(sink)
                            .map_err(|error| {
                                error
                            })
                            .and_then(|(a, sink)| {
                                sink.send(OwnedMessage::Close(None))
                            })
                    });

                handle.spawn(f.map_err(move |e| println!("Err"))
                    .map(move |_| println!("Done")));
                Ok(())
            });
    core.run(f).expect("somerror");
}

如评论中所述,使用 stream 作为输入工作正常。使用 rx 时,编译器会提示有关错误类型的类型不匹配(我相信):

error[E0271]: type mismatch resolving `<futures::stream::SplitSink<websocket::client::async::Framed<tokio_core::net::TcpStream, websocket::async::MessageCodec<websocket::OwnedMessage>>> as futures::Sink>::SinkError == ()`
  --> src/main.rs:47:26
   |
47 |                         .forward(sink)
   |                          ^^^^^^^ expected enum `websocket::WebSocketError`, found ()
   |
   = note: expected type `websocket::WebSocketError`
              found type `()`

error[E0599]: no method named `map_err` found for type `futures::stream::Forward<futures::sync::mpsc::Receiver<websocket::OwnedMessage>, futures::stream::SplitSink<websocket::client::async::Framed<tokio_core::net::TcpStream, websocket::async::MessageCodec<websocket::OwnedMessage>>>>` in the current scope
  --> src/main.rs:48:26
   |
48 |                         .map_err(|error| {
   |                          ^^^^^^^
   |
   = note: the method `map_err` exists but the following trait bounds were not satisfied:
           `futures::stream::Forward<futures::sync::mpsc::Receiver<websocket::OwnedMessage>, futures::stream::SplitSink<websocket::client::async::Framed<tokio_core::net::TcpStream, websocket::async::MessageCodec<websocket::OwnedMessage>>>> : futures::Future`

这些是我的依赖项:

[dependencies]
websocket = "0.20.0"
futures = "0.1"
tokio-core = "0.1"

我在这里错过了什么?

最佳答案

error[E0271]: type mismatch resolving
  `<futures::stream::SplitSink<
        websocket::client::async::Framed<
            tokio_core::net::TcpStream,
            websocket::async::MessageCodec<websocket::OwnedMessage>>>
   as futures::Sink>::SinkError == ()`

我们这里有两种类型:<futures::stream::SplitSink<...> as futures::Sink>::SinkError() .这两种类型从何而来?此外,第一个是未解析的关联类型;也许我们可以解决它以获得更多见解?让我们一步步追溯。

首先,我们需要弄清楚为什么编译器首先要尝试匹配这两种类型。如果我们查看 forward 的签名,我们将看到约束 Self::Error: From<S::SinkError> . Self是我们调用的流的类型 forward上,同时 S是作为参数传递给 forward 的接收器的类型.

我们正在调用 forwardrx , 其类型为 futures::sync::mpsc::Receiver .关于documentation page for Receiver ,我们可以看到以下内容:

impl<T> Stream for Receiver<T>
  type Item = T
  type Error = ()

这向我们展示了 () 在哪里来自。让我们看看sink现在争论。

sink 的类型是futures::stream::SplitSink<websocket::client::async::Framed<tokio_core::net::TcpStream, websocket::async::MessageCodec<websocket::OwnedMessage>>> (我们从错误消息中知道这一点;RLS 也证实了这一点)。关于documentation page for SplitSink ,我们有:

impl<S: Sink> Sink for SplitSink<S>
  type SinkItem = S::SinkItem
  type SinkError = S::SinkError

所以 SplitSinkSinkError与其内部水槽的 SinkError 相同.内水槽的类型是 websocket::client::async::Framed<tokio_core::net::TcpStream, websocket::async::MessageCodec<websocket::OwnedMessage>> . documentation for Framed 是什么意思说?

impl<T, U> Sink for Framed<T, U>
where
    T: AsyncWrite,
    U: Encoder,
    <U as Encoder>::Error: From<Error>,
  type SinkItem = <U as Encoder>::Item
  type SinkError = <U as Encoder>::Error

Framed有两个类型参数,但我们只需要看第二个,即websocket::async::MessageCodec<websocket::OwnedMessage>在这里,确定SinkError类型。我们来看看 MessageCodec 现在。 (注意:websocket::codec::ws::MessageCodecreexported 作为 websocket::async::MessageCodec)

impl<M> Decoder for MessageCodec<M>
where
    M: MessageTrait,
  type Item = OwnedMessage
  type Error = WebSocketError

啊哈!接收器产生 WebSocketError 类型的错误.


现在我们已经弄清楚了类型,让我们回到我们最初关心类型的原因。我们试图理解为什么约束 Self::Error: From<S::SinkError>在调用 forward 时未遇到.我们现在知道编译器正在尝试解析 (): From<WebSocketError> .好像没有impl From<WebSocketError> for () .让我们验证一下:

extern crate websocket;

fn main() {
    let a = websocket::result::WebSocketError::NoDataAvailable;
    let () = From::from(a);
}

确实,编译失败:

error[E0277]: the trait bound `(): std::convert::From<websocket::WebSocketError>` is not satisfied
 --> src/main.rs:5:14
  |
5 |     let () = From::from(a);
  |              ^^^^^^^^^^ the trait `std::convert::From<websocket::WebSocketError>` is not implemented for `()`
  |
  = note: required by `std::convert::From::from`

我们可以使用 sink_map_err 解决缺少的实现改变sink的错误类型。

let (sink, stream) = s.split();
let sink = sink.sink_map_err(|_| ()); // <<<<<
rx
    .forward(sink)
    .and_then(|(a, sink)| {
        sink.send(OwnedMessage::Close(None))
    })

这解决了对 forward 的调用, 但现在这个闭包的结果不与 upgrade.use_protocol("rust-websocket").accept() 组成, 其中还有 WebSocketError作为它的错误类型。更改 rx 更有意义的错误类型。但是我们如何构造一个WebSocketError来自() , 不携带任何信息?

您可能想知道,为什么 Receiver使用 ()它的错误类型?如果我们看一下 source code ,我们可以看到其实poll 从不 返回错误。我认为如果错误类型是 ! 会更合适(never 类型)或其他一些 void 类型,以清楚地表明错误是不可能的;有一个issue open on futures为 future 0.2 请求此更改。

因为错误是不可能的,我们不需要构造 WebSocketError ;我们可以改为分散,例如通过 panic 。

fn worker(rx: Receiver<OwnedMessage>) {
    let rx = rx.map_err(|()| panic!("Receiver should never fail!"));
    let mut core = Core::new().unwrap();
    let handle = core.handle();
    // bind to the server
    let server = Server::bind("127.0.0.1:9000", &handle).unwrap();
    let f = server.incoming()
            // we don't wanna save the stream if it drops
            .map_err(|InvalidConnection { error, .. }| error)
            .for_each(|(upgrade, addr)| {
                // accept the request to be a ws connection if it does
                let f = upgrade
                    .use_protocol("rust-websocket")
                    .accept()
                    .and_then(|(s, _)| {
                        let (sink, stream) = s.split();
                        rx
                            .forward(sink)
                            .and_then(|(a, sink)| {
                                sink.send(OwnedMessage::Close(None))
                            })
                    });

                handle.spawn(f.map_err(move |e| println!("Err"))
                    .map(move |_| println!("Done")));
                Ok(())
            });
    core.run(f).expect("somerror");
}

现在,还是有错误:

error[E0507]: cannot move out of captured outer variable in an `FnMut` closure
  --> src/main.rs:43:31
   |
30 |     let rx = rx.map_err(|()| panic!("Receiver should never fail!"));
   |         -- captured outer variable
...
43 |                     .and_then(|(s, _)| {
   |                               ^^^^^^^^ cannot move out of captured outer variable in an `FnMut` closure

为什么关闭试图移动 rx ?因为forward需要 self按值(value)。为什么闭包是 FnMut ?当心, Future::and_then 需要 FnOnce (将值从捕获的变量移动到 FnOnce 闭包中是有效的),但是 Stream::for_each 需要 FnMut .这是有道理的:for_each将为每个传入连接调用一次闭包!

您正在使用的 channel 是 multi-producer, single-consumer (因此得名 mpsc),但您试图在此处拥有多个消费者(每个连接都试图从接收方读取数据)。我会把它留给你来解决你程序中的这个设计问题。请记住,可以有多个并发客户端连接!

关于rust - 类型不匹配解决从 future channel 转发消息到 WebSocket Sink 时的错误类型,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/47582586/

相关文章:

firebase - flutter/Dart 错误 : The argument type 'Future<File>' can't be assigned to the parameter type 'File'

java - 当 ExecutorService 中的任何 Thread/Runnable/Callable 在等待终止时失败时如何捕获异常

rust - 如何创建一个 tokio 专用传输来覆盖默认的 tick 实现?

async-await - 如何在 Tokio 中使用 async/await 语法?

rust - 当多个 future 使用相同的基础套接字时,为什么不能将它们唤醒?

macros - 是否可以防止 Rust 中的宏重复相同的参数?

enums - 我应该使用枚举还是盒装特征对象来模拟多态性?

rust - 手动删除引用包装器的Rust生命周期问题

rust - 如何访问通过 UDP 发送的数据?

scala - 在 Scala Future 完成后进行清理的最佳实践