Created Mojo BudgetService implementation in chrome/browser

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>
 
@@ skipping 13 matching lines @@
 #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"
 #include "content/public/browser/storage_partition.h"
 #include "content/public/browser/web_contents.h"
 #include "content/public/common/notification_resources.h"
 #include "content/public/common/url_constants.h"
 #include "net/base/registry_controlled_domains/registry_controlled_domain.h"
+#include "third_party/WebKit/public/platform/modules/budget_service/budget_service.mojom.h"
 #include "ui/base/l10n/l10n_util.h"
 #include "url/gurl.h"
 
 #if BUILDFLAG(ANDROID_JAVA_UI)
 #include "chrome/browser/ui/android/tab_model/tab_model.h"
 #include "chrome/browser/ui/android/tab_model/tab_model_list.h"
 #else
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/browser_list.h"
 #include "chrome/browser/ui/tabs/tab_strip_model.h"
@@ skipping 161 matching lines @@
     const double budget) {
   // Record the budget available any time the budget is queried.
   UMA_HISTOGRAM_COUNTS_100("PushMessaging.BackgroundBudget", budget);
 
   // Get the site engagement score. Only used for UMA recording.
   SiteEngagementService* ses_service = SiteEngagementService::Get(profile_);
   double ses_score = ses_service->GetScore(origin);
 
   // Generate histograms for the GetBudget calls which would return "no budget"
   // or "low budget" if an API was available to app developers.
-  double cost = BudgetManager::GetCost(BudgetManager::CostType::SILENT_PUSH);
+  double cost =
+      BudgetManager::GetCost(blink::mojom::BudgetOperationType::SILENT_PUSH);
   if (budget < cost)
     UMA_HISTOGRAM_COUNTS_100("PushMessaging.SESForNoBudgetOrigin", ses_score);
   else if (budget < 2.0 * cost)
     UMA_HISTOGRAM_COUNTS_100("PushMessaging.SESForLowBudgetOrigin", ses_score);
 
   if (notification_needed && !notification_shown) {
     // If the worker needed to show a notification and didn't, check the budget
     // and take appropriate action.
     CheckForMissedNotification(origin, service_worker_registration_id,
                                message_handled_closure, budget);
@@ skipping 49 matching lines @@
 
 void PushMessagingNotificationManager::CheckForMissedNotification(
     const GURL& origin,
     int64_t service_worker_registration_id,
     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.
-  double cost = BudgetManager::GetCost(BudgetManager::CostType::SILENT_PUSH);
+  double cost =
+      BudgetManager::GetCost(blink::mojom::BudgetOperationType::SILENT_PUSH);
   if (budget >= cost) {
     RecordUserVisibleStatus(
         content::PUSH_USER_VISIBLE_STATUS_REQUIRED_BUT_NOT_SHOWN_USED_GRACE);
 
     BudgetManager* manager = BudgetManagerFactory::GetForProfile(profile_);
     // Update the stored budget.
     manager->StoreBudget(origin, budget - cost, message_handled_closure);
 
     return;
   }
@@ skipping 54 matching lines @@
   // Do not pass service worker scope. The origin will be used instead of the
   // service worker scope to determine whether a notification should be
   // attributed to a WebAPK on Android. This is OK because this code path is hit
   // rarely.
   PlatformNotificationServiceImpl::GetInstance()->DisplayPersistentNotification(
       profile_, persistent_notification_id, GURL() /* service_worker_scope */,
       origin, notification_data, NotificationResources());
 
   message_handled_closure.Run();
 }