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

53 lines
1.3 KiB
Rust
Raw Normal View History

use actix_web::{get, middleware, post, web, App, HttpResponse, HttpServer};
2020-01-26 00:49:19 +01:00
use futures::executor;
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::<()>();
let bind = "127.0.0.1:8080";
// 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();
// 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();
// stop server gracefully
executor::block_on(srv.stop(true))
});
// run server
server.await
}