ClickHouse/base/poco/Foundation/include/Poco/Condition.h

//
// 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