| 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_TASK_SCHEDULER_IMPL_H_ | 5 #ifndef BASE_TASK_SCHEDULER_TASK_SCHEDULER_IMPL_H_ |
| 6 #define BASE_TASK_SCHEDULER_TASK_SCHEDULER_IMPL_H_ | 6 #define BASE_TASK_SCHEDULER_TASK_SCHEDULER_IMPL_H_ |
| 7 | 7 |
| 8 #include <memory> | 8 #include <memory> |
| 9 #include <vector> | 9 #include <vector> |
| 10 | 10 |
| 11 #include "base/base_export.h" | 11 #include "base/base_export.h" |
| 12 #include "base/callback.h" | 12 #include "base/callback.h" |
| 13 #include "base/logging.h" | 13 #include "base/logging.h" |
| 14 #include "base/macros.h" | 14 #include "base/macros.h" |
| 15 #include "base/memory/ptr_util.h" | |
| 16 #include "base/memory/ref_counted.h" | 15 #include "base/memory/ref_counted.h" |
| 17 #include "base/strings/string_piece.h" | 16 #include "base/strings/string_piece.h" |
| 18 #include "base/synchronization/atomic_flag.h" | 17 #include "base/synchronization/atomic_flag.h" |
| 19 #include "base/task_scheduler/delayed_task_manager.h" | 18 #include "base/task_scheduler/delayed_task_manager.h" |
| 20 #include "base/task_scheduler/scheduler_single_thread_task_runner_manager.h" | 19 #include "base/task_scheduler/scheduler_single_thread_task_runner_manager.h" |
| 21 #include "base/task_scheduler/scheduler_worker_pool_impl.h" | 20 #include "base/task_scheduler/scheduler_worker_pool_impl.h" |
| 22 #include "base/task_scheduler/single_thread_task_runner_thread_mode.h" | 21 #include "base/task_scheduler/single_thread_task_runner_thread_mode.h" |
| 23 #include "base/task_scheduler/task_scheduler.h" | 22 #include "base/task_scheduler/task_scheduler.h" |
| 24 #include "base/task_scheduler/task_tracker.h" | 23 #include "base/task_scheduler/task_tracker.h" |
| 25 #include "base/task_scheduler/task_traits.h" | 24 #include "base/task_scheduler/task_traits.h" |
| 26 #include "base/threading/thread.h" | 25 #include "base/threading/thread.h" |
| 27 #include "build/build_config.h" | 26 #include "build/build_config.h" |
| 28 | 27 |
| 29 #if defined(OS_POSIX) && !defined(OS_NACL_SFI) | 28 #if defined(OS_POSIX) && !defined(OS_NACL_SFI) |
| 30 #include "base/task_scheduler/task_tracker_posix.h" | 29 #include "base/task_scheduler/task_tracker_posix.h" |
| 31 #endif | 30 #endif |
| 32 | 31 |
| 33 namespace base { | 32 namespace base { |
| 34 | 33 |
| 35 class HistogramBase; | 34 class HistogramBase; |
| 36 | 35 |
| 37 namespace internal { | 36 namespace internal { |
| 38 | 37 |
| 39 // Default TaskScheduler implementation. This class is thread-safe. | 38 // Default TaskScheduler implementation. This class is thread-safe. |
| 40 class BASE_EXPORT TaskSchedulerImpl : public TaskScheduler { | 39 class BASE_EXPORT TaskSchedulerImpl : public TaskScheduler { |
| 41 public: | 40 public: |
| 42 using TaskTrackerType = | 41 // |name| is used to label threads and histograms. |
| 43 #if defined(OS_POSIX) && !defined(OS_NACL_SFI) | 42 explicit TaskSchedulerImpl(StringPiece name); |
| 44 TaskTrackerPosix; | |
| 45 #else | |
| 46 TaskTracker; | |
| 47 #endif | |
| 48 | |
| 49 // |name| is used to label threads and histograms. |task_tracker| can be used | |
| 50 // for tests that need more execution control. By default, you get the | |
| 51 // production TaskTracker. | |
| 52 explicit TaskSchedulerImpl(StringPiece name, | |
| 53 std::unique_ptr<TaskTrackerType> task_tracker = | |
| 54 MakeUnique<TaskTrackerType>()); | |
| 55 ~TaskSchedulerImpl() override; | 43 ~TaskSchedulerImpl() override; |
| 56 | 44 |
| 57 // TaskScheduler: | 45 // TaskScheduler: |
| 58 void Start(const TaskScheduler::InitParams& init_params) override; | 46 void Start(const TaskScheduler::InitParams& init_params) override; |
| 59 void PostDelayedTaskWithTraits(const tracked_objects::Location& from_here, | 47 void PostDelayedTaskWithTraits(const tracked_objects::Location& from_here, |
| 60 const TaskTraits& traits, | 48 const TaskTraits& traits, |
| 61 OnceClosure task, | 49 OnceClosure task, |
| 62 TimeDelta delay) override; | 50 TimeDelta delay) override; |
| 63 scoped_refptr<TaskRunner> CreateTaskRunnerWithTraits( | 51 scoped_refptr<TaskRunner> CreateTaskRunnerWithTraits( |
| 64 const TaskTraits& traits) override; | 52 const TaskTraits& traits) override; |
| (...skipping 14 matching lines...) Expand all Loading... |
| 79 void FlushForTesting() override; | 67 void FlushForTesting() override; |
| 80 void JoinForTesting() override; | 68 void JoinForTesting() override; |
| 81 | 69 |
| 82 private: | 70 private: |
| 83 // Returns the worker pool that runs Tasks with |traits|. | 71 // Returns the worker pool that runs Tasks with |traits|. |
| 84 SchedulerWorkerPoolImpl* GetWorkerPoolForTraits( | 72 SchedulerWorkerPoolImpl* GetWorkerPoolForTraits( |
| 85 const TaskTraits& traits) const; | 73 const TaskTraits& traits) const; |
| 86 | 74 |
| 87 const std::string name_; | 75 const std::string name_; |
| 88 Thread service_thread_; | 76 Thread service_thread_; |
| 89 const std::unique_ptr<TaskTrackerType> task_tracker_; | 77 #if defined(OS_POSIX) && !defined(OS_NACL_SFI) |
| 78 TaskTrackerPosix task_tracker_; |
| 79 #else |
| 80 TaskTracker task_tracker_; |
| 81 #endif |
| 90 DelayedTaskManager delayed_task_manager_; | 82 DelayedTaskManager delayed_task_manager_; |
| 91 SchedulerSingleThreadTaskRunnerManager single_thread_task_runner_manager_; | 83 SchedulerSingleThreadTaskRunnerManager single_thread_task_runner_manager_; |
| 92 | 84 |
| 93 // There are 4 SchedulerWorkerPoolImpl in this array to match the 4 | 85 // There are 4 SchedulerWorkerPoolImpl in this array to match the 4 |
| 94 // SchedulerWorkerPoolParams in TaskScheduler::InitParams. | 86 // SchedulerWorkerPoolParams in TaskScheduler::InitParams. |
| 95 std::unique_ptr<SchedulerWorkerPoolImpl> worker_pools_[4]; | 87 std::unique_ptr<SchedulerWorkerPoolImpl> worker_pools_[4]; |
| 96 | 88 |
| 97 #if DCHECK_IS_ON() | 89 #if DCHECK_IS_ON() |
| 98 // Set once JoinForTesting() has returned. | 90 // Set once JoinForTesting() has returned. |
| 99 AtomicFlag join_for_testing_returned_; | 91 AtomicFlag join_for_testing_returned_; |
| 100 #endif | 92 #endif |
| 101 | 93 |
| 102 DISALLOW_COPY_AND_ASSIGN(TaskSchedulerImpl); | 94 DISALLOW_COPY_AND_ASSIGN(TaskSchedulerImpl); |
| 103 }; | 95 }; |
| 104 | 96 |
| 105 } // namespace internal | 97 } // namespace internal |
| 106 } // namespace base | 98 } // namespace base |
| 107 | 99 |
| 108 #endif // BASE_TASK_SCHEDULER_TASK_SCHEDULER_IMPL_H_ | 100 #endif // BASE_TASK_SCHEDULER_TASK_SCHEDULER_IMPL_H_ |
| OLD | NEW |