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
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
|
// 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/download/internal/background_service/download_store.h"
#include <memory>
#include "base/functional/bind.h"
#include "base/functional/callback_helpers.h"
#include "components/download/internal/background_service/entry.h"
#include "components/download/internal/background_service/proto/entry.pb.h"
#include "components/download/internal/background_service/proto_conversions.h"
#include "components/leveldb_proto/public/proto_database_provider.h"
namespace download {
namespace {
using KeyVector = std::vector<std::string>;
using ProtoEntryVector = std::vector<protodb::Entry>;
using KeyProtoEntryVector = std::vector<std::pair<std::string, protodb::Entry>>;
leveldb_env::Options GetDownloadDBOptions() {
// These options reduce memory consumption.
leveldb_env::Options options = leveldb_proto::CreateSimpleOptions();
options.reuse_logs = false;
options.write_buffer_size = 64 << 10; // 64 KiB
return options;
}
} // namespace
DownloadStore::DownloadStore(
std::unique_ptr<leveldb_proto::ProtoDatabase<protodb::Entry>> db)
: db_(std::move(db)), is_initialized_(false) {}
DownloadStore::~DownloadStore() = default;
bool DownloadStore::IsInitialized() {
return is_initialized_;
}
void DownloadStore::Initialize(InitCallback callback) {
DCHECK(!IsInitialized());
db_->Init(GetDownloadDBOptions(),
base::BindOnce(&DownloadStore::OnDatabaseInited,
weak_factory_.GetWeakPtr(), std::move(callback)));
}
void DownloadStore::HardRecover(StoreCallback callback) {
is_initialized_ = false;
db_->Destroy(base::BindOnce(&DownloadStore::OnDatabaseDestroyed,
weak_factory_.GetWeakPtr(), std::move(callback)));
}
void DownloadStore::OnDatabaseInited(InitCallback callback,
leveldb_proto::Enums::InitStatus status) {
if (status != leveldb_proto::Enums::InitStatus::kOK) {
std::move(callback).Run(false, std::make_unique<std::vector<Entry>>());
return;
}
db_->LoadEntries(base::BindOnce(&DownloadStore::OnDatabaseLoaded,
weak_factory_.GetWeakPtr(),
std::move(callback)));
}
void DownloadStore::OnDatabaseLoaded(InitCallback callback,
bool success,
std::unique_ptr<ProtoEntryVector> protos) {
if (!success) {
std::move(callback).Run(success, std::make_unique<std::vector<Entry>>());
return;
}
auto entries = ProtoConversions::EntryVectorFromProto(std::move(protos));
is_initialized_ = true;
std::move(callback).Run(success, std::move(entries));
}
void DownloadStore::OnDatabaseDestroyed(StoreCallback callback, bool success) {
if (!success) {
std::move(callback).Run(success);
return;
}
db_->Init(GetDownloadDBOptions(),
base::BindOnce(&DownloadStore::OnDatabaseInitedAfterDestroy,
weak_factory_.GetWeakPtr(), std::move(callback)));
}
void DownloadStore::OnDatabaseInitedAfterDestroy(
StoreCallback callback,
leveldb_proto::Enums::InitStatus status) {
is_initialized_ = status == leveldb_proto::Enums::InitStatus::kOK;
std::move(callback).Run(is_initialized_);
}
void DownloadStore::Update(const Entry& entry, StoreCallback callback) {
DCHECK(IsInitialized());
auto entries_to_save = std::make_unique<KeyProtoEntryVector>();
protodb::Entry proto = ProtoConversions::EntryToProto(entry);
entries_to_save->emplace_back(entry.guid, std::move(proto));
db_->UpdateEntries(std::move(entries_to_save), std::make_unique<KeyVector>(),
std::move(callback));
}
void DownloadStore::Remove(const std::string& guid, StoreCallback callback) {
DCHECK(IsInitialized());
auto keys_to_remove = std::make_unique<KeyVector>();
keys_to_remove->push_back(guid);
db_->UpdateEntries(std::make_unique<KeyProtoEntryVector>(),
std::move(keys_to_remove), std::move(callback));
}
} // namespace download
|
