Reduce blocking of threads when saving state

Wiebe Cazemier
1 parent cd429cb7
Showing 3 changed files with 34 additions and 2 deletions
mainapp.cpp
mainapp.h
subscriptionstore.cpp
@@ -210,7 +210,7 @@ MainApp::MainApp(const std::string &amp;configFilePath) :
         subscriptionStore->loadSessionsAndSubscriptions(settings->getSessionsDBFile());
     }
-    auto fSaveState = std::bind(&MainApp::saveState, this);
+    auto fSaveState = std::bind(&MainApp::saveStateInThread, this);
     timer.addCallback(fSaveState, 900000, "Save state.");
 }
@@ -355,8 +355,31 @@ void MainApp::queuePublishStatsOnDollarTopic()
     }
 }
+/**
+ * @brief MainApp::saveStateInThread starts a thread for disk IO, because file IO is not async.
+ */
+void MainApp::saveStateInThread()
+{
+    // Prevent queueing it again when it's still running.
+    std::unique_lock<std::mutex> locker(saveStateMutex, std::try_to_lock);
+    if (!locker.owns_lock())
+        return;
+
+    // Join previous instances.
+    if (saveStateThread.joinable())
+        saveStateThread.join();
+
+    auto f = std::bind(&MainApp::saveState, this);
+    saveStateThread = std::thread(f);
+
+    pthread_t native = saveStateThread.native_handle();
+    pthread_setname_np(native, "SaveState");
+}
+
 void MainApp::saveState()
 {
+    std::lock_guard<std::mutex> lg(saveStateMutex);
+
     try
     {
         if (!settings->storageDir.empty())
@@ -366,6 +389,8 @@ void MainApp::saveState()
             const std::string sessionsDBPath = settings->getSessionsDBFile();
             subscriptionStore->saveSessionsAndSubscriptions(sessionsDBPath);
+
+            logger->logf(LOG_INFO, "Saving states done");
         }
     }
     catch(std::exception &ex)
@@ -669,6 +694,9 @@ void MainApp::start()
     }
     saveState();
+
+    if (saveStateThread.joinable())
+        saveStateThread.join();
 }
 void MainApp::quit()
@@ -71,6 +71,9 @@ class MainApp
     Logger *logger = Logger::getInstance();
+    std::thread saveStateThread;
+    std::mutex saveStateMutex;
+
     void setlimits();
     void loadConfig();
     void reloadConfig();
@@ -84,6 +87,7 @@ class MainApp
     void setFuzzFile(const std::string &fuzzFilePath);
     void queuePublishStatsOnDollarTopic();
     void saveState();
+    void saveStateInThread();
     MainApp(const std::string &configFilePath);
 public:
@@ -634,7 +634,7 @@ void SubscriptionStore::saveSessionsAndSubscriptions(const std::string &amp;filePath
     logger->logf(LOG_INFO, "Saving sessions and subscriptions to '%s'", filePath.c_str());
     RWLockGuard lock_guard(&subscriptionsRwlock);
-    lock_guard.wrlock();
+    lock_guard.rdlock();
     // First copy the sessions...