mirror of https://github.com/Icinga/icinga2.git
128 lines
3.7 KiB
C++
128 lines
3.7 KiB
C++
/******************************************************************************
|
|
* Icinga 2 *
|
|
* Copyright (C) 2012 Icinga Development Team (http://www.icinga.org/) *
|
|
* *
|
|
* 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 "i2-base.h"
|
|
|
|
using namespace icinga;
|
|
|
|
/**
|
|
* Constructor for the CondVar class.
|
|
*/
|
|
CondVar::CondVar(void)
|
|
{
|
|
#ifdef _WIN32
|
|
InitializeConditionVariable(&m_CondVar);
|
|
#else /* _WIN32 */
|
|
|
|
#endif /* _WIN32 */
|
|
}
|
|
|
|
/**
|
|
* Destructor for the CondVar class.
|
|
*/
|
|
CondVar::~CondVar(void)
|
|
{
|
|
#ifdef _WIN32
|
|
/* nothing to do here */
|
|
#else /* _WIN32 */
|
|
|
|
#endif /* _WIN32 */
|
|
}
|
|
|
|
/**
|
|
* Waits for the condition variable to be signaled. Releases the specified mutex
|
|
* before it begins to wait and re-acquires the mutex after waiting.
|
|
*
|
|
* @param mtx The mutex that should be released during waiting.
|
|
* @param timeoutMilliseconds The timeout in milliseconds. Use the special
|
|
* constant CondVar::TimeoutInfinite for an
|
|
* infinite timeout.
|
|
* @returns false if a timeout occured, true otherwise.
|
|
*/
|
|
bool CondVar::Wait(Mutex& mtx, WaitTimeout timeoutMilliseconds)
|
|
{
|
|
#ifdef _WIN32
|
|
return (SleepConditionVariableCS(&m_CondVar, mtx.Get(),
|
|
timeoutMilliseconds) != FALSE);
|
|
#else /* _WIN32 */
|
|
if (timeoutMilliseconds == TimeoutInfinite)
|
|
pthread_cond_wait(&m_CondVar, mtx.Get());
|
|
else {
|
|
timeval now;
|
|
timespec ts;
|
|
|
|
if (gettimeofday(&now, NULL) < 0)
|
|
throw PosixException("gettimeofday failed.", errno);
|
|
|
|
ts.tv_sec = now.tv_sec;
|
|
ts.tv_nsec = now.tv_usec * 1000;
|
|
ts.tv_nsec += timeoutMilliseconds * 1000;
|
|
|
|
int rc = pthread_cond_timedwait(&m_CondVar, mtx.Get(), &ts);
|
|
|
|
if (rc == 0)
|
|
return true;
|
|
|
|
if (errno != ETIMEDOUT)
|
|
throw PosixException("pthread_cond_timedwait failed",
|
|
errno);
|
|
|
|
return false;
|
|
}
|
|
#endif /* _WIN32 */
|
|
}
|
|
|
|
/**
|
|
* Wakes up at least one waiting thread.
|
|
*/
|
|
void CondVar::Signal(void)
|
|
{
|
|
#ifdef _WIN32
|
|
WakeConditionVariable(&m_CondVar);
|
|
#else /* _WIN32 */
|
|
pthread_cond_signal(&m_CondVar);
|
|
#endif /* _WIN32 */
|
|
}
|
|
|
|
/**
|
|
* Wakes up all waiting threads.
|
|
*/
|
|
void CondVar::Broadcast(void)
|
|
{
|
|
#ifdef _WIN32
|
|
WakeAllConditionVariable(&m_CondVar);
|
|
#else /* _WIN32 */
|
|
pthread_cond_broadcast(&m_CondVar);
|
|
#endif /* _WIN32 */
|
|
}
|
|
|
|
/**
|
|
* Retrieves the platform-specific condition variable handle.
|
|
*
|
|
* @returns The platform-specific condition variable handle.
|
|
*/
|
|
#ifdef _WIN32
|
|
CONDITION_VARIABLE *CondVar::Get(void)
|
|
#else /* _WIN32 */
|
|
pthread_cond_t *CondVar::Get(void)
|
|
#endif /* _WIN32 */
|
|
{
|
|
return &m_CondVar;
|
|
}
|