123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117 |
- // Copyright 2018 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 COMPONENTS_LEVELDB_PROTO_INTERNAL_UNIQUE_PROTO_DATABASE_H_
- #define COMPONENTS_LEVELDB_PROTO_INTERNAL_UNIQUE_PROTO_DATABASE_H_
- #include <memory>
- #include <string>
- #include "base/bind.h"
- #include "base/component_export.h"
- #include "base/sequence_checker.h"
- #include "components/leveldb_proto/internal/proto_leveldb_wrapper.h"
- #include "components/leveldb_proto/public/proto_database.h"
- namespace leveldb_proto {
- // An implementation of ProtoDatabase<std::string> that manages the lifecycle of
- // a unique LevelDB instance.
- class COMPONENT_EXPORT(LEVELDB_PROTO) UniqueProtoDatabase {
- public:
- explicit UniqueProtoDatabase(
- const scoped_refptr<base::SequencedTaskRunner>& task_runner);
- explicit UniqueProtoDatabase(std::unique_ptr<ProtoLevelDBWrapper>);
- virtual ~UniqueProtoDatabase();
- UniqueProtoDatabase(
- const base::FilePath& database_dir,
- const leveldb_env::Options& options,
- const scoped_refptr<base::SequencedTaskRunner>& task_runner);
- virtual void Init(const std::string& client_name,
- Callbacks::InitStatusCallback callback);
- virtual void InitWithDatabase(LevelDB* database,
- const base::FilePath& database_dir,
- const leveldb_env::Options& options,
- bool destroy_on_corruption,
- Callbacks::InitStatusCallback callback);
- virtual void UpdateEntries(std::unique_ptr<KeyValueVector> entries_to_save,
- std::unique_ptr<KeyVector> keys_to_remove,
- Callbacks::UpdateCallback callback);
- virtual void UpdateEntriesWithRemoveFilter(
- std::unique_ptr<KeyValueVector> entries_to_save,
- const KeyFilter& delete_key_filter,
- Callbacks::UpdateCallback callback);
- virtual void UpdateEntriesWithRemoveFilter(
- std::unique_ptr<KeyValueVector> entries_to_save,
- const KeyFilter& delete_key_filter,
- const std::string& target_prefix,
- Callbacks::UpdateCallback callback);
- virtual void LoadEntries(typename Callbacks::LoadCallback callback);
- virtual void LoadEntriesWithFilter(const KeyFilter& filter,
- typename Callbacks::LoadCallback callback);
- virtual void LoadEntriesWithFilter(const KeyFilter& key_filter,
- const leveldb::ReadOptions& options,
- const std::string& target_prefix,
- typename Callbacks::LoadCallback callback);
- virtual void LoadKeysAndEntries(
- typename Callbacks::LoadKeysAndEntriesCallback callback);
- virtual void LoadKeysAndEntriesWithFilter(
- const KeyFilter& filter,
- typename Callbacks::LoadKeysAndEntriesCallback callback);
- virtual void LoadKeysAndEntriesWithFilter(
- const KeyFilter& filter,
- const leveldb::ReadOptions& options,
- const std::string& target_prefix,
- typename Callbacks::LoadKeysAndEntriesCallback callback);
- virtual void LoadKeysAndEntriesInRange(
- const std::string& start,
- const std::string& end,
- typename Callbacks::LoadKeysAndEntriesCallback callback);
- virtual void LoadKeysAndEntriesWhile(
- const std::string& start,
- const KeyIteratorController& controller,
- typename Callbacks::LoadKeysAndEntriesCallback callback);
- virtual void LoadKeys(Callbacks::LoadKeysCallback callback);
- virtual void LoadKeys(const std::string& target_prefix,
- Callbacks::LoadKeysCallback callback);
- virtual void GetEntry(const std::string& key,
- typename Callbacks::GetCallback callback);
- virtual void Destroy(Callbacks::DestroyCallback callback);
- void RemoveKeysForTesting(const KeyFilter& key_filter,
- const std::string& target_prefix,
- Callbacks::UpdateCallback callback);
- bool GetApproximateMemoryUse(uint64_t* approx_mem_use);
- // Sets the identifier used by the underlying LevelDB wrapper to record
- // metrics.
- void SetMetricsId(const std::string& id);
- protected:
- std::unique_ptr<ProtoLevelDBWrapper> db_wrapper_;
- private:
- SEQUENCE_CHECKER(sequence_checker_);
- base::FilePath database_dir_;
- leveldb_env::Options options_;
- std::unique_ptr<LevelDB> db_;
- };
- } // namespace leveldb_proto
- #endif // COMPONENTS_LEVELDB_PROTO_INTERNAL_UNIQUE_PROTO_DATABASE_H_
|