Merge pull request #22 from wiktor-k/reintroduce-service-binding

Add `Agent::bind` to accept `service_binding::Listener`s
wiktor-k · Mar 15, 2024 · e494afa · e494afa
2 parents 2edc1fb + 79a5a28
commit e494afa
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 0 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -24,6 +24,7 @@ futures = { version = "0.3.30", optional = true }
 log = { version = "0.4.6", optional = true }
 tokio = { version = "1", optional = true, features = ["rt", "net"] }
 tokio-util = { version = "0.7.1", optional = true, features = ["codec"] }
+service-binding = "1.1.0"
 
 [features]
 default = ["agent"]

diff --git a/src/agent.rs b/src/agent.rs
@@ -132,6 +132,16 @@ pub trait Agent: 'static + Sync + Send + Sized {
             }
         }
     }
+    async fn bind(mut self, listener: service_binding::Listener) -> Result<(), AgentError> {
+        match listener {
+            service_binding::Listener::Unix(listener) => {
+                self.listen(UnixListener::from_std(listener)?).await
+            }
+            service_binding::Listener::Tcp(listener) => {
+                self.listen(TcpListener::from_std(listener)?).await
+            }
+        }
+    }
 }
 
 impl<T> Agent for T