fix(transport): Add Connected impl for TcpStream (#245)

hyperium · Jan 20, 2020 · cfdf0af · cfdf0af
1 parent 6de0b4d
commit cfdf0af
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/tonic/src/transport/server/conn.rs b/tonic/src/transport/server/conn.rs
@@ -1,6 +1,7 @@
 use crate::transport::Certificate;
 use hyper::server::conn::AddrStream;
 use std::net::SocketAddr;
+use tokio::net::TcpStream;
 #[cfg(feature = "tls")]
 use tokio_rustls::{rustls::Session, server::TlsStream};
 
@@ -27,6 +28,12 @@ impl Connected for AddrStream {
     }
 }
 
+impl Connected for TcpStream {
+    fn remote_addr(&self) -> Option<SocketAddr> {
+        self.peer_addr().ok()
+    }
+}
+
 #[cfg(feature = "tls")]
 impl<T: Connected> Connected for TlsStream<T> {
     fn remote_addr(&self) -> Option<SocketAddr> {