106 lines
2.3 KiB
C++
Executable File
106 lines
2.3 KiB
C++
Executable File
/**
|
|
* @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
|