1
0
mirror of https://github.com/actix/examples synced 2025-06-29 02:10:36 +02:00

enhanced, renamed shutdown example and modified async_pg readme

This commit is contained in:
dowwie
2020-01-31 07:10:56 -05:00
parent 78a753a06a
commit da72aef69c
7 changed files with 42 additions and 13 deletions

View File

@ -49,6 +49,7 @@ mod errors {
fn error_response(&self) -> HttpResponse {
match *self {
MyError::NotFound => HttpResponse::NotFound().finish(),
MyError::PoolError(ref err) => HttpResponse::InternalServerError().body(err.to_string()),
_ => HttpResponse::InternalServerError().finish(),
}
}
@ -97,7 +98,7 @@ mod handlers {
let client: Client =
db_pool.get().await.map_err(|err| MyError::PoolError(err))?;
let new_user = db::add_user(&client, user_info).await?;
Ok(HttpResponse::Ok().json(new_user))
@ -108,6 +109,8 @@ use actix_web::{web, App, HttpServer};
use dotenv::dotenv;
use handlers::add_user;
use tokio_postgres::NoTls;
use tokio::signal::unix::{signal, SignalKind};
#[actix_rt::main]
async fn main() -> std::io::Result<()> {
@ -125,5 +128,15 @@ async fn main() -> std::io::Result<()> {
.run();
println!("Server running at http://{}/", config.server_addr);
let srv = server.clone();
let mut stream = signal(SignalKind::interrupt())?;
actix_rt::spawn(async move {
loop {
stream.recv().await;
println!("\nSIGINT Received. Stopping server.\n");
srv.stop(true).await;
}
});
server.await
}