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
|
// Copyright 2018 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/public/task/task_manager_impl.h"
namespace download {
namespace {
bool HasDuplicateParams(
const std::map<DownloadTaskType, TaskManager::TaskParams>& map,
DownloadTaskType task_type,
const TaskManager::TaskParams& params) {
auto it = map.find(task_type);
return it != map.end() && it->second == params;
}
} // namespace
TaskManagerImpl::TaskManagerImpl(std::unique_ptr<TaskScheduler> task_scheduler)
: task_scheduler_(std::move(task_scheduler)) {}
TaskManagerImpl::~TaskManagerImpl() = default;
void TaskManagerImpl::ScheduleTask(DownloadTaskType task_type,
const TaskParams& params) {
if (HasDuplicateParams(current_task_params_, task_type, params) ||
HasDuplicateParams(pending_task_params_, task_type, params)) {
return;
}
pending_task_params_[task_type] = params;
if (IsRunningTask(task_type))
return;
task_scheduler_->ScheduleTask(
task_type, params.require_unmetered_network, params.require_charging,
params.optimal_battery_percentage, params.window_start_time_seconds,
params.window_end_time_seconds);
}
void TaskManagerImpl::UnscheduleTask(DownloadTaskType task_type) {
pending_task_params_.erase(task_type);
if (IsRunningTask(task_type))
return;
task_scheduler_->CancelTask(task_type);
}
void TaskManagerImpl::OnStartScheduledTask(DownloadTaskType task_type,
TaskFinishedCallback callback) {
if (pending_task_params_.find(task_type) != pending_task_params_.end())
current_task_params_[task_type] = pending_task_params_[task_type];
pending_task_params_.erase(task_type);
task_finished_callbacks_[task_type] = std::move(callback);
}
void TaskManagerImpl::OnStopScheduledTask(DownloadTaskType task_type) {
DCHECK(IsRunningTask(task_type));
current_task_params_.erase(task_type);
task_finished_callbacks_.erase(task_type);
if (pending_task_params_.find(task_type) != pending_task_params_.end()) {
auto params = pending_task_params_[task_type];
pending_task_params_.erase(task_type);
ScheduleTask(task_type, params);
}
}
bool TaskManagerImpl::IsRunningTask(DownloadTaskType task_type) const {
return task_finished_callbacks_.find(task_type) !=
task_finished_callbacks_.end();
}
void TaskManagerImpl::NotifyTaskFinished(DownloadTaskType task_type,
bool needs_reschedule) {
if (!IsRunningTask(task_type))
return;
current_task_params_.erase(task_type);
bool has_pending_params =
pending_task_params_.find(task_type) != pending_task_params_.end();
std::move(task_finished_callbacks_[task_type])
.Run(needs_reschedule && !has_pending_params);
task_finished_callbacks_.erase(task_type);
if (has_pending_params) {
auto params = pending_task_params_[task_type];
pending_task_params_.erase(task_type);
ScheduleTask(task_type, params);
}
}
} // namespace download
|
