ClosureCounter.h

//------------------------------------------------------------------------------
/*
    This file is part of rippled: https://github.com/ripple/rippled
    Copyright (c) 2017 Ripple Labs Inc.
 
    Permission to use, copy, modify, and/or distribute this software for any
    purpose  with  or without fee is hereby granted, provided that the above
    copyright notice and this permission notice appear in all copies.
 
    THE  SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
    WITH  REGARD  TO  THIS  SOFTWARE  INCLUDING  ALL  IMPLIED  WARRANTIES  OF
    MERCHANTABILITY  AND  FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
    ANY  SPECIAL ,  DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
    WHATSOEVER  RESULTING  FROM  LOSS  OF USE, DATA OR PROFITS, WHETHER IN AN
    ACTION  OF  CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
    OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
*/
//==============================================================================
 
#ifndef RIPPLE_CORE_CLOSURE_COUNTER_H_INCLUDED
#define RIPPLE_CORE_CLOSURE_COUNTER_H_INCLUDED
 
#include <ripple/basics/Log.h>
#include <boost/optional.hpp>
#include <atomic>
#include <condition_variable>
#include <mutex>
#include <type_traits>
 
namespace ripple {
 
// A class that does reference counting for postponed closures -- a closure
// who's execution is delayed by a timer or queue.  The reference counting
// allows a Stoppable to assure that all such postponed closures are
// completed before the Stoppable declares itself stopped().
//
// Ret_t is the type that the counted closure returns.
// Args_t are the types of the arguments that will be passed to the closure.
template <typename Ret_t, typename... Args_t>
class ClosureCounter
{
private:
    std::mutex mutable mutex_ {};
    std::condition_variable allClosuresDoneCond_ {};  // guard with mutex_
    bool waitForClosures_ {false};                    // guard with mutex_
    std::atomic<int> closureCount_ {0};
 
    // Increment the count.
    ClosureCounter& operator++()
    {
        ++closureCount_;
        return *this;
    }
 
    // Decrement the count.  If we're stopping and the count drops to zero
    // notify allClosuresDoneCond_.
    ClosureCounter&  operator--()
    {
        // Even though closureCount_ is atomic, we decrement its value under
        // a lock.  This removes a small timing window that occurs if the
        // waiting thread is handling a spurious wakeup when closureCount_
        // drops to zero.
        std::lock_guard lock {mutex_};
 
        // Update closureCount_.  Notify if stopping and closureCount_ == 0.
        if ((--closureCount_ == 0) && waitForClosures_)
            allClosuresDoneCond_.notify_all();
        return *this;
    }
 
    // A private template class that helps count the number of closures
    // in flight.  This allows Stoppables to hold off declaring stopped()
    // until all their postponed closures are dispatched.
    template <typename Closure>
    class Wrapper
    {
    private:
        ClosureCounter& counter_;
        std::remove_reference_t<Closure> closure_;
 
        static_assert (
            std::is_same<decltype(
                closure_(std::declval<Args_t>()...)), Ret_t>::value,
                "Closure arguments don't match ClosureCounter Ret_t or Args_t");
 
    public:
        Wrapper() = delete;
 
        Wrapper (Wrapper const& rhs)
        : counter_ (rhs.counter_)
        , closure_ (rhs.closure_)
        {
            ++counter_;
        }
 
        Wrapper (Wrapper&& rhs) noexcept(
          std::is_nothrow_move_constructible<Closure>::value)
        : counter_ (rhs.counter_)
        , closure_ (std::move (rhs.closure_))
        {
            ++counter_;
        }
 
        Wrapper (ClosureCounter& counter, Closure&& closure)
        : counter_ (counter)
        , closure_ (std::forward<Closure> (closure))
        {
            ++counter_;
        }
 
        Wrapper& operator=(Wrapper const& rhs) = delete;
        Wrapper& operator=(Wrapper&& rhs) = delete;
 
        ~Wrapper()
        {
            --counter_;
        }
 
        // Note that Args_t is not deduced, it is explicit.  So Args_t&&
        // would be an rvalue reference, not a forwarding reference.  We
        // want to forward exactly what the user declared.
        Ret_t operator ()(Args_t... args)
        {
            return closure_ (std::forward<Args_t>(args)...);
        }
    };
 
public:
    ClosureCounter() = default;
    // Not copyable or movable.  Outstanding counts would be hard to sort out.
    ClosureCounter (ClosureCounter const&) = delete;
 
    ClosureCounter& operator=(ClosureCounter const&) = delete;
 
    ~ClosureCounter()
    {
        using namespace std::chrono_literals;
        join ("ClosureCounter", 1s, debugLog());
    }
 
    void join (char const* name,
        std::chrono::milliseconds wait, beast::Journal j)
    {
        std::unique_lock<std::mutex> lock {mutex_};
        waitForClosures_ = true;
        if (closureCount_ > 0)
        {
            if (! allClosuresDoneCond_.wait_for (
                lock, wait, [this] { return closureCount_ == 0; }))
            {
                if (auto stream = j.error())
                    stream << name
                        << " waiting for ClosureCounter::join().";
                allClosuresDoneCond_.wait (
                    lock, [this] { return closureCount_ == 0; });
            }
        }
    }
 
    template <class Closure>
    boost::optional<Wrapper<Closure>>
    wrap (Closure&& closure)
    {
        boost::optional<Wrapper<Closure>> ret;
 
        std::lock_guard lock {mutex_};
        if (! waitForClosures_)
            ret.emplace (*this, std::forward<Closure> (closure));
 
        return ret;
    }
 
    int count() const
    {
        return closureCount_;
    }
 
    bool joined() const
    {
        std::lock_guard lock {mutex_};
        return waitForClosures_;
    }
};
 
} // ripple
 
#endif // RIPPLE_CORE_CLOSURE_COUNTER_H_INCLUDED