62 lines
1.6 KiB
Rust
62 lines
1.6 KiB
Rust
use std::net::SocketAddr;
|
|
|
|
use tracing_subscriber::{layer::SubscriberExt, util::SubscriberInitExt};
|
|
use what2watch::get_db_pool;
|
|
|
|
fn main() {
|
|
tracing_subscriber::registry()
|
|
.with(
|
|
tracing_subscriber::EnvFilter::try_from_default_env()
|
|
.unwrap_or_else(|_| "what2watch=debug,axum::routing=debug".into()),
|
|
)
|
|
.with(tracing_subscriber::fmt::layer())
|
|
.init();
|
|
|
|
//tracing_subscriber::fmt().with_target(false).pretty().init();
|
|
|
|
let pool = get_db_pool();
|
|
|
|
let rt = tokio::runtime::Builder::new_multi_thread()
|
|
.enable_all()
|
|
.build()
|
|
.unwrap();
|
|
|
|
let app = rt.block_on(what2watch::app(pool.clone()));
|
|
|
|
rt.block_on(async {
|
|
let addr: SocketAddr = ([0, 0, 0, 0], 3000).into();
|
|
tracing::debug!("binding to {addr:?}");
|
|
let listener = tokio::net::TcpListener::bind(&addr).await.unwrap();
|
|
axum::serve(listener, app)
|
|
.with_graceful_shutdown(graceful_shutdown())
|
|
.await
|
|
.unwrap()
|
|
});
|
|
|
|
rt.block_on(pool.close());
|
|
}
|
|
|
|
async fn graceful_shutdown() {
|
|
use tokio::signal;
|
|
let ctrl_c = async {
|
|
signal::ctrl_c()
|
|
.await
|
|
.expect("failed to install Ctrl+C handler");
|
|
};
|
|
|
|
#[cfg(unix)]
|
|
let terminate = async {
|
|
signal::unix::signal(signal::unix::SignalKind::terminate())
|
|
.expect("failed to install signal handler")
|
|
.recv()
|
|
.await;
|
|
};
|
|
|
|
#[cfg(not(unix))]
|
|
let terminate = std::future::pending::<()>();
|
|
|
|
tokio::select! {
|
|
_ = ctrl_c => {tracing::info!("shutting down")},
|
|
_ = terminate => {},
|
|
}
|
|
}
|