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
|
// 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.
#include "components/offline_pages/core/prefetch/tasks/get_operation_task.h"
#include <memory>
#include <utility>
#include "base/bind.h"
#include "base/callback.h"
#include "components/offline_pages/core/prefetch/prefetch_dispatcher.h"
#include "components/offline_pages/core/prefetch/prefetch_network_request_factory.h"
#include "components/offline_pages/core/prefetch/store/prefetch_store.h"
#include "sql/database.h"
#include "sql/statement.h"
#include "sql/transaction.h"
namespace offline_pages {
namespace {
GetOperationTask::OperationResultList MakeError() {
return nullptr;
}
bool UpdatePrefetchItemsForOperationSync(sql::Database* db,
const std::string& operation_name) {
static const char kSql[] =
"UPDATE prefetch_items SET"
" state = ?,"
" get_operation_attempts = get_operation_attempts + 1"
" WHERE state = ? AND operation_name = ?";
sql::Statement statement(db->GetCachedStatement(SQL_FROM_HERE, kSql));
statement.BindInt(0, static_cast<int>(PrefetchItemState::SENT_GET_OPERATION));
statement.BindInt(1, static_cast<int>(PrefetchItemState::RECEIVED_GCM));
statement.BindString(2, operation_name);
return statement.Run();
}
std::unique_ptr<std::vector<std::string>> AvailableOperationsSync(
sql::Database* db) {
static const char kSql[] =
"SELECT DISTINCT operation_name FROM prefetch_items WHERE state = ?";
sql::Statement statement(db->GetCachedStatement(SQL_FROM_HERE, kSql));
statement.BindInt(0, static_cast<int>(PrefetchItemState::RECEIVED_GCM));
auto operations = std::make_unique<std::vector<std::string>>();
while (statement.Step()) {
operations->push_back(statement.ColumnString(0)); // operation_name
}
return operations;
}
GetOperationTask::OperationResultList SelectOperationsToFetch(
sql::Database* db) {
sql::Transaction transaction(db);
if (!transaction.Begin())
return MakeError();
auto operations = AvailableOperationsSync(db);
if (!operations)
return MakeError();
for (std::string operation : *operations) {
if (!UpdatePrefetchItemsForOperationSync(db, operation))
return MakeError(); // Rollback.
}
if (!transaction.Commit())
return MakeError();
return operations;
}
} // namespace
GetOperationTask::GetOperationTask(
PrefetchStore* store,
PrefetchNetworkRequestFactory* request_factory,
GetOperationFinishedCallback callback)
: prefetch_store_(store),
request_factory_(request_factory),
callback_(std::move(callback)),
weak_factory_(this) {}
GetOperationTask::~GetOperationTask() {}
void GetOperationTask::Run() {
prefetch_store_->Execute(
base::BindOnce(&SelectOperationsToFetch),
base::BindOnce(&GetOperationTask::StartGetOperationRequests,
weak_factory_.GetWeakPtr()),
MakeError());
}
void GetOperationTask::StartGetOperationRequests(
OperationResultList operation_names) {
if (operation_names) {
for (std::string& operation : *operation_names) {
request_factory_->MakeGetOperationRequest(
operation, PrefetchRequestFinishedCallback(callback_));
}
}
TaskComplete();
}
} // namespace offline_pages
|