metrics_plugin_impl.h

#pragma once

#include "nighthawk/adaptive_load/metrics_plugin.h"

#include "external/envoy/source/common/common/logger.h"

#include "api/adaptive_load/metrics_plugin_impl.pb.h"
#include "api/client/options.pb.h"
#include "api/client/output.pb.h"

#include "absl/container/flat_hash_map.h"
#include "absl/strings/string_view.h"

namespace Nighthawk {

/**
 * Emulated MetricPlugin that wraps already collected Nighthawk Service counters and stats in a
 * MetricPlugin interface. This class is not registered with the Envoy registry mechanism. It will
 * be constructed on the fly from each Nighthawk Service result.
 */
class NighthawkStatsEmulatedMetricsPlugin
    : public MetricsPlugin,
      public Envoy::Logger::Loggable<Envoy::Logger::Id::main> {
public:
  /**
   * Constructs the MetricsPlugin from the given Nighthawk Service output.
   *
   * @param nighthawk_output Proto describing benchmark results from Nighthawk Service.
   */
  explicit NighthawkStatsEmulatedMetricsPlugin(const nighthawk::client::Output& nighthawk_output);
  absl::StatusOr<double> GetMetricByName(absl::string_view metric_name) override;
  const std::vector<std::string> GetAllSupportedMetricNames() const override;

private:
  absl::flat_hash_map<std::string, double> metric_from_name_;
  std::vector<std::string> errors_;
};

} // namespace Nighthawk