Initial commit

This commit is contained in:
dobiadi
2024-04-21 23:02:59 +02:00
commit 0428a21912
4 changed files with 99 additions and 0 deletions

1
.gitignore vendored Normal file
View File

@@ -0,0 +1 @@
/target

15
Cargo.toml Normal file
View File

@@ -0,0 +1,15 @@
[package]
name = "web-template"
version = "0.1.0"
edition = "2021"
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
[dependencies]
axum = "0"
libc = "0"
tokio = { version = "1", features = ["full"] }
tower = "0"
tower-http = { version = "0", features = ["trace", "timeout"] }
tracing = "0"
tracing-subscriber = { version = "0", features = ["json", "env-filter"] }

19
Dockerfile Normal file
View File

@@ -0,0 +1,19 @@
ARG ALPINE_VERSION=latest
FROM alpine:$ALPINE_VERSION as builder
COPY Cargo.lock /usr/src/
COPY Cargo.toml /usr/src/
COPY src/ /usr/src/src/
WORKDIR /usr/src
RUN apk add --no-cache rustup build-base && \
rustup-init -qy --profile=minimal && \
source "$HOME/.cargo/env" && \
cargo build --release --target=x86_64-unknown-linux-musl
FROM alpine:$ALPINE_VERSION
COPY --from=builder /usr/src/target/x86_64-unknown-linux-musl/release/web-template /usr/bin/
CMD [ "/usr/bin/web-template" ]

64
src/main.rs Normal file
View File

@@ -0,0 +1,64 @@
use std::time::Duration;
use axum::{routing, Router};
use tokio::{
net::TcpListener,
signal::{self, unix::SignalKind},
task::JoinSet,
};
use tower_http::{timeout::TimeoutLayer, trace::TraceLayer};
use tracing_subscriber::{layer::SubscriberExt, util::SubscriberInitExt};
#[tokio::main]
async fn main() -> Result<(), std::io::Error> {
tracing_subscriber::registry()
.with(tracing_subscriber::EnvFilter::from_env("LOG_LEVEL"))
.with(tracing_subscriber::fmt::layer().json())
.init();
let app = Router::new()
.route("/", routing::get(|| async { "Hello world!" }))
.layer((
TraceLayer::new_for_http(),
TimeoutLayer::new(Duration::from_secs(10)),
));
let listener = TcpListener::bind("0.0.0.0:8080").await.unwrap();
axum::serve(listener, app)
.with_graceful_shutdown(shutdown_signal())
.await
}
const SIGNALS: &'static [(i32, &'static str)] = &[
(libc::SIGTERM, "SIGTERM"),
(libc::SIGQUIT, "SIGQUIT"),
(libc::SIGINT, "SIGINT"),
];
async fn shutdown_signal() {
let term_signals = [
SignalKind::terminate(),
SignalKind::quit(),
SignalKind::interrupt(),
];
let mut futures = JoinSet::new();
for term_signal in term_signals {
futures.spawn(async move {
signal::unix::signal(term_signal)
.expect("failed to install signal handler")
.recv()
.await;
term_signal.as_raw_value()
});
}
let kind = futures.join_next().await;
if let Some(Ok(kind)) = kind {
let name = SIGNALS.iter().find(|e| e.0 == kind).unwrap().1;
tracing::info!("{} received, shutting down gracefully", name);
}
}