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 { + self.peer_addr().ok() + } +} + #[cfg(feature = "tls")] impl Connected for TlsStream { fn remote_addr(&self) -> Option {