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

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

Issue 1906083002: TaskScheduler: Remove base/task_scheduler/utils.h/.cc (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@sched_2_stack
Patch Set: typos Created 4 years, 8 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 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 #ifndef BASE_TASK_SCHEDULER_SCHEDULER_THREAD_POOL_H_ 5 #ifndef BASE_TASK_SCHEDULER_SCHEDULER_THREAD_POOL_IMPL_H_
6 #define BASE_TASK_SCHEDULER_SCHEDULER_THREAD_POOL_H_ 6 #define BASE_TASK_SCHEDULER_SCHEDULER_THREAD_POOL_IMPL_H_
7 7
8 #include <stddef.h> 8 #include <stddef.h>
9 9
10 #include <memory> 10 #include <memory>
11 #include <vector> 11 #include <vector>
12 12
13 #include "base/base_export.h" 13 #include "base/base_export.h"
14 #include "base/callback.h" 14 #include "base/callback.h"
15 #include "base/macros.h" 15 #include "base/macros.h"
16 #include "base/memory/ref_counted.h" 16 #include "base/memory/ref_counted.h"
17 #include "base/synchronization/condition_variable.h" 17 #include "base/synchronization/condition_variable.h"
18 #include "base/task_runner.h" 18 #include "base/task_runner.h"
19 #include "base/task_scheduler/priority_queue.h" 19 #include "base/task_scheduler/priority_queue.h"
20 #include "base/task_scheduler/scheduler_lock.h" 20 #include "base/task_scheduler/scheduler_lock.h"
21 #include "base/task_scheduler/scheduler_task_executor.h" 21 #include "base/task_scheduler/scheduler_thread_pool.h"
22 #include "base/task_scheduler/scheduler_worker_thread.h" 22 #include "base/task_scheduler/scheduler_worker_thread.h"
23 #include "base/task_scheduler/scheduler_worker_thread_stack.h" 23 #include "base/task_scheduler/scheduler_worker_thread_stack.h"
24 #include "base/task_scheduler/sequence.h" 24 #include "base/task_scheduler/sequence.h"
25 #include "base/task_scheduler/task.h" 25 #include "base/task_scheduler/task.h"
26 #include "base/task_scheduler/task_traits.h" 26 #include "base/task_scheduler/task_traits.h"
27 #include "base/threading/platform_thread.h" 27 #include "base/threading/platform_thread.h"
28 28
29 namespace base { 29 namespace base {
30 namespace internal { 30 namespace internal {
31 31
32 class DelayedTaskManager; 32 class DelayedTaskManager;
33 struct SequenceSortKey; 33 struct SequenceSortKey;
34 class TaskTracker; 34 class TaskTracker;
35 35
36 // A pool of threads that run Tasks. This class is thread-safe. 36 // A pool of threads that run Tasks. This class is thread-safe.
37 class BASE_EXPORT SchedulerThreadPool : public SchedulerTaskExecutor { 37 class BASE_EXPORT SchedulerThreadPoolImpl : public SchedulerThreadPool {
38 public: 38 public:
39 // Callback invoked when a Sequence isn't empty after a worker thread pops a 39 // Callback invoked when a Sequence isn't empty after a worker thread pops a
40 // Task from it. 40 // Task from it.
41 using EnqueueSequenceCallback = Callback<void(scoped_refptr<Sequence>)>; 41 using ReEnqueueSequenceCallback = Callback<void(scoped_refptr<Sequence>)>;
42 42
43 // Destroying a SchedulerThreadPool returned by CreateThreadPool() is not 43 // Destroying a SchedulerThreadPool returned by CreateThreadPool() is not
44 // allowed in production; it is always leaked. In tests, it can only be 44 // allowed in production; it is always leaked. In tests, it can only be
45 // destroyed after JoinForTesting() has returned. 45 // destroyed after JoinForTesting() has returned.
46 ~SchedulerThreadPool() override; 46 ~SchedulerThreadPoolImpl() override;
47 47
48 // Creates a SchedulerThreadPool with up to |max_threads| threads of priority 48 // Creates a SchedulerThreadPool with up to |max_threads| threads of priority
49 // |thread_priority|. |enqueue_sequence_callback| will be invoked after a 49 // |thread_priority|. |re_enqueue_sequence_callback| will be invoked after a
50 // thread of this thread pool tries to run a Task. |task_tracker| is used to 50 // thread of this thread pool tries to run a Task. |task_tracker| is used to
51 // handle shutdown behavior of Tasks. |delayed_task_manager| handles Tasks 51 // handle shutdown behavior of Tasks. |delayed_task_manager| handles Tasks
52 // posted with a delay. Returns nullptr on failure to create a thread pool 52 // posted with a delay. Returns nullptr on failure to create a thread pool
53 // with at least one thread. 53 // with at least one thread.
54 static std::unique_ptr<SchedulerThreadPool> CreateThreadPool( 54 static std::unique_ptr<SchedulerThreadPoolImpl> CreateThreadPool(
55 ThreadPriority thread_priority, 55 ThreadPriority thread_priority,
56 size_t max_threads, 56 size_t max_threads,
57 const EnqueueSequenceCallback& enqueue_sequence_callback, 57 const ReEnqueueSequenceCallback& re_enqueue_sequence_callback,
58 TaskTracker* task_tracker, 58 TaskTracker* task_tracker,
59 DelayedTaskManager* delayed_task_manager); 59 DelayedTaskManager* delayed_task_manager);
60 60
61 // Returns a TaskRunner whose PostTask invocations will result in scheduling
62 // Tasks with |traits| and |execution_mode| in this thread pool.
63 scoped_refptr<TaskRunner> CreateTaskRunnerWithTraits(
64 const TaskTraits& traits,
65 ExecutionMode execution_mode);
66
67 // Inserts |sequence| into this thread pool's shared priority queue with
68 // |sequence_sort_key|. Must only be called from a worker thread to put
69 // |sequence| back into a PriorityQueue after running a Task from it. The
70 // worker thread doesn't have to belong to this thread pool.
71 void EnqueueSequence(scoped_refptr<Sequence> sequence,
72 const SequenceSortKey& sequence_sort_key);
73
74 // Waits until all threads are idle. 61 // Waits until all threads are idle.
75 void WaitForAllWorkerThreadsIdleForTesting(); 62 void WaitForAllWorkerThreadsIdleForTesting();
76 63
77 // Joins all threads of this thread pool. Tasks that are already running are 64 // Joins all threads of this thread pool. Tasks that are already running are
78 // allowed to complete their execution. This can only be called once. 65 // allowed to complete their execution. This can only be called once.
79 void JoinForTesting(); 66 void JoinForTesting();
80 67
81 // SchedulerTaskExecutor: 68 // SchedulerThreadPool:
82 void PostTaskWithSequence(std::unique_ptr<Task> task, 69 scoped_refptr<TaskRunner> CreateTaskRunnerWithTraits(
70 const TaskTraits& traits,
71 ExecutionMode execution_mode) override;
72 void ReEnqueueSequence(scoped_refptr<Sequence> sequence,
73 const SequenceSortKey& sequence_sort_key) override;
74 bool PostTaskWithSequence(std::unique_ptr<Task> task,
83 scoped_refptr<Sequence> sequence) override; 75 scoped_refptr<Sequence> sequence) override;
76 void PostTaskWithSequenceNow(std::unique_ptr<Task> task,
77 scoped_refptr<Sequence> sequence) override;
84 78
85 private: 79 private:
86 class SchedulerWorkerThreadDelegateImpl; 80 class SchedulerWorkerThreadDelegateImpl;
87 81
88 SchedulerThreadPool(const EnqueueSequenceCallback& enqueue_sequence_callback, 82 SchedulerThreadPoolImpl(
89 TaskTracker* task_tracker, 83 const ReEnqueueSequenceCallback& re_enqueue_sequence_callback,
90 DelayedTaskManager* delayed_task_manager); 84 TaskTracker* task_tracker,
85 DelayedTaskManager* delayed_task_manager);
91 86
92 bool Initialize(ThreadPriority thread_priority, size_t max_threads); 87 bool Initialize(ThreadPriority thread_priority, size_t max_threads);
93 88
94 // Wakes up the last thread from this thread pool to go idle, if any. 89 // Wakes up the last thread from this thread pool to go idle, if any.
95 void WakeUpOneThread(); 90 void WakeUpOneThread();
96 91
97 // Adds |worker_thread| to |idle_worker_threads_stack_|. 92 // Adds |worker_thread| to |idle_worker_threads_stack_|.
98 void AddToIdleWorkerThreadsStack(SchedulerWorkerThread* worker_thread); 93 void AddToIdleWorkerThreadsStack(SchedulerWorkerThread* worker_thread);
99 94
100 // PriorityQueue from which all threads of this thread pool get work. 95 // PriorityQueue from which all threads of this thread pool get work.
(...skipping 18 matching lines...) Expand all
119 114
120 // Signaled once JoinForTesting() has returned. 115 // Signaled once JoinForTesting() has returned.
121 WaitableEvent join_for_testing_returned_; 116 WaitableEvent join_for_testing_returned_;
122 117
123 // Delegate for all worker threads in this pool. 118 // Delegate for all worker threads in this pool.
124 std::unique_ptr<SchedulerWorkerThread::Delegate> worker_thread_delegate_; 119 std::unique_ptr<SchedulerWorkerThread::Delegate> worker_thread_delegate_;
125 120
126 TaskTracker* const task_tracker_; 121 TaskTracker* const task_tracker_;
127 DelayedTaskManager* const delayed_task_manager_; 122 DelayedTaskManager* const delayed_task_manager_;
128 123
129 DISALLOW_COPY_AND_ASSIGN(SchedulerThreadPool); 124 DISALLOW_COPY_AND_ASSIGN(SchedulerThreadPoolImpl);
130 }; 125 };
131 126
132 } // namespace internal 127 } // namespace internal
133 } // namespace base 128 } // namespace base
134 129
135 #endif // BASE_TASK_SCHEDULER_SCHEDULER_THREAD_POOL_H_ 130 #endif // BASE_TASK_SCHEDULER_SCHEDULER_THREAD_POOL_IMPL_H_
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698