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
|
// Copyright 2014 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef COMPONENTS_DOMAIN_RELIABILITY_DISPATCHER_H_
#define COMPONENTS_DOMAIN_RELIABILITY_DISPATCHER_H_
#include <memory>
#include <set>
#include "base/containers/unique_ptr_adapters.h"
#include "base/functional/callback_forward.h"
#include "base/memory/raw_ptr.h"
#include "base/time/time.h"
#include "components/domain_reliability/domain_reliability_export.h"
namespace domain_reliability {
class MockableTime;
// Runs tasks during a specified interval. Calling |RunEligibleTasks| gives any
// task a chance to run early (if the minimum delay has already passed); tasks
// that aren't run early will be run once their maximum delay has passed.
//
// (See scheduler.h for an explanation of how the intervals are chosen.)
class DOMAIN_RELIABILITY_EXPORT DomainReliabilityDispatcher {
public:
explicit DomainReliabilityDispatcher(MockableTime* time);
DomainReliabilityDispatcher(const DomainReliabilityDispatcher&) = delete;
DomainReliabilityDispatcher& operator=(const DomainReliabilityDispatcher&) =
delete;
~DomainReliabilityDispatcher();
// Schedules |task| to be executed between |min_delay| and |max_delay| from
// now. The task will be run at most |max_delay| from now; once |min_delay|
// has passed, any call to |RunEligibleTasks| will run the task earlier than
// that.
void ScheduleTask(base::OnceClosure task,
base::TimeDelta min_delay,
base::TimeDelta max_delay);
// Runs all existing tasks whose minimum delay has already passed. Does not
// run tasks added by those existing tasks, even if their minimum delay has
// already passed.
void RunEligibleTasks();
// Runs all waiting or eligible tasks, regardless of whether their minimum
// delay has passed.
void RunAllTasksForTesting();
private:
struct Task;
// Adds |task| to the set of all tasks, but not the set of eligible tasks.
void MakeTaskWaiting(Task* task);
// Adds |task| to the set of eligible tasks, and also the set of all tasks
// if not already there.
void MakeTaskEligible(Task* task);
// Runs |task|'s callback, removes it from both sets, and deletes it.
void RunAndDeleteTask(Task* task);
raw_ptr<MockableTime> time_;
std::set<std::unique_ptr<Task>, base::UniquePtrComparator> tasks_;
std::set<raw_ptr<Task, SetExperimental>> eligible_tasks_;
};
} // namespace domain_reliability
#endif // COMPONENTS_DOMAIN_RELIABILITY_DISPATCHER_H_
|
