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 "base/callback_forward.h"
+#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 currently
 // implements a grace period of 1 non-visual notification in 10.
-class BackgroundBudgetService {
+class BackgroundBudgetService : public KeyedService {
  public:
+  explicit BackgroundBudgetService(Profile* profile);
+  ~BackgroundBudgetService() override {}
+
   static void RegisterProfilePrefs(user_prefs::PrefRegistrySyncable* registry);
 
   // Get the budget associated with the origin. This currently tracks a string
   // which holds 0 and 1 for the last 10 push messages and whether they
   // triggered a visual notification.
-  static std::string GetBudget(Profile* profile, const GURL& origin);
+  std::string GetBudget(const GURL& origin);
 
   // Store the budget associated with the origin. notifications_shown is
   // expected to be a string encoding whether the last 10 push messages
   // triggered
   // a visual notification.
-  static void StoreBudget(Profile* profile,
-                          const GURL& origin,
-                          const std::string& notifications_shown);
+  void StoreBudget(const GURL& origin, const std::string& notifications_shown);
 
  private:
-  DISALLOW_IMPLICIT_CONSTRUCTORS(BackgroundBudgetService);
+  Profile* profile_;
+  DISALLOW_COPY_AND_ASSIGN(BackgroundBudgetService);
 };
 
 #endif  // CHROME_BROWSER_PUSH_MESSAGING_BACKGROUND_BUDGET_SERVICE_H_