107 lines
2.7 KiB
Rust
107 lines
2.7 KiB
Rust
|
use std::{net::SocketAddr, ops::Deref, sync::Arc};
|
||
|
|
|
||
|
|
use anyhow::Context;
|
||
|
|
use axum::extract::MatchedPath;
|
||
|
|
use axum::http::Request;
|
||
|
|
use axum::Router;
|
||
|
|
use axum::{extract::FromRef, routing::get};
|
||
|
|
use clap::{Parser, Subcommand};
|
||
|
|
use sqlx::{Pool, Postgres};
|
||
|
|
use tower_http::trace::TraceLayer;
|
||
|
|
|
||
|
|
#[derive(Parser)]
|
||
|
|
#[command(author, version, about, long_about = None, subcommand_required = true)]
|
||
|
|
struct Command {
|
||
|
|
#[command(subcommand)]
|
||
|
|
command: Option<Commands>,
|
||
|
|
}
|
||
|
|
|
||
|
|
#[derive(Subcommand)]
|
||
|
|
enum Commands {
|
||
|
|
Serve {
|
||
|
|
#[arg(env = "SERVICE_HOST", long, default_value = "127.0.0.1:3000")]
|
||
|
|
host: SocketAddr,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
#[tokio::main]
|
||
|
|
async fn main() -> anyhow::Result<()> {
|
||
|
|
dotenv::dotenv().ok();
|
||
|
|
tracing_subscriber::fmt::init();
|
||
|
|
|
||
|
|
let cli = Command::parse();
|
||
|
|
|
||
|
|
if let Some(Commands::Serve { host }) = cli.command {
|
||
|
|
tracing::info!("Starting service");
|
||
|
|
|
||
|
|
let state = SharedState(Arc::new(State::new().await?));
|
||
|
|
|
||
|
|
let app = Router::new()
|
||
|
|
.route("/", get(root))
|
||
|
|
.with_state(state.clone())
|
||
|
|
.layer(
|
||
|
|
TraceLayer::new_for_http().make_span_with(|request: &Request<_>| {
|
||
|
|
// Log the matched route's path (with placeholders not filled in).
|
||
|
|
// Use request.uri() or OriginalUri if you want the real path.
|
||
|
|
let matched_path = request
|
||
|
|
.extensions()
|
||
|
|
.get::<MatchedPath>()
|
||
|
|
.map(MatchedPath::as_str);
|
||
|
|
|
||
|
|
tracing::info_span!(
|
||
|
|
"http_request",
|
||
|
|
method = ?request.method(),
|
||
|
|
matched_path,
|
||
|
|
some_other_field = tracing::field::Empty,
|
||
|
|
)
|
||
|
|
}), // ...
|
||
|
|
);
|
||
|
|
|
||
|
|
tracing::info!("listening on {}", host);
|
||
|
|
let listener = tokio::net::TcpListener::bind(host).await.unwrap();
|
||
|
|
axum::serve(listener, app.into_make_service())
|
||
|
|
.await
|
||
|
|
.unwrap();
|
||
|
|
}
|
||
|
|
|
||
|
|
Ok(())
|
||
|
|
}
|
||
|
|
|
||
|
|
async fn root() -> &'static str {
|
||
|
|
"Hello, %%name%%!"
|
||
|
|
}
|
||
|
|
|
||
|
|
#[derive(Clone)]
|
||
|
|
pub struct SharedState(Arc<State>);
|
||
|
|
|
||
|
|
impl Deref for SharedState {
|
||
|
|
type Target = Arc<State>;
|
||
|
|
|
||
|
|
fn deref(&self) -> &Self::Target {
|
||
|
|
&self.0
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
pub struct State {
|
||
|
|
pub db: Pool<Postgres>,
|
||
|
|
}
|
||
|
|
|
||
|
|
impl State {
|
||
|
|
pub async fn new() -> anyhow::Result<Self> {
|
||
|
|
let db = sqlx::PgPool::connect(
|
||
|
|
&std::env::var("DATABASE_URL").context("DATABASE_URL is not set")?,
|
||
|
|
)
|
||
|
|
.await?;
|
||
|
|
|
||
|
|
sqlx::migrate!("migrations/crdb")
|
||
|
|
.set_locking(false)
|
||
|
|
.run(&db)
|
||
|
|
.await?;
|
||
|
|
|
||
|
|
let _ = sqlx::query("SELECT 1;").fetch_one(&db).await?;
|
||
|
|
|
||
|
|
Ok(Self { db })
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|