support listening on uds

Author: niklasf

Cargo.lock

@@ -9,7 +9,7 @@ keywords = ["ip", "proxy", "geolocation", "lichess"]
 edition = "2021"
 
 [dependencies]
-axum = "0.8.0-rc.1"
+axum = "0.8"
 tokio = { version = "1", features = ["full"] }
 serde =  { version = "1", features = ["derive"] }
 serde_with = "3"

Cargo.toml

@@ -10,7 +10,7 @@ use listenfd::ListenFd;
 use serde::{Deserialize, Serialize};
 use serde_with::{formats::CommaSeparator, serde_as, StringWithSeparator};
 use tikv_jemallocator::Jemalloc;
-use tokio::net::TcpListener;
+use tokio::net::{TcpListener, UnixListener};
 
 #[global_allocator]
 static GLOBAL: Jemalloc = Jemalloc;
@@ -93,16 +93,17 @@ async fn main() {
         .route("/batch", get(move |query| batch_query(db, query)))
         .route("/status", get(move || status(db)));
 
-    let listener = match ListenFd::from_env()
-        .take_tcp_listener(0)
-        .expect("tcp listener")
-    {
-        Some(std_listener) => {
-            std_listener.set_nonblocking(true).expect("set nonblocking");
-            TcpListener::from_std(std_listener).expect("listener")
-        }
-        None => TcpListener::bind(&opt.bind).await.expect("bind"),
-    };
-
-    axum::serve(listener, app).await.expect("serve");
+    let mut fds = ListenFd::from_env();
+    if let Ok(Some(uds)) = fds.take_unix_listener(0) {
+        uds.set_nonblocking(true).expect("set nonblocking");
+        let listener = UnixListener::from_std(uds).expect("listener");
+        axum::serve(listener, app).await.expect("serve");
+    } else if let Ok(Some(tcp)) = fds.take_tcp_listener(0) {
+        tcp.set_nonblocking(true).expect("set nonblocking");
+        let listener = TcpListener::from_std(tcp).expect("listener");
+        axum::serve(listener, app).await.expect("serve");
+    } else {
+        let listener = TcpListener::bind(&opt.bind).await.expect("bind");
+        axum::serve(listener, app).await.expect("serve");
+    }
 }