proxmox-rest-server: improve logging

And rename server_state_init() into catch_shutdown_and_reload_signals().

diff --git a/proxmox-rest-server/src/daemon.rs b/proxmox-rest-server/src/daemon.rs
index d96c18fd70b51c403c7096bd235dad0596ae21fa..f22e4cf1a087e5787dc8a8e2a518d3394b15b90b 100644 (file)
--- a/proxmox-rest-server/src/daemon.rs
+++ b/proxmox-rest-server/src/daemon.rs
@@ -152,9 +152,9 @@ impl Reloader {
                             self.do_reexec(new_args)
                         })
                         {
-                            Ok(Ok(())) => eprintln!("do_reexec returned!"),
-                            Ok(Err(err)) => eprintln!("do_reexec failed: {}", err),
-                            Err(_) => eprintln!("panic in re-exec"),
+                            Ok(Ok(())) => log::error!("do_reexec returned!"),
+                            Ok(Err(err)) => log::error!("do_reexec failed: {}", err),
+                            Err(_) => log::error!("panic in re-exec"),
                         }
                     }
                     Ok(ForkResult::Parent { child }) => {
@@ -307,7 +307,7 @@ where
         wait_service_is_not_state(service_name, "reloading").await?;
     }
 
-    log::info!("daemon shut down...");
+    log::info!("daemon shut down.");
     Ok(())
 }
 
@@ -360,6 +360,11 @@ pub enum SystemdNotify {
 
 /// Tells systemd the startup state of the service (see: ``man sd_notify``)
 pub fn systemd_notify(state: SystemdNotify) -> Result<(), Error> {
+
+    if let SystemdNotify::Ready = &state  {
+        log::info!("service is ready");
+    }
+
     let message = match state {
         SystemdNotify::Ready => CString::new("READY=1"),
         SystemdNotify::Reloading => CString::new("RELOADING=1"),
@@ -374,5 +379,6 @@ pub fn systemd_notify(state: SystemdNotify) -> Result<(), Error> {
             std::io::Error::from_raw_os_error(-rc),
         );
     }
+
     Ok(())
 }

diff --git a/proxmox-rest-server/src/lib.rs b/proxmox-rest-server/src/lib.rs
index 9c2dbb08e87eff068737c4c6b32e23abada6881b..5f1cefbb601a8d81bb4299008694bb5445364b53 100644 (file)
--- a/proxmox-rest-server/src/lib.rs
+++ b/proxmox-rest-server/src/lib.rs
@@ -96,7 +96,6 @@ pub fn our_ctrl_sock() -> String {
 static SHUTDOWN_REQUESTED: AtomicBool = AtomicBool::new(false);
 
 pub fn request_shutdown() {
-    println!("request_shutdown");
     SHUTDOWN_REQUESTED.store(true, Ordering::SeqCst);
     crate::server_shutdown();
 }

diff --git a/proxmox-rest-server/src/state.rs b/proxmox-rest-server/src/state.rs
index 468ef0aa2297a55c4bdc3b8f52a3b031443a972c..715bc1c95f3d4bb85694161c0adf1a44058cd601 100644 (file)
--- a/proxmox-rest-server/src/state.rs
+++ b/proxmox-rest-server/src/state.rs
@@ -34,13 +34,18 @@ lazy_static! {
     });
 }
 
-pub fn server_state_init() -> Result<(), Error> {
+/// Listen to ``SIGINT`` and ``SIGHUP`` signals
+///
+/// * ``SIGINT``: start server shutdown
+///
+/// * ``SIGHUP``: start server reload
+pub fn catch_shutdown_and_reload_signals() -> Result<(), Error> {
 
     let mut stream = signal(SignalKind::interrupt())?;
 
     let future = async move {
         while stream.recv().await.is_some() {
-            println!("got shutdown request (SIGINT)");
+            log::info!("got shutdown request (SIGINT)");
             SERVER_STATE.lock().unwrap().reload_request = false;
             crate::request_shutdown();
         }
@@ -55,7 +60,7 @@ pub fn server_state_init() -> Result<(), Error> {
 
     let future = async move {
         while stream.recv().await.is_some() {
-            println!("got reload request (SIGHUP)");
+            log::info!("got reload request (SIGHUP)");
             SERVER_STATE.lock().unwrap().reload_request = true;
             crate::request_shutdown();
         }
@@ -78,7 +83,7 @@ pub fn is_reload_request() -> bool {
 pub fn server_shutdown() {
     let mut data = SERVER_STATE.lock().unwrap();
 
-    println!("SET SHUTDOWN MODE");
+    log::info!("request_shutdown");
 
     data.mode = ServerMode::Shutdown;
 

diff --git a/src/bin/proxmox-backup-api.rs b/src/bin/proxmox-backup-api.rs
index 3474d1ca108141889d312fd4fb0bbacd16038718..38d075ee6334b968037d5d124a61981165905459 100644 (file)
--- a/src/bin/proxmox-backup-api.rs
+++ b/src/bin/proxmox-backup-api.rs
@@ -127,7 +127,7 @@ async fn run() -> Result<(), Error> {
     let init_result: Result<(), Error> = try_block!({
         proxmox_rest_server::register_task_control_commands(&mut commando_sock)?;
         commando_sock.spawn()?;
-        proxmox_rest_server::server_state_init()?;
+        proxmox_rest_server::catch_shutdown_and_reload_signals()?;
         Ok(())
     });
 

diff --git a/src/bin/proxmox-backup-proxy.rs b/src/bin/proxmox-backup-proxy.rs
index 4d9765f777af699a7e6cabc605a2f2459082d3fa..59694486cb257c04de3aec13d9ae11ce5c7068ba 100644 (file)
--- a/src/bin/proxmox-backup-proxy.rs
+++ b/src/bin/proxmox-backup-proxy.rs
@@ -270,7 +270,7 @@ async fn run() -> Result<(), Error> {
     let init_result: Result<(), Error> = try_block!({
         proxmox_rest_server::register_task_control_commands(&mut commando_sock)?;
         commando_sock.spawn()?;
-        proxmox_rest_server::server_state_init()?;
+        proxmox_rest_server::catch_shutdown_and_reload_signals()?;
         Ok(())
     });
 

diff --git a/tests/worker-task-abort.rs b/tests/worker-task-abort.rs
index 8502aca9a42c0003299a5c1c1410470158b483e6..75c226d382802aa1bb2d6aa340871d5c20d0f116 100644 (file)
--- a/tests/worker-task-abort.rs
+++ b/tests/worker-task-abort.rs
@@ -53,7 +53,7 @@ fn worker_task_abort() -> Result<(), Error> {
 
         let init_result: Result<(), Error> = try_block!({
             proxmox_rest_server::register_task_control_commands(&mut commando_sock)?;
-            proxmox_rest_server::server_state_init()?;
+            proxmox_rest_server::catch_shutdown_and_reload_signals()?;
             Ok(())
         });