Make BackgroundBudgetService a KeyedService

chrome/browser/push_messaging/background_budget_service_factory.cc

+// 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.
+
+#include "chrome/browser/push_messaging/background_budget_service_factory.h"
+
+#include "chrome/browser/engagement/site_engagement_service_factory.h"
+#include "chrome/browser/profiles/profile.h"
+#include "chrome/browser/push_messaging/background_budget_service.h"
+#include "components/keyed_service/content/browser_context_dependency_manager.h"
+#include "content/public/browser/browser_context.h"
+
+// static
+BackgroundBudgetService* BackgroundBudgetServiceFactory::GetForProfile(
+    content::BrowserContext* profile) {
+  // Budget tracking is not supported in incognito mode.
+  if (profile->IsOffTheRecord())

[Peter Beverloo, 2016/04/15 12:22:24]

Why not? The dependency (SiteEngagementServiceFactory) is available in
incognito, so I see no reason to derive here?

+    return nullptr;
+
+  return static_cast<BackgroundBudgetService*>(
+      GetInstance()->GetServiceForBrowserContext(profile, true));
+}
+
+// static
+BackgroundBudgetServiceFactory* BackgroundBudgetServiceFactory::GetInstance() {
+  return base::Singleton<BackgroundBudgetServiceFactory>::get();
+}
+
+BackgroundBudgetServiceFactory::BackgroundBudgetServiceFactory()
+    : BrowserContextKeyedServiceFactory(
+          "BackgroundBudgetService",
+          BrowserContextDependencyManager::GetInstance()) {
+  DependsOn(SiteEngagementServiceFactory::GetInstance());
+}
+
+KeyedService* BackgroundBudgetServiceFactory::BuildServiceInstanceFor(
+    content::BrowserContext* profile) const {
+  return new BackgroundBudgetService(static_cast<Profile*>(profile));
+}