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 #include "base/task_scheduler/task_scheduler.h" | 5 #include "base/task_scheduler/task_scheduler.h" |
6 | 6 |
7 #include <algorithm> | 7 #include <algorithm> |
8 | 8 |
9 #include "base/bind.h" | 9 #include "base/bind.h" |
10 #include "base/logging.h" | 10 #include "base/logging.h" |
11 #include "base/sys_info.h" | 11 #include "base/sys_info.h" |
12 #include "base/task_scheduler/scheduler_worker_pool_params.h" | 12 #include "base/task_scheduler/scheduler_worker_pool_params.h" |
13 #include "base/task_scheduler/task_scheduler_impl.h" | 13 #include "base/task_scheduler/task_scheduler_impl.h" |
14 #include "base/threading/platform_thread.h" | 14 #include "base/threading/platform_thread.h" |
15 #include "base/time/time.h" | 15 #include "base/time/time.h" |
16 | 16 |
17 namespace base { | 17 namespace base { |
18 | 18 |
19 namespace { | 19 namespace { |
20 | 20 |
21 // |g_task_scheduler| is intentionally leaked on shutdown. | 21 // |g_task_scheduler| is intentionally leaked on shutdown. |
22 TaskScheduler* g_task_scheduler = nullptr; | 22 TaskScheduler* g_task_scheduler = nullptr; |
23 | 23 |
24 } // namespace | 24 } // namespace |
25 | 25 |
| 26 TaskScheduler::InitParams::InitParams( |
| 27 const SchedulerWorkerPoolParams& background_worker_pool_params_in, |
| 28 const SchedulerWorkerPoolParams& background_blocking_worker_pool_params_in, |
| 29 const SchedulerWorkerPoolParams& foreground_worker_pool_params_in, |
| 30 const SchedulerWorkerPoolParams& foreground_blocking_worker_pool_params_in) |
| 31 : background_worker_pool_params(background_worker_pool_params_in), |
| 32 background_blocking_worker_pool_params( |
| 33 background_blocking_worker_pool_params_in), |
| 34 foreground_worker_pool_params(foreground_worker_pool_params_in), |
| 35 foreground_blocking_worker_pool_params( |
| 36 foreground_blocking_worker_pool_params_in) {} |
| 37 |
| 38 TaskScheduler::InitParams::~InitParams() = default; |
| 39 |
26 #if !defined(OS_NACL) | 40 #if !defined(OS_NACL) |
27 // static | 41 // static |
28 void TaskScheduler::CreateAndSetSimpleTaskScheduler(const std::string& name) { | 42 void TaskScheduler::CreateAndSetSimpleTaskScheduler(const std::string& name) { |
29 constexpr int kMinNumThreads = 1; | 43 constexpr int kMinNumThreads = 1; |
30 std::vector<SchedulerWorkerPoolParams> worker_pool_params_vector; | 44 std::vector<SchedulerWorkerPoolParams> worker_pool_params_vector; |
31 worker_pool_params_vector.emplace_back( | 45 worker_pool_params_vector.emplace_back( |
32 name, ThreadPriority::NORMAL, | 46 name, ThreadPriority::NORMAL, |
33 SchedulerWorkerPoolParams::StandbyThreadPolicy::LAZY, | 47 SchedulerWorkerPoolParams::StandbyThreadPolicy::LAZY, |
34 std::max(kMinNumThreads, base::SysInfo::NumberOfProcessors()), | 48 std::max(kMinNumThreads, SysInfo::NumberOfProcessors()), |
35 TimeDelta::FromSeconds(30)); | 49 TimeDelta::FromSeconds(30)); |
36 CreateAndSetDefaultTaskScheduler( | 50 CreateAndSetDefaultTaskScheduler( |
37 worker_pool_params_vector, | 51 worker_pool_params_vector, |
38 Bind([](const TaskTraits&) -> size_t { return 0; })); | 52 Bind([](const TaskTraits&) -> size_t { return 0; })); |
39 } | 53 } |
40 #endif // !defined(OS_NACL) | 54 #endif // !defined(OS_NACL) |
41 | 55 |
42 // static | 56 // static |
43 void TaskScheduler::CreateAndSetDefaultTaskScheduler( | 57 void TaskScheduler::CreateAndSetDefaultTaskScheduler( |
44 const std::vector<SchedulerWorkerPoolParams>& worker_pool_params_vector, | 58 const std::vector<SchedulerWorkerPoolParams>& worker_pool_params_vector, |
45 const WorkerPoolIndexForTraitsCallback& | 59 const WorkerPoolIndexForTraitsCallback& |
46 worker_pool_index_for_traits_callback) { | 60 worker_pool_index_for_traits_callback) { |
47 SetInstance(internal::TaskSchedulerImpl::Create( | 61 SetInstance(internal::TaskSchedulerImpl::Create( |
48 worker_pool_params_vector, worker_pool_index_for_traits_callback)); | 62 worker_pool_params_vector, worker_pool_index_for_traits_callback)); |
49 } | 63 } |
50 | 64 |
| 65 void TaskScheduler::CreateAndSetDefaultTaskScheduler( |
| 66 const std::string& name, |
| 67 const InitParams& init_params) { |
| 68 SetInstance(internal::TaskSchedulerImpl::Create(name, init_params)); |
| 69 } |
| 70 |
51 // static | 71 // static |
52 void TaskScheduler::SetInstance(std::unique_ptr<TaskScheduler> task_scheduler) { | 72 void TaskScheduler::SetInstance(std::unique_ptr<TaskScheduler> task_scheduler) { |
53 delete g_task_scheduler; | 73 delete g_task_scheduler; |
54 g_task_scheduler = task_scheduler.release(); | 74 g_task_scheduler = task_scheduler.release(); |
55 } | 75 } |
56 | 76 |
57 // static | 77 // static |
58 TaskScheduler* TaskScheduler::GetInstance() { | 78 TaskScheduler* TaskScheduler::GetInstance() { |
59 return g_task_scheduler; | 79 return g_task_scheduler; |
60 } | 80 } |
61 | 81 |
62 } // namespace base | 82 } // namespace base |
OLD | NEW |