support creating TcpListener from std/tokio listeners or socket2 socket

closes #405

Cargo.toml

@@ -71,6 +71,7 @@ percent-encoding = "2.1"
 pin-project-lite = "0.2.13"
 rustls-pki-types = "^1"
 proc-macro2 = "1.0"
+socket2 = "0.5.8"
 opentelemetry = { version = "0.27.0", default-features = false, features = [
     "trace",
 ] }

rama-net/Cargo.toml

@@ -42,6 +42,7 @@ rama-utils = { version = "0.2.0-alpha.7", path = "../rama-utils" }
 rustls = { workspace = true, optional = true }
 serde = { workspace = true, features = ["derive"] }
 sha2 = { workspace = true, optional = true }
+socket2 = { workspace = true }
 tokio = { workspace = true, features = ["macros", "fs", "io-std", "io-util", "net"] }
 tracing = { workspace = true }
 venndb = { workspace = true, optional = true }

rama-net/src/lib.rs

@@ -38,3 +38,6 @@ pub mod tls;
 
 #[cfg(any(feature = "tls", feature = "http"))]
 pub mod fingerprint;
+
+#[cfg(any(windows, unix))]
+pub use ::socket2 as socket;

rama-tcp/src/server/listener.rs

@@ -194,6 +194,49 @@ impl<S> TcpListener<S> {
     }
 }
 
+impl From<TokioTcpListener> for TcpListener<()> {
+    fn from(value: TokioTcpListener) -> Self {
+        Self {
+            inner: value,
+            state: (),
+        }
+    }
+}
+
+#[cfg(any(windows, unix))]
+impl TryFrom<rama_net::socket::Socket> for TcpListener<()> {
+    type Error = std::io::Error;
+
+    #[inline]
+    fn try_from(value: rama_net::socket::Socket) -> Result<Self, Self::Error> {
+        let listener = std::net::TcpListener::from(value);
+        listener.try_into()
+    }
+}
+
+impl TryFrom<std::net::TcpListener> for TcpListener<()> {
+    type Error = std::io::Error;
+
+    fn try_from(value: std::net::TcpListener) -> Result<Self, Self::Error> {
+        value.set_nonblocking(true)?;
+        Ok(Self {
+            inner: TokioTcpListener::from_std(value)?,
+            state: (),
+        })
+    }
+}
+
+impl TcpListener<()> {
+    /// Define the TcpListener's state after it was created,
+    /// useful in case it wasn't built using the builder.
+    pub fn with_state<S>(self, state: S) -> TcpListener<S> {
+        TcpListener {
+            inner: self.inner,
+            state,
+        }
+    }
+}
+
 impl<State> TcpListener<State>
 where
     State: Clone + Send + Sync + 'static,