mdebray/tvl-depot
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat(tvix/store/listener): implement connect_info()

Browse source 
For Unix and TCP streams, we can actually give more info than just ().

Change-Id: I80ca261b9e7a880389ec2029ea5893180314f0b2
Reviewed-on: https://cl.tvl.fyi/c/depot/+/9561
Autosubmit: flokli <flokli@flokli.de>
Tested-by: BuildkiteCI
Reviewed-by: Connor Brewster <cbrewster@hey.com>
This commit is contained in:
Florian Klink 2023-10-07 13:33:44 +02:00 committed by flokli
parent 3b620e6d0c
commit 09328c79d5
1 changed files with 19 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
tvix/store/src/listener/mod.rs
View file

@ -9,7 +9,7 @@ use futures::Stream;
use pin_project_lite::pin_project;
use tokio::io::{AsyncRead, AsyncWrite};
use tokio_listener::{Listener, ListenerAddress};
use tonic::transport::server::Connected;
use tonic::transport::server::{Connected, TcpConnectInfo, UdsConnectInfo};
/// A wrapper around a [Listener] which implements the [Stream] trait.
/// Mainly used to bridge [tokio_listener] with [tonic].
@ -72,11 +72,27 @@ impl DerefMut for Connection {
}
}
#[derive(Clone)]
pub enum ListenerConnectInfo {
TCP(TcpConnectInfo),
Unix(UdsConnectInfo),
Stdio,
Other,
}
impl Connected for Connection {
type ConnectInfo = ();
type ConnectInfo = ListenerConnectInfo;
fn connect_info(&self) -> Self::ConnectInfo {
()
if let Some(tcp_stream) = self.try_borrow_tcp() {
ListenerConnectInfo::TCP(tcp_stream.connect_info())
} else if let Some(unix_stream) = self.try_borrow_unix() {
ListenerConnectInfo::Unix(unix_stream.connect_info())
} else if let Some(_) = self.try_borrow_stdio() {
ListenerConnectInfo::Stdio
} else {
ListenerConnectInfo::Other
}
}
}