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) {
@@ skipping 38 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) {
+  if (!core()->refresh_scheduler()) {
+    invalidation_replayer_.SetInvalidationInfo(version, payload);
+    return;
+  }
+  DCHECK(core()->client());
+  core()->client()->SetInvalidationInfo(version, payload);
+}
+
+void CloudPolicyManager::InvalidatePolicy() {
+  if (!core()->refresh_scheduler()) {
+    invalidation_replayer_.InvalidatePolicy();
+    return;
+  }
+  core()->refresh_scheduler()->RefreshSoon();
+}
+
+void CloudPolicyManager::OnInvalidatorStateChanged(bool invalidations_enabled) {
+  if (!core()->refresh_scheduler()) {
+    invalidation_replayer_.OnInvalidatorStateChanged(invalidations_enabled);
+    return;
+  }
+  // Enable this code once method is added to CloudPolicyRefreshScheduler.
+  // 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();
+  invalidation_replayer_.Replay(this);
+}
+
 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