heartd/daemon/heartbeat_manager.h

// Copyright 2024 The ChromiumOS Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef HEARTD_DAEMON_HEARTBEAT_MANAGER_H_
#define HEARTD_DAEMON_HEARTBEAT_MANAGER_H_

#include <memory>
#include <unordered_map>

#include <mojo/public/cpp/bindings/pending_receiver.h>

#include "heartd/daemon/action_runner.h"
#include "heartd/daemon/heartbeat_tracker.h"
#include "heartd/mojom/heartd.mojom.h"

namespace heartd {

class HeartbeatManager {
 public:
  explicit HeartbeatManager(ActionRunner* action_runner);
  HeartbeatManager(const HeartbeatManager&) = delete;
  HeartbeatManager& operator=(const HeartbeatManager&) = delete;
  virtual ~HeartbeatManager();

  // Returns if the pacemaker is bound for the service. This is used to check if
  // it's a repeated registration.
  bool IsPacemakerBound(ash::heartd::mojom::ServiceName name);

  // Establishes the HeartbeatTracker for the registration.
  void EstablishHeartbeatTracker(
      ash::heartd::mojom::ServiceName name,
      mojo::PendingReceiver<ash::heartd::mojom::Pacemaker> receiver,
      ash::heartd::mojom::HeartbeatServiceArgumentPtr argument);

  // Returns if there is any active heartbeat trackers.
  virtual bool AnyHeartbeatTracker();

  // Asks each heartbeat tracker instances to verify the heartbeat, and takes
  // actions if needed.
  virtual void VerifyHeartbeatAndTakeAction();

 private:
  // Removes the heartbeat tracker instances while |IsStopMonitor()| returning
  // true.
  void RemoveUnusedHeartbeatTrackers();

  void DryRun(ash::heartd::mojom::ServiceName name,
              HeartbeatTracker* heartbeat_tracker);

 private:
  // Unowned pointer. Should outlive this instance.
  // Used to run actions.
  ActionRunner* const action_runner_;
  // Used to hold all heartbeat trackers.
  std::unordered_map<ash::heartd::mojom::ServiceName,
                     std::unique_ptr<HeartbeatTracker>>
      heartbeat_trackers_;
};

}  // namespace heartd

#endif  // HEARTD_DAEMON_HEARTBEAT_MANAGER_H_