2013-11-05 17:58:43 -08:00
|
|
|
// Copyright (c) 2012-2013 The Bitcoin developers
|
2012-09-03 10:05:30 -07:00
|
|
|
// Distributed under the MIT/X11 software license, see the accompanying
|
|
|
|
// file COPYING or http://www.opensource.org/licenses/mit-license.php.
|
2013-11-05 17:58:43 -08:00
|
|
|
|
|
|
|
#ifndef BITCOIN_LEVELDBWRAPPER_H
|
|
|
|
#define BITCOIN_LEVELDBWRAPPER_H
|
2012-09-03 10:05:30 -07:00
|
|
|
|
|
|
|
#include "serialize.h"
|
2013-09-18 03:38:08 -07:00
|
|
|
#include "util.h"
|
2013-04-12 22:13:08 -07:00
|
|
|
#include "version.h"
|
2012-09-03 10:05:30 -07:00
|
|
|
|
2013-04-12 22:13:08 -07:00
|
|
|
#include <boost/filesystem/path.hpp>
|
2012-09-03 10:05:30 -07:00
|
|
|
#include <leveldb/db.h>
|
|
|
|
#include <leveldb/write_batch.h>
|
|
|
|
|
2013-01-28 16:44:19 -08:00
|
|
|
class leveldb_error : public std::runtime_error
|
|
|
|
{
|
|
|
|
public:
|
|
|
|
leveldb_error(const std::string &msg) : std::runtime_error(msg) {}
|
|
|
|
};
|
|
|
|
|
|
|
|
void HandleError(const leveldb::Status &status) throw(leveldb_error);
|
|
|
|
|
2013-11-05 17:58:43 -08:00
|
|
|
// Batch of changes queued to be written to a CLevelDBWrapper
|
2012-09-03 10:05:30 -07:00
|
|
|
class CLevelDBBatch
|
|
|
|
{
|
2013-11-05 17:58:43 -08:00
|
|
|
friend class CLevelDBWrapper;
|
2012-09-03 10:05:30 -07:00
|
|
|
|
|
|
|
private:
|
|
|
|
leveldb::WriteBatch batch;
|
|
|
|
|
|
|
|
public:
|
|
|
|
template<typename K, typename V> void Write(const K& key, const V& value) {
|
|
|
|
CDataStream ssKey(SER_DISK, CLIENT_VERSION);
|
|
|
|
ssKey.reserve(ssKey.GetSerializeSize(key));
|
|
|
|
ssKey << key;
|
|
|
|
leveldb::Slice slKey(&ssKey[0], ssKey.size());
|
|
|
|
|
|
|
|
CDataStream ssValue(SER_DISK, CLIENT_VERSION);
|
|
|
|
ssValue.reserve(ssValue.GetSerializeSize(value));
|
|
|
|
ssValue << value;
|
|
|
|
leveldb::Slice slValue(&ssValue[0], ssValue.size());
|
|
|
|
|
|
|
|
batch.Put(slKey, slValue);
|
|
|
|
}
|
|
|
|
|
|
|
|
template<typename K> void Erase(const K& key) {
|
|
|
|
CDataStream ssKey(SER_DISK, CLIENT_VERSION);
|
|
|
|
ssKey.reserve(ssKey.GetSerializeSize(key));
|
|
|
|
ssKey << key;
|
|
|
|
leveldb::Slice slKey(&ssKey[0], ssKey.size());
|
|
|
|
|
|
|
|
batch.Delete(slKey);
|
|
|
|
}
|
|
|
|
};
|
|
|
|
|
2013-11-05 17:58:43 -08:00
|
|
|
class CLevelDBWrapper
|
2012-09-03 10:05:30 -07:00
|
|
|
{
|
|
|
|
private:
|
|
|
|
// custom environment this database is using (may be NULL in case of default environment)
|
|
|
|
leveldb::Env *penv;
|
|
|
|
|
|
|
|
// database options used
|
|
|
|
leveldb::Options options;
|
|
|
|
|
|
|
|
// options used when reading from the database
|
|
|
|
leveldb::ReadOptions readoptions;
|
|
|
|
|
|
|
|
// options used when iterating over values of the database
|
|
|
|
leveldb::ReadOptions iteroptions;
|
|
|
|
|
|
|
|
// options used when writing to the database
|
|
|
|
leveldb::WriteOptions writeoptions;
|
|
|
|
|
|
|
|
// options used when sync writing to the database
|
|
|
|
leveldb::WriteOptions syncoptions;
|
|
|
|
|
|
|
|
// the database itself
|
|
|
|
leveldb::DB *pdb;
|
|
|
|
|
|
|
|
public:
|
2013-11-05 17:58:43 -08:00
|
|
|
CLevelDBWrapper(const boost::filesystem::path &path, size_t nCacheSize, bool fMemory = false, bool fWipe = false);
|
|
|
|
~CLevelDBWrapper();
|
2012-09-03 10:05:30 -07:00
|
|
|
|
2013-01-28 16:44:19 -08:00
|
|
|
template<typename K, typename V> bool Read(const K& key, V& value) throw(leveldb_error) {
|
2012-09-03 10:05:30 -07:00
|
|
|
CDataStream ssKey(SER_DISK, CLIENT_VERSION);
|
|
|
|
ssKey.reserve(ssKey.GetSerializeSize(key));
|
|
|
|
ssKey << key;
|
|
|
|
leveldb::Slice slKey(&ssKey[0], ssKey.size());
|
|
|
|
|
|
|
|
std::string strValue;
|
|
|
|
leveldb::Status status = pdb->Get(readoptions, slKey, &strValue);
|
|
|
|
if (!status.ok()) {
|
|
|
|
if (status.IsNotFound())
|
|
|
|
return false;
|
2013-09-18 03:38:08 -07:00
|
|
|
LogPrintf("LevelDB read failure: %s\n", status.ToString().c_str());
|
2013-01-28 16:44:19 -08:00
|
|
|
HandleError(status);
|
2012-09-03 10:05:30 -07:00
|
|
|
}
|
|
|
|
try {
|
|
|
|
CDataStream ssValue(strValue.data(), strValue.data() + strValue.size(), SER_DISK, CLIENT_VERSION);
|
|
|
|
ssValue >> value;
|
|
|
|
} catch(std::exception &e) {
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2013-01-28 16:44:19 -08:00
|
|
|
template<typename K, typename V> bool Write(const K& key, const V& value, bool fSync = false) throw(leveldb_error) {
|
2012-09-03 10:05:30 -07:00
|
|
|
CLevelDBBatch batch;
|
|
|
|
batch.Write(key, value);
|
|
|
|
return WriteBatch(batch, fSync);
|
|
|
|
}
|
|
|
|
|
2013-01-28 16:44:19 -08:00
|
|
|
template<typename K> bool Exists(const K& key) throw(leveldb_error) {
|
2012-09-03 10:05:30 -07:00
|
|
|
CDataStream ssKey(SER_DISK, CLIENT_VERSION);
|
|
|
|
ssKey.reserve(ssKey.GetSerializeSize(key));
|
|
|
|
ssKey << key;
|
|
|
|
leveldb::Slice slKey(&ssKey[0], ssKey.size());
|
|
|
|
|
|
|
|
std::string strValue;
|
|
|
|
leveldb::Status status = pdb->Get(readoptions, slKey, &strValue);
|
|
|
|
if (!status.ok()) {
|
|
|
|
if (status.IsNotFound())
|
|
|
|
return false;
|
2013-09-18 03:38:08 -07:00
|
|
|
LogPrintf("LevelDB read failure: %s\n", status.ToString().c_str());
|
2013-01-28 16:44:19 -08:00
|
|
|
HandleError(status);
|
2012-09-03 10:05:30 -07:00
|
|
|
}
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2013-01-28 16:44:19 -08:00
|
|
|
template<typename K> bool Erase(const K& key, bool fSync = false) throw(leveldb_error) {
|
2012-09-03 10:05:30 -07:00
|
|
|
CLevelDBBatch batch;
|
|
|
|
batch.Erase(key);
|
|
|
|
return WriteBatch(batch, fSync);
|
|
|
|
}
|
|
|
|
|
2013-01-28 16:44:19 -08:00
|
|
|
bool WriteBatch(CLevelDBBatch &batch, bool fSync = false) throw(leveldb_error);
|
2012-09-03 10:05:30 -07:00
|
|
|
|
|
|
|
// not available for LevelDB; provide for compatibility with BDB
|
|
|
|
bool Flush() {
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2013-01-28 16:44:19 -08:00
|
|
|
bool Sync() throw(leveldb_error) {
|
2012-09-03 10:05:30 -07:00
|
|
|
CLevelDBBatch batch;
|
|
|
|
return WriteBatch(batch, true);
|
|
|
|
}
|
|
|
|
|
|
|
|
// not exactly clean encapsulation, but it's easiest for now
|
|
|
|
leveldb::Iterator *NewIterator() {
|
|
|
|
return pdb->NewIterator(iteroptions);
|
|
|
|
}
|
|
|
|
};
|
|
|
|
|
2013-11-05 17:58:43 -08:00
|
|
|
#endif // BITCOIN_LEVELDBWRAPPER_H
|