| 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 #ifndef BASE_TASK_SCHEDULER_SCHEDULER_WORKER_THREAD_H_ | 5 #ifndef BASE_TASK_SCHEDULER_SCHEDULER_WORKER_THREAD_H_ |
| 6 #define BASE_TASK_SCHEDULER_SCHEDULER_WORKER_THREAD_H_ | 6 #define BASE_TASK_SCHEDULER_SCHEDULER_WORKER_THREAD_H_ |
| 7 | 7 |
| 8 #include <memory> | 8 #include <memory> |
| 9 | 9 |
| 10 #include "base/base_export.h" | 10 #include "base/base_export.h" |
| (...skipping 35 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 46 | 46 |
| 47 // Called when |sequence| isn't empty after the SchedulerWorkerThread pops a | 47 // Called when |sequence| isn't empty after the SchedulerWorkerThread pops a |
| 48 // Task from it. |sequence| is the last Sequence returned by GetWork(). | 48 // Task from it. |sequence| is the last Sequence returned by GetWork(). |
| 49 virtual void EnqueueSequence(scoped_refptr<Sequence> sequence) = 0; | 49 virtual void EnqueueSequence(scoped_refptr<Sequence> sequence) = 0; |
| 50 }; | 50 }; |
| 51 | 51 |
| 52 // Creates a SchedulerWorkerThread with priority |thread_priority| that runs | 52 // Creates a SchedulerWorkerThread with priority |thread_priority| that runs |
| 53 // Tasks from Sequences returned by |delegate|. |task_tracker| is used to | 53 // Tasks from Sequences returned by |delegate|. |task_tracker| is used to |
| 54 // handle shutdown behavior of Tasks. Returns nullptr if creating the | 54 // handle shutdown behavior of Tasks. Returns nullptr if creating the |
| 55 // underlying platform thread fails. | 55 // underlying platform thread fails. |
| 56 static std::unique_ptr<SchedulerWorkerThread> CreateSchedulerWorkerThread( | 56 static std::unique_ptr<SchedulerWorkerThread> CreateWorkerThread( |
| 57 ThreadPriority thread_priority, | 57 ThreadPriority thread_priority, |
| 58 Delegate* delegate, | 58 std::unique_ptr<Delegate> delegate, |
| 59 TaskTracker* task_tracker); | 59 TaskTracker* task_tracker); |
| 60 | 60 |
| 61 // Destroying a SchedulerWorkerThread in production is not allowed; it is | 61 // Destroying a SchedulerWorkerThread in production is not allowed; it is |
| 62 // always leaked. In tests, it can only be destroyed after JoinForTesting() | 62 // always leaked. In tests, it can only be destroyed after JoinForTesting() |
| 63 // has returned. | 63 // has returned. |
| 64 ~SchedulerWorkerThread() override; | 64 ~SchedulerWorkerThread() override; |
| 65 | 65 |
| 66 // Wakes up this SchedulerWorkerThread if it wasn't already awake. After this | 66 // Wakes up this SchedulerWorkerThread if it wasn't already awake. After this |
| 67 // is called, this SchedulerWorkerThread will run Tasks from Sequences | 67 // is called, this SchedulerWorkerThread will run Tasks from Sequences |
| 68 // returned by the GetWork() method of its delegate until it returns nullptr. | 68 // returned by the GetWork() method of its delegate until it returns nullptr. |
| 69 void WakeUp(); | 69 void WakeUp(); |
| 70 | 70 |
| 71 // Joins this SchedulerWorkerThread. If a Task is already running, it will be | 71 // Joins this SchedulerWorkerThread. If a Task is already running, it will be |
| 72 // allowed to complete its execution. This can only be called once. | 72 // allowed to complete its execution. This can only be called once. |
| 73 void JoinForTesting(); | 73 void JoinForTesting(); |
| 74 | 74 |
| 75 private: | 75 private: |
| 76 SchedulerWorkerThread(ThreadPriority thread_priority, | 76 SchedulerWorkerThread(ThreadPriority thread_priority, |
| 77 Delegate* delegate, | 77 std::unique_ptr<Delegate> delegate, |
| 78 TaskTracker* task_tracker); | 78 TaskTracker* task_tracker); |
| 79 | 79 |
| 80 // PlatformThread::Delegate: | 80 // PlatformThread::Delegate: |
| 81 void ThreadMain() override; | 81 void ThreadMain() override; |
| 82 | 82 |
| 83 bool ShouldExitForTesting() const; | 83 bool ShouldExitForTesting() const; |
| 84 | 84 |
| 85 // Platform thread managed by this SchedulerWorkerThread. | 85 // Platform thread managed by this SchedulerWorkerThread. |
| 86 PlatformThreadHandle thread_handle_; | 86 PlatformThreadHandle thread_handle_; |
| 87 | 87 |
| 88 // Event signaled to wake up this SchedulerWorkerThread. | 88 // Event signaled to wake up this SchedulerWorkerThread. |
| 89 WaitableEvent wake_up_event_; | 89 WaitableEvent wake_up_event_; |
| 90 | 90 |
| 91 Delegate* const delegate_; | 91 const std::unique_ptr<Delegate> delegate_; |
| 92 TaskTracker* const task_tracker_; | 92 TaskTracker* const task_tracker_; |
| 93 | 93 |
| 94 // Synchronizes access to |should_exit_for_testing_|. | 94 // Synchronizes access to |should_exit_for_testing_|. |
| 95 mutable SchedulerLock should_exit_for_testing_lock_; | 95 mutable SchedulerLock should_exit_for_testing_lock_; |
| 96 | 96 |
| 97 // True once JoinForTesting() has been called. | 97 // True once JoinForTesting() has been called. |
| 98 bool should_exit_for_testing_ = false; | 98 bool should_exit_for_testing_ = false; |
| 99 | 99 |
| 100 DISALLOW_COPY_AND_ASSIGN(SchedulerWorkerThread); | 100 DISALLOW_COPY_AND_ASSIGN(SchedulerWorkerThread); |
| 101 }; | 101 }; |
| 102 | 102 |
| 103 } // namespace internal | 103 } // namespace internal |
| 104 } // namespace base | 104 } // namespace base |
| 105 | 105 |
| 106 #endif // BASE_TASK_SCHEDULER_SCHEDULER_WORKER_THREAD_H_ | 106 #endif // BASE_TASK_SCHEDULER_SCHEDULER_WORKER_THREAD_H_ |
| OLD | NEW |