1
0
mirror of https://github.com/actix/examples synced 2024-11-24 06:43:00 +01:00
examples/shutdown-server/src/main.rs

52 lines
1.3 KiB
Rust
Raw Normal View History

use actix_web::{get, middleware, post, web, App, HttpResponse, HttpServer};
2020-01-31 13:21:39 +01:00
use std::{sync::mpsc, thread};
2020-01-26 00:49:19 +01:00
#[get("/hello")]
async fn hello() -> &'static str {
"Hello world!"
}
#[post("/stop")]
async fn stop(stopper: web::Data<mpsc::Sender<()>>) -> HttpResponse {
// make request that sends message through the Sender
stopper.send(()).unwrap();
HttpResponse::NoContent().finish()
}
2020-09-12 17:49:45 +02:00
#[actix_web::main]
2020-01-26 00:49:19 +01:00
async fn main() -> std::io::Result<()> {
std::env::set_var("RUST_LOG", "actix_server=debug,actix_web=debug");
env_logger::init();
// create a channel
let (tx, rx) = mpsc::channel::<()>();
2022-02-17 21:22:36 +01:00
let bind = ("127.0.0.1", 8080);
2020-01-26 00:49:19 +01:00
// start server as normal but don't .await after .run() yet
let server = HttpServer::new(move || {
// give the server a Sender in .data
App::new()
.app_data(web::Data::new(tx.clone()))
2020-01-26 00:49:19 +01:00
.wrap(middleware::Logger::default())
.service(hello)
.service(stop)
})
.bind(&bind)?
.run();
2022-03-06 01:15:16 +01:00
// clone the server handle
let srv = server.handle();
2020-01-26 00:49:19 +01:00
thread::spawn(move || {
// wait for shutdown signal
rx.recv().unwrap();
2022-03-06 01:15:16 +01:00
// send stop server gracefully command
srv.stop(true)
2020-01-26 00:49:19 +01:00
});
2022-03-06 01:15:16 +01:00
// run server until stopped (either by ctrl-c or stop endpoint)
2020-01-26 00:49:19 +01:00
server.await
}