Full hookup of BudgetManager interfaces to BudgetDatabase.

chrome/browser/budget_service/budget_database.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_BUDGET_SERVICE_BUDGET_DATABASE_H_
 #define CHROME_BROWSER_BUDGET_SERVICE_BUDGET_DATABASE_H_
 
 #include <list>
 #include <memory>
 #include <unordered_map>
 
 #include "base/callback_forward.h"
 #include "base/macros.h"
 #include "base/memory/weak_ptr.h"
 #include "components/leveldb_proto/proto_database.h"
+#include "third_party/WebKit/public/platform/modules/budget_service/budget_service.mojom.h"
 
 namespace base {
 class Clock;
 class SequencedTaskRunner;
 class Time;
 }
 
 namespace budget_service {
 class Budget;
 }
 
 class GURL;
 class Profile;
 
 // A class used to asynchronously read and write details of the budget
 // assigned to an origin. The class uses an underlying LevelDB.
 class BudgetDatabase {
  public:
-  // Data structure to hold information about the cumulative amount of budget
-  // at a particular point in time in the future. The budget described by a
-  // BudgetStatus represents the budget from zero or more BudgetChunk objects.
-  struct BudgetStatus {
-    BudgetStatus(double budget_at, base::Time time)
-        : budget_at(budget_at), time(time) {}
-    BudgetStatus(const BudgetStatus& other)
-        : budget_at(other.budget_at), time(other.time) {}
-
-    double budget_at;
-    base::Time time;
-  };
-
-  // Data structure for returing the budget decay prediction to the caller.
-  using BudgetPrediction = std::list<BudgetStatus>;
-
   // Callback for setting a budget value.
   using StoreBudgetCallback = base::Callback<void(bool success)>;
 
   // Callback for getting a list of all budget chunks.
-  using GetBudgetDetailsCallback =
-      base::Callback<void(bool success, const BudgetPrediction& prediction)>;
+  using GetBudgetCallback = blink::mojom::BudgetService::GetBudgetCallback;
 
   // The database_dir specifies the location of the budget information on
   // disk. The task_runner is used by the ProtoDatabase to handle all blocking
   // calls and disk access.
   BudgetDatabase(Profile* profile,
                  const base::FilePath& database_dir,
                  const scoped_refptr<base::SequencedTaskRunner>& task_runner);
   ~BudgetDatabase();
 
   // Get the full budget expectation for the origin. This will return a
   // sequence of time points and the expected budget at those times.
-  void GetBudgetDetails(const GURL& origin,
-                        const GetBudgetDetailsCallback& callback);
+  void GetBudgetDetails(const GURL& origin, const GetBudgetCallback& callback);
 
   // Spend a particular amount of budget for an origin. The callback takes
   // a boolean which indicates whether the origin had enough budget to spend.
   // If it returns success, then the budget was deducted and the result written
   // to the database.
   void SpendBudget(const GURL& origin,
                    double amount,
                    const StoreBudgetCallback& callback);
 
  private:
@@ skipping 31 matching lines @@
     DISALLOW_COPY_AND_ASSIGN(BudgetInfo);
   };
 
   using AddToCacheCallback = base::Callback<void(bool success)>;
   using SyncCacheCallback = base::Callback<void(bool success)>;
 
   void OnDatabaseInit(bool success);
 
   bool IsCached(const GURL& origin) const;
 
+  double GetBudget(const GURL& origin) const;
+
   void AddToCache(const GURL& origin,
                   const AddToCacheCallback& callback,
                   bool success,
                   std::unique_ptr<budget_service::Budget> budget);
 
   void GetBudgetAfterSync(const GURL& origin,
-                          const GetBudgetDetailsCallback& callback,
+                          const GetBudgetCallback& callback,
                           bool success);
 
   void SpendBudgetAfterSync(const GURL& origin,
                             double amount,
                             const StoreBudgetCallback& callback,
                             bool success);
 
   void WriteCachedValuesToDatabase(const GURL& origin,
                                    const StoreBudgetCallback& callback);
 
@@ skipping 20 matching lines @@
 
   // The clock used to vend times.
   std::unique_ptr<base::Clock> clock_;
 
   base::WeakPtrFactory<BudgetDatabase> weak_ptr_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(BudgetDatabase);
 };
 
 #endif  // CHROME_BROWSER_BUDGET_SERVICE_BUDGET_DATABASE_H_