src/local_aggregation_1_1/aggregation_procedures/count_aggregation_procedure.h

// Copyright 2020 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef COBALT_SRC_LOCAL_AGGREGATION_1_1_AGGREGATION_PROCEDURES_COUNT_AGGREGATION_PROCEDURE_H_
#define COBALT_SRC_LOCAL_AGGREGATION_1_1_AGGREGATION_PROCEDURES_COUNT_AGGREGATION_PROCEDURE_H_

#include "src/lib/statusor/statusor.h"
#include "src/local_aggregation_1_1/aggregation_procedures/hourly_aggregation_procedure.h"
#include "src/registry/metric_definition.pb.h"
#include "src/registry/report_definition.pb.h"

namespace cobalt::local_aggregation {

class CountAggregationProcedure : public AggregationProcedure {
 public:
  CountAggregationProcedure(const MetricDefinition &metric, const ReportDefinition &report)
      : AggregationProcedure(metric, report) {}

  [[nodiscard]] bool IsDaily() const override;

  void UpdateAggregateData(const logger::EventRecord &event_record, AggregateData *aggregate_data,
                           EventCodeAggregate *aggregate) override;

  lib::statusor::StatusOr<std::unique_ptr<Observation>> GenerateSingleObservation(
      const std::vector<EventCodeAggregate *> &aggregates,
      const std::set<std::vector<uint32_t>> &event_vectors) override;

  [[nodiscard]] std::string DebugString() const override;
};

}  // namespace cobalt::local_aggregation

#endif  // COBALT_SRC_LOCAL_AGGREGATION_1_1_AGGREGATION_PROCEDURES_COUNT_AGGREGATION_PROCEDURE_H_