WIP: start work on aquatic_http_private

aquatic_http_private/src/workers/socket/mod.rs

@@ -0,0 +1,67 @@
+mod db;
+mod routes;
+
+use std::net::{Ipv4Addr, SocketAddr, SocketAddrV4, TcpListener};
+
+use anyhow::Context;
+use axum::{routing::get, Extension, Router};
+use sqlx::mysql::MySqlPoolOptions;
+
+pub fn run_socket_worker() -> anyhow::Result<()> {
+    let addr = SocketAddr::V4(SocketAddrV4::new(Ipv4Addr::LOCALHOST, 3000));
+
+    let tcp_listener = create_tcp_listener(addr, false)?;
+
+    let runtime = tokio::runtime::Builder::new_current_thread()
+        .enable_all()
+        .build()?;
+
+    runtime.block_on(run_app(tcp_listener))?;
+
+    Ok(())
+}
+
+async fn run_app(tcp_listener: TcpListener) -> anyhow::Result<()> {
+    let db_url = ::std::env::var("DATABASE_URL").unwrap();
+
+    let pool = MySqlPoolOptions::new()
+        .max_connections(5)
+        .connect(&db_url)
+        .await?;
+
+    let app = Router::new()
+        .route("/:user_token/announce/", get(routes::announce))
+        .layer(Extension(pool));
+
+    axum::Server::from_tcp(tcp_listener)?
+        .serve(app.into_make_service())
+        .await?;
+
+    Ok(())
+}
+
+fn create_tcp_listener(addr: SocketAddr, only_ipv6: bool) -> anyhow::Result<TcpListener> {
+    let domain = if addr.is_ipv4() {
+        socket2::Domain::IPV4
+    } else {
+        socket2::Domain::IPV6
+    };
+
+    let socket = socket2::Socket::new(domain, socket2::Type::STREAM, Some(socket2::Protocol::TCP))?;
+
+    if only_ipv6 {
+        socket.set_only_v6(true).with_context(|| "set only_ipv6")?;
+    }
+
+    socket
+        .set_reuse_port(true)
+        .with_context(|| "set_reuse_port")?;
+    socket
+        .bind(&addr.into())
+        .with_context(|| format!("bind to {}", addr))?;
+    socket
+        .listen(1024)
+        .with_context(|| format!("listen on {}", addr))?;
+
+    Ok(socket.into())
+}