From cfdf0aff549196af0c3b7f6e531dbeacfb6990dc Mon Sep 17 00:00:00 2001
From: michael-a-cliqz <michael.a@cliqz.com>
Date: Mon, 20 Jan 2020 17:29:12 +0100
Subject: [PATCH] fix(transport): Add Connected impl for TcpStream (#245)

---
 tonic/src/transport/server/conn.rs | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/tonic/src/transport/server/conn.rs b/tonic/src/transport/server/conn.rs
index 05f2ec0..1f4900a 100644
--- 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> {