Shutdown callback for service interfaces

Author: rovo89

libxbot-service-interface/include/xbot-service-interface/XbotServiceInterface.hpp

@@ -5,6 +5,8 @@
 #ifndef XBOT_FRAMEWORK_XBOTSERVICEINTERFACE_HPP
 #define XBOT_FRAMEWORK_XBOTSERVICEINTERFACE_HPP
 
+#include <functional>
+
 #include <xbot-service-interface/ServiceDiscovery.hpp>
 #include <xbot-service-interface/ServiceIO.hpp>
 
@@ -15,6 +17,14 @@ struct Context {
   void *ctx = nullptr;
 };
 
+using ShutdownCallback = std::function<void()>;
+
+/**
+ * Register a callback invoked after Stop() when SIGINT/SIGTERM is received.
+ * Must be called before Start() if register_signal_handlers is true.
+ */
+void SetShutdownCallback(ShutdownCallback callback);
+
 /**
  * Call this method to start xbot_framework.
  *

libxbot-service-interface/src/XbotServiceInterface.cpp

@@ -26,10 +26,18 @@ hub::CrowToSpeedlogHandler logger{};
 std::unique_ptr<PlotJugglerBridge> pjb = nullptr;
 std::unique_ptr<crow::SimpleApp> crow_app = nullptr;
 
+ShutdownCallback shutdown_callback{};
+
 void SignalHandler(int signal) {
   Stop();
+  if (shutdown_callback) {
+    shutdown_callback();
+  }
+}
+
+void xbot::serviceif::SetShutdownCallback(ShutdownCallback callback) {
+  shutdown_callback = std::move(callback);
 }
-struct sigaction act;
 
 xbot::serviceif::Context xbot::serviceif::Start(bool register_handlers, std::string bind_ip, bool start_rest_api) {
   std::unique_lock lk{mtx};
@@ -104,7 +112,9 @@ xbot::serviceif::Context xbot::serviceif::Start(bool register_handlers, std::str
 
   if (register_handlers) {
     // Register own signal handlers
+    struct sigaction act {};
     act.sa_handler = SignalHandler;
+    sigemptyset(&act.sa_mask);
     sigaction(SIGINT, &act, nullptr);
     sigaction(SIGTERM, &act, nullptr);
   }