content/browser/indexed_db/leveldb/leveldb_database.h

   1 // Copyright (c) 2013 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #ifndef CONTENT_BROWSER_INDEXED_DB_LEVELDB_LEVELDB_DATABASE_H_
   6 #define CONTENT_BROWSER_INDEXED_DB_LEVELDB_LEVELDB_DATABASE_H_
   7
   8 #include <string>
   9
  10 #include "base/files/file_path.h"
  11 #include "base/memory/scoped_ptr.h"
  12 #include "base/strings/string16.h"
  13 #include "base/strings/string_piece.h"
  14 #include "content/common/content_export.h"
  15 #include "third_party/leveldatabase/src/include/leveldb/comparator.h"
  16 #include "third_party/leveldatabase/src/include/leveldb/status.h"
  17
  18 namespace leveldb {
  19 class Comparator;
  20 class DB;
  21 class Env;
  22 class Snapshot;
  23 }
  24
  25 namespace content {
  26
  27 class LevelDBComparator;
  28 class LevelDBDatabase;
  29 class LevelDBIterator;
  30 class LevelDBWriteBatch;
  31
  32 class LevelDBSnapshot {
  33  private:
  34   friend class LevelDBDatabase;
  35   friend class LevelDBTransaction;
  36
  37   explicit LevelDBSnapshot(LevelDBDatabase* db);
  38   ~LevelDBSnapshot();
  39
  40   leveldb::DB* db_;
  41   const leveldb::Snapshot* snapshot_;
  42 };
  43
  44 class CONTENT_EXPORT LevelDBLock {
  45 public:
  46   virtual ~LevelDBLock() {}
  47 };
  48
  49 class CONTENT_EXPORT LevelDBDatabase {
  50  public:
  51   class ComparatorAdapter : public leveldb::Comparator {
  52    public:
  53     explicit ComparatorAdapter(const LevelDBComparator* comparator);
  54
  55     virtual int Compare(const leveldb::Slice& a,
  56                         const leveldb::Slice& b) const OVERRIDE;
  57
  58     virtual const char* Name() const OVERRIDE;
  59
  60     virtual void FindShortestSeparator(std::string* start,
  61                                        const leveldb::Slice& limit) const
  62         OVERRIDE;
  63     virtual void FindShortSuccessor(std::string* key) const OVERRIDE;
  64
  65    private:
  66     const LevelDBComparator* comparator_;
  67   };
  68
  69   static leveldb::Status Open(const base::FilePath& file_name,
  70                               const LevelDBComparator* comparator,
  71                               scoped_ptr<LevelDBDatabase>* db,
  72                               bool* is_disk_full = 0);
  73   static scoped_ptr<LevelDBDatabase> OpenInMemory(
  74       const LevelDBComparator* comparator);
  75   static leveldb::Status Destroy(const base::FilePath& file_name);
  76   static scoped_ptr<LevelDBLock> LockForTesting(
  77       const base::FilePath& file_name);
  78   virtual ~LevelDBDatabase();
  79
  80   leveldb::Status Put(const base::StringPiece& key, std::string* value);
  81   leveldb::Status Remove(const base::StringPiece& key);
  82   virtual leveldb::Status Get(const base::StringPiece& key,
  83                               std::string* value,
  84                               bool* found,
  85                               const LevelDBSnapshot* = 0);
  86   leveldb::Status Write(const LevelDBWriteBatch& write_batch);
  87   scoped_ptr<LevelDBIterator> CreateIterator(const LevelDBSnapshot* = 0);
  88   const LevelDBComparator* Comparator() const;
  89   void Compact(const base::StringPiece& start, const base::StringPiece& stop);
  90   void CompactAll();
  91
  92  protected:
  93   LevelDBDatabase();
  94
  95  private:
  96   friend class LevelDBSnapshot;
  97
  98   scoped_ptr<leveldb::Env> env_;
  99   scoped_ptr<leveldb::Comparator> comparator_adapter_;
 100   scoped_ptr<leveldb::DB> db_;
 101   const LevelDBComparator* comparator_;
 102 };
 103
 104 }  // namespace content
 105
 106 #endif  // CONTENT_BROWSER_INDEXED_DB_LEVELDB_LEVELDB_DATABASE_H_