Initial commit

Lts/Services/RoutingServiceNG/data_writer_task.cpp

@@ -0,0 +1,105 @@
+/**
+  * @RDE_HEADER@
+  * Routing Test - data writer thread
+  * $Id$
+  * $HeadURL$
+  * $Author$
+  */
+
+#include "data_writer_task.h"
+
+namespace RoutingService
+{
+DataWriterTask::DataWriterTask()
+  : m_doShutdown(false)
+  , m_logger(getLogger("DataWriterTask"))
+{
+  DEBUG_LOG(m_logger) << this << ": init ";
+}
+
+DataWriterTask::~DataWriterTask()
+{
+  std::lock_guard<std::mutex> lock(m_queueLock);
+  DEBUG_LOG(m_logger) << this << ": Cleaning up remaining writers: " << m_writerQueue.size();
+  while (!m_writerQueue.empty())
+  {
+    delete m_writerQueue.front();
+    m_writerQueue.pop();
+  }
+}
+
+void DataWriterTask::enqueue(SampleWriterInterface *swi)
+{
+  std::lock_guard<std::mutex> lock(m_queueLock);
+  TRACE_LOG(m_logger) << this << ": enqueue " << swi << " shutdown: " << m_doShutdown;
+  if (m_doShutdown)
+  {
+    delete swi;
+    return;
+  }
+  m_writerQueue.push(swi);
+  m_waitForDataCondition.notify_one();
+}
+
+void DataWriterTask::shutdown()
+{
+  std::lock_guard<std::mutex> l(m_queueLock);
+  m_doShutdown = true;
+  DEBUG_LOG(m_logger) << this << ": Cleaning up remaining writers: " << m_writerQueue.size();
+  while (!m_writerQueue.empty())
+  {
+    delete m_writerQueue.front();
+    m_writerQueue.pop();
+  }
+
+  m_waitForDataCondition.notify_one();
+}
+
+bool DataWriterTask::waitForData()
+{
+  TRACE_LOG(m_logger) << this << ": waiting for data ";
+  std::unique_lock<std::mutex> waitCondLock(m_queueLock);
+  while (true)
+  {
+    if (m_doShutdown)
+      return false;
+
+    if (!m_writerQueue.empty())
+      return true;
+
+    m_waitForDataCondition.wait(waitCondLock); // m_queueLock gets unlocked inside wait()
+    // m_queueLock is locked now
+  }
+  return true;  // dead code ...
+}
+
+SampleWriterInterface *DataWriterTask::getNextSampleWriter()
+{
+  std::lock_guard<std::mutex> lock(m_queueLock);
+  if (m_writerQueue.empty())
+    return nullptr;
+
+  TRACE_LOG(m_logger) << this << ": getData() queue size: " << m_writerQueue.size();
+  auto ret = m_writerQueue.front();
+  m_writerQueue.pop();
+  return ret;
+}
+
+void DataWriterTask::writeSamples()
+{
+  while (true)
+  {
+    if (!waitForData())
+      return;
+
+    SampleWriterInterface *swi = nullptr;
+    while ((swi = getNextSampleWriter()) != nullptr)
+    {
+      swi->writeSamples();
+      TRACE_LOG(swi->logger()) << this << ": delete " << swi;
+      delete swi;
+    }
+  }
+}
+
+} // namespace RoutingService