1/*
2 * Copyright (C) 2011 Google Inc. All rights reserved.
3 *
4 * Redistribution and use in source and binary forms, with or without
5 * modification, are permitted provided that the following conditions
6 * are met:
7 *
8 * 1.  Redistributions of source code must retain the above copyright
9 *     notice, this list of conditions and the following disclaimer.
10 * 2.  Redistributions in binary form must reproduce the above copyright
11 *     notice, this list of conditions and the following disclaimer in the
12 *     documentation and/or other materials provided with the distribution.
13 *
14 * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY
15 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
16 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
17 * DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY
18 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
19 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
20 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
21 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
22 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
23 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
24 */
25
26#ifndef LevelDBDatabase_h
27#define LevelDBDatabase_h
28
29#if USE(LEVELDB)
30
31#include <wtf/OwnPtr.h>
32#include <wtf/PassOwnPtr.h>
33#include <wtf/Vector.h>
34#include <wtf/text/WTFString.h>
35
36namespace leveldb {
37class Comparator;
38class DB;
39class Env;
40class Snapshot;
41}
42
43namespace WebCore {
44
45class LevelDBComparator;
46class LevelDBDatabase;
47class LevelDBIterator;
48class LevelDBSlice;
49class LevelDBWriteBatch;
50
51class LevelDBSnapshot {
52private:
53    friend class LevelDBDatabase;
54    friend class LevelDBTransaction;
55
56    explicit LevelDBSnapshot(LevelDBDatabase*);
57    ~LevelDBSnapshot();
58
59    leveldb::DB* m_db;
60    const leveldb::Snapshot* m_snapshot;
61};
62
63class LevelDBDatabase {
64public:
65    static PassOwnPtr<LevelDBDatabase> open(const String& fileName, const LevelDBComparator*);
66    static PassOwnPtr<LevelDBDatabase> openInMemory(const LevelDBComparator*);
67    static bool destroy(const String& fileName);
68    virtual ~LevelDBDatabase();
69
70    bool put(const LevelDBSlice& key, const Vector<char>& value);
71    bool remove(const LevelDBSlice& key);
72    virtual bool safeGet(const LevelDBSlice& key, Vector<char>& value, bool& found, const LevelDBSnapshot* = 0);
73    bool write(LevelDBWriteBatch&);
74    PassOwnPtr<LevelDBIterator> createIterator(const LevelDBSnapshot* = 0);
75    const LevelDBComparator* comparator() const;
76
77protected:
78    LevelDBDatabase();
79
80private:
81    friend class LevelDBSnapshot;
82
83    OwnPtr<leveldb::Env> m_env;
84    OwnPtr<leveldb::Comparator> m_comparatorAdapter;
85    OwnPtr<leveldb::DB> m_db;
86    const LevelDBComparator* m_comparator;
87};
88
89}
90
91#endif
92#endif
93