| Index: chrome/browser/budget_service/budget_database.h
|
| diff --git a/chrome/browser/budget_service/budget_database.h b/chrome/browser/budget_service/budget_database.h
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..155f23d22c8f6e708c1f7282669e55b7ce0a9ae0
|
| --- /dev/null
|
| +++ b/chrome/browser/budget_service/budget_database.h
|
| @@ -0,0 +1,60 @@
|
| +// 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 <memory>
|
| +
|
| +#include "base/callback_forward.h"
|
| +#include "base/macros.h"
|
| +#include "base/memory/weak_ptr.h"
|
| +#include "components/leveldb_proto/proto_database.h"
|
| +
|
| +namespace base {
|
| +class SequencedTaskRunner;
|
| +}
|
| +
|
| +namespace budget_service {
|
| +class Budget;
|
| +}
|
| +
|
| +class GURL;
|
| +
|
| +// 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:
|
| + // Callback for the basic GetBudget call.
|
| + using GetValueCallback =
|
| + base::Callback<void(bool success,
|
| + std::unique_ptr<budget_service::Budget>)>;
|
| +
|
| + // Callback for setting a budget value.
|
| + using SetValueCallback = base::Callback<void(bool success)>;
|
| +
|
| + // 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(const base::FilePath& database_dir,
|
| + const scoped_refptr<base::SequencedTaskRunner>& task_runner);
|
| + ~BudgetDatabase();
|
| +
|
| + void GetValue(const GURL& origin, const GetValueCallback& callback);
|
| + void SetValue(const GURL& origin,
|
| + const budget_service::Budget& budget,
|
| + const SetValueCallback& callback);
|
| +
|
| + private:
|
| + void OnDatabaseInit(bool success);
|
| +
|
| + // The database for storing budget information.
|
| + std::unique_ptr<leveldb_proto::ProtoDatabase<budget_service::Budget>> db_;
|
| +
|
| + base::WeakPtrFactory<BudgetDatabase> weak_ptr_factory_;
|
| +
|
| + DISALLOW_COPY_AND_ASSIGN(BudgetDatabase);
|
| +};
|
| +
|
| +#endif // CHROME_BROWSER_BUDGET_SERVICE_BUDGET_DATABASE_H_
|
|
|