diff options
Diffstat (limited to 'src/cmd/stream.rs')
| -rw-r--r-- | src/cmd/stream.rs | 18 |
1 files changed, 10 insertions, 8 deletions
diff --git a/src/cmd/stream.rs b/src/cmd/stream.rs index 95c1191..11634d3 100644 --- a/src/cmd/stream.rs +++ b/src/cmd/stream.rs @@ -95,13 +95,15 @@ impl crate::config::Config for Config { let connector = connector.clone(); let connector = tokio_tls::TlsConnector::from(connector); let stream = tokio::net::tcp::TcpStream::connect(&address); - Box::new(stream.context(crate::error::Connect).and_then( - move |stream| { - connector - .connect(&host, stream) - .context(crate::error::ConnectTls) - }, - )) + Box::new( + stream + .context(crate::error::Connect { address }) + .and_then(move |stream| { + connector + .connect(&host, stream) + .context(crate::error::ConnectTls { host }) + }), + ) }); Box::new(StreamSession::new( &self.command, @@ -114,7 +116,7 @@ impl crate::config::Config for Config { let connect: crate::client::Connector<_> = Box::new(move || { Box::new( tokio::net::tcp::TcpStream::connect(&address) - .context(crate::error::Connect), + .context(crate::error::Connect { address }), ) }); Box::new(StreamSession::new( |