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
|
// 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/public/background_service/basic_task_scheduler.h"
#include "base/functional/bind.h"
#include "base/task/sequenced_task_runner.h"
#include "base/task/single_thread_task_runner.h"
#include "base/task/task_traits.h"
#include "base/time/time.h"
#include "components/download/public/background_service/background_download_service.h"
namespace download {
BasicTaskScheduler::BasicTaskScheduler(
const base::RepeatingCallback<BackgroundDownloadService*()>&
get_download_service)
: get_download_service_(get_download_service) {}
BasicTaskScheduler::~BasicTaskScheduler() = default;
void BasicTaskScheduler::ScheduleTask(download::DownloadTaskType task_type,
bool require_unmetered_network,
bool require_charging,
int optimal_battery_percentage,
int64_t window_start_time_seconds,
int64_t window_end_time_seconds) {
scheduled_tasks_[task_type].Reset(
base::BindOnce(&BasicTaskScheduler::RunScheduledTask,
weak_factory_.GetWeakPtr(), task_type));
base::SingleThreadTaskRunner::GetCurrentDefault()->PostDelayedTask(
FROM_HERE, scheduled_tasks_[task_type].callback(),
base::Seconds(window_start_time_seconds));
}
void BasicTaskScheduler::CancelTask(download::DownloadTaskType task_type) {
scheduled_tasks_[task_type].Cancel();
}
void BasicTaskScheduler::RunScheduledTask(
download::DownloadTaskType task_type) {
get_download_service_.Run()->OnStartScheduledTask(
task_type, base::BindOnce(&BasicTaskScheduler::OnTaskFinished,
weak_factory_.GetWeakPtr()));
}
void BasicTaskScheduler::OnTaskFinished(bool reschedule) {
// TODO(shaktisahu): Cache the original scheduling params and re-post task in
// case it needs reschedule.
}
} // namespace download
|
