blob: adb397b6b3cf41543c4e9f5665387e6ebc2f368b [file] [log] [blame]
// Copyright 2017 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 <string>
#include <vector>
#include "base/files/file_path.h"
#include "base/macros.h"
#include "base/memory/weak_ptr.h"
#include "components/download/internal/store.h"
#include "components/leveldb_proto/proto_database.h"
namespace protodb {
class Entry;
} // namespace protodb
namespace download {
// DownloadStore provides a layer around a LevelDB proto database that persists
// the download request, scheduling params and metadata to the disk. The data is
// read during initialization and presented to the caller after converting to
// Entry entries.
class DownloadStore : public Store {
const base::FilePath& database_dir,
std::unique_ptr<leveldb_proto::ProtoDatabase<protodb::Entry>> db);
~DownloadStore() override;
// Store implementation.
bool IsInitialized() override;
void Initialize(InitCallback callback) override;
void HardRecover(StoreCallback callback) override;
void Update(const Entry& entry, StoreCallback callback) override;
void Remove(const std::string& guid, StoreCallback callback) override;
void OnDatabaseInited(InitCallback callback, bool success);
void OnDatabaseLoaded(InitCallback callback,
bool success,
std::unique_ptr<std::vector<protodb::Entry>> protos);
void OnDatabaseDestroyed(StoreCallback callback, bool success);
void OnDatabaseInitedAfterDestroy(StoreCallback callback, bool success);
std::unique_ptr<leveldb_proto::ProtoDatabase<protodb::Entry>> db_;
base::FilePath database_dir_;
bool is_initialized_;
base::WeakPtrFactory<DownloadStore> weak_factory_;
} // namespace download