| 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/task_scheduler/scheduler_worker_params.h" | 11 #include "base/task_scheduler/scheduler_worker_params.h" | 
| 12 #include "base/task_scheduler/switches.h" | 12 #include "base/task_scheduler/switches.h" | 
| 13 #include "base/threading/sequenced_worker_pool.h" | 13 #include "base/threading/sequenced_worker_pool.h" | 
| 14 #include "components/task_scheduler_util/common/variations_util.h" | 14 #include "components/task_scheduler_util/common/variations_util.h" | 
| 15 #include "components/variations/variations_associated_data.h" | 15 #include "components/variations/variations_associated_data.h" | 
| 16 | 16 | 
| 17 namespace task_scheduler_util { | 17 namespace task_scheduler_util { | 
| 18 | 18 | 
| 19 namespace { | 19 namespace { | 
| 20 | 20 | 
| 21 constexpr char kFieldTrialName[] = "BrowserScheduler"; | 21 constexpr char kFieldTrialName[] = "BrowserScheduler"; | 
| 22 | 22 | 
| 23 enum WorkerPoolType : size_t { |  | 
| 24   BACKGROUND = 0, |  | 
| 25   BACKGROUND_BLOCKING, |  | 
| 26   FOREGROUND, |  | 
| 27   FOREGROUND_BLOCKING, |  | 
| 28   WORKER_POOL_COUNT  // Always last. |  | 
| 29 }; |  | 
| 30 |  | 
| 31 }  // namespace | 23 }  // namespace | 
| 32 | 24 | 
| 33 std::unique_ptr<base::TaskScheduler::InitParams> | 25 std::unique_ptr<base::TaskScheduler::InitParams> | 
| 34 GetBrowserTaskSchedulerInitParamsFromVariations() { | 26 GetBrowserTaskSchedulerInitParamsFromVariations() { | 
| 35   std::map<std::string, std::string> variation_params; | 27   std::map<std::string, std::string> variation_params; | 
| 36   if (!::variations::GetVariationParams(kFieldTrialName, &variation_params)) | 28   if (!::variations::GetVariationParams(kFieldTrialName, &variation_params)) | 
| 37     return nullptr; | 29     return nullptr; | 
| 38 | 30 | 
| 39   return GetTaskSchedulerInitParams( | 31   return GetTaskSchedulerInitParams( | 
| 40       "", variation_params, base::SchedulerBackwardCompatibility::INIT_COM_STA); | 32       "", variation_params, base::SchedulerBackwardCompatibility::INIT_COM_STA); | 
| 41 } | 33 } | 
| 42 | 34 | 
| 43 std::vector<base::SchedulerWorkerPoolParams> |  | 
| 44 GetBrowserWorkerPoolParamsFromVariations() { |  | 
| 45   const auto init_params = GetBrowserTaskSchedulerInitParamsFromVariations(); |  | 
| 46   if (!init_params) |  | 
| 47     return std::vector<base::SchedulerWorkerPoolParams>(); |  | 
| 48 |  | 
| 49   return std::vector<base::SchedulerWorkerPoolParams>{ |  | 
| 50       init_params->background_worker_pool_params, |  | 
| 51       init_params->background_blocking_worker_pool_params, |  | 
| 52       init_params->foreground_worker_pool_params, |  | 
| 53       init_params->foreground_blocking_worker_pool_params}; |  | 
| 54 } |  | 
| 55 |  | 
| 56 size_t BrowserWorkerPoolIndexForTraits(const base::TaskTraits& traits) { |  | 
| 57   const bool is_background = |  | 
| 58       traits.priority() == base::TaskPriority::BACKGROUND; |  | 
| 59   if (traits.may_block() || traits.with_base_sync_primitives()) |  | 
| 60     return is_background ? BACKGROUND_BLOCKING : FOREGROUND_BLOCKING; |  | 
| 61   return is_background ? BACKGROUND : FOREGROUND; |  | 
| 62 } |  | 
| 63 |  | 
| 64 void MaybePerformBrowserTaskSchedulerRedirection() { | 35 void MaybePerformBrowserTaskSchedulerRedirection() { | 
| 65   // TODO(gab): Remove this when http://crbug.com/622400 concludes. | 36   // TODO(gab): Remove this when http://crbug.com/622400 concludes. | 
| 66   if (!base::CommandLine::ForCurrentProcess()->HasSwitch( | 37   if (!base::CommandLine::ForCurrentProcess()->HasSwitch( | 
| 67           switches::kDisableBrowserTaskScheduler) && | 38           switches::kDisableBrowserTaskScheduler) && | 
| 68       variations::GetVariationParamValue( | 39       variations::GetVariationParamValue( | 
| 69           kFieldTrialName, "RedirectSequencedWorkerPools") == "true") { | 40           kFieldTrialName, "RedirectSequencedWorkerPools") == "true") { | 
| 70     const base::TaskPriority max_task_priority = | 41     const base::TaskPriority max_task_priority = | 
| 71         variations::GetVariationParamValue( | 42         variations::GetVariationParamValue( | 
| 72             kFieldTrialName, "CapSequencedWorkerPoolsAtUserVisible") == "true" | 43             kFieldTrialName, "CapSequencedWorkerPoolsAtUserVisible") == "true" | 
| 73             ? base::TaskPriority::USER_VISIBLE | 44             ? base::TaskPriority::USER_VISIBLE | 
| 74             : base::TaskPriority::HIGHEST; | 45             : base::TaskPriority::HIGHEST; | 
| 75     base::SequencedWorkerPool::EnableWithRedirectionToTaskSchedulerForProcess( | 46     base::SequencedWorkerPool::EnableWithRedirectionToTaskSchedulerForProcess( | 
| 76         max_task_priority); | 47         max_task_priority); | 
| 77   } | 48   } | 
| 78 } | 49 } | 
| 79 | 50 | 
| 80 }  // namespace task_scheduler_util | 51 }  // namespace task_scheduler_util | 
| OLD | NEW | 
|---|