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 DISALLOW_COPY_AND_ASSIGN(LevelDBSnapshot); 44 }; 45 46 class CONTENT_EXPORT LevelDBLock { 47 public: LevelDBLock()48 LevelDBLock() {} ~LevelDBLock()49 virtual ~LevelDBLock() {} 50 51 private: 52 DISALLOW_COPY_AND_ASSIGN(LevelDBLock); 53 }; 54 55 class 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