Make BackgroundBudgetService a KeyedService

chrome/browser/push_messaging/push_messaging_notification_manager.cc

 // Copyright 2015 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/push_messaging/push_messaging_notification_manager.h"
 
 #include <stddef.h>
 
 #include <bitset>
 
 #include "base/metrics/histogram_macros.h"
 #include "base/strings/utf_string_conversions.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/notifications/platform_notification_service_impl.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/push_messaging/background_budget_service.h"
+#include "chrome/browser/push_messaging/background_budget_service_factory.h"
 #include "chrome/browser/push_messaging/push_messaging_constants.h"
 #include "chrome/common/features.h"
 #include "chrome/grit/generated_resources.h"
 #include "components/rappor/rappor_utils.h"
 #include "components/url_formatter/elide_url.h"
 #include "content/public/browser/browser_context.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/browser/platform_notification_context.h"
 #include "content/public/browser/push_messaging_service.h"
 #include "content/public/browser/render_frame_host.h"
@@ skipping 155 matching lines @@
           notification_database_data.origin, false /* by_user */);
 
       break;
     }
   }
 
   if (notification_needed && !notification_shown) {
     // Only track budget for messages that needed to show a notification but
     // did not.
     double budget = 0;
-    BackgroundBudgetService::GetBudget(profile_, origin, budget);
+    BackgroundBudgetService* service =
+        BackgroundBudgetServiceFactory::GetForProfile(profile_);
+    service->GetBudget(origin, budget);
     DidGetBudget(origin, service_worker_registration_id,
                  message_handled_closure, budget);
     return;
   }
 
   if (notification_needed && notification_shown) {
     RecordUserVisibleStatus(
         content::PUSH_USER_VISIBLE_STATUS_REQUIRED_AND_SHOWN);
   } else if (!notification_needed && !notification_shown) {
     RecordUserVisibleStatus(
@@ skipping 45 matching lines @@
     const base::Closure& message_handled_closure,
     const double budget) {
   DCHECK_CURRENTLY_ON(BrowserThread::UI);
 
   // If the service needed to show a notification but did not, update the
   // budget. Currently this just subtracts 1.0, but we may expand the cost to
   // include things like whether wifi is available in the mobile case.
   double cost = 1.0;
   if (budget >= cost) {
     // Update the stored budget.
-    BackgroundBudgetService::StoreBudget(profile_, origin, budget - cost);
+    BackgroundBudgetService* service =
+        BackgroundBudgetServiceFactory::GetForProfile(profile_);
+    service->StoreBudget(origin, budget - cost);
 
     RecordUserVisibleStatus(
         content::PUSH_USER_VISIBLE_STATUS_REQUIRED_BUT_NOT_SHOWN_USED_GRACE);
     message_handled_closure.Run();
     return;
   }
 
   RecordUserVisibleStatus(
       content::PUSH_USER_VISIBLE_STATUS_REQUIRED_BUT_NOT_SHOWN_GRACE_EXCEEDED);
   rappor::SampleDomainAndRegistryFromGURL(
@@ skipping 46 matching lines @@
     message_handled_closure.Run();
     return;
   }
 
   PlatformNotificationServiceImpl::GetInstance()->DisplayPersistentNotification(
       profile_, persistent_notification_id, origin, notification_data,
       NotificationResources());
 
   message_handled_closure.Run();
 }