leveldb_database.h revision f8ee788a64d60abd8f2d742a5fdedde054ecd910 
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
18namespace leveldb {
19class Comparator;
20class DB;
21class Env;
22class Snapshot;
23}
24
25namespace content {
26
27class LevelDBComparator;
28class LevelDBDatabase;
29class LevelDBIterator;
30class LevelDBWriteBatch;
31
32class 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  DISALLOW_COPY_AND_ASSIGN(LevelDBSnapshot);
44};
45
46class CONTENT_EXPORT LevelDBLock {
47 public:
48  LevelDBLock() {}
49  virtual ~LevelDBLock() {}
50
51 private:
52  DISALLOW_COPY_AND_ASSIGN(LevelDBLock);
53};
54
55class CONTENT_EXPORT LevelDBDatabase {
56 public:
57  class ComparatorAdapter : public leveldb::Comparator {
58   public:
59    explicit ComparatorAdapter(const LevelDBComparator* comparator);
60
61    virtual int Compare(const leveldb::Slice& a,
62                        const leveldb::Slice& b) const OVERRIDE;
63
64    virtual const char* Name() const OVERRIDE;
65
66    virtual void FindShortestSeparator(std::string* start,
67                                       const leveldb::Slice& limit) const
68        OVERRIDE;
69    virtual void FindShortSuccessor(std::string* key) const OVERRIDE;
70
71   private:
72    const LevelDBComparator* comparator_;
73  };
74
75  static leveldb::Status Open(const base::FilePath& file_name,
76                              const LevelDBComparator* comparator,
77                              scoped_ptr<LevelDBDatabase>* db,
78                              bool* is_disk_full = 0);
79  static scoped_ptr<LevelDBDatabase> OpenInMemory(
80      const LevelDBComparator* comparator);
81  static leveldb::Status Destroy(const base::FilePath& file_name);
82  static scoped_ptr<LevelDBLock> LockForTesting(
83      const base::FilePath& file_name);
84  virtual ~LevelDBDatabase();
85
86  leveldb::Status Put(const base::StringPiece& key, std::string* value);
87  leveldb::Status Remove(const base::StringPiece& key);
88  virtual leveldb::Status Get(const base::StringPiece& key,
89                              std::string* value,
90                              bool* found,
91                              const LevelDBSnapshot* = 0);
92  leveldb::Status Write(const LevelDBWriteBatch& write_batch);
93  scoped_ptr<LevelDBIterator> CreateIterator(const LevelDBSnapshot* = 0);
94  const LevelDBComparator* Comparator() const;
95  void Compact(const base::StringPiece& start, const base::StringPiece& stop);
96  void CompactAll();
97
98 protected:
99  LevelDBDatabase();
100
101 private:
102  friend class LevelDBSnapshot;
103
104  scoped_ptr<leveldb::Env> env_;
105  scoped_ptr<leveldb::Comparator> comparator_adapter_;
106  scoped_ptr<leveldb::DB> db_;
107  const LevelDBComparator* comparator_;
108};
109
110}  // namespace content
111
112#endif  // CONTENT_BROWSER_INDEXED_DB_LEVELDB_LEVELDB_DATABASE_H_
113