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
|
// Copyright (c) 2012 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 "base/threading/worker_pool.h"
#include <utility>
#include "base/bind.h"
#include "base/compiler_specific.h"
#include "base/debug/leak_annotations.h"
#include "base/macros.h"
#include "base/task_runner.h"
#include "base/threading/post_task_and_reply_impl.h"
#include "base/tracked_objects.h"
namespace base {
namespace {
class PostTaskAndReplyWorkerPool : public internal::PostTaskAndReplyImpl {
public:
explicit PostTaskAndReplyWorkerPool(bool task_is_slow)
: task_is_slow_(task_is_slow) {
}
~PostTaskAndReplyWorkerPool() override = default;
private:
bool PostTask(const tracked_objects::Location& from_here,
Closure task) override {
return WorkerPool::PostTask(from_here, std::move(task), task_is_slow_);
}
bool task_is_slow_;
};
// WorkerPoolTaskRunner ---------------------------------------------
// A TaskRunner which posts tasks to a WorkerPool with a
// fixed ShutdownBehavior.
//
// Note that this class is RefCountedThreadSafe (inherited from TaskRunner).
class WorkerPoolTaskRunner : public TaskRunner {
public:
explicit WorkerPoolTaskRunner(bool tasks_are_slow);
// TaskRunner implementation
bool PostDelayedTask(const tracked_objects::Location& from_here,
Closure task,
TimeDelta delay) override;
bool RunsTasksOnCurrentThread() const override;
private:
~WorkerPoolTaskRunner() override;
// Helper function for posting a delayed task. Asserts that the delay is
// zero because non-zero delays are not supported.
bool PostDelayedTaskAssertZeroDelay(
const tracked_objects::Location& from_here,
Closure task,
base::TimeDelta delay);
const bool tasks_are_slow_;
DISALLOW_COPY_AND_ASSIGN(WorkerPoolTaskRunner);
};
WorkerPoolTaskRunner::WorkerPoolTaskRunner(bool tasks_are_slow)
: tasks_are_slow_(tasks_are_slow) {
}
WorkerPoolTaskRunner::~WorkerPoolTaskRunner() {
}
bool WorkerPoolTaskRunner::PostDelayedTask(
const tracked_objects::Location& from_here,
Closure task,
TimeDelta delay) {
return PostDelayedTaskAssertZeroDelay(from_here, std::move(task), delay);
}
bool WorkerPoolTaskRunner::RunsTasksOnCurrentThread() const {
return WorkerPool::RunsTasksOnCurrentThread();
}
bool WorkerPoolTaskRunner::PostDelayedTaskAssertZeroDelay(
const tracked_objects::Location& from_here,
Closure task,
base::TimeDelta delay) {
DCHECK_EQ(delay.InMillisecondsRoundedUp(), 0)
<< "WorkerPoolTaskRunner does not support non-zero delays";
return WorkerPool::PostTask(from_here, std::move(task), tasks_are_slow_);
}
struct TaskRunnerHolder {
TaskRunnerHolder() {
taskrunners_[0] = new WorkerPoolTaskRunner(false);
taskrunners_[1] = new WorkerPoolTaskRunner(true);
}
scoped_refptr<TaskRunner> taskrunners_[2];
};
} // namespace
bool WorkerPool::PostTaskAndReply(const tracked_objects::Location& from_here,
Closure task,
Closure reply,
bool task_is_slow) {
// Do not report PostTaskAndReplyRelay leaks in tests. There's nothing we can
// do about them because WorkerPool doesn't have a flushing API.
// http://crbug.com/248513
// http://crbug.com/290897
// Note: this annotation does not cover tasks posted through a TaskRunner.
ANNOTATE_SCOPED_MEMORY_LEAK;
return PostTaskAndReplyWorkerPool(task_is_slow)
.PostTaskAndReply(from_here, std::move(task), std::move(reply));
}
// static
const scoped_refptr<TaskRunner>&
WorkerPool::GetTaskRunner(bool tasks_are_slow) {
static auto* task_runner_holder = new TaskRunnerHolder();
return task_runner_holder->taskrunners_[tasks_are_slow];
}
} // namespace base
|
