Add the future storage framework for the BudgetDatabase.

chrome/browser/budget_service/budget_database.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_database.h"
+
+#include "chrome/browser/budget_service/budget.pb.h"
+#include "components/leveldb_proto/proto_database_impl.h"
+#include "content/public/browser/browser_thread.h"
+
+namespace {
+
+// UMA are logged for the database with this string as part of the name.
+// They will be LevelDB.*.BackgroundBudgetService. Changes here should be
+// synchronized with histograms.xml.
+const char kDatabaseUMAName[] = "BackgroundBudgetService";
+
+}  // namespace
+
+BudgetDatabase::BudgetDatabase(
+    const base::FilePath& database_dir,
+    const scoped_refptr<base::SequencedTaskRunner>& task_runner)
+    : db_(new leveldb_proto::ProtoDatabaseImpl<budget_service::Budget>(
+          task_runner)),
+      weak_ptr_factory_(this) {
+  db_->Init(kDatabaseUMAName, database_dir,
+            base::Bind(&BudgetDatabase::OnDatabaseInit,
+                       weak_ptr_factory_.GetWeakPtr()));
+}
+
+BudgetDatabase::~BudgetDatabase() {}
+
+void BudgetDatabase::OnDatabaseInit(bool success) {
+  // TODO(harkness): Consider caching the budget database now?
+}
+
+void BudgetDatabase::GetValue(const GURL& origin,
+                              const GetValueCallback& callback) {
+  DCHECK_EQ(origin.GetOrigin(), origin);
+  db_->GetEntry(origin.spec(), callback);
+}
+
+void BudgetDatabase::SetValue(const GURL& origin,
+                              const budget_service::Budget& budget,
+                              const SetValueCallback& callback) {
+  DCHECK_EQ(origin.GetOrigin(), origin);
+
+  // Build structures to hold the updated values.
+  std::unique_ptr<
+      leveldb_proto::ProtoDatabase<budget_service::Budget>::KeyEntryVector>
+      entries(new leveldb_proto::ProtoDatabase<
+              budget_service::Budget>::KeyEntryVector());
+  entries->push_back(std::make_pair(origin.spec(), budget));
+  std::unique_ptr<std::vector<std::string>> keys_to_remove(
+      new std::vector<std::string>());
+
+  // Send the updates to the database.
+  db_->UpdateEntries(std::move(entries), std::move(keys_to_remove), callback);
+}