| Index: chrome/browser/chromeos/gdata/gdata_leveldb.cc
|
| ===================================================================
|
| --- chrome/browser/chromeos/gdata/gdata_leveldb.cc (revision 133818)
|
| +++ chrome/browser/chromeos/gdata/gdata_leveldb.cc (working copy)
|
| @@ -1,204 +0,0 @@
|
| -// Copyright (c) 2012 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/chromeos/gdata/gdata_leveldb.h"
|
| -
|
| -#include <string>
|
| -
|
| -#include "base/logging.h"
|
| -#include "chrome/browser/chromeos/gdata/gdata_files.h"
|
| -#include "leveldb/write_batch.h"
|
| -
|
| -namespace gdata {
|
| -namespace {
|
| -
|
| -const char kResourceIdPrefix[] = "id:";
|
| -const char kPathPrefix[] = "path:";
|
| -
|
| -// Append prefix id: to |resource_id|.
|
| -std::string ResourceIdToKey(const std::string& resource_id) {
|
| - return std::string(kResourceIdPrefix) + resource_id;
|
| -}
|
| -
|
| -// Append prefix path: to |path|.
|
| -std::string PathToKey(const FilePath& path) {
|
| - return std::string(kPathPrefix) + path.value();
|
| -}
|
| -
|
| -GDataDB::Status GetStatus(const leveldb::Status& db_status) {
|
| - if (db_status.ok())
|
| - return GDataDB::DB_OK;
|
| -
|
| - if (db_status.IsNotFound())
|
| - return GDataDB::DB_KEY_NOT_FOUND;
|
| -
|
| - if (db_status.IsCorruption())
|
| - return GDataDB::DB_CORRUPTION;
|
| -
|
| - if (db_status.IsIOError())
|
| - return GDataDB::DB_IO_ERROR;
|
| -
|
| - NOTREACHED();
|
| - return GDataDB::DB_INTERNAL_ERROR;
|
| -}
|
| -
|
| -} // namespace
|
| -
|
| -GDataLevelDB::GDataLevelDB() {
|
| -}
|
| -
|
| -GDataLevelDB::~GDataLevelDB() {
|
| -}
|
| -
|
| -void GDataLevelDB::Init(const FilePath& db_path) {
|
| - leveldb::DB* level_db = NULL;
|
| - leveldb::Options options;
|
| - options.create_if_missing = true;
|
| - leveldb::Status db_status = leveldb::DB::Open(options,
|
| - db_path.Append("level_db").value(), &level_db);
|
| - DCHECK(level_db);
|
| - // TODO(achuith): If db cannot be opened, we should try to recover it.
|
| - // If that fails, we should just delete it and create a new file.
|
| - DCHECK(db_status.ok());
|
| - level_db_.reset(level_db);
|
| -}
|
| -
|
| -GDataDB::Status GDataLevelDB::Put(const GDataEntry& entry) {
|
| - // Write the serialized proto.
|
| - std::string serialized_proto;
|
| - entry.SerializeToString(&serialized_proto);
|
| -
|
| - leveldb::WriteBatch batch;
|
| - const std::string resource_id_key =
|
| - ResourceIdToKey(entry.resource_id());
|
| - const std::string path_key = PathToKey(entry.GetFilePath());
|
| - batch.Put(leveldb::Slice(resource_id_key), leveldb::Slice(serialized_proto));
|
| - // Note we store the resource_id without prefix when it's the value.
|
| - batch.Put(leveldb::Slice(path_key), leveldb::Slice(entry.resource_id()));
|
| - leveldb::Status db_status = level_db_->Write(
|
| - leveldb::WriteOptions(),
|
| - &batch);
|
| -
|
| - DVLOG(1) << "GDataLevelDB::Put resource_id key = " << resource_id_key
|
| - << ", path key = " << path_key;
|
| - return GetStatus(db_status);
|
| -}
|
| -
|
| -GDataDB::Status GDataLevelDB::DeleteByResourceId(
|
| - const std::string& resource_id) {
|
| - scoped_ptr<GDataEntry> entry;
|
| - Status status = GetByResourceId(resource_id, &entry);
|
| - if (status == DB_KEY_NOT_FOUND)
|
| - return DB_OK;
|
| - else if (status != DB_OK)
|
| - return status;
|
| -
|
| - leveldb::WriteBatch batch;
|
| - const std::string resource_id_key = ResourceIdToKey(resource_id);
|
| - const std::string path_key = PathToKey(entry->GetFilePath());
|
| - batch.Delete(leveldb::Slice(resource_id_key));
|
| - batch.Delete(leveldb::Slice(path_key));
|
| -
|
| - leveldb::Status db_status = level_db_->Write(leveldb::WriteOptions(),
|
| - &batch);
|
| - return GetStatus(db_status);
|
| -}
|
| -
|
| -GDataDB::Status GDataLevelDB::DeleteByPath(
|
| - const FilePath& path) {
|
| - std::string resource_id;
|
| - const Status status = ResourceIdForPath(path, &resource_id);
|
| - if (status != DB_OK)
|
| - return status;
|
| - return DeleteByResourceId(resource_id);
|
| -}
|
| -
|
| -GDataDB::Status GDataLevelDB::GetByResourceId(const std::string& resource_id,
|
| - scoped_ptr<GDataEntry>* entry) {
|
| - entry->reset();
|
| - std::string serialized_proto;
|
| - const std::string resource_id_key = ResourceIdToKey(resource_id);
|
| - const leveldb::Status db_status = level_db_->Get(leveldb::ReadOptions(),
|
| - leveldb::Slice(resource_id_key), &serialized_proto);
|
| -
|
| - if (db_status.IsNotFound())
|
| - return DB_KEY_NOT_FOUND;
|
| -
|
| - if (db_status.ok()) {
|
| - DCHECK(!serialized_proto.empty());
|
| - *entry = GDataEntry::FromProtoString(serialized_proto);
|
| - DCHECK(entry->get());
|
| - return DB_OK;
|
| - }
|
| - return GetStatus(db_status);
|
| -}
|
| -
|
| -GDataDB::Status GDataLevelDB::GetByPath(const FilePath& path,
|
| - scoped_ptr<GDataEntry>* entry) {
|
| - entry->reset();
|
| - std::string resource_id;
|
| - const Status status = ResourceIdForPath(path, &resource_id);
|
| - if (status != DB_OK)
|
| - return status;
|
| - return GetByResourceId(resource_id, entry);
|
| -}
|
| -
|
| -GDataDB::Status GDataLevelDB::ResourceIdForPath(const FilePath& path,
|
| - std::string* resource_id) {
|
| - const std::string path_key = PathToKey(path);
|
| - const leveldb::Status db_status = level_db_->Get(
|
| - leveldb::ReadOptions(), path_key, resource_id);
|
| -
|
| - return GetStatus(db_status);
|
| -}
|
| -
|
| -scoped_ptr<GDataDBIter> GDataLevelDB::CreateIterator(const FilePath& path) {
|
| - return scoped_ptr<GDataDBIter>(new GDataLevelDBIter(
|
| - scoped_ptr<leveldb::Iterator>(
|
| - level_db_->NewIterator(leveldb::ReadOptions())),
|
| - this,
|
| - path));
|
| -}
|
| -
|
| -GDataLevelDBIter::GDataLevelDBIter(scoped_ptr<leveldb::Iterator> level_db_iter,
|
| - GDataDB* db,
|
| - const FilePath& path)
|
| - : level_db_iter_(level_db_iter.Pass()),
|
| - db_(db),
|
| - path_(path) {
|
| - const std::string path_key = PathToKey(path);
|
| - level_db_iter_->Seek(leveldb::Slice(path_key));
|
| -}
|
| -
|
| -GDataLevelDBIter::~GDataLevelDBIter() {
|
| -}
|
| -
|
| -bool GDataLevelDBIter::GetNext(std::string* path,
|
| - scoped_ptr<GDataEntry>* entry) {
|
| - DCHECK(path);
|
| - DCHECK(entry);
|
| - path->clear();
|
| - entry->reset();
|
| -
|
| - if (!level_db_iter_->Valid())
|
| - return false;
|
| -
|
| - // Only consider keys under |path|.
|
| - const std::string path_key = PathToKey(path_);
|
| - leveldb::Slice key_slice(level_db_iter_->key());
|
| - if (!key_slice.starts_with(path_key))
|
| - return false;
|
| -
|
| - GDataDB::Status status =
|
| - db_->GetByResourceId(level_db_iter_->value().ToString(), entry);
|
| - DCHECK_EQ(GDataDB::DB_OK, status);
|
| -
|
| - key_slice.remove_prefix(sizeof(kPathPrefix) - 1);
|
| - path->assign(key_slice.ToString());
|
| -
|
| - level_db_iter_->Next();
|
| - return true;
|
| -}
|
| -
|
| -} // namespace gdata
|
|
|