src/local_aggregation_1_1/aggregation_procedures/at_least_once_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_AT_LEAST_ONCE_AGGREGATION_PROCEDURE_H_
#define COBALT_SRC_LOCAL_AGGREGATION_1_1_AGGREGATION_PROCEDURES_AT_LEAST_ONCE_AGGREGATION_PROCEDURE_H_

#include <set>
#include <vector>

#include "src/local_aggregation_1_1/aggregation_procedures/aggregation_procedure.h"
#include "src/public/lib/statusor/statusor.h"

namespace cobalt::local_aggregation {

class AtLeastOnceAggregationProcedure : public AggregationProcedure {
 public:
  explicit AtLeastOnceAggregationProcedure(const MetricDefinition &metric,
                                           const ReportDefinition &report);

  void UpdateAggregateData(const logger::EventRecord & /*event_record*/,
                           AggregateData *aggregate_data,
                           AggregationPeriodBucket * /*bucket*/) override;
  [[nodiscard]] std::string DebugString() const override;
  [[nodiscard]] bool IsDaily() const override { return true; }
  // Call observation generation more frequently so that data can be sent as soon as possible.
  [[nodiscard]] bool IsExpedited() const override { return is_expedited_; }
  void ObservationsCommitted(ReportAggregate *aggregate, util::TimeInfo info,
                             uint64_t system_profile_hash) const override;

  lib::statusor::StatusOr<std::unique_ptr<Observation>> GenerateSingleObservation(
      const std::vector<AggregateDataToGenerate> &buckets,
      const std::set<std::vector<uint32_t>> &event_vectors,
      const util::TimeInfo &time_info) override;

 private:
  bool is_expedited_;
};

}  // namespace cobalt::local_aggregation

#endif  // COBALT_SRC_LOCAL_AGGREGATION_1_1_AGGREGATION_PROCEDURES_AT_LEAST_ONCE_AGGREGATION_PROCEDURE_H_