atuin/atuin-server/src/lib.rs

52 lines
1.1 KiB
Rust
Raw Normal View History

#![forbid(unsafe_code)]
2022-04-12 16:06:19 -06:00
use std::net::{IpAddr, SocketAddr};
2022-04-12 16:06:19 -06:00
use axum::Server;
use database::Postgres;
use eyre::{Context, Result};
use crate::settings::Settings;
use tokio::signal;
pub mod auth;
pub mod calendar;
pub mod database;
pub mod handlers;
pub mod models;
pub mod router;
pub mod settings;
pub mod utils;
async fn shutdown_signal() {
let terminate = async {
signal::unix::signal(signal::unix::SignalKind::terminate())
.expect("failed to register signal handler")
.recv()
.await;
};
tokio::select! {
_ = terminate => (),
}
eprintln!("Shutting down gracefully...");
}
2022-04-12 16:06:19 -06:00
pub async fn launch(settings: Settings, host: String, port: u16) -> Result<()> {
let host = host.parse::<IpAddr>()?;
let postgres = Postgres::new(settings.clone())
2022-04-12 16:06:19 -06:00
.await
.wrap_err_with(|| format!("failed to connect to db: {}", settings.db_uri))?;
2022-04-12 16:06:19 -06:00
let r = router::router(postgres, settings);
Server::bind(&SocketAddr::new(host, port))
.serve(r.into_make_service())
.with_graceful_shutdown(shutdown_signal())
2022-04-12 16:06:19 -06:00
.await?;
Ok(())
}