Even better waits

Wolkabout · Feb 24, 2022 · eb46ba2 · eb46ba2
1 parent ded5265
commit eb46ba2
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 14 deletions.
diff --git a/core/utilities/Buffer.h b/core/utilities/Buffer.h
@@ -62,7 +62,10 @@ template <class T> Buffer<T>::~Buffer()
 
 template <class T> void Buffer<T>::stop()
 {
-    m_exitCondition = true;
+    {
+        std::lock_guard<std::mutex> lg{m_pushLock};
+        m_exitCondition = true;
+    }
     m_condition.notify_one();
 }
 
@@ -92,7 +95,7 @@ template <class T> T Buffer<T>::pop()
 
     if (m_popQueue.empty())
     {
-        LOG(ERROR) << "Poping from empty buffer";
+        LOG(ERROR) << "Popping from empty buffer";
         return {};
     }
 
@@ -106,10 +109,8 @@ template <class T> void Buffer<T>::swapBuffers()
 {
     std::unique_lock<std::mutex> pushGuard(m_pushLock);
 
-    if (m_pushQueue.empty() && !m_exitCondition)
-    {
+    if (m_pushQueue.empty())
         m_condition.wait(pushGuard, [&] { return !m_pushQueue.empty() || m_exitCondition; });
-    }
 
     std::lock_guard<std::mutex> popGuard{m_popLock};
     std::swap(m_pushQueue, m_popQueue);

diff --git a/core/utilities/CommandBuffer.cpp b/core/utilities/CommandBuffer.cpp
@@ -16,8 +16,6 @@
 
 #include "CommandBuffer.h"
 
-#include <atomic>
-#include <condition_variable>
 #include <functional>
 #include <memory>
 #include <mutex>
@@ -36,7 +34,7 @@ CommandBuffer::~CommandBuffer()
     stop();
 }
 
-void CommandBuffer::pushCommand(std::shared_ptr<Command> command)
+void CommandBuffer::pushCommand(const std::shared_ptr<Command>& command)
 {
     std::unique_lock<std::mutex> unique_lock(m_lock);
 
@@ -47,7 +45,10 @@ void CommandBuffer::pushCommand(std::shared_ptr<Command> command)
 
 void CommandBuffer::stop()
 {
-    m_isRunning = false;
+    {
+        std::lock_guard<std::mutex> lg{m_lock};
+        m_isRunning = false;
+    }
     notify();
 
     if (m_worker && m_worker->joinable())
@@ -79,10 +80,8 @@ void CommandBuffer::switchBuffers()
 {
     std::unique_lock<std::mutex> unique_lock(m_lock);
 
-    if (m_pushCommandQueue.empty() && m_isRunning)
-    {
-        m_condition.wait(unique_lock);
-    }
+    if (m_pushCommandQueue.empty())
+        m_condition.wait(unique_lock, [&] { return !m_pushCommandQueue.empty() || !m_isRunning; });
 
     std::swap(m_pushCommandQueue, m_popCommandQueue);
 }

diff --git a/core/utilities/CommandBuffer.h b/core/utilities/CommandBuffer.h
@@ -35,7 +35,7 @@ class CommandBuffer
     CommandBuffer();
     virtual ~CommandBuffer();
 
-    void pushCommand(std::shared_ptr<Command> command);
+    void pushCommand(const std::shared_ptr<Command>& command);
 
     void stop();