Variations Service now creates field trials from variations seed.

chrome/browser/metrics/variations_service.cc

 // Copyright (c) 2012 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 "chrome/browser/metrics/variations_service.h"
 
 #include "base/base64.h"
 #include "base/build_time.h"
 #include "base/version.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/memory/singleton.h"
+#include "base/metrics/field_trial.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/metrics/proto/trials_seed.pb.h"
 #include "chrome/browser/prefs/pref_service.h"
 #include "chrome/common/pref_names.h"
 #include "content/public/common/url_fetcher.h"
 #include "googleurl/src/gurl.h"
 #include "net/base/load_flags.h"
 #include "net/url_request/url_request_status.h"
 
 namespace {
@@ skipping 21 matching lines @@
   return chrome::VersionInfo::CHANNEL_UNKNOWN;
 }
 
 }  // namespace
 
 // Static
 VariationsService* VariationsService::GetInstance() {
   return Singleton<VariationsService>::get();
 }
 
-void VariationsService::LoadVariationsSeed(PrefService* local_prefs) {
-  std::string base64_seed_data = local_prefs->GetString(prefs::kVariationsSeed);
-  std::string seed_data;
+bool VariationsService::CreateTrialsFromSeed(PrefService* local_prefs) {
+  chrome_variations::TrialsSeed seed;
+  if (!LoadVariationsSeed(local_prefs, &seed))
+      return false;
 
-  // If the decode process fails, assume the pref value is corrupt, and clear
-  // it.
-  if (!base::Base64Decode(base64_seed_data, &seed_data) ||
-      !variations_seed_.ParseFromString(seed_data)) {
-    VLOG(1) << "Variations Seed data in local pref is corrupt, clearing the "
-            << "pref.";
-    local_prefs->ClearPref(prefs::kVariationsSeed);
-  }
+  for (int i = 0; i < seed.study_size(); ++i)
+    CreateTrialFromStudy(seed.study(i));
+
+  return true;
 }
 
 void VariationsService::StartFetchingVariationsSeed() {
   pending_seed_request_.reset(content::URLFetcher::Create(
       GURL(kDefaultVariationsServer), content::URLFetcher::GET, this));
   pending_seed_request_->SetLoadFlags(net::LOAD_DO_NOT_SEND_COOKIES |
                                       net::LOAD_DO_NOT_SAVE_COOKIES);
   pending_seed_request_->SetRequestContext(
       g_browser_process->system_request_context());
+  // TODO(jwd): pull max retries into a contsant
   pending_seed_request_->SetMaxRetries(5);
   pending_seed_request_->Start();
 }
 
 void VariationsService::OnURLFetchComplete(const content::URLFetcher* source) {
   DCHECK_EQ(pending_seed_request_.get(), source);
   // When we're done handling the request, the fetcher will be deleted.
   scoped_ptr<const content::URLFetcher> request(
       pending_seed_request_.release());
   if (request->GetStatus().status() != net::URLRequestStatus::SUCCESS ||
@@ skipping 55 matching lines @@
   }
 
   DVLOG(1) << "Kept study " << study.name() << ".";
   return true;
 }
 
 // static
 bool VariationsService::CheckStudyChannel(
     const chrome_variations::Study& study,
     chrome::VersionInfo::Channel channel) {
+  if (study.channel_size() == 0) {
+    // An empty channel list matches all channels.
+    return true;
+  }
+
   for (int i = 0; i < study.channel_size(); ++i) {
     if (ConvertStudyChannelToVersionChannel(study.channel(i)) == channel)
       return true;
   }
   return false;
 }
 
 // static
 bool VariationsService::CheckStudyVersion(const chrome_variations::Study& study,
                                           const std::string& version_string) {
@@ skipping 22 matching lines @@
   return true;
 }
 
 // static
 bool VariationsService::CheckStudyDate(const chrome_variations::Study& study,
                                        const base::Time& date_time) {
   const base::Time epoch = base::Time::UnixEpoch();
 
   if (study.has_start_date()) {
     const base::Time start_date =
-        epoch + base::TimeDelta::FromMilliseconds(study.start_date());
+        epoch + base::TimeDelta::FromSeconds(study.start_date());
     if (date_time < start_date)
       return false;
   }
 
   if (study.has_expiry_date()) {
     const base::Time expiry_date =
-        epoch + base::TimeDelta::FromMilliseconds(study.expiry_date());
+        epoch + base::TimeDelta::FromSeconds(study.expiry_date());
     if (date_time >= expiry_date)
       return false;
   }
 
   return true;
 }
+
+bool VariationsService::LoadVariationsSeed(
+    PrefService* local_prefs,
+    chrome_variations::TrialsSeed* seed) {
+  std::string base64_seed_data = local_prefs->GetString(prefs::kVariationsSeed);
+  std::string seed_data;
+
+  // If the decode process fails, assume the pref value is corrupt, and clear
+  // it.
+  if (!base::Base64Decode(base64_seed_data, &seed_data) ||
+      !seed->ParseFromString(seed_data)) {
+    VLOG(1) << "Variations Seed data in local pref is corrupt, clearing the "
+            << "pref.";
+    local_prefs->ClearPref(prefs::kVariationsSeed);
+    return false;
+  }
+  return true;
+}
+
+void VariationsService::CreateTrialFromStudy(
+    const chrome_variations::Study& study) {
+  if (!ShouldAddStudy(study))
+    return;
+
+  // At the moment, a missing default_experiment_name makes the study invalid.
+  if (!study.has_default_experiment_name()) {
+    DVLOG(1) << study.name() << " has no default experiment defined.";
+    return;
+  }
+
+  const std::string& default_group_name = study.default_experiment_name();
+  base::FieldTrial::Probability divisor = 0;
+
+  bool found_default_group = false;
+  for (int i = 0; i < study.experiment_size(); ++i) {
+    divisor += study.experiment(i).probability_weight();
+    if (study.experiment(i).name() == default_group_name) {
+      found_default_group = true;
+    }
+  }
+  if (!found_default_group){

[Alexei Svitkine (slow), 2012/05/08 03:40:13]

Space before {.

[jwd, 2012/05/08 03:47:24]

Aw man, I'm missing all of these...
Done.

+    DVLOG(1) << study.name() << " is missing default experiment in it's "
+             << "experiment list";
+    // The default group was not found in the list of groups. This study is not
+    //  valid.

[Alexei Svitkine (slow), 2012/05/08 03:40:13]

Nit remove space before "valid".

[jwd, 2012/05/08 03:47:24]

Done.

+    return;
+  }
+
+  const base::Time epoch = base::Time::UnixEpoch();
+  const base::Time expiry_date =
+      epoch + base::TimeDelta::FromSeconds(study.expiry_date());
+  base::Time::Exploded exploded_end_date;
+  expiry_date.UTCExplode(&exploded_end_date);
+
+  scoped_refptr<base::FieldTrial> trial(
+      base::FieldTrialList::FactoryGetFieldTrial(
+          study.name(), divisor, default_group_name, exploded_end_date.year,
+          exploded_end_date.month, exploded_end_date.day_of_month,  NULL));

[Alexei Svitkine (slow), 2012/05/08 03:40:13]

Nit: remove space before NULL.

[jwd, 2012/05/08 03:47:24]

Done.

+
+  if (study.has_consistency() &&
+      study.consistency() == chrome_variations::Study_Consistency_PERMANENT)

[Alexei Svitkine (slow), 2012/05/08 03:40:13]

Nit: Add {'s.

[jwd, 2012/05/08 03:47:24]

Done.

+    trial->UseOneTimeRandomization();
+
+  for (int i = 0; i < study.experiment_size(); ++i) {
+    if (study.experiment(i).name() != default_group_name)
+      trial->AppendGroup(study.experiment(i).name(),
+                        study.experiment(i).probability_weight());

[Alexei Svitkine (slow), 2012/05/08 03:40:13]

Nit: align.

[SteveT, 2012/05/08 03:44:28]

Nit: Braces around if body, since it spans 2 physical lines.

[jwd, 2012/05/08 03:47:24]

Done.

+  }
+
+  trial->SetForced();
+}