Merge pull request #63 from najamelan/cleanup

Cleanup
pull/1/head
Daniel Abramov 5 years ago committed by GitHub
commit 9cf7243860
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 12
      src/connect.rs
  2. 2
      src/lib.rs

@ -67,7 +67,7 @@ mod encryption {
}
pub fn wrap_stream<S>(socket: S, domain: String, mode: Mode)
-> Box<Future<Item=AutoStream<S>, Error=Error> + Send>
-> Box<dyn Future<Item=AutoStream<S>, Error=Error> + Send>
where
S: 'static + AsyncRead + AsyncWrite + Send,
{
@ -77,8 +77,8 @@ mod encryption {
Box::new(future::result(TlsConnector::new())
.map(TokioTlsConnector::from)
.and_then(move |connector| connector.connect(&domain, socket))
.map(|s| StreamSwitcher::Tls(s))
.map_err(|e| Error::Tls(e)))
.map(StreamSwitcher::Tls)
.map_err(Error::Tls))
}
}
}
@ -123,7 +123,7 @@ fn domain(request: &Request) -> Result<String, Error> {
/// Creates a WebSocket handshake from a request and a stream,
/// upgrading the stream to TLS if required.
pub fn client_async_tls<R, S>(request: R, stream: S)
-> Box<Future<Item=(WebSocketStream<AutoStream<S>>, Response), Error=Error> + Send>
-> Box<dyn Future<Item=(WebSocketStream<AutoStream<S>>, Response), Error=Error> + Send>
where
R: Into<Request<'static>>,
S: 'static + AsyncRead + AsyncWrite + NoDelay + Send,
@ -138,7 +138,7 @@ where
// Make sure we check domain and mode first. URL must be valid.
let mode = match url_mode(&request.url) {
Ok(m) => m,
Err(e) => return Box::new(future::err(e.into())),
Err(e) => return Box::new(future::err(e)),
};
Box::new(wrap_stream(stream, domain, mode)
@ -152,7 +152,7 @@ where
/// Connect to a given URL.
pub fn connect_async<R>(request: R)
-> Box<Future<Item=(WebSocketStream<AutoStream<TcpStream>>, Response), Error=Error> + Send>
-> Box<dyn Future<Item=(WebSocketStream<AutoStream<TcpStream>>, Response), Error=Error> + Send>
where
R: Into<Request<'static>>
{

@ -215,7 +215,7 @@ impl<T> Stream for WebSocketStream<T> where T: AsyncRead + AsyncWrite {
fn poll(&mut self) -> Poll<Option<Message>, WsError> {
self.inner
.read_message()
.map(|msg| Some(msg))
.map(Some)
.to_async()
.or_else(|err| match err {
WsError::ConnectionClosed => Ok(Async::Ready(None)),

Loading…
Cancel
Save