Quit threads by using task queue

This fixes the 100ms wait time requirement.

Quit threads by using task queue
This fixes the 100ms wait time requirement.
Wiebe Cazemier
1 parent b724f979
Showing 5 changed files with 41 additions and 3 deletions
authplugin.cpp
authplugin.h
mainapp.cpp
threaddata.cpp
threaddata.h
@@ -97,6 +97,9 @@ void AuthPlugin::init()
     if (settings.authPluginSerializeInit)
         lock.lock();
+    if (quitting)
+        return;
+
     AuthOptCompatWrap &authOpts = settings.getAuthOptsCompat();
     int result = init_v2(&pluginData, authOpts.head(), authOpts.size());
     if (result != 0)
@@ -125,6 +128,9 @@ void AuthPlugin::securityInit(bool reloading)
     if (settings.authPluginSerializeInit)
         lock.lock();
+    if (quitting)
+        return;
+
     AuthOptCompatWrap &authOpts = settings.getAuthOptsCompat();
     int result = security_init_v2(pluginData, authOpts.head(), authOpts.size(), reloading);
     if (result != 0)
@@ -201,6 +207,11 @@ AuthResult AuthPlugin::unPwdCheck(const std::string &amp;username, const std::string
     return r;
 }
+void AuthPlugin::setQuitting()
+{
+    this->quitting = true;
+}
+
 std::string AuthResultToString(AuthResult r)
 {
     {
@@ -63,6 +63,7 @@ class AuthPlugin
     Logger *logger = nullptr;
     bool initialized = false;
     bool wanted = false;
+    bool quitting = false;
     void *loadSymbol(void *handle, const char *symbol) const;
 public:
@@ -79,6 +80,8 @@ public:
     AuthResult aclCheck(const std::string &clientid, const std::string &username, const std::string &topic, AclAccess access);
     AuthResult unPwdCheck(const std::string &username, const std::string &password);
+    void setQuitting();
+
 };
 #endif // AUTHPLUGIN_H
@@ -440,7 +440,12 @@ void MainApp::start()
     for(std::shared_ptr<ThreadData> &thread : threads)
     {
-        thread->quit();
+        thread->queueQuit();
+    }
+
+    for(std::shared_ptr<ThreadData> &thread : threads)
+    {
+        thread->waitForQuit();
     }
     for(auto pair : listenerMap)
@@ -52,7 +52,6 @@ void ThreadData::start(thread_f f)
 void ThreadData::quit()
 {
     running = false;
-    thread.join();
 }
 void ThreadData::giveClient(Client_p client)
@@ -109,6 +108,23 @@ void ThreadData::queueDoKeepAliveCheck()
     wakeUpThread();
 }
+void ThreadData::queueQuit()
+{
+    std::lock_guard<std::mutex> locker(taskQueueMutex);
+
+    auto f = std::bind(&ThreadData::quit, this);
+    taskQueue.push_front(f);
+
+    authPlugin.setQuitting();
+
+    wakeUpThread();
+}
+
+void ThreadData::waitForQuit()
+{
+    thread.join();
+}
+
 // TODO: profile how fast hash iteration is. Perhaps having a second list/vector is beneficial?
 void ThreadData::doKeepAliveCheck()
 {
@@ -33,6 +33,7 @@ class ThreadData
     void reload(std::shared_ptr<Settings> settings);
     void wakeUpThread();
     void doKeepAliveCheck();
+    void quit();
 public:
     Settings settingsLocalCopy; // Is updated on reload, within the thread loop.
@@ -50,7 +51,7 @@ public:
     ThreadData(ThreadData &&other) = delete;
     void start(thread_f f);
-    void quit();
+
     void giveClient(Client_p client);
     Client_p getClient(int fd);
     void removeClient(Client_p client);
@@ -60,6 +61,8 @@ public:
     void initAuthPlugin();
     void queueReload(std::shared_ptr<Settings> settings);
     void queueDoKeepAliveCheck();
+    void queueQuit();
+    void waitForQuit();
 };