From 6a26d9e5a8d4d3e3b2c44b436adb298f5543725c Mon Sep 17 00:00:00 2001
From: Michael Apozyan <michael.a@cliqz.com>
Date: Mon, 20 Jan 2020 11:17:40 +0100
Subject: [PATCH] Impl transport::server::Connected for tokio::net::TcpStream

---
 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 05f2ec0ca..1f4900a69 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> {