mirror of https://github.com/Icinga/icinga2.git
140 lines
3.9 KiB
C++
140 lines
3.9 KiB
C++
/******************************************************************************
|
|
* Icinga 2 *
|
|
* Copyright (C) 2012-2018 Icinga Development Team (https://icinga.com/) *
|
|
* *
|
|
* 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/exception.hpp"
|
|
#include "base/io-engine.hpp"
|
|
#include "base/lazy-init.hpp"
|
|
#include "base/logger.hpp"
|
|
#include <exception>
|
|
#include <memory>
|
|
#include <thread>
|
|
#include <boost/asio/io_service.hpp>
|
|
#include <boost/asio/spawn.hpp>
|
|
#include <boost/date_time/posix_time/ptime.hpp>
|
|
|
|
using namespace icinga;
|
|
|
|
CpuBoundWork::CpuBoundWork(boost::asio::yield_context yc)
|
|
: m_Done(false)
|
|
{
|
|
auto& ioEngine (IoEngine::Get());
|
|
|
|
for (;;) {
|
|
auto availableSlots (ioEngine.m_CpuBoundSemaphore.fetch_sub(1));
|
|
|
|
if (availableSlots < 1) {
|
|
ioEngine.m_CpuBoundSemaphore.fetch_add(1);
|
|
ioEngine.m_AlreadyExpiredTimer.async_wait(yc);
|
|
continue;
|
|
}
|
|
|
|
break;
|
|
}
|
|
}
|
|
|
|
CpuBoundWork::~CpuBoundWork()
|
|
{
|
|
if (!m_Done) {
|
|
IoEngine::Get().m_CpuBoundSemaphore.fetch_add(1);
|
|
}
|
|
}
|
|
|
|
void CpuBoundWork::Done()
|
|
{
|
|
if (!m_Done) {
|
|
IoEngine::Get().m_CpuBoundSemaphore.fetch_add(1);
|
|
|
|
m_Done = true;
|
|
}
|
|
}
|
|
|
|
IoBoundWorkSlot::IoBoundWorkSlot(boost::asio::yield_context yc)
|
|
: yc(yc)
|
|
{
|
|
IoEngine::Get().m_CpuBoundSemaphore.fetch_add(1);
|
|
}
|
|
|
|
IoBoundWorkSlot::~IoBoundWorkSlot()
|
|
{
|
|
auto& ioEngine (IoEngine::Get());
|
|
|
|
for (;;) {
|
|
auto availableSlots (ioEngine.m_CpuBoundSemaphore.fetch_sub(1));
|
|
|
|
if (availableSlots < 1) {
|
|
ioEngine.m_CpuBoundSemaphore.fetch_add(1);
|
|
ioEngine.m_AlreadyExpiredTimer.async_wait(yc);
|
|
continue;
|
|
}
|
|
|
|
break;
|
|
}
|
|
}
|
|
|
|
LazyInit<std::unique_ptr<IoEngine>> IoEngine::m_Instance ([]() { return std::unique_ptr<IoEngine>(new IoEngine()); });
|
|
|
|
IoEngine& IoEngine::Get()
|
|
{
|
|
return *m_Instance.Get();
|
|
}
|
|
|
|
boost::asio::io_service& IoEngine::GetIoService()
|
|
{
|
|
return m_IoService;
|
|
}
|
|
|
|
IoEngine::IoEngine() : m_IoService(), m_KeepAlive(m_IoService), m_Threads(decltype(m_Threads)::size_type(std::thread::hardware_concurrency() * 2u)), m_AlreadyExpiredTimer(m_IoService)
|
|
{
|
|
m_AlreadyExpiredTimer.expires_at(boost::posix_time::neg_infin);
|
|
m_CpuBoundSemaphore.store(std::thread::hardware_concurrency() * 3u / 2u);
|
|
|
|
for (auto& thread : m_Threads) {
|
|
thread = std::thread(&IoEngine::RunEventLoop, this);
|
|
}
|
|
}
|
|
|
|
IoEngine::~IoEngine()
|
|
{
|
|
for (auto& thread : m_Threads) {
|
|
m_IoService.post([]() {
|
|
throw TerminateIoThread();
|
|
});
|
|
}
|
|
|
|
for (auto& thread : m_Threads) {
|
|
thread.join();
|
|
}
|
|
}
|
|
|
|
void IoEngine::RunEventLoop()
|
|
{
|
|
for (;;) {
|
|
try {
|
|
m_IoService.run();
|
|
|
|
break;
|
|
} catch (const TerminateIoThread&) {
|
|
break;
|
|
} catch (const std::exception& e) {
|
|
Log(LogCritical, "IoEngine", "Exception during I/O operation!");
|
|
Log(LogDebug, "IoEngine") << "Exception during I/O operation: " << DiagnosticInformation(e);
|
|
}
|
|
}
|
|
}
|