Implement cloud policy invalidations using the invalidation service framework.

chrome/browser/policy/cloud/cloud_policy_manager.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/policy/cloud/cloud_policy_manager.h"
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
 #include "base/logging.h"
 #include "base/prefs/pref_service.h"
+#include "chrome/browser/policy/cloud/cloud_policy_refresh_scheduler.h"
 #include "chrome/browser/policy/cloud/cloud_policy_service.h"
 #include "chrome/browser/policy/policy_bundle.h"
 #include "chrome/browser/policy/policy_map.h"
 
 namespace policy {
 
 CloudPolicyManager::CloudPolicyManager(const PolicyNamespaceKey& policy_ns_key,
                                        CloudPolicyStore* cloud_policy_store)
     : core_(policy_ns_key, cloud_policy_store),
       waiting_for_policy_refresh_(false) {
   store()->AddObserver(this);
 
   // If the underlying store is already initialized, publish the loaded
   // policy. Otherwise, request a load now.
   if (store()->is_initialized())
     CheckAndPublishPolicy();
   else
     store()->Load();
 }
 
 CloudPolicyManager::~CloudPolicyManager() {}
 
+void CloudPolicyManager::EnableInvalidations(
+    const base::Closure& initialize_invalidator) {
+  // If the refresh scheduler is already running, initialize the invalidator
+  // right away. Otherwise, store the closure so it can be invoked when the
+  // refresh scheduler starts.
+  if (core()->refresh_scheduler())
+    initialize_invalidator.Run();
+  else
+    initialize_invalidator_ = initialize_invalidator;
+}
+
 void CloudPolicyManager::Shutdown() {
   core_.Disconnect();
   store()->RemoveObserver(this);
   ConfigurationPolicyProvider::Shutdown();
 }
 
 bool CloudPolicyManager::IsInitializationComplete(PolicyDomain domain) const {
   if (domain == POLICY_DOMAIN_CHROME)
     return store()->is_initialized();
   return true;
@@ skipping 16 matching lines @@
 }
 
 void CloudPolicyManager::OnStoreError(CloudPolicyStore* cloud_policy_store) {
   DCHECK_EQ(store(), cloud_policy_store);
   // Publish policy (even though it hasn't changed) in order to signal load
   // complete on the ConfigurationPolicyProvider interface. Technically, this
   // is only required on the first load, but doesn't hurt in any case.
   CheckAndPublishPolicy();
 }
 
+void CloudPolicyManager::SetInvalidationInfo(
+    int64 version,
+    const std::string& payload) {
+  DCHECK(core()->client());
+  core()->client()->SetInvalidationInfo(version, payload);
+}
+
+void CloudPolicyManager::InvalidatePolicy() {
+  DCHECK(core()->refresh_scheduler());
+  core()->refresh_scheduler()->RefreshSoon();
+}
+
+void CloudPolicyManager::OnInvalidatorStateChanged(bool invalidations_enabled) {
+  DCHECK(core()->refresh_scheduler());
+  // Enable this code once method is added to CloudPolicyRefreshScheduler.

[rlarocque, 2013/07/29 18:29:54]

nit: Add a TODO with your name on it.

[Steve Condie, 2013/07/29 18:53:07]

The code in question got checked in, so these lines can be uncommented now.

+  // core()->refresh_scheduler()->SetInvalidationServiceAvailability(
+  //     invalidations_enabled);
+}
+
 void CloudPolicyManager::CheckAndPublishPolicy() {
   if (IsInitializationComplete(POLICY_DOMAIN_CHROME) &&
       !waiting_for_policy_refresh_) {
     UpdatePolicy(CreatePolicyBundle());
   }
 }
 
+void CloudPolicyManager::StartRefreshScheduler() {
+  DCHECK(!core()->refresh_scheduler());
+  core()->StartRefreshScheduler();
+  // Initialize the invalidator if EnableInvalidations has been called.
+  if (!initialize_invalidator_.is_null())
+    initialize_invalidator_.Run();
+}
+
 scoped_ptr<PolicyBundle> CloudPolicyManager::CreatePolicyBundle() {
   scoped_ptr<PolicyBundle> bundle(new PolicyBundle);
   bundle->Get(PolicyNamespace(POLICY_DOMAIN_CHROME, std::string()))
       .CopyFrom(store()->policy_map());
   return bundle.Pass();
 }
 
 void CloudPolicyManager::OnRefreshComplete(bool success) {
   waiting_for_policy_refresh_ = false;
   CheckAndPublishPolicy();
 }
 
 }  // namespace policy