Connect Mojo budget_service to BudgetManager implementation of Reserve.

chrome/browser/budget_service/budget_manager.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/budget_service/budget_manager.h"
 
 #include <stdint.h>
 
 #include "base/callback.h"
 #include "base/memory/ptr_util.h"
@@ skipping 56 matching lines @@
 }
 
 void BudgetManager::GetBudget(const url::Origin& origin,
                               const GetBudgetCallback& callback) {
   db_.GetBudgetDetails(origin, callback);
 }
 
 void BudgetManager::Reserve(const url::Origin& origin,
                             blink::mojom::BudgetOperationType type,
                             const ReserveCallback& callback) {
-  db_.SpendBudget(
-      origin, GetCost(type),
-      base::Bind(&BudgetManager::DidReserve, weak_ptr_factory_.GetWeakPtr(),
-                 origin, type, callback));
+  db_.SpendBudget(origin, GetCost(type),
+                  base::Bind(&BudgetManager::DidReserve,
+                             weak_ptr_factory_.GetWeakPtr(), origin, callback));
 }
 
 void BudgetManager::Consume(const url::Origin& origin,
                             blink::mojom::BudgetOperationType type,
                             const ConsumeCallback& callback) {
   bool found_reservation = false;
 
   // First, see if there is a reservation already.
-  auto count = reservation_map_.find(origin.host());
+  auto count = reservation_map_.find(origin);
   if (count != reservation_map_.end()) {
     if (count->second == 1)
-      reservation_map_.erase(origin.host());
+      reservation_map_.erase(origin);
     else
-      reservation_map_[origin.host()]--;
+      reservation_map_[origin]--;
     found_reservation = true;
   }
 
   if (found_reservation) {
     callback.Run(true);
     return;
   }
 
   // If there wasn't a reservation already, try to directly consume budget.
   // The callback will return directly to the caller.
-  db_.SpendBudget(origin, GetCost(type), callback);
+  db_.SpendBudget(origin, GetCost(type),
+                  base::Bind(&BudgetManager::DidConsume,
+                             weak_ptr_factory_.GetWeakPtr(), callback));
+}
+
+void BudgetManager::DidConsume(const ConsumeCallback& callback,
+                               blink::mojom::BudgetServiceErrorType error,
+                               bool success) {
+  // The caller of Consume only cares whether it succeeded or failed and not
+  // why. So, only return a combined bool.
+  if (error != blink::mojom::BudgetServiceErrorType::NONE) {
+    callback.Run(false /* success */);
+    return;
+  }
+  callback.Run(success);
 }
 
 void BudgetManager::DidReserve(const url::Origin& origin,
-                               blink::mojom::BudgetOperationType type,
                                const ReserveCallback& callback,
+                               blink::mojom::BudgetServiceErrorType error,
                                bool success) {
-  if (!success) {
-    callback.Run(false);
-    return;
-  }
+  // If the call succeeded, write the new reservation into the map.
+  if (success && error == blink::mojom::BudgetServiceErrorType::NONE)
+    reservation_map_[origin]++;
 
-  // Write the new reservation into the map.
-  reservation_map_[origin.host()]++;
-  callback.Run(true);
+  callback.Run(error, success);
 }