150 lines
4.7 KiB
C++
150 lines
4.7 KiB
C++
// Copyright (c) 2009-2010 Satoshi Nakamoto
|
|
// Copyright (c) 2009-2018 The Bitcoin Core developers
|
|
// Distributed under the MIT software license, see the accompanying
|
|
// file COPYING or http://www.opensource.org/licenses/mit-license.php.
|
|
|
|
#ifndef BITCOIN_TXDB_H
|
|
#define BITCOIN_TXDB_H
|
|
|
|
#include <coins.h>
|
|
#include <chain.h>
|
|
#include <primitives/block.h>
|
|
|
|
#include <map>
|
|
#include <memory>
|
|
#include <string>
|
|
#include <utility>
|
|
#include <vector>
|
|
|
|
#include <sqlite/sqlite3.h>
|
|
|
|
namespace sqlite {
|
|
inline int bind_col_in_db(sqlite3_stmt* stmt, int inx, const CScript& val) {
|
|
return sqlite3_bind_blob(stmt, inx, val.data(), int(val.size()), SQLITE_STATIC);
|
|
}
|
|
|
|
inline void store_result_in_db(sqlite3_context* db, const CScript& val) {
|
|
sqlite3_result_blob(db, val.data(), int(val.size()), SQLITE_TRANSIENT);
|
|
}
|
|
}
|
|
|
|
#include <sqlite.h>
|
|
|
|
namespace sqlite {
|
|
template<>
|
|
struct has_sqlite_type<CScript, SQLITE_BLOB, void> : std::true_type {};
|
|
|
|
inline CScript get_col_from_db(sqlite3_stmt* stmt, int inx, result_type<CScript>) {
|
|
auto ptr = (const unsigned char*)sqlite3_column_blob(stmt, inx);
|
|
if (!ptr) return {};
|
|
int bytes = sqlite3_column_bytes(stmt, inx);
|
|
assert(bytes >= 0);
|
|
return CScript(ptr, ptr + bytes);
|
|
}
|
|
}
|
|
|
|
class CBlockIndex;
|
|
class CCoinsViewDBCursor;
|
|
class uint256;
|
|
|
|
//! No need to periodic flush if at least this much space still available.
|
|
static constexpr int MAX_BLOCK_COINSDB_USAGE = 10;
|
|
//! -dbcache default (MiB)
|
|
static const int64_t nDefaultDbCache = 480;
|
|
//! -dbbatchsize default (bytes)
|
|
static const int64_t nDefaultDbBatchSize = 16 << 20;
|
|
//! max. -dbcache (MiB)
|
|
static const int64_t nMaxDbCache = sizeof(void*) > 4 ? 16384 : 1024;
|
|
//! min. -dbcache (MiB)
|
|
static const int64_t nMinDbCache = 4;
|
|
//! Max memory allocated to block tree DB specific cache
|
|
static const int64_t nMaxBlockDBCache = 260;
|
|
//! Max memory allocated to coin DB specific cache (MiB)
|
|
static const int64_t nMaxCoinsDBCache = 40;
|
|
|
|
|
|
struct CDiskTxPos : public CDiskBlockPos
|
|
{
|
|
unsigned int nTxOffset; // after header
|
|
|
|
ADD_SERIALIZE_METHODS;
|
|
|
|
template <typename Stream, typename Operation>
|
|
inline void SerializationOp(Stream& s, Operation ser_action) {
|
|
READWRITEAS(CDiskBlockPos, *this);
|
|
READWRITE(VARINT(nTxOffset));
|
|
}
|
|
|
|
CDiskTxPos(const CDiskBlockPos &blockIn, unsigned int nTxOffsetIn) : CDiskBlockPos(blockIn.nFile, blockIn.nPos), nTxOffset(nTxOffsetIn) {
|
|
}
|
|
|
|
CDiskTxPos() {
|
|
SetNull();
|
|
}
|
|
|
|
void SetNull() {
|
|
CDiskBlockPos::SetNull();
|
|
nTxOffset = 0;
|
|
}
|
|
};
|
|
|
|
/** CCoinsView backed by the coin database (chainstate/) */
|
|
class CCoinsViewDB final : public CCoinsView
|
|
{
|
|
friend CCoinsViewDBCursor;
|
|
mutable sqlite::database db;
|
|
|
|
public:
|
|
explicit CCoinsViewDB(size_t nCacheSize, bool fMemory = false, bool fWipe = false);
|
|
|
|
bool GetCoin(const COutPoint &outpoint, Coin &coin) const override;
|
|
bool HaveCoin(const COutPoint &outpoint) const override;
|
|
uint256 GetBestBlock() const override;
|
|
std::vector<uint256> GetHeadBlocks() const override;
|
|
bool BatchWrite(CCoinsMap &mapCoins, const uint256 &hashBlock, bool sync) override;
|
|
CCoinsViewCursor *Cursor() const override;
|
|
size_t EstimateSize() const override;
|
|
};
|
|
|
|
/** Specialization of CCoinsViewCursor to iterate over a CCoinsViewDB */
|
|
class CCoinsViewDBCursor: public CCoinsViewCursor
|
|
{
|
|
public:
|
|
~CCoinsViewDBCursor() noexcept override {}
|
|
|
|
bool GetKey(COutPoint &key) const override;
|
|
bool GetValue(Coin &coin) const override;
|
|
|
|
bool Valid() const override;
|
|
void Next() override;
|
|
|
|
private:
|
|
friend CCoinsViewDB;
|
|
explicit CCoinsViewDBCursor(const uint256 &hashBlockIn, const CCoinsViewDB* view);
|
|
const CCoinsViewDB* owner;
|
|
mutable sqlite::database_binder query;
|
|
mutable sqlite::row_iterator iter;
|
|
};
|
|
|
|
/** Access to the block database (blocks/index/) */
|
|
class CBlockTreeDB
|
|
{
|
|
sqlite::database db;
|
|
|
|
public:
|
|
explicit CBlockTreeDB(size_t nCacheSize, bool fMemory = false, bool fWipe = false);
|
|
|
|
bool BatchWrite(const std::vector<std::pair<int, const CBlockFileInfo*> >& fileInfo,
|
|
int nLastFile, const std::vector<const CBlockIndex*>& blockInfo, bool sync);
|
|
bool ReadBlockFileInfo(int nFile, CBlockFileInfo &info);
|
|
bool ReadLastBlockFile(int &nFile);
|
|
bool WriteReindexing(bool fReindexing);
|
|
void ReadReindexing(bool &fReindexing);
|
|
bool ReadTxIndex(const uint256 &txid, CDiskTxPos &pos);
|
|
bool WriteTxIndex(const std::vector<std::pair<uint256, CDiskTxPos>> &list);
|
|
bool WriteFlag(const std::string &name, bool fValue);
|
|
bool ReadFlag(const std::string &name, bool &fValue);
|
|
bool LoadBlockIndexGuts(const Consensus::Params& consensusParams, std::function<CBlockIndex*(const uint256&)> insertBlockIndex);
|
|
};
|
|
|
|
#endif // BITCOIN_TXDB_H
|