Control TaskScheduler initialization params in renderers via field trial.

components/task_scheduler_util/common/variations_util.cc

 // Copyright 2016 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "components/task_scheduler_util/common/variations_util.h"
 
+#include "base/command_line.h"
 #include "base/logging.h"
 #include "base/strings/string_number_conversions.h"
 #include "base/strings/string_piece.h"
 #include "base/strings/string_split.h"
+#include "base/strings/string_util.h"
 #include "base/task_scheduler/initialization_util.h"
 #include "base/time/time.h"
 
 namespace task_scheduler_util {
 
 namespace {
 
 struct SchedulerCustomizableWorkerPoolParams {
   base::SchedulerWorkerPoolParams::StandbyThreadPolicy standby_thread_policy;
   int max_threads = 0;
   base::TimeDelta detach_period;
 };
 
+constexpr char kTaskSchedulerVariationParamsSwitch[] =
+    "task-scheduler-variation-params";
+
+constexpr char kSeparator[] = "|";
+
+bool ContainsSeparator(const std::string& str) {
+  return str.find(kSeparator) != std::string::npos;
+}
+
 // Converts |pool_descriptor| to a SchedulerWorkerPoolVariableParams. Returns a
 // default SchedulerWorkerPoolVariableParams on failure.
 //
 // |pool_descriptor| is a semi-colon separated value string with the following
 // items:
 // 0. Minimum Thread Count (int)
 // 1. Maximum Thread Count (int)
 // 2. Thread Count Multiplier (double)
 // 3. Thread Count Offset (int)
 // 4. Detach Time in Milliseconds (int)
@@ skipping 64 matching lines @@
     }
     worker_pool_params_vector.emplace_back(
         worker_pool_name, constant_worker_pool_params.priority_hint,
         variable_worker_pool_params.standby_thread_policy,
         variable_worker_pool_params.max_threads,
         variable_worker_pool_params.detach_period);
   }
   return worker_pool_params_vector;
 }
 
+void AddVariationParamsToCommandLine(
+    const std::map<std::string, std::string> variation_params,
+    base::StringPiece key_prefix,
+    base::CommandLine* command_line) {
+  DCHECK(command_line);
+
+  std::vector<std::string> parts;
+  for (const auto& key_value : variation_params) {
+    if (base::StartsWith(key_value.first, key_prefix,
+                         base::CompareCase::SENSITIVE)) {
+      if (ContainsSeparator(key_value.first) ||
+          ContainsSeparator(key_value.second)) {
+        DLOG(ERROR)
+            << "Unexpected Character in Task Scheduler Variation Params: "
+            << key_value.first << " [" << key_value.second << "]";
+        return;
+      }
+      parts.push_back(key_value.first);
+      parts.push_back(key_value.second);
+    }
+  }
+
+  if (!parts.empty()) {
+    command_line->AppendSwitchASCII(kTaskSchedulerVariationParamsSwitch,
+                                    base::JoinString(parts, kSeparator));
+  }
+}
+
+std::map<std::string, std::string> GetVariationParamsFromCommandLine(
+    const base::CommandLine& command_line) {
+  const auto serialized_variation_params =
+      command_line.GetSwitchValueASCII(kTaskSchedulerVariationParamsSwitch);
+  const auto parts =
+      base::SplitStringPiece(serialized_variation_params, kSeparator,
+                             base::KEEP_WHITESPACE, base::SPLIT_WANT_NONEMPTY);
+  std::map<std::string, std::string> variation_params;
+  for (auto it = parts.begin(); it != parts.end(); ++it) {
+    auto key = *it;
+    ++it;
+    if (it == parts.end())
+      return std::map<std::string, std::string>();
+    auto value = *it;
+    variation_params[key.as_string()] = value.as_string();
+  }
+  return variation_params;
+}
+
 }  // namespace task_scheduler_util