icinga2/lib/base/workqueue.cpp

141 lines
3.6 KiB
C++
Raw Normal View History

2013-09-18 09:09:16 +02:00
/******************************************************************************
* Icinga 2 *
2013-09-25 07:43:57 +02:00
* Copyright (C) 2012-2013 Icinga Development Team (http://www.icinga.org/) *
2013-09-18 09:09:16 +02:00
* *
* This program is free software; you can redistribute it and/or *
* modify it under the terms of the GNU General Public License *
* as published by the Free Software Foundation; either version 2 *
* of the License, or (at your option) any later version. *
* *
* This program is distributed in the hope that it will be useful, *
* but WITHOUT ANY WARRANTY; without even the implied warranty of *
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
* GNU General Public License for more details. *
* *
* You should have received a copy of the GNU General Public License *
* along with this program; if not, write to the Free Software Foundation *
* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA. *
******************************************************************************/
#include "base/workqueue.h"
#include "base/utility.h"
2013-10-17 10:19:17 +02:00
#include "base/debug.h"
#include "base/logger_fwd.h"
2013-09-18 09:09:16 +02:00
#include <boost/bind.hpp>
using namespace icinga;
2013-10-17 10:19:17 +02:00
int WorkQueue::m_NextID = 1;
WorkQueue::WorkQueue(size_t maxItems)
: m_ID(m_NextID++), m_MaxItems(maxItems), m_Joined(false),
m_Stopped(false), m_ExceptionCallback(WorkQueue::DefaultExceptionCallback)
2013-10-17 10:19:17 +02:00
{
m_Thread = boost::thread(boost::bind(&WorkQueue::WorkerThreadProc, this));
}
2013-09-21 17:53:14 +02:00
2013-09-18 09:09:16 +02:00
WorkQueue::~WorkQueue(void)
{
Join();
2013-10-17 10:19:17 +02:00
ASSERT(m_Stopped);
2013-09-18 09:09:16 +02:00
}
/**
* Enqueues a work item. Work items are guaranteed to be executed in the order
* they were enqueued in.
*/
void WorkQueue::Enqueue(const WorkCallback& item)
{
boost::mutex::scoped_lock lock(m_Mutex);
2013-09-21 17:53:14 +02:00
ASSERT(!m_Stopped);
2013-10-17 10:19:17 +02:00
2013-11-22 10:32:53 +01:00
bool wq_thread = (boost::this_thread::get_id() == GetThreadId());
if (!wq_thread) {
while (m_Items.size() >= m_MaxItems)
m_CV.wait(lock);
}
2013-10-17 10:19:17 +02:00
2013-09-18 09:09:16 +02:00
m_Items.push_back(item);
2013-11-22 10:32:53 +01:00
if (wq_thread)
ProcessItems(lock);
else
m_CV.notify_all();
2013-09-18 09:09:16 +02:00
}
void WorkQueue::Join(void)
{
boost::mutex::scoped_lock lock(m_Mutex);
2013-10-17 10:19:17 +02:00
m_Joined = true;
2013-10-17 15:49:53 +02:00
m_CV.notify_all();
2013-10-17 10:19:17 +02:00
while (!m_Stopped)
2013-09-18 09:09:16 +02:00
m_CV.wait(lock);
}
boost::thread::id WorkQueue::GetThreadId(void) const
{
return m_Thread.get_id();
}
void WorkQueue::SetExceptionCallback(const ExceptionCallback& callback)
{
boost::mutex::scoped_lock lock(m_Mutex);
m_ExceptionCallback = callback;
}
void WorkQueue::DefaultExceptionCallback(boost::exception_ptr exp)
{
throw;
}
2013-11-22 10:32:53 +01:00
void WorkQueue::ProcessItems(boost::mutex::scoped_lock& lock)
2013-09-18 09:09:16 +02:00
{
2013-11-22 10:32:53 +01:00
while (!m_Items.empty()) {
2013-09-18 09:09:16 +02:00
try {
WorkCallback wi = m_Items.front();
m_Items.pop_front();
2013-09-21 17:53:14 +02:00
m_CV.notify_all();
2013-09-18 09:09:16 +02:00
lock.unlock();
wi();
2013-10-17 10:19:17 +02:00
} catch (const std::exception& ex) {
lock.lock();
ExceptionCallback callback = m_ExceptionCallback;
lock.unlock();
2013-10-17 10:19:17 +02:00
callback(boost::current_exception());
2013-09-18 09:09:16 +02:00
}
2013-10-17 10:19:17 +02:00
lock.lock();
2013-09-18 09:09:16 +02:00
}
2013-11-22 10:32:53 +01:00
}
void WorkQueue::WorkerThreadProc(void)
{
boost::mutex::scoped_lock lock(m_Mutex);
std::ostringstream idbuf;
idbuf << "WQ #" << m_ID;
Utility::SetThreadName(idbuf.str());
for (;;) {
while (m_Items.empty() && !m_Joined)
m_CV.wait(lock);
if (m_Joined)
break;
ProcessItems(lock);
}
2013-09-18 09:09:16 +02:00
2013-10-17 10:19:17 +02:00
m_Stopped = true;
2013-10-17 09:14:06 +02:00
m_CV.notify_all();
2013-09-18 09:09:16 +02:00
}