improve error messages

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(