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 "components/task_scheduler_util/browser/initialization.h" | 5 #include "components/task_scheduler_util/browser/initialization.h" |
6 | 6 |
7 #include <map> | 7 #include <map> |
8 #include <string> | 8 #include <string> |
9 | 9 |
10 #include "base/command_line.h" | 10 #include "base/command_line.h" |
11 #include "base/logging.h" | |
12 #include "base/task_scheduler/scheduler_worker_params.h" | 11 #include "base/task_scheduler/scheduler_worker_params.h" |
13 #include "base/task_scheduler/switches.h" | 12 #include "base/task_scheduler/switches.h" |
14 #include "base/task_scheduler/task_traits.h" | |
15 #include "base/threading/platform_thread.h" | |
16 #include "base/threading/sequenced_worker_pool.h" | 13 #include "base/threading/sequenced_worker_pool.h" |
17 #include "components/task_scheduler_util/common/variations_util.h" | 14 #include "components/task_scheduler_util/common/variations_util.h" |
18 #include "components/variations/variations_associated_data.h" | 15 #include "components/variations/variations_associated_data.h" |
19 | 16 |
20 namespace task_scheduler_util { | 17 namespace task_scheduler_util { |
21 | 18 |
22 namespace { | 19 namespace { |
23 | 20 |
24 constexpr char kFieldTrialName[] = "BrowserScheduler"; | 21 constexpr char kFieldTrialName[] = "BrowserScheduler"; |
25 | 22 |
26 enum WorkerPoolType : size_t { | 23 enum WorkerPoolType : size_t { |
robliao
2017/04/03 19:13:11
This can be removed.
fdoray
2017/04/03 19:21:45
Still used in BrowserWorkerPoolIndexForTraits.
robliao
2017/04/03 19:23:46
Indeed. My mistake!
| |
27 BACKGROUND = 0, | 24 BACKGROUND = 0, |
28 BACKGROUND_BLOCKING, | 25 BACKGROUND_BLOCKING, |
29 FOREGROUND, | 26 FOREGROUND, |
30 FOREGROUND_BLOCKING, | 27 FOREGROUND_BLOCKING, |
31 WORKER_POOL_COUNT // Always last. | 28 WORKER_POOL_COUNT // Always last. |
32 }; | 29 }; |
33 | 30 |
34 } // namespace | 31 } // namespace |
35 | 32 |
33 std::unique_ptr<base::TaskScheduler::InitParams> | |
34 GetBrowserTaskSchedulerInitParamsFromVariations() { | |
35 std::map<std::string, std::string> variation_params; | |
36 if (!::variations::GetVariationParams(kFieldTrialName, &variation_params)) | |
37 return nullptr; | |
38 | |
39 return GetTaskSchedulerInitParams( | |
40 "", variation_params, base::SchedulerBackwardCompatibility::INIT_COM_STA); | |
41 } | |
42 | |
36 std::vector<base::SchedulerWorkerPoolParams> | 43 std::vector<base::SchedulerWorkerPoolParams> |
37 GetBrowserWorkerPoolParamsFromVariations() { | 44 GetBrowserWorkerPoolParamsFromVariations() { |
38 using ThreadPriority = base::ThreadPriority; | 45 const auto init_params = GetBrowserTaskSchedulerInitParamsFromVariations(); |
39 | 46 if (!init_params) |
40 std::map<std::string, std::string> variation_params; | |
41 if (!::variations::GetVariationParams(kFieldTrialName, &variation_params)) | |
42 return std::vector<base::SchedulerWorkerPoolParams>(); | 47 return std::vector<base::SchedulerWorkerPoolParams>(); |
43 | 48 |
44 std::vector<SchedulerImmutableWorkerPoolParams> immutable_worker_pool_params; | 49 return std::vector<base::SchedulerWorkerPoolParams>{ |
45 DCHECK_EQ(BACKGROUND, immutable_worker_pool_params.size()); | 50 init_params->background_worker_pool_params, |
46 immutable_worker_pool_params.emplace_back("Background", | 51 init_params->background_blocking_worker_pool_params, |
47 ThreadPriority::BACKGROUND); | 52 init_params->foreground_worker_pool_params, |
48 DCHECK_EQ(BACKGROUND_BLOCKING, immutable_worker_pool_params.size()); | 53 init_params->foreground_blocking_worker_pool_params}; |
49 immutable_worker_pool_params.emplace_back("BackgroundBlocking", | |
50 ThreadPriority::BACKGROUND); | |
51 DCHECK_EQ(FOREGROUND, immutable_worker_pool_params.size()); | |
52 immutable_worker_pool_params.emplace_back("Foreground", | |
53 ThreadPriority::NORMAL); | |
54 // Tasks posted to SequencedWorkerPool or BrowserThreadImpl may be redirected | |
55 // to this pool. Since COM STA is initialized in these environments, it must | |
56 // also be initialized in this pool. | |
57 DCHECK_EQ(FOREGROUND_BLOCKING, immutable_worker_pool_params.size()); | |
58 immutable_worker_pool_params.emplace_back( | |
59 "ForegroundBlocking", ThreadPriority::NORMAL, | |
60 base::SchedulerBackwardCompatibility::INIT_COM_STA); | |
61 | |
62 return GetWorkerPoolParams(immutable_worker_pool_params, variation_params); | |
63 } | 54 } |
64 | 55 |
65 size_t BrowserWorkerPoolIndexForTraits(const base::TaskTraits& traits) { | 56 size_t BrowserWorkerPoolIndexForTraits(const base::TaskTraits& traits) { |
66 const bool is_background = | 57 const bool is_background = |
67 traits.priority() == base::TaskPriority::BACKGROUND; | 58 traits.priority() == base::TaskPriority::BACKGROUND; |
68 if (traits.may_block() || traits.with_base_sync_primitives()) | 59 if (traits.may_block() || traits.with_base_sync_primitives()) |
69 return is_background ? BACKGROUND_BLOCKING : FOREGROUND_BLOCKING; | 60 return is_background ? BACKGROUND_BLOCKING : FOREGROUND_BLOCKING; |
70 return is_background ? BACKGROUND : FOREGROUND; | 61 return is_background ? BACKGROUND : FOREGROUND; |
71 } | 62 } |
72 | 63 |
73 void MaybePerformBrowserTaskSchedulerRedirection() { | 64 void MaybePerformBrowserTaskSchedulerRedirection() { |
74 // TODO(gab): Remove this when http://crbug.com/622400 concludes. | 65 // TODO(gab): Remove this when http://crbug.com/622400 concludes. |
75 if (!base::CommandLine::ForCurrentProcess()->HasSwitch( | 66 if (!base::CommandLine::ForCurrentProcess()->HasSwitch( |
76 switches::kDisableBrowserTaskScheduler) && | 67 switches::kDisableBrowserTaskScheduler) && |
77 variations::GetVariationParamValue( | 68 variations::GetVariationParamValue( |
78 kFieldTrialName, "RedirectSequencedWorkerPools") == "true") { | 69 kFieldTrialName, "RedirectSequencedWorkerPools") == "true") { |
79 const base::TaskPriority max_task_priority = | 70 const base::TaskPriority max_task_priority = |
80 variations::GetVariationParamValue( | 71 variations::GetVariationParamValue( |
81 kFieldTrialName, "CapSequencedWorkerPoolsAtUserVisible") == "true" | 72 kFieldTrialName, "CapSequencedWorkerPoolsAtUserVisible") == "true" |
82 ? base::TaskPriority::USER_VISIBLE | 73 ? base::TaskPriority::USER_VISIBLE |
83 : base::TaskPriority::HIGHEST; | 74 : base::TaskPriority::HIGHEST; |
84 base::SequencedWorkerPool::EnableWithRedirectionToTaskSchedulerForProcess( | 75 base::SequencedWorkerPool::EnableWithRedirectionToTaskSchedulerForProcess( |
85 max_task_priority); | 76 max_task_priority); |
86 } | 77 } |
87 } | 78 } |
88 | 79 |
89 } // namespace task_scheduler_util | 80 } // namespace task_scheduler_util |
OLD | NEW |