rippled/include/xrpl/beast/clock/manual_clock.h

#ifndef BEAST_CHRONO_MANUAL_CLOCK_H_INCLUDED
#define BEAST_CHRONO_MANUAL_CLOCK_H_INCLUDED

#include <xrpl/beast/clock/abstract_clock.h>
#include <xrpl/beast/utility/instrumentation.h>

#include <chrono>

namespace beast {

/** Manual clock implementation.

    This concrete class implements the @ref abstract_clock interface and
    allows the time to be advanced manually, mainly for the purpose of
    providing a clock in unit tests.

    @tparam Clock A type meeting these requirements:
        http://en.cppreference.com/w/cpp/concept/Clock
*/
template <class Clock>
class manual_clock : public abstract_clock<Clock>
{
public:
    using typename abstract_clock<Clock>::rep;
    using typename abstract_clock<Clock>::duration;
    using typename abstract_clock<Clock>::time_point;

private:
    time_point now_;

public:
    explicit manual_clock(time_point const& now = time_point(duration(0)))
        : now_(now)
    {
    }

    time_point
    now() const override
    {
        return now_;
    }

    /** Set the current time of the manual clock. */
    void
    set(time_point const& when)
    {
        XRPL_ASSERT(
            !Clock::is_steady || when >= now_,
            "beast::manual_clock::set(time_point) : forward input");
        now_ = when;
    }

    /** Convenience for setting the time in seconds from epoch. */
    template <class Integer>
    void
    set(Integer seconds_from_epoch)
    {
        set(time_point(duration(std::chrono::seconds(seconds_from_epoch))));
    }

    /** Advance the clock by a duration. */
    template <class Rep, class Period>
    void
    advance(std::chrono::duration<Rep, Period> const& elapsed)
    {
        XRPL_ASSERT(
            !Clock::is_steady || (now_ + elapsed) >= now_,
            "beast::manual_clock::advance(duration) : forward input");
        now_ += elapsed;
    }

    /** Convenience for advancing the clock by one second. */
    manual_clock&
    operator++()
    {
        advance(std::chrono::seconds(1));
        return *this;
    }
};

}  // namespace beast

#endif