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
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
|
// Copyright 2016 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_NTP_SNIPPETS_REMOTE_REMOTE_SUGGESTIONS_DATABASE_H_
#define COMPONENTS_NTP_SNIPPETS_REMOTE_REMOTE_SUGGESTIONS_DATABASE_H_
#include <memory>
#include <set>
#include <string>
#include <utility>
#include <vector>
#include "base/callback.h"
#include "base/macros.h"
#include "base/memory/ref_counted.h"
#include "base/memory/weak_ptr.h"
#include "base/sequenced_task_runner.h"
#include "components/leveldb_proto/proto_database.h"
#include "components/ntp_snippets/remote/ntp_snippet.h"
namespace base {
class FilePath;
} // namespace base
namespace ntp_snippets {
class SnippetImageProto;
class SnippetProto;
class RemoteSuggestionsDatabase {
public:
using SnippetsCallback = base::Callback<void(NTPSnippet::PtrVector)>;
using SnippetImageCallback = base::Callback<void(std::string)>;
RemoteSuggestionsDatabase(
const base::FilePath& database_dir,
scoped_refptr<base::SequencedTaskRunner> file_task_runner);
~RemoteSuggestionsDatabase();
// Returns whether the database has finished initialization. While this is
// false, loads may already be started (they'll be serviced after
// initialization finishes), but no updates are allowed.
bool IsInitialized() const;
// Returns whether the database is in an (unrecoverable) error state. If this
// is true, the database must not be used anymore
bool IsErrorState() const;
// Set a callback to be called when the database enters an error state.
void SetErrorCallback(const base::Closure& error_callback);
// Loads all snippets from storage and passes them to |callback|.
void LoadSnippets(const SnippetsCallback& callback);
// Adds or updates the given snippet.
void SaveSnippet(const NTPSnippet& snippet);
// Adds or updates all the given snippets.
void SaveSnippets(const NTPSnippet::PtrVector& snippets);
// Deletes the snippet with the given ID.
void DeleteSnippet(const std::string& snippet_id);
// Deletes all the given snippets (identified by their IDs).
void DeleteSnippets(std::unique_ptr<std::vector<std::string>> snippet_ids);
// Loads the image data for the snippet with the given ID and passes it to
// |callback|. Passes an empty string if not found.
void LoadImage(const std::string& snippet_id,
const SnippetImageCallback& callback);
// Adds or updates the image data for the given snippet ID.
void SaveImage(const std::string& snippet_id, const std::string& image_data);
// Deletes the image data for the given snippet ID.
void DeleteImage(const std::string& snippet_id);
// Deletes the image data for the given snippets (identified by their IDs).
void DeleteImages(std::unique_ptr<std::vector<std::string>> snippet_ids);
// Deletes all images which are not associated with any of the provided
// snippets.
void GarbageCollectImages(
std::unique_ptr<std::set<std::string>> alive_snippet_ids);
private:
friend class RemoteSuggestionsDatabaseTest;
using KeyEntryVector =
leveldb_proto::ProtoDatabase<SnippetProto>::KeyEntryVector;
using ImageKeyEntryVector =
leveldb_proto::ProtoDatabase<SnippetImageProto>::KeyEntryVector;
// Callbacks for ProtoDatabase<SnippetProto> operations.
void OnDatabaseInited(bool success);
void OnDatabaseLoaded(const SnippetsCallback& callback,
bool success,
std::unique_ptr<std::vector<SnippetProto>> entries);
void OnDatabaseSaved(bool success);
// Callbacks for ProtoDatabase<SnippetImageProto> operations.
void OnImageDatabaseInited(bool success);
void OnImageDatabaseLoaded(const SnippetImageCallback& callback,
bool success,
std::unique_ptr<SnippetImageProto> entry);
void OnImageDatabaseSaved(bool success);
void OnDatabaseError();
void ProcessPendingLoads();
void LoadSnippetsImpl(const SnippetsCallback& callback);
void SaveSnippetsImpl(std::unique_ptr<KeyEntryVector> entries_to_save);
void LoadImageImpl(const std::string& snippet_id,
const SnippetImageCallback& callback);
void DeleteUnreferencedImages(
std::unique_ptr<std::set<std::string>> references,
bool load_keys_success,
std::unique_ptr<std::vector<std::string>> image_keys);
std::unique_ptr<leveldb_proto::ProtoDatabase<SnippetProto>> database_;
bool database_initialized_;
std::vector<SnippetsCallback> pending_snippets_callbacks_;
std::unique_ptr<leveldb_proto::ProtoDatabase<SnippetImageProto>>
image_database_;
bool image_database_initialized_;
std::vector<std::pair<std::string, SnippetImageCallback>>
pending_image_callbacks_;
base::Closure error_callback_;
base::WeakPtrFactory<RemoteSuggestionsDatabase> weak_ptr_factory_;
DISALLOW_COPY_AND_ASSIGN(RemoteSuggestionsDatabase);
};
} // namespace ntp_snippets
#endif // COMPONENTS_NTP_SNIPPETS_REMOTE_REMOTE_SUGGESTIONS_DATABASE_H_
|
