OLD | NEW |
1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "base/test/scoped_task_scheduler.h" | 5 #include "base/test/scoped_task_scheduler.h" |
6 | 6 |
7 #include <memory> | 7 #include <memory> |
8 #include <utility> | 8 #include <utility> |
9 #include <vector> | 9 #include <vector> |
10 | 10 |
11 #include "base/bind.h" | 11 #include "base/bind.h" |
12 #include "base/callback.h" | 12 #include "base/callback.h" |
13 #include "base/logging.h" | 13 #include "base/logging.h" |
14 #include "base/memory/ptr_util.h" | 14 #include "base/memory/ptr_util.h" |
15 #include "base/memory/ref_counted.h" | 15 #include "base/memory/ref_counted.h" |
16 #include "base/message_loop/message_loop.h" | 16 #include "base/message_loop/message_loop.h" |
17 #include "base/metrics/histogram_base.h" | 17 #include "base/metrics/histogram_base.h" |
18 #include "base/run_loop.h" | 18 #include "base/run_loop.h" |
19 #include "base/sequence_token.h" | 19 #include "base/sequence_token.h" |
20 #include "base/sequenced_task_runner.h" | 20 #include "base/sequenced_task_runner.h" |
21 #include "base/single_thread_task_runner.h" | 21 #include "base/single_thread_task_runner.h" |
22 #include "base/task_runner.h" | 22 #include "base/task_runner.h" |
| 23 #include "base/task_scheduler/single_thread_task_runner_thread_mode.h" |
23 #include "base/task_scheduler/task.h" | 24 #include "base/task_scheduler/task.h" |
24 #include "base/task_scheduler/task_scheduler.h" | 25 #include "base/task_scheduler/task_scheduler.h" |
25 #include "base/task_scheduler/task_tracker.h" | 26 #include "base/task_scheduler/task_tracker.h" |
26 #include "base/threading/thread_task_runner_handle.h" | 27 #include "base/threading/thread_task_runner_handle.h" |
27 #include "base/time/time.h" | 28 #include "base/time/time.h" |
28 #include "build/build_config.h" | 29 #include "build/build_config.h" |
29 | 30 |
30 #if defined(OS_WIN) | 31 #if defined(OS_WIN) |
31 #include "base/win/scoped_com_initializer.h" | 32 #include "base/win/scoped_com_initializer.h" |
32 #endif // defined(OS_WIN) | 33 #endif // defined(OS_WIN) |
(...skipping 20 matching lines...) Expand all Loading... |
53 void Start(const TaskScheduler::InitParams& init_params) override; | 54 void Start(const TaskScheduler::InitParams& init_params) override; |
54 void PostDelayedTaskWithTraits(const tracked_objects::Location& from_here, | 55 void PostDelayedTaskWithTraits(const tracked_objects::Location& from_here, |
55 const TaskTraits& traits, | 56 const TaskTraits& traits, |
56 OnceClosure task, | 57 OnceClosure task, |
57 TimeDelta delay) override; | 58 TimeDelta delay) override; |
58 scoped_refptr<TaskRunner> CreateTaskRunnerWithTraits( | 59 scoped_refptr<TaskRunner> CreateTaskRunnerWithTraits( |
59 const TaskTraits& traits) override; | 60 const TaskTraits& traits) override; |
60 scoped_refptr<SequencedTaskRunner> CreateSequencedTaskRunnerWithTraits( | 61 scoped_refptr<SequencedTaskRunner> CreateSequencedTaskRunnerWithTraits( |
61 const TaskTraits& traits) override; | 62 const TaskTraits& traits) override; |
62 scoped_refptr<SingleThreadTaskRunner> CreateSingleThreadTaskRunnerWithTraits( | 63 scoped_refptr<SingleThreadTaskRunner> CreateSingleThreadTaskRunnerWithTraits( |
63 const TaskTraits& traits) override; | 64 const TaskTraits& traits, |
| 65 SingleThreadTaskRunnerThreadMode thread_mode) override; |
64 #if defined(OS_WIN) | 66 #if defined(OS_WIN) |
65 scoped_refptr<SingleThreadTaskRunner> CreateCOMSTATaskRunnerWithTraits( | 67 scoped_refptr<SingleThreadTaskRunner> CreateCOMSTATaskRunnerWithTraits( |
66 const TaskTraits& traits) override; | 68 const TaskTraits& traits, |
| 69 SingleThreadTaskRunnerThreadMode thread_mode) override; |
67 #endif // defined(OS_WIN) | 70 #endif // defined(OS_WIN) |
68 std::vector<const HistogramBase*> GetHistograms() const override; | 71 std::vector<const HistogramBase*> GetHistograms() const override; |
69 int GetMaxConcurrentTasksWithTraitsDeprecated( | 72 int GetMaxConcurrentTasksWithTraitsDeprecated( |
70 const TaskTraits& traits) const override; | 73 const TaskTraits& traits) const override; |
71 void Shutdown() override; | 74 void Shutdown() override; |
72 void FlushForTesting() override; | 75 void FlushForTesting() override; |
73 void JoinForTesting() override; | 76 void JoinForTesting() override; |
74 | 77 |
75 // Posts |task| to this TaskScheduler with |sequence_token|. Returns true on | 78 // Posts |task| to this TaskScheduler with |sequence_token|. Returns true on |
76 // success. | 79 // success. |
(...skipping 112 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
189 | 192 |
190 scoped_refptr<SequencedTaskRunner> | 193 scoped_refptr<SequencedTaskRunner> |
191 TestTaskScheduler::CreateSequencedTaskRunnerWithTraits( | 194 TestTaskScheduler::CreateSequencedTaskRunnerWithTraits( |
192 const TaskTraits& traits) { | 195 const TaskTraits& traits) { |
193 return make_scoped_refptr( | 196 return make_scoped_refptr( |
194 new TestTaskSchedulerTaskRunner(this, ExecutionMode::SEQUENCED, traits)); | 197 new TestTaskSchedulerTaskRunner(this, ExecutionMode::SEQUENCED, traits)); |
195 } | 198 } |
196 | 199 |
197 scoped_refptr<SingleThreadTaskRunner> | 200 scoped_refptr<SingleThreadTaskRunner> |
198 TestTaskScheduler::CreateSingleThreadTaskRunnerWithTraits( | 201 TestTaskScheduler::CreateSingleThreadTaskRunnerWithTraits( |
199 const TaskTraits& traits) { | 202 const TaskTraits& traits, |
| 203 SingleThreadTaskRunnerThreadMode thread_mode) { |
200 return make_scoped_refptr(new TestTaskSchedulerTaskRunner( | 204 return make_scoped_refptr(new TestTaskSchedulerTaskRunner( |
201 this, ExecutionMode::SINGLE_THREADED, traits)); | 205 this, ExecutionMode::SINGLE_THREADED, traits)); |
202 } | 206 } |
203 | 207 |
204 #if defined(OS_WIN) | 208 #if defined(OS_WIN) |
205 scoped_refptr<SingleThreadTaskRunner> | 209 scoped_refptr<SingleThreadTaskRunner> |
206 TestTaskScheduler::CreateCOMSTATaskRunnerWithTraits(const TaskTraits& traits) { | 210 TestTaskScheduler::CreateCOMSTATaskRunnerWithTraits( |
| 211 const TaskTraits& traits, |
| 212 SingleThreadTaskRunnerThreadMode thread_mode) { |
207 EnsureCOMSTA(); | 213 EnsureCOMSTA(); |
208 return make_scoped_refptr(new TestTaskSchedulerTaskRunner( | 214 return make_scoped_refptr(new TestTaskSchedulerTaskRunner( |
209 this, ExecutionMode::SINGLE_THREADED, traits)); | 215 this, ExecutionMode::SINGLE_THREADED, traits)); |
210 } | 216 } |
211 #endif // defined(OS_WIN) | 217 #endif // defined(OS_WIN) |
212 | 218 |
213 std::vector<const HistogramBase*> TestTaskScheduler::GetHistograms() const { | 219 std::vector<const HistogramBase*> TestTaskScheduler::GetHistograms() const { |
214 NOTREACHED(); | 220 NOTREACHED(); |
215 return std::vector<const HistogramBase*>(); | 221 return std::vector<const HistogramBase*>(); |
216 } | 222 } |
(...skipping 114 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
331 DCHECK_EQ(task_scheduler_, TaskScheduler::GetInstance()); | 337 DCHECK_EQ(task_scheduler_, TaskScheduler::GetInstance()); |
332 | 338 |
333 // Per contract, call JoinForTesting() before deleting the TaskScheduler. | 339 // Per contract, call JoinForTesting() before deleting the TaskScheduler. |
334 TaskScheduler::GetInstance()->JoinForTesting(); | 340 TaskScheduler::GetInstance()->JoinForTesting(); |
335 | 341 |
336 TaskScheduler::SetInstance(nullptr); | 342 TaskScheduler::SetInstance(nullptr); |
337 } | 343 } |
338 | 344 |
339 } // namespace test | 345 } // namespace test |
340 } // namespace base | 346 } // namespace base |
OLD | NEW |