Add support for Omaha cohorts to the component updater.

components/update_client/persisted_data.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/update_client/persisted_data.h"
 
 #include <string>
 #include <vector>
 
 #include "base/guid.h"
@@ skipping 54 matching lines @@
   DictionaryPrefUpdate update(pref_service_, kPersistedDataPreference);
   for (auto id : ids) {
     // We assume ids do not contain '.' characters.
     DCHECK_EQ(std::string::npos, id.find('.'));
     update->SetInteger(base::StringPrintf("apps.%s.dlrc", id.c_str()), datenum);
     update->SetString(base::StringPrintf("apps.%s.pf", id.c_str()),
                       base::GenerateGUID());
   }
 }
 
+std::string PersistedData::GetCohort(const std::string& id) const {
+  DCHECK(thread_checker_.CalledOnValidThread());
+  if (!pref_service_)
+    return std::string();
+  std::string cohort;
+  const base::DictionaryValue* dict =
+      pref_service_->GetDictionary(kPersistedDataPreference);
+  // We assume ids do not contain '.' characters.
+  DCHECK_EQ(std::string::npos, id.find('.'));

[Sorin Jianu, 2016/08/18 21:15:15]

If the dcheck is a pre-condition for the argument we can move it closer to the
top of the function. Then, we could declare cohort closer to where it is used.

[waffles, 2016/08/18 22:29:23]

Done.

+  if (!dict->GetString(base::StringPrintf("apps.%s.cohort", id.c_str()),
+                       &cohort))
+    return std::string();
+  return cohort;
+}
+
+std::string PersistedData::GetCohortName(const std::string& id) const {
+  DCHECK(thread_checker_.CalledOnValidThread());

[Sorin Jianu, 2016/08/18 21:15:15]

Is there a way to reduce dublication of code in these functions?

[waffles, 2016/08/18 22:29:23]

Done.

+  if (!pref_service_)
+    return std::string();
+  std::string cohort_name;
+  const base::DictionaryValue* dict =
+      pref_service_->GetDictionary(kPersistedDataPreference);
+  // We assume ids do not contain '.' characters.
+  DCHECK_EQ(std::string::npos, id.find('.'));
+  if (!dict->GetString(base::StringPrintf("apps.%s.cohortname", id.c_str()),
+                       &cohort_name))
+    return std::string();
+  return cohort_name;
+}
+
+std::string PersistedData::GetCohortHint(const std::string& id) const {
+  DCHECK(thread_checker_.CalledOnValidThread());
+  if (!pref_service_)
+    return std::string();
+  std::string cohort_hint;
+  const base::DictionaryValue* dict =
+      pref_service_->GetDictionary(kPersistedDataPreference);
+  // We assume ids do not contain '.' characters.
+  DCHECK_EQ(std::string::npos, id.find('.'));
+  if (!dict->GetString(base::StringPrintf("apps.%s.cohorthint", id.c_str()),
+                       &cohort_hint))
+    return std::string();
+  return cohort_hint;
+}
+
+void PersistedData::SetCohort(const std::string& id,
+                              const std::string& cohort) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+  if (!pref_service_)
+    return;
+  DictionaryPrefUpdate update(pref_service_, kPersistedDataPreference);
+  update->SetString(base::StringPrintf("apps.%s.cohort", id.c_str()), cohort);
+}
+
+void PersistedData::SetCohortName(const std::string& id,
+                                  const std::string& cohort_name) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+  if (!pref_service_)
+    return;
+  DictionaryPrefUpdate update(pref_service_, kPersistedDataPreference);
+  update->SetString(base::StringPrintf("apps.%s.cohortname", id.c_str()),
+                    cohort_name);
+}
+
+void PersistedData::SetCohortHint(const std::string& id,
+                                  const std::string& cohort_hint) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+  if (!pref_service_)
+    return;
+  DictionaryPrefUpdate update(pref_service_, kPersistedDataPreference);
+  update->SetString(base::StringPrintf("apps.%s.cohorthint", id.c_str()),
+                    cohort_hint);
+}
+
 void PersistedData::RegisterPrefs(PrefRegistrySimple* registry) {
   registry->RegisterDictionaryPref(kPersistedDataPreference);
 }
 
 }  // namespace update_client