clock.hpp

#pragma once
 
#include <chrono>
#include <thread>
 
namespace algotradelib {
namespace runner {

class SteadyClock {
public:
  std::chrono::system_clock::time_point now() const noexcept {
    return std::chrono::system_clock::now();
  }

  std::chrono::steady_clock::time_point monotonic_now() const noexcept {
    return std::chrono::steady_clock::now();
  }

  void wait_for(std::chrono::nanoseconds duration) const {
    if (duration <= std::chrono::nanoseconds::zero()) {
      return;
    }
    std::this_thread::sleep_for(duration);
  }
};

template <typename Clock>
class ClockReference {
public:
  explicit ClockReference(Clock& clock) noexcept : clock_(&clock) {}

  std::chrono::system_clock::time_point now() const { return clock_->now(); }

  std::chrono::steady_clock::time_point monotonic_now() const { return clock_->monotonic_now(); }

  void wait_for(std::chrono::nanoseconds duration) { clock_->wait_for(duration); }
 
private:
  Clock* clock_;
};

class ManualClock {
public:
  explicit ManualClock(std::chrono::system_clock::time_point start = {}) : now_(start) {}

  std::chrono::system_clock::time_point now() const noexcept { return now_; }

  std::chrono::steady_clock::time_point monotonic_now() const noexcept { return monotonic_; }

  void wait_for(std::chrono::nanoseconds duration) noexcept {
    ++waits_;
    if (duration > std::chrono::nanoseconds::zero()) {
      now_ += std::chrono::duration_cast<std::chrono::system_clock::duration>(duration);
      monotonic_ += std::chrono::duration_cast<std::chrono::steady_clock::duration>(duration);
      waited_ += duration;
    }
  }

  void advance(std::chrono::nanoseconds duration) noexcept {
    now_ += std::chrono::duration_cast<std::chrono::system_clock::duration>(duration);
    if (duration > std::chrono::nanoseconds::zero()) {
      monotonic_ += std::chrono::duration_cast<std::chrono::steady_clock::duration>(duration);
    }
  }

  void jump(std::chrono::nanoseconds duration) noexcept {
    now_ += std::chrono::duration_cast<std::chrono::system_clock::duration>(duration);
  }

  std::size_t waits() const noexcept { return waits_; }

  std::chrono::nanoseconds waited() const noexcept { return waited_; }
 
private:
  std::chrono::system_clock::time_point now_;
  std::chrono::steady_clock::time_point monotonic_{};
  std::size_t waits_{0};
  std::chrono::nanoseconds waited_{std::chrono::nanoseconds::zero()};
};
 
} // namespace runner
} // namespace algotradelib