summaryrefslogtreecommitdiff
path: root/chromium/components/feature_engagement/internal/availability_model_impl.h
blob: 23721d77615179d5513f9ac761f0e4167acf089f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
// 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.

#ifndef COMPONENTS_FEATURE_ENGAGEMENT_INTERNAL_AVAILABILITY_MODEL_IMPL_H_
#define COMPONENTS_FEATURE_ENGAGEMENT_INTERNAL_AVAILABILITY_MODEL_IMPL_H_

#include <stdint.h>

#include <map>
#include <memory>
#include <string>

#include "base/callback.h"
#include "base/macros.h"
#include "base/memory/weak_ptr.h"
#include "components/feature_engagement/internal/availability_model.h"
#include "components/feature_engagement/internal/persistent_availability_store.h"

namespace base {
struct Feature;
}  // namespace base

namespace feature_engagement {
// An AvailabilityModel which supports loading data from an
// PersistentAvailabilityStore.
class AvailabilityModelImpl : public AvailabilityModel {
 public:
  using StoreLoadCallback =
      base::OnceCallback<void(PersistentAvailabilityStore::OnLoadedCallback,
                              uint32_t current_day)>;

  explicit AvailabilityModelImpl(StoreLoadCallback load_callback);
  ~AvailabilityModelImpl() override;

  // AvailabilityModel implementation.
  void Initialize(OnInitializedCallback callback,
                  uint32_t current_day) override;
  bool IsReady() const override;
  absl::optional<uint32_t> GetAvailability(
      const base::Feature& feature) const override;

 private:
  // This is invoked when the store has completed loading.
  void OnStoreLoadComplete(
      OnInitializedCallback on_initialized_callback,
      bool success,
      std::unique_ptr<std::map<std::string, uint32_t>> feature_availabilities);

  // Stores the day number since epoch (1970-01-01) in the local timezone for
  // when the particular feature was made available. The key is the feature
  // name.
  std::map<std::string, uint32_t> feature_availabilities_;

  // Whether the model has successfully initialized.
  bool ready_;

  // A callback for loading availability data from the store. This is reset
  // as soon as it is invoked.
  StoreLoadCallback store_load_callback_;

  base::WeakPtrFactory<AvailabilityModelImpl> weak_ptr_factory_{this};

  DISALLOW_COPY_AND_ASSIGN(AvailabilityModelImpl);
};

}  // namespace feature_engagement

#endif  // COMPONENTS_FEATURE_ENGAGEMENT_INTERNAL_AVAILABILITY_MODEL_IMPL_H_