server: use generalized commando socket for worker tasks commands

[proxmox-backup.git] / src / bin / proxmox-backup-api.rs

diff --git a/src/bin/proxmox-backup-api.rs b/src/bin/proxmox-backup-api.rs
index 7c4ccfd9c25b07dc85f7680a174ff529e19797b1..934562574e4926757486e225774d1947220a9369 100644 (file)
--- a/src/bin/proxmox-backup-api.rs
+++ b/src/bin/proxmox-backup-api.rs
@@ -1,30 +1,28 @@
+use anyhow::{bail, Error};
+use futures::*;
+
+use proxmox::try_block;
+use proxmox::api::RpcEnvironmentType;
+
 //use proxmox_backup::tools;
-use proxmox_backup::api_schema::router::*;
-use proxmox_backup::api_schema::config::*;
+//use proxmox_backup::api_schema::config::*;
 use proxmox_backup::server::rest::*;
 use proxmox_backup::server;
 use proxmox_backup::tools::daemon;
 use proxmox_backup::auth_helpers::*;
 use proxmox_backup::config;
-
-use failure::*;
-use lazy_static::lazy_static;
-use proxmox::tools::try_block;
-
-use futures::*;
-use futures::future::Future;
-
-use hyper;
+use proxmox_backup::buildcfg;
 
 fn main() {
+    proxmox_backup::tools::setup_safe_path_env();
 
-    if let Err(err) = run() {
+    if let Err(err) = proxmox_backup::tools::runtime::main(run()) {
         eprintln!("Error: {}", err);
         std::process::exit(-1);
     }
 }
 
-fn run() -> Result<(), Error> {
+async fn run() -> Result<(), Error> {
     if let Err(err) = syslog::init(
         syslog::Facility::LOG_DAEMON,
         log::LevelFilter::Info,
@@ -36,6 +34,11 @@ fn run() -> Result<(), Error> {
 
     config::create_configdir()?;
 
+    config::update_self_signed_cert(false)?;
+
+    proxmox_backup::rrd::create_rrdb_dir()?;
+    proxmox_backup::server::jobstate::create_jobstate_dir()?;
+
     if let Err(err) = generate_auth_key() {
         bail!("unable to generate auth key - {}", err);
     }
@@ -46,48 +49,53 @@ fn run() -> Result<(), Error> {
     }
     let _ = csrf_secret(); // load with lazy_static
 
-    lazy_static!{
-       static ref ROUTER: Router = proxmox_backup::api2::router();
-    }
+    let mut config = server::ApiConfig::new(
+        buildcfg::JS_DIR, &proxmox_backup::api2::ROUTER, RpcEnvironmentType::PRIVILEGED)?;
+
+    let mut commando_sock = server::CommandoSocket::new(server::our_ctrl_sock());
 
-    let config = ApiConfig::new(
-        env!("PROXMOX_JSDIR"), &ROUTER, RpcEnvironmentType::PRIVILEGED);
+    config.enable_file_log(buildcfg::API_ACCESS_LOG_FN)?;
 
     let rest_server = RestServer::new(config);
 
     // http server future:
     let server = daemon::create_daemon(
         ([127,0,0,1], 82).into(),
-        |listener| {
-            Ok(hyper::Server::builder(listener.incoming())
-               .serve(rest_server)
-               .with_graceful_shutdown(server::shutdown_future())
-               .map_err(|e| eprintln!("server error: {}", e))
+        move |listener, ready| {
+            let incoming = proxmox_backup::tools::async_io::StaticIncoming::from(listener);
+            Ok(ready
+                .and_then(|_| hyper::Server::builder(incoming)
+                    .serve(rest_server)
+                    .with_graceful_shutdown(server::shutdown_future())
+                    .map_err(Error::from)
+                )
+                .map(|e| {
+                    if let Err(e) = e {
+                        eprintln!("server error: {}", e);
+                    }
+                })
             )
         },
-    )?;
+    );
 
     daemon::systemd_notify(daemon::SystemdNotify::Ready)?;
 
-    tokio::run(lazy(||  {
+    let init_result: Result<(), Error> = try_block!({
+        server::register_task_control_commands(&mut commando_sock)?;
+        commando_sock.spawn()?;
+        server::server_state_init()?;
+        Ok(())
+    });
 
-        let init_result: Result<(), Error> = try_block!({
-            server::create_task_control_socket()?;
-            server::server_state_init()?;
-            Ok(())
-        });
+    if let Err(err) = init_result {
+        bail!("unable to start daemon - {}", err);
+    }
 
-        if let Err(err) = init_result {
-            eprintln!("unable to start daemon - {}", err);
-        } else {
-            tokio::spawn(server.then(|_| {
-                log::info!("done - exit server");
-                Ok(())
-            }));
-        }
+    server.await?;
+    log::info!("server shutting down, waiting for active workers to complete");
+    proxmox_backup::server::last_worker_future().await?;
 
-        Ok(())
-    }));
+    log::info!("done - exit server");
 
     Ok(())
 }