blob: b7dd0683a1fe5d0c1d0472e1c3a06fae1266a392 [file] [log] [blame]
// Copyright 2017 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include <string>
#include <unordered_map>
#include <unordered_set>
#include "base/macros.h"
#include "components/feature_engagement/internal/event_storage_validator.h"
#include "components/feature_engagement/public/feature_list.h"
namespace feature_engagement {
class Configuration;
struct EventConfig;
struct FeatureConfig;
// A EventStorageValidator that uses the FeatureConfig as the source of truth.
class FeatureConfigEventStorageValidator : public EventStorageValidator {
~FeatureConfigEventStorageValidator() override;
// EventStorageValidator implementation.
bool ShouldStore(const std::string& event_name) const override;
bool ShouldKeep(const std::string& event_name,
uint32_t event_day,
uint32_t current_day) const override;
// Set up internal configuration required for the given |features|.
void InitializeFeatures(FeatureVector features,
const Configuration& configuration);
// Resets the full state of this EventStorageValidator. After calling this
// method it is valid to call InitializeFeatures() again.
void ClearForTesting();
// Updates the internal configuration with conditions from the given
// |feature_config|.
void InitializeFeatureConfig(const FeatureConfig& feature_config);
// Updates the internal configuration with conditions from the given
// |event_config|.
void InitializeEventConfig(const EventConfig& event_config);
// Contains an entry for each of the events that any EventConfig required to
// be stored.
std::unordered_set<std::string> should_store_event_names_;
// Contains the longest time to store each event across all EventConfigs,
// as a number of days.
std::unordered_map<std::string, uint32_t> longest_storage_times_;
} // namespace feature_engagement