Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(405)

Side by Side Diff: base/task_scheduler/scheduler_single_thread_task_runner_manager.h

Issue 2765703002: Add the COM STA Task Runner (Closed)
Patch Set: Created 3 years, 9 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
1 // Copyright 2017 The Chromium Authors. All rights reserved. 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 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 #ifndef BASE_TASK_SCHEDULER_SCHEDULER_SINGLE_THREAD_TASK_RUNNER_MANAGER_H_ 5 #ifndef BASE_TASK_SCHEDULER_SCHEDULER_SINGLE_THREAD_TASK_RUNNER_MANAGER_H_
6 #define BASE_TASK_SCHEDULER_SCHEDULER_SINGLE_THREAD_TASK_RUNNER_MANAGER_H_ 6 #define BASE_TASK_SCHEDULER_SCHEDULER_SINGLE_THREAD_TASK_RUNNER_MANAGER_H_
7 7
8 #include <vector> 8 #include <vector>
9 9
10 #include "base/atomicops.h" 10 #include "base/atomicops.h"
11 #include "base/base_export.h" 11 #include "base/base_export.h"
12 #include "base/logging.h" 12 #include "base/logging.h"
13 #include "base/macros.h" 13 #include "base/macros.h"
14 #include "base/memory/ref_counted.h" 14 #include "base/memory/ref_counted.h"
15 #include "base/task_scheduler/scheduler_lock.h" 15 #include "base/task_scheduler/scheduler_lock.h"
16 #include "base/task_scheduler/scheduler_worker_pool_params.h" 16 #include "base/task_scheduler/scheduler_worker_pool_params.h"
17 #include "base/task_scheduler/task_scheduler.h" 17 #include "base/task_scheduler/task_scheduler.h"
fdoray 2017/03/21 15:23:12 #include "build/build_config.h"
robliao 2017/03/21 20:00:20 Done.
18 18
19 namespace base { 19 namespace base {
20 20
21 class TaskTraits; 21 class TaskTraits;
22 class SingleThreadTaskRunner; 22 class SingleThreadTaskRunner;
23 23
24 namespace internal { 24 namespace internal {
25 25
26 class DelayedTaskManager; 26 class DelayedTaskManager;
27 class SchedulerWorker; 27 class SchedulerWorker;
28 class TaskTracker; 28 class TaskTracker;
29 29
30 class BASE_EXPORT SchedulerSingleThreadTaskRunnerManager final { 30 class BASE_EXPORT SchedulerSingleThreadTaskRunnerManager final {
31 public: 31 public:
32 SchedulerSingleThreadTaskRunnerManager( 32 SchedulerSingleThreadTaskRunnerManager(
33 const std::vector<SchedulerWorkerPoolParams>& worker_pool_params_vector, 33 const std::vector<SchedulerWorkerPoolParams>& worker_pool_params_vector,
34 const TaskScheduler::WorkerPoolIndexForTraitsCallback& 34 const TaskScheduler::WorkerPoolIndexForTraitsCallback&
35 worker_pool_index_for_traits_callback, 35 worker_pool_index_for_traits_callback,
36 TaskTracker* task_tracker, 36 TaskTracker* task_tracker,
37 DelayedTaskManager* delayed_task_manager); 37 DelayedTaskManager* delayed_task_manager);
38 ~SchedulerSingleThreadTaskRunnerManager(); 38 ~SchedulerSingleThreadTaskRunnerManager();
39 39
40 scoped_refptr<SingleThreadTaskRunner> CreateSingleThreadTaskRunnerWithTraits( 40 scoped_refptr<SingleThreadTaskRunner> CreateSingleThreadTaskRunnerWithTraits(
41 const TaskTraits& traits); 41 const TaskTraits& traits);
42 42
43 #if defined(OS_WIN)
44 scoped_refptr<SingleThreadTaskRunner> CreateCOMSTATaskRunnerWithTraits(
45 const TaskTraits& traits);
46 #endif
47
43 void JoinForTesting(); 48 void JoinForTesting();
44 49
45 private: 50 private:
46 class SchedulerSingleThreadTaskRunner; 51 class SchedulerSingleThreadTaskRunner;
47 52
53 enum DelegateType {
fdoray 2017/03/21 15:23:12 enum class
robliao 2017/03/21 20:00:20 Done.
54 REGULAR,
gab 2017/03/21 21:09:34 How about "DEFAULT"? Sounds more natural to me tha
robliao 2017/03/21 22:25:32 sgtm. Done.
55 #if defined(OS_WIN)
56 COM_STA,
57 #endif
58 };
59
48 SchedulerWorker* CreateAndRegisterSchedulerWorker( 60 SchedulerWorker* CreateAndRegisterSchedulerWorker(
49 const SchedulerWorkerPoolParams& params); 61 const SchedulerWorkerPoolParams& params,
62 DelegateType delegate_type);
50 63
51 void UnregisterSchedulerWorker(SchedulerWorker* worker); 64 void UnregisterSchedulerWorker(SchedulerWorker* worker);
52 65
53 const std::vector<SchedulerWorkerPoolParams> worker_pool_params_vector_; 66 const std::vector<SchedulerWorkerPoolParams> worker_pool_params_vector_;
54 const TaskScheduler::WorkerPoolIndexForTraitsCallback 67 const TaskScheduler::WorkerPoolIndexForTraitsCallback
55 worker_pool_index_for_traits_callback_; 68 worker_pool_index_for_traits_callback_;
56 TaskTracker* const task_tracker_; 69 TaskTracker* const task_tracker_;
57 DelayedTaskManager* const delayed_task_manager_; 70 DelayedTaskManager* const delayed_task_manager_;
58 71
59 // Synchronizes access to |workers_| and |worker_id_|. 72 // Synchronizes access to |workers_| and |worker_id_|.
60 SchedulerLock workers_lock_; 73 SchedulerLock workers_lock_;
61 std::vector<scoped_refptr<SchedulerWorker>> workers_; 74 std::vector<scoped_refptr<SchedulerWorker>> workers_;
62 int next_worker_id_ = 0; 75 int next_worker_id_ = 0;
63 76
64 #if DCHECK_IS_ON() 77 #if DCHECK_IS_ON()
65 subtle::Atomic32 workers_unregistered_during_join_ = 0; 78 subtle::Atomic32 workers_unregistered_during_join_ = 0;
66 #endif 79 #endif
67 80
68 DISALLOW_COPY_AND_ASSIGN(SchedulerSingleThreadTaskRunnerManager); 81 DISALLOW_COPY_AND_ASSIGN(SchedulerSingleThreadTaskRunnerManager);
69 }; 82 };
70 83
71 } // namespace internal 84 } // namespace internal
72 } // namespace base 85 } // namespace base
73 86
74 #endif // BASE_TASK_SCHEDULER_SCHEDULER_SINGLE_THREAD_TASK_RUNNER_MANAGER_H_ 87 #endif // BASE_TASK_SCHEDULER_SCHEDULER_SINGLE_THREAD_TASK_RUNNER_MANAGER_H_
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698