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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
|
// Copyright 2017 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "components/feature_engagement/internal/persistent_event_store.h"
#include <vector>
#include "base/functional/bind.h"
#include "components/feature_engagement/internal/stats.h"
#include "components/leveldb_proto/public/proto_database.h"
namespace feature_engagement {
namespace {
using KeyEventPair = std::pair<std::string, Event>;
using KeyEventList = std::vector<KeyEventPair>;
void NoopUpdateCallback(bool success) {
stats::RecordDbUpdate(success, stats::StoreType::EVENTS_STORE);
}
} // namespace
PersistentEventStore::PersistentEventStore(
std::unique_ptr<leveldb_proto::ProtoDatabase<Event>> db)
: db_(std::move(db)), ready_(false) {}
PersistentEventStore::~PersistentEventStore() = default;
void PersistentEventStore::Load(OnLoadedCallback callback) {
DCHECK(!ready_);
db_->Init(base::BindOnce(&PersistentEventStore::OnInitComplete,
weak_ptr_factory_.GetWeakPtr(),
std::move(callback)));
}
bool PersistentEventStore::IsReady() const {
return ready_;
}
void PersistentEventStore::WriteEvent(const Event& event) {
DCHECK(IsReady());
std::unique_ptr<KeyEventList> entries = std::make_unique<KeyEventList>();
entries->push_back(KeyEventPair(event.name(), event));
db_->UpdateEntries(std::move(entries),
std::make_unique<std::vector<std::string>>(),
base::BindOnce(&NoopUpdateCallback));
}
void PersistentEventStore::DeleteEvent(const std::string& event_name) {
DCHECK(IsReady());
auto deletes = std::make_unique<std::vector<std::string>>();
deletes->push_back(event_name);
db_->UpdateEntries(std::make_unique<KeyEventList>(), std::move(deletes),
base::BindOnce(&NoopUpdateCallback));
}
void PersistentEventStore::OnInitComplete(
OnLoadedCallback callback,
leveldb_proto::Enums::InitStatus status) {
bool success = status == leveldb_proto::Enums::InitStatus::kOK;
stats::RecordDbInitEvent(success, stats::StoreType::EVENTS_STORE);
if (!success) {
std::move(callback).Run(false, std::make_unique<std::vector<Event>>());
return;
}
db_->LoadEntries(base::BindOnce(&PersistentEventStore::OnLoadComplete,
weak_ptr_factory_.GetWeakPtr(),
std::move(callback)));
}
void PersistentEventStore::OnLoadComplete(
OnLoadedCallback callback,
bool success,
std::unique_ptr<std::vector<Event>> entries) {
stats::RecordEventDbLoadEvent(success, *entries);
ready_ = success;
std::move(callback).Run(success, std::move(entries));
}
} // namespace feature_engagement
|
