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
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
|
// Copyright 2014 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 GOOGLE_APIS_GCM_ENGINE_GCM_STORE_IMPL_H_
#define GOOGLE_APIS_GCM_ENGINE_GCM_STORE_IMPL_H_
#include "base/basictypes.h"
#include "base/memory/ref_counted.h"
#include "base/memory/weak_ptr.h"
#include "google_apis/gcm/base/gcm_export.h"
#include "google_apis/gcm/engine/gcm_store.h"
namespace base {
class FilePath;
class SequencedTaskRunner;
} // namespace base
namespace gcm {
class Encryptor;
// An implementation of GCM Store that uses LevelDB for persistence.
// It performs all blocking operations on the blocking task runner, and posts
// all callbacks to the thread on which the GCMStoreImpl is created.
class GCM_EXPORT GCMStoreImpl : public GCMStore {
public:
GCMStoreImpl(const base::FilePath& path,
scoped_refptr<base::SequencedTaskRunner> blocking_task_runner,
scoped_ptr<Encryptor> encryptor);
~GCMStoreImpl() override;
// Load the directory and pass the initial state back to caller.
void Load(const LoadCallback& callback) override;
// Closes the GCM store.
void Close() override;
// Clears the GCM store of all data and destroys any LevelDB files associated
// with this store.
// WARNING: this will permanently destroy any pending outgoing messages
// and require the device to re-create credentials and serial number mapping
// tables.
void Destroy(const UpdateCallback& callback) override;
// Sets this device's messaging credentials.
void SetDeviceCredentials(uint64 device_android_id,
uint64 device_security_token,
const UpdateCallback& callback) override;
// Registration info.
void AddRegistration(const std::string& app_id,
const linked_ptr<RegistrationInfo>& registration,
const UpdateCallback& callback) override;
void RemoveRegistration(const std::string& app_id,
const UpdateCallback& callback) override;
// Unacknowledged incoming message handling.
void AddIncomingMessage(const std::string& persistent_id,
const UpdateCallback& callback) override;
void RemoveIncomingMessage(const std::string& persistent_id,
const UpdateCallback& callback) override;
void RemoveIncomingMessages(const PersistentIdList& persistent_ids,
const UpdateCallback& callback) override;
// Unacknowledged outgoing messages handling.
bool AddOutgoingMessage(const std::string& persistent_id,
const MCSMessage& message,
const UpdateCallback& callback) override;
void OverwriteOutgoingMessage(const std::string& persistent_id,
const MCSMessage& message,
const UpdateCallback& callback) override;
void RemoveOutgoingMessage(const std::string& persistent_id,
const UpdateCallback& callback) override;
void RemoveOutgoingMessages(const PersistentIdList& persistent_ids,
const UpdateCallback& callback) override;
// Sets last device's checkin information.
void SetLastCheckinInfo(const base::Time& time,
const std::set<std::string>& accounts,
const UpdateCallback& callback) override;
// G-service settings handling.
void SetGServicesSettings(const std::map<std::string, std::string>& settings,
const std::string& settings_digest,
const UpdateCallback& callback) override;
// Sets the account information related to device to account mapping.
void AddAccountMapping(const AccountMapping& account_mapping,
const UpdateCallback& callback) override;
void RemoveAccountMapping(const std::string& account_id,
const UpdateCallback& callback) override;
// Sets last token fetch time.
void SetLastTokenFetchTime(const base::Time& time,
const UpdateCallback& callback) override;
// Sets the custom client heartbeat interval for the scope.
void AddHeartbeatInterval(const std::string& scope,
int interval_ms,
const UpdateCallback& callback) override;
void RemoveHeartbeatInterval(const std::string& scope,
const UpdateCallback& callback) override;
// Instance ID data.
void AddInstanceIDData(const std::string& app_id,
const std::string& instance_id_data,
const UpdateCallback& callback) override;
void RemoveInstanceIDData(const std::string& app_id,
const UpdateCallback& callback) override;
// Injects a value to database. Only to be used for testing.
void SetValueForTesting(const std::string& key,
const std::string& value,
const UpdateCallback& callback);
private:
typedef std::map<std::string, int> AppIdToMessageCountMap;
// Continuation to update the per-app message counts after a load.
void LoadContinuation(const LoadCallback& callback,
scoped_ptr<LoadResult> result);
// Continuation to update the per-app message counts when adding messages.
// In particular, if a message fails to add, the message count is decremented.
void AddOutgoingMessageContinuation(const UpdateCallback& callback,
const std::string& app_id,
bool success);
// Continuation to update the per-app message counts when removing messages.
// Note: if doing a read-then-write when removing messages proves expensive,
// an in-memory mapping of persisted message id to app could be maintained
// instead.
void RemoveOutgoingMessagesContinuation(
const UpdateCallback& callback,
bool success,
const std::map<std::string, int>& removed_message_counts);
class Backend;
// Map of App ids to their message counts.
AppIdToMessageCountMap app_message_counts_;
scoped_refptr<Backend> backend_;
scoped_refptr<base::SequencedTaskRunner> blocking_task_runner_;
base::WeakPtrFactory<GCMStoreImpl> weak_ptr_factory_;
DISALLOW_COPY_AND_ASSIGN(GCMStoreImpl);
};
} // namespace gcm
#endif // GOOGLE_APIS_GCM_ENGINE_GCM_STORE_IMPL_H_
|
