#pragma once

#include <atomic>

#include "envoy/server/watchdog.h"

namespace Envoy {
namespace Server {

/**
 * This class stores the actual data about when the WatchDog was last touched
 * along with thread metadata.
 */
class WatchDogImpl : public WatchDog {
public:
  /**
   * @param thread_id ThreadId of the monitored thread
   */
  WatchDogImpl(Thread::ThreadId thread_id) : thread_id_(thread_id) {}

  Thread::ThreadId threadId() const override { return thread_id_; }
  // Used by GuardDogImpl determine if the watchdog was touched recently and reset the touch status.
  bool getTouchedAndReset() { return touched_.exchange(false, std::memory_order_relaxed); }

  // Server::WatchDog
  void touch() override {
    // Set touched_ if not already set.
    bool expected = false;
    touched_.compare_exchange_strong(expected, true, std::memory_order_relaxed);
  }

private:
  const Thread::ThreadId thread_id_;
  std::atomic<bool> touched_{false};
};

} // namespace Server
} // namespace Envoy