Skeleton GCMAppHandler for Push API

components/gcm_driver/gcm_driver.cc

 // Copyright 2014 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/gcm_driver/gcm_driver.h"
 
 #include <algorithm>
 
 #include "base/logging.h"
 #include "components/gcm_driver/gcm_app_handler.h"
 
 namespace gcm {
 
 GCMDriver::GCMDriver() {
 }
 
 GCMDriver::~GCMDriver() {
 }
 
+void GCMDriver::SetLazyLoadAppHandlersClosure(
+    const base::Closure& lazy_load_app_handlers) {
+  lazy_load_app_handlers_ = lazy_load_app_handlers;
+}
+
 void GCMDriver::Register(const std::string& app_id,
                          const std::vector<std::string>& sender_ids,
                          const RegisterCallback& callback) {
   DCHECK(!app_id.empty());
   DCHECK(!sender_ids.empty());
   DCHECK(!callback.is_null());
 
   GCMClient::Result result = EnsureStarted();
   if (result != GCMClient::SUCCESS) {
     callback.Run(std::string(), result);
@@ skipping 111 matching lines @@
        iter != app_handlers_.end(); ++iter) {
     iter->second->ShutdownHandler();
   }
   app_handlers_.clear();
 }
 
 void GCMDriver::AddAppHandler(const std::string& app_id,
                               GCMAppHandler* handler) {
   DCHECK(!app_id.empty());
   DCHECK(handler);
-  DCHECK(app_handlers_.find(app_id) == app_handlers_.end());
+  DCHECK(app_handlers_.find(app_id) == app_handlers_.end() ||
+         app_handlers_[app_id] == handler);
   app_handlers_[app_id] = handler;
 }
 
 void GCMDriver::RemoveAppHandler(const std::string& app_id) {
   DCHECK(!app_id.empty());
   app_handlers_.erase(app_id);
 }
 
 GCMAppHandler* GCMDriver::GetAppHandler(const std::string& app_id) {
+  if (!lazy_load_app_handlers_.is_null()) {
+    lazy_load_app_handlers_.Run();
+    lazy_load_app_handlers_.Reset();
+  }
   GCMAppHandlerMap::const_iterator iter = app_handlers_.find(app_id);
   return iter == app_handlers_.end() ? &default_app_handler_ : iter->second;
 }
 
 bool GCMDriver::HasRegisterCallback(const std::string& app_id) {
   return register_callbacks_.find(app_id) != register_callbacks_.end();
 }
 
 void GCMDriver::ClearCallbacks() {
   register_callbacks_.clear();
   unregister_callbacks_.clear();
   send_callbacks_.clear();
 }
 
 bool GCMDriver::IsAsyncOperationPending(const std::string& app_id) const {
   return register_callbacks_.find(app_id) != register_callbacks_.end() ||
          unregister_callbacks_.find(app_id) != unregister_callbacks_.end();
 }
 
 }  // namespace gcm