]> git.proxmox.com Git - proxmox-backup.git/blob - src/bin/proxmox-backup-api.rs
proxy/api: await running workers before shutdown
[proxmox-backup.git] / src / bin / proxmox-backup-api.rs
1 use failure::*;
2 use futures::*;
3
4 use proxmox::tools::try_block;
5 use proxmox::api::RpcEnvironmentType;
6
7 //use proxmox_backup::tools;
8 //use proxmox_backup::api_schema::config::*;
9 use proxmox_backup::server::rest::*;
10 use proxmox_backup::server;
11 use proxmox_backup::tools::daemon;
12 use proxmox_backup::auth_helpers::*;
13 use proxmox_backup::config;
14 use proxmox_backup::buildcfg;
15
16 #[tokio::main]
17 async fn main() {
18 if let Err(err) = run().await {
19 eprintln!("Error: {}", err);
20 std::process::exit(-1);
21 }
22 }
23
24 async fn run() -> Result<(), Error> {
25 if let Err(err) = syslog::init(
26 syslog::Facility::LOG_DAEMON,
27 log::LevelFilter::Info,
28 Some("proxmox-backup-api")) {
29 bail!("unable to inititialize syslog - {}", err);
30 }
31
32 server::create_task_log_dirs()?;
33
34 config::create_configdir()?;
35
36 if let Err(err) = generate_auth_key() {
37 bail!("unable to generate auth key - {}", err);
38 }
39 let _ = private_auth_key(); // load with lazy_static
40
41 if let Err(err) = generate_csrf_key() {
42 bail!("unable to generate csrf key - {}", err);
43 }
44 let _ = csrf_secret(); // load with lazy_static
45
46 let config = server::ApiConfig::new(
47 buildcfg::JS_DIR, &proxmox_backup::api2::ROUTER, RpcEnvironmentType::PRIVILEGED);
48
49 let rest_server = RestServer::new(config);
50
51 // http server future:
52 let server = daemon::create_daemon(
53 ([127,0,0,1], 82).into(),
54 move |listener, ready| {
55 let incoming = proxmox_backup::tools::async_io::StaticIncoming::from(listener);
56 Ok(ready
57 .and_then(|_| hyper::Server::builder(incoming)
58 .serve(rest_server)
59 .with_graceful_shutdown(server::shutdown_future())
60 .map_err(Error::from)
61 )
62 .map(|e| {
63 if let Err(e) = e {
64 eprintln!("server error: {}", e);
65 }
66 })
67 )
68 },
69 );
70
71 daemon::systemd_notify(daemon::SystemdNotify::Ready)?;
72
73 let init_result: Result<(), Error> = try_block!({
74 server::create_task_control_socket()?;
75 server::server_state_init()?;
76 Ok(())
77 });
78
79 if let Err(err) = init_result {
80 bail!("unable to start daemon - {}", err);
81 }
82
83 server.await?;
84 log::info!("server shutting down, waiting for active workers to complete");
85 proxmox_backup::server::last_worker_future().await?;
86
87 log::info!("done - exit server");
88
89 Ok(())
90 }