OLD | NEW |
| (Empty) |
1 // Copyright 2017 The Chromium Authors. All rights reserved. | |
2 // Use of this source code is governed by a BSD-style license that can be | |
3 // found in the LICENSE file. | |
4 | |
5 #ifndef BASE_TASK_SCHEDULER_SCHEDULER_SINGLE_THREAD_TASK_RUNNER_MANAGER_H_ | |
6 #define BASE_TASK_SCHEDULER_SCHEDULER_SINGLE_THREAD_TASK_RUNNER_MANAGER_H_ | |
7 | |
8 #include <vector> | |
9 | |
10 #include "base/base_export.h" | |
11 #include "base/macros.h" | |
12 #include "base/memory/ref_counted.h" | |
13 #include "base/task_scheduler/scheduler_lock.h" | |
14 #include "base/task_scheduler/scheduler_worker_pool_params.h" | |
15 #include "base/task_scheduler/task_scheduler.h" | |
16 | |
17 namespace base { | |
18 | |
19 class TaskTraits; | |
20 class SingleThreadTaskRunner; | |
21 | |
22 namespace internal { | |
23 | |
24 class DelayedTaskManager; | |
25 class SchedulerWorker; | |
26 class TaskTracker; | |
27 | |
28 class BASE_EXPORT SchedulerSingleThreadTaskRunnerManager final { | |
29 public: | |
30 SchedulerSingleThreadTaskRunnerManager( | |
31 const std::vector<SchedulerWorkerPoolParams>& worker_pool_params_vector, | |
32 const TaskScheduler::WorkerPoolIndexForTraitsCallback& | |
33 worker_pool_index_for_traits_callback, | |
34 TaskTracker* task_tracker, | |
35 DelayedTaskManager* delayed_task_manager); | |
36 ~SchedulerSingleThreadTaskRunnerManager(); | |
37 | |
38 scoped_refptr<SingleThreadTaskRunner> CreateSingleThreadTaskRunnerWithTraits( | |
39 const TaskTraits& traits); | |
40 | |
41 void JoinForTesting(); | |
42 | |
43 private: | |
44 class SchedulerSingleThreadTaskRunner; | |
45 | |
46 SchedulerWorker* CreateAndRegisterSchedulerWorker( | |
47 const SchedulerWorkerPoolParams& params); | |
48 | |
49 void UnregisterSchedulerWorker(SchedulerWorker* worker); | |
50 | |
51 const std::vector<SchedulerWorkerPoolParams> worker_pool_params_vector_; | |
52 const TaskScheduler::WorkerPoolIndexForTraitsCallback | |
53 worker_pool_index_for_traits_callback_; | |
54 TaskTracker* const task_tracker_; | |
55 DelayedTaskManager* const delayed_task_manager_; | |
56 | |
57 // Synchronizes access to |workers_| and |worker_id_|. | |
58 SchedulerLock workers_lock_; | |
59 std::vector<scoped_refptr<SchedulerWorker>> workers_; | |
60 int next_worker_id_ = 0; | |
61 | |
62 DISALLOW_COPY_AND_ASSIGN(SchedulerSingleThreadTaskRunnerManager); | |
63 }; | |
64 | |
65 } // namespace internal | |
66 } // namespace base | |
67 | |
68 #endif // BASE_TASK_SCHEDULER_SCHEDULER_SINGLE_THREAD_TASK_RUNNER_MANAGER_H_ | |
OLD | NEW |