| Index: trunk/src/chrome/browser/drive/drive_notification_manager.cc
|
| ===================================================================
|
| --- trunk/src/chrome/browser/drive/drive_notification_manager.cc (revision 208346)
|
| +++ trunk/src/chrome/browser/drive/drive_notification_manager.cc (working copy)
|
| @@ -6,9 +6,9 @@
|
|
|
| #include "base/metrics/histogram.h"
|
| #include "chrome/browser/drive/drive_notification_observer.h"
|
| -#include "chrome/browser/invalidation/invalidation_service.h"
|
| -#include "chrome/browser/invalidation/invalidation_service_factory.h"
|
| #include "chrome/browser/profiles/profile.h"
|
| +#include "chrome/browser/sync/profile_sync_service.h"
|
| +#include "chrome/browser/sync/profile_sync_service_factory.h"
|
| #include "google/cacheinvalidation/types.pb.h"
|
|
|
| namespace drive {
|
| @@ -42,16 +42,15 @@
|
|
|
| void DriveNotificationManager::Shutdown() {
|
| // Unregister for Drive notifications.
|
| - invalidation::InvalidationService* invalidation_service =
|
| - invalidation::InvalidationServiceFactory::GetForProfile(profile_);
|
| - if (!invalidation_service || !push_notification_registered_) {
|
| + ProfileSyncService* profile_sync_service =
|
| + ProfileSyncServiceFactory::GetForProfile(profile_);
|
| + if (!profile_sync_service || !push_notification_registered_) {
|
| return;
|
| }
|
|
|
| - // We unregister the handler without updating unregistering our IDs on
|
| - // purpose. See the class comment on the InvalidationService interface for
|
| - // more information.
|
| - invalidation_service->UnregisterInvalidationHandler(this);
|
| + profile_sync_service->UpdateRegisteredInvalidationIds(
|
| + this, syncer::ObjectIdSet());
|
| + profile_sync_service->UnregisterInvalidationHandler(this);
|
| }
|
|
|
| void DriveNotificationManager::OnInvalidatorStateChange(
|
| @@ -77,10 +76,9 @@
|
|
|
| // TODO(dcheng): Only acknowledge the invalidation once the fetch has
|
| // completed. http://crbug.com/156843
|
| - invalidation::InvalidationService* invalidation_service =
|
| - invalidation::InvalidationServiceFactory::GetForProfile(profile_);
|
| - DCHECK(invalidation_service);
|
| - invalidation_service->AcknowledgeInvalidation(
|
| + ProfileSyncService* profile_sync_service =
|
| + ProfileSyncServiceFactory::GetForProfile(profile_);
|
| + profile_sync_service->AcknowledgeInvalidation(
|
| invalidation_map.begin()->first,
|
| invalidation_map.begin()->second.ack_handle);
|
|
|
| @@ -130,19 +128,19 @@
|
| void DriveNotificationManager::RegisterDriveNotifications() {
|
| DCHECK(!push_notification_enabled_);
|
|
|
| - invalidation::InvalidationService* invalidation_service =
|
| - invalidation::InvalidationServiceFactory::GetForProfile(profile_);
|
| - if (!invalidation_service)
|
| + ProfileSyncService* profile_sync_service =
|
| + ProfileSyncServiceFactory::GetForProfile(profile_);
|
| + if (!profile_sync_service)
|
| return;
|
|
|
| - invalidation_service->RegisterInvalidationHandler(this);
|
| + profile_sync_service->RegisterInvalidationHandler(this);
|
| syncer::ObjectIdSet ids;
|
| ids.insert(invalidation::ObjectId(
|
| ipc::invalidation::ObjectSource::COSMO_CHANGELOG,
|
| kDriveInvalidationObjectId));
|
| - invalidation_service->UpdateRegisteredInvalidationIds(this, ids);
|
| + profile_sync_service->UpdateRegisteredInvalidationIds(this, ids);
|
| push_notification_registered_ = true;
|
| - OnInvalidatorStateChange(invalidation_service->GetInvalidatorState());
|
| + OnInvalidatorStateChange(profile_sync_service->GetInvalidatorState());
|
|
|
| UMA_HISTOGRAM_BOOLEAN("Drive.PushNotificationRegistered",
|
| push_notification_registered_);
|
|
|