mirror of https://github.com/fafhrd91/actix-net
50 lines
1.5 KiB
Rust
50 lines
1.5 KiB
Rust
//! Demonstrates use of the `ServerBuilder::shutdown_signal` method using `tokio-util`s
|
|
//! `CancellationToken` helper using a nonsensical timer. In practice, this cancellation token would
|
|
//! be wired throughout your application and typically triggered by OS signals elsewhere.
|
|
|
|
use std::{io, time::Duration};
|
|
|
|
use actix_rt::net::TcpStream;
|
|
use actix_server::Server;
|
|
use actix_service::fn_service;
|
|
use tokio_util::sync::CancellationToken;
|
|
|
|
async fn run(stop_signal: CancellationToken) -> io::Result<()> {
|
|
pretty_env_logger::formatted_timed_builder()
|
|
.parse_env(pretty_env_logger::env_logger::Env::default().default_filter_or("info"));
|
|
|
|
let addr = ("127.0.0.1", 8080);
|
|
tracing::info!("starting server on port: {}", &addr.0);
|
|
|
|
// let (tx, rx) = tokio::sync::oneshot::channel::<()>();
|
|
// let (tx, mut rx) = tokio::sync::broadcast::channel::<()>(1);
|
|
|
|
Server::build()
|
|
.bind("shutdown-signal", addr, || {
|
|
fn_service(|_stream: TcpStream| async { Ok::<_, io::Error>(()) })
|
|
})?
|
|
.shutdown_signal(stop_signal.cancelled_owned())
|
|
// .shutdown_signal(async move {
|
|
// rx.await;
|
|
// // rx.recv().await;
|
|
// })
|
|
.run()
|
|
.await
|
|
}
|
|
|
|
#[tokio::main]
|
|
async fn main() -> io::Result<()> {
|
|
let stop_signal = CancellationToken::new();
|
|
|
|
tokio::spawn({
|
|
let stop_signal = stop_signal.clone();
|
|
async move {
|
|
tokio::time::sleep(Duration::from_secs(10)).await;
|
|
stop_signal.cancel();
|
|
}
|
|
});
|
|
|
|
run(stop_signal).await?;
|
|
Ok(())
|
|
}
|