mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-14 10:22:10 +00:00
139 lines
3.5 KiB
C++
139 lines
3.5 KiB
C++
//
|
|
// Condition.h
|
|
//
|
|
// Library: Foundation
|
|
// Package: Threading
|
|
// Module: Condition
|
|
//
|
|
// Definition of the Condition class template.
|
|
//
|
|
// Copyright (c) 2007, Applied Informatics Software Engineering GmbH.
|
|
// and Contributors.
|
|
//
|
|
// SPDX-License-Identifier: BSL-1.0
|
|
//
|
|
|
|
|
|
#ifndef Foundation_Condition_INCLUDED
|
|
#define Foundation_Condition_INCLUDED
|
|
|
|
|
|
#include <deque>
|
|
#include "Poco/Event.h"
|
|
#include "Poco/Exception.h"
|
|
#include "Poco/Foundation.h"
|
|
#include "Poco/Mutex.h"
|
|
#include "Poco/ScopedUnlock.h"
|
|
|
|
|
|
namespace Poco
|
|
{
|
|
|
|
|
|
class Foundation_API Condition
|
|
/// A Condition is a synchronization object used to block a thread
|
|
/// until a particular condition is met.
|
|
/// A Condition object is always used in conjunction with
|
|
/// a Mutex (or FastMutex) object.
|
|
///
|
|
/// Condition objects are similar to POSIX condition variables, which the
|
|
/// difference that Condition is not subject to spurious wakeups.
|
|
///
|
|
/// Threads waiting on a Condition are resumed in FIFO order.
|
|
{
|
|
public:
|
|
Condition();
|
|
/// Creates the Condition.
|
|
|
|
~Condition();
|
|
/// Destroys the Condition.
|
|
|
|
template <class Mtx>
|
|
void wait(Mtx & mutex)
|
|
/// Unlocks the mutex (which must be locked upon calling
|
|
/// wait()) and waits until the Condition is signalled.
|
|
///
|
|
/// The given mutex will be locked again upon
|
|
/// leaving the function, even in case of an exception.
|
|
{
|
|
ScopedUnlock<Mtx> unlock(mutex, false);
|
|
Event event;
|
|
{
|
|
FastMutex::ScopedLock lock(_mutex);
|
|
mutex.unlock();
|
|
enqueue(event);
|
|
}
|
|
event.wait();
|
|
}
|
|
|
|
template <class Mtx>
|
|
void wait(Mtx & mutex, long milliseconds)
|
|
/// Unlocks the mutex (which must be locked upon calling
|
|
/// wait()) and waits for the given time until the Condition is signalled.
|
|
///
|
|
/// The given mutex will be locked again upon successfully leaving the
|
|
/// function, even in case of an exception.
|
|
///
|
|
/// Throws a TimeoutException if the Condition is not signalled
|
|
/// within the given time interval.
|
|
{
|
|
if (!tryWait(mutex, milliseconds))
|
|
throw TimeoutException();
|
|
}
|
|
|
|
template <class Mtx>
|
|
bool tryWait(Mtx & mutex, long milliseconds)
|
|
/// Unlocks the mutex (which must be locked upon calling
|
|
/// tryWait()) and waits for the given time until the Condition is signalled.
|
|
///
|
|
/// The given mutex will be locked again upon leaving the
|
|
/// function, even in case of an exception.
|
|
///
|
|
/// Returns true if the Condition has been signalled
|
|
/// within the given time interval, otherwise false.
|
|
{
|
|
ScopedUnlock<Mtx> unlock(mutex, false);
|
|
Event event;
|
|
{
|
|
FastMutex::ScopedLock lock(_mutex);
|
|
mutex.unlock();
|
|
enqueue(event);
|
|
}
|
|
if (!event.tryWait(milliseconds))
|
|
{
|
|
FastMutex::ScopedLock lock(_mutex);
|
|
dequeue(event);
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
void signal();
|
|
/// Signals the Condition and allows one waiting thread
|
|
/// to continue execution.
|
|
|
|
void broadcast();
|
|
/// Signals the Condition and allows all waiting
|
|
/// threads to continue their execution.
|
|
|
|
protected:
|
|
void enqueue(Event & event);
|
|
void dequeue();
|
|
void dequeue(Event & event);
|
|
|
|
private:
|
|
Condition(const Condition &);
|
|
Condition & operator=(const Condition &);
|
|
|
|
typedef std::deque<Event *> WaitQueue;
|
|
|
|
FastMutex _mutex;
|
|
WaitQueue _waitQueue;
|
|
};
|
|
|
|
|
|
} // namespace Poco
|
|
|
|
|
|
#endif // Foundation_Condition_INCLUDED
|