通过target_os使用Unix套接字连接器通信 (apache#102)

* 使用Unix套接字连接器通信demo

* 通过target_os使用Unix套接字连接器通信

Co-authored-by: qunwei <[email protected]>

Author: daedalus2022

dubbo/src/triple/client/builder.rs

@@ -60,6 +60,13 @@ impl ClientBuilder {
         }
     }
 
+    pub fn with_connector(self, connector: &'static str) -> Self {
+        Self {
+            connector: connector,
+            ..self
+        }
+    }
+
     pub fn connect(self) -> ClientBoxService {
         let builder = ServiceBuilder::new();
         let timeout = self.timeout.unwrap_or(5);

dubbo/src/triple/transport/connection.rs

@@ -87,6 +87,7 @@ where
         let uri = self.host.clone();
         let fut = async move {
             let mut con = connector.call(uri).await.unwrap();
+
             con.call(req)
                 .await
                 .map_err(|err| err.into())

dubbo/src/triple/transport/connector/mod.rs

@@ -16,7 +16,7 @@
  */
 
 pub mod http_connector;
-#[cfg(target_os = "unix")]
+#[cfg(any(target_os = "macos", target_os = "unix"))]
 pub mod unix_connector;
 
 use hyper::Uri;
@@ -79,7 +79,7 @@ pub fn get_connector(connector: &'static str) -> BoxCloneService<Uri, BoxIO, cra
             let c = http_connector::HttpConnector::new();
             BoxCloneService::new(Connector::new(c))
         }
-        #[cfg(target_os = "unix")]
+        #[cfg(any(target_os = "macos", target_os = "unix"))]
         "unix" => {
             let c = unix_connector::UnixConnector::new();
             BoxCloneService::new(Connector::new(c))

dubbo/src/triple/transport/listener/mod.rs

@@ -16,7 +16,7 @@
  */
 
 pub mod tcp_listener;
-#[cfg(target_os = "unix")]
+#[cfg(any(target_os = "macos", target_os = "unix"))]
 pub mod unix_listener;
 
 use std::net::SocketAddr;
@@ -64,7 +64,7 @@ impl<T: Listener> Listener for WrappedListener<T> {
 pub async fn get_listener(name: String, addr: SocketAddr) -> Result<BoxListener, crate::Error> {
     match name.as_str() {
         "tcp" => Ok(TcpListener::bind(addr).await?.boxed()),
-        #[cfg(target_os = "unix")]
+        #[cfg(any(target_os = "macos", target_os = "unix"))]
         "unix" => Ok(unix_listener::UnixListener::bind(addr).await?.boxed()),
         _ => {
             tracing::warn!("no support listener: {:?}", name);

examples/echo/src/echo/client.rs

@@ -31,7 +31,10 @@ impl Filter for FakeFilter {
 
 #[tokio::main]
 async fn main() {
-    let mut cli = EchoClient::connect("http://127.0.0.1:8888".to_string());
+    let builder = ClientBuilder::new()
+        .with_connector("unix")
+        .with_host("unix://127.0.0.1:8888");
+    let mut cli = EchoClient::build(builder);
     // let mut unary_cli = cli.clone().with_filter(FakeFilter {});
     // let mut cli = EchoClient::build(ClientBuilder::from_static("http://127.0.0.1:8888"));
     let resp = cli

examples/echo/src/echo/server.rs

@@ -80,9 +80,9 @@ async fn main() {
 
     // 3. 通过serverbuilder来初始化Server
     let builder = ServerBuilder::new()
-        .with_listener("tcp".to_string())
+        .with_listener("unix".to_string())
         .with_service_names(vec!["grpc.examples.echo.Echo".to_string()])
-        .with_addr("0.0.0.0:8888");
+        .with_addr("127.0.0.1:8888");
     builder.build().serve().await.unwrap();
 }