Make BackgroundBudgetService a KeyedService

chrome/browser/push_messaging/background_budget_service.h

 // Copyright 2016 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.
 
 #ifndef CHROME_BROWSER_PUSH_MESSAGING_BACKGROUND_BUDGET_SERVICE_H_
 #define CHROME_BROWSER_PUSH_MESSAGING_BACKGROUND_BUDGET_SERVICE_H_
 
 #include <string>
 
+#include "components/keyed_service/core/keyed_service.h"
 #include "url/gurl.h"
 
 class Profile;
 
 namespace user_prefs {
 class PrefRegistrySyncable;
 }
 
 // A budget service to help Chrome decide how much background work a service
 // worker should be able to do on behalf of the user. The budget is calculated
 // based on the Site Engagment Score and is consumed when a service worker
 // doesn't show a notification when it needed to.
-class BackgroundBudgetService {
+class BackgroundBudgetService : public KeyedService {
  public:
+  explicit BackgroundBudgetService(Profile* profile);

[Peter Beverloo, 2016/04/14 16:01:18]

nit: virtual destructor

[harkness, 2016/04/14 17:44:19]

Done.

+
   static void RegisterProfilePrefs(user_prefs::PrefRegistrySyncable* registry);
 
   // Get the budget associated with the origin. This is returned as the double
   // budget.
-  static void GetBudget(Profile* profile, const GURL& origin, double& budget);
+  void GetBudget(const GURL& origin, double& budget);
 
   // Store the budget associated with the origin.
-  static void StoreBudget(Profile* profile,
-                          const GURL& origin,
-                          const double budget);
+  void StoreBudget(const GURL& origin, const double budget);
 
  private:
-  DISALLOW_IMPLICIT_CONSTRUCTORS(BackgroundBudgetService);
+  Profile* profile_;
+
+  DISALLOW_COPY_AND_ASSIGN(BackgroundBudgetService);
 };
 
 #endif  // CHROME_BROWSER_PUSH_MESSAGING_BACKGROUND_BUDGET_SERVICE_H_