Git-Mirrors/monero
1
0
Fork 0
mirror of https://github.com/monero-project/monero.git synced 2024-10-01 11:49:47 -04:00
Code Issues Packages Projects Releases Wiki Activity

Another take on migration

Browse Source 
Delete old indices and recreate them, rather than updating them
Maybe not quite as slow as before.
This commit is contained in:
Howard Chu 2016-04-10 17:25:13 +01:00
parent 66b1e13aa7
commit 2b0fa05f0d
2 changed files with 115 additions and 591 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
src/blockchain_db/blockchain_db.h
View File

@ -462,18 +462,6 @@ private:
*/ */
void pop_block(); void pop_block();
/**
* @brief helper function for add_transactions, to add each individual transaction
*
* This function is called by add_transactions() for each transaction to be
* added.
*
* @param blk_hash hash of the block which has the transaction
* @param tx the transaction to add
* @param tx_hash_ptr the hash of the transaction, if already calculated
*/
void add_transaction(const crypto::hash& blk_hash, const transaction& tx, const crypto::hash* tx_hash_ptr = NULL);
// helper function to remove transaction from blockchain // helper function to remove transaction from blockchain
/** /**
* @brief helper function to remove transaction from the blockchain * @brief helper function to remove transaction from the blockchain
@ -492,6 +480,18 @@ private:
protected: protected:
/**
* @brief helper function for add_transactions, to add each individual transaction
*
* This function is called by add_transactions() for each transaction to be
* added.
*
* @param blk_hash hash of the block which has the transaction
* @param tx the transaction to add
* @param tx_hash_ptr the hash of the transaction, if already calculated
*/
void add_transaction(const crypto::hash& blk_hash, const transaction& tx, const crypto::hash* tx_hash_ptr = NULL);
mutable uint64_t time_tx_exists = 0; //!< a performance metric mutable uint64_t time_tx_exists = 0; //!< a performance metric
uint64_t time_commit1 = 0; //!< a performance metric uint64_t time_commit1 = 0; //!< a performance metric
bool m_auto_remove_logs = true; //!< whether or not to automatically remove old logs bool m_auto_remove_logs = true; //!< whether or not to automatically remove old logs

682
src/blockchain_db/lmdb/db_lmdb.cpp
View File

@ -1142,8 +1142,8 @@ void BlockchainLMDB::open(const std::string& filename, const int mdb_flags)
// See commit e5d2680094ee15889934fe28901e4e133cda56f2 2015/07/10 // See commit e5d2680094ee15889934fe28901e4e133cda56f2 2015/07/10
// We don't handle the old format previous to that commit. // We don't handle the old format previous to that commit.
txn.commit(); txn.commit();
migrate(*(const uint32_t *)v.mv_data);
m_open = true; m_open = true;
migrate(*(const uint32_t *)v.mv_data);
return; return;
} }
#endif #endif
@ -2777,13 +2777,6 @@ void BlockchainLMDB::fixup()
BlockchainDB::fixup(); BlockchainDB::fixup();
} }
static int compare_amtidx(const MDB_val *a, const MDB_val *b)
{
const uint64_t *pa = (const uint64_t *)a->mv_data;
const uint64_t *pb = (const uint64_t *)b->mv_data;
return (pa[0] < pb[1]) ? -1 : pa[0] > pb[1];
}
#define RENAME_DB(name) \ #define RENAME_DB(name) \
k.mv_data = (void *)name; \ k.mv_data = (void *)name; \
k.mv_size = sizeof(name)-1; \ k.mv_size = sizeof(name)-1; \
@ -3125,395 +3118,63 @@ void BlockchainLMDB::migrate_0_1()
txn.commit(); txn.commit();
} while(0); } while(0);
LOG_PRINT_L0("Total number of outputs: " << m_num_outputs);
LOG_PRINT_L1("outputs migration will update output_amounts and output_txs...");
do { do {
LOG_PRINT_L1("migrating output_amounts:"); LOG_PRINT_L1("deleting old indices:");
MDB_dbi keys; /* Delete all other tables, we're just going to recreate them */
MDB_dbi dbi;
result = mdb_txn_begin(m_env, NULL, 0, txn); result = mdb_txn_begin(m_env, NULL, 0, txn);
if (result) if (result)
throw0(DB_ERROR(lmdb_error("Failed to create a transaction for the db: ", result).c_str())); throw0(DB_ERROR(lmdb_error("Failed to create a transaction for the db: ", result).c_str()));
result = mdb_dbi_open(txn, "output_keys", 0, &keys);
result = mdb_dbi_open(txn, "tx_unlocks", 0, &dbi);
if (result == MDB_NOTFOUND) { if (result == MDB_NOTFOUND) {
txn.abort(); txn.abort();
LOG_PRINT_L1(" output_amounts already migrated"); LOG_PRINT_L1(" old indices already deleted");
break;
}
MDB_dbi o_amts;
outkey ok;
MDB_val_set(nv, ok);
/* the output_amounts table name is the same but the old version and new version
* have incompatible DB flags. Create a new table with the right flags.
*/
o_amts = m_output_amounts;
lmdb_db_open(txn, "output_amountr", MDB_INTEGERKEY | MDB_CREATE | MDB_DUPSORT | MDB_DUPFIXED, m_output_amounts, "Failed to open db handle for output_amountr");
mdb_set_dupsort(txn, m_output_amounts, compare_uint64);
MDB_cursor *c_oamts, *c_keys, *c_cur;
i = 0;
z = m_num_outputs;
uint64_t j;
mdb_size_t num_elems;
MDB_val v2;
while(1) {
if (!(i % 1000)) {
if (i) {
LOGIF(1) {
std::cout << i << " / " << z << " \r" << std::flush;
}
txn.commit();
result = mdb_txn_begin(m_env, NULL, 0, txn);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to create a transaction for the db: ", result).c_str()));
}
result = mdb_cursor_open(txn, m_output_amounts, &c_cur);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to open a cursor for output_amountr: ", result).c_str()));
result = mdb_cursor_open(txn, o_amts, &c_oamts);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to open a cursor for output_amounts: ", result).c_str()));
result = mdb_cursor_open(txn, keys, &c_keys);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to open a cursor for output_keys: ", result).c_str()));
if (!i) {
MDB_stat ms;
mdb_stat(txn, m_output_amounts, &ms);
i = ms.ms_entries;
}
}
result = mdb_cursor_get(c_oamts, &k, &v, MDB_FIRST);
if (result == MDB_NOTFOUND) {
txn.commit();
break; break;
} else if (result)
throw0(DB_ERROR(lmdb_error("Failed to get a record from output_amounts: ", result).c_str()));
result = mdb_cursor_count(c_oamts, &num_elems);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to get number of outputs for amount: ", result).c_str()));
for (j=0; j<num_elems; j++,i++) {
if (!(i % 1000)) {
uint64_t amt = *(uint64_t *)k.mv_data;
uint64_t oid = *(uint64_t *)v.mv_data;
LOGIF(1) {
std::cout << i << " / " << z << " \r" << std::flush;
}
txn.commit();
result = mdb_txn_begin(m_env, NULL, 0, txn);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to create a transaction for the db: ", result).c_str()));
result = mdb_cursor_open(txn, m_output_amounts, &c_cur);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to open a cursor for output_amountr: ", result).c_str()));
result = mdb_cursor_open(txn, o_amts, &c_oamts);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to open a cursor for output_amounts: ", result).c_str()));
result = mdb_cursor_open(txn, keys, &c_keys);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to open a cursor for output_keys: ", result).c_str()));
k.mv_data = (void *)&amt;
v.mv_data = (void *)&oid;
result = mdb_cursor_get(c_oamts, &k, &v, MDB_GET_BOTH);
/* should never fail since we already had this record before the commit */
if (result)
throw0(DB_ERROR(lmdb_error("Failed to get a record from output_amounts: ", result).c_str()));
}
result = mdb_cursor_get(c_keys, &v, &v2, MDB_SET);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to get output_key for amount: ", result).c_str()));
ok.amount_index = j;
ok.output_id = *(uint64_t *)v.mv_data;
ok.data = *(output_data_t *)v2.mv_data;
result = mdb_cursor_put(c_cur, &k, &nv, MDB_APPENDDUP);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to put a record into output_amountr: ", result).c_str()));
result = mdb_cursor_del(c_keys, 0);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to delete a record from output_keys: ", result).c_str()));
result = mdb_cursor_get(c_oamts, &k, &v, MDB_NEXT_DUP);
if (result == MDB_NOTFOUND)
break;
else if (result)
throw0(DB_ERROR(lmdb_error("Failed to get a record from output_amounts: ", result).c_str()));
}
result = mdb_cursor_del(c_oamts, MDB_NODUPDATA);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to delete a record from output_amounts: ", result).c_str()));
} }
txn.abort();
#define DELETE_DB(x) do { \
LOG_PRINT_L1(" " x ":"); \
result = mdb_txn_begin(m_env, NULL, 0, txn); \
if (result) \
throw0(DB_ERROR(lmdb_error("Failed to create a transaction for the db: ", result).c_str())); \
result = mdb_dbi_open(txn, x, 0, &dbi); \
if (!result) { \
result = mdb_drop(txn, dbi, 1); \
if (result) \
throw0(DB_ERROR(lmdb_error("Failed to delete " x ": ", result).c_str())); \
txn.commit(); \
} } while(0)
DELETE_DB("tx_heights");
DELETE_DB("output_txs");
DELETE_DB("output_indices");
DELETE_DB("output_keys");
DELETE_DB("spent_keys");
DELETE_DB("output_amounts");
DELETE_DB("tx_outputs");
DELETE_DB("tx_unlocks");
/* reopen new DBs with correct flags */
result = mdb_txn_begin(m_env, NULL, 0, txn); result = mdb_txn_begin(m_env, NULL, 0, txn);
if (result) if (result)
throw0(DB_ERROR(lmdb_error("Failed to create a transaction for the db: ", result).c_str())); throw0(DB_ERROR(lmdb_error("Failed to create a transaction for the db: ", result).c_str()));
result = mdb_drop(txn, keys, 1); lmdb_db_open(txn, LMDB_OUTPUT_TXS, MDB_INTEGERKEY | MDB_CREATE | MDB_DUPSORT | MDB_DUPFIXED, m_output_txs, "Failed to open db handle for m_output_txs");
if (result) mdb_set_dupsort(txn, m_output_txs, compare_uint64);
throw0(DB_ERROR(lmdb_error("Failed to delete output_keys from the db: ", result).c_str())); lmdb_db_open(txn, LMDB_TX_OUTPUTS, MDB_INTEGERKEY | MDB_CREATE, m_tx_outputs, "Failed to open db handle for m_tx_outputs");
result = mdb_drop(txn, o_amts, 1); lmdb_db_open(txn, LMDB_SPENT_KEYS, MDB_INTEGERKEY | MDB_CREATE | MDB_DUPSORT | MDB_DUPFIXED, m_spent_keys, "Failed to open db handle for m_spent_keys");
if (result) mdb_set_dupsort(txn, m_spent_keys, compare_hash32);
throw0(DB_ERROR(lmdb_error("Failed to delete output_amounts from the db: ", result).c_str())); lmdb_db_open(txn, LMDB_OUTPUT_AMOUNTS, MDB_INTEGERKEY | MDB_DUPSORT | MDB_DUPFIXED | MDB_CREATE, m_output_amounts, "Failed to open db handle for m_output_amounts");
RENAME_DB("output_amountr");
/* close and reopen to get old dbi slot back */
mdb_dbi_close(m_env, m_output_amounts);
lmdb_db_open(txn, "output_amounts", MDB_INTEGERKEY | MDB_DUPSORT | MDB_DUPFIXED, m_output_amounts, "Failed to open db handle for output_amounts");
mdb_set_dupsort(txn, m_output_amounts, compare_uint64); mdb_set_dupsort(txn, m_output_amounts, compare_uint64);
txn.commit(); txn.commit();
m_num_txs = 0;
m_num_outputs = 0;
} while(0); } while(0);
do { do {
LOG_PRINT_L1("migrating output_txs:"); LOG_PRINT_L1("migrating txs and outputs:");
MDB_dbi indices;
result = mdb_txn_begin(m_env, NULL, 0, txn);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to create a transaction for the db: ", result).c_str()));
result = mdb_dbi_open(txn, "output_indices", 0, &indices);
if (result == MDB_NOTFOUND) {
txn.abort();
LOG_PRINT_L1(" output_txs already migrated");
break;
}
MDB_dbi o_otxs;
outtx ot;
MDB_val_set(nv, ot);
/* the output_txs table name is the same but the old version and new version
* have incompatible DB flags. Create a new table with the right flags.
*/
o_otxs = m_output_txs;
lmdb_db_open(txn, "output_txr", MDB_INTEGERKEY | MDB_CREATE | MDB_DUPSORT | MDB_DUPFIXED, m_output_txs, "Failed to open db handle for output_txr");
mdb_set_dupsort(txn, m_output_txs, compare_uint64);
MDB_cursor *c_otxs, *c_oixs, *c_cur;
i = 0;
z = m_num_outputs;
while(1) {
if (!(i % 2000)) {
if (i) {
LOGIF(1) {
std::cout << i << " / " << z << " \r" << std::flush;
}
txn.commit();
result = mdb_txn_begin(m_env, NULL, 0, txn);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to create a transaction for the db: ", result).c_str()));
}
result = mdb_cursor_open(txn, m_output_txs, &c_cur);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to open a cursor for output_txr: ", result).c_str()));
result = mdb_cursor_open(txn, o_otxs, &c_otxs);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to open a cursor for output_txs: ", result).c_str()));
result = mdb_cursor_open(txn, indices, &c_oixs);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to open a cursor for output_indices: ", result).c_str()));
if (!i) {
MDB_stat ms;
mdb_stat(txn, m_output_txs, &ms);
i = ms.ms_entries;
}
}
result = mdb_cursor_get(c_otxs, &k, &v, MDB_NEXT);
if (result == MDB_NOTFOUND) {
txn.commit();
break;
} else if (result)
throw0(DB_ERROR(lmdb_error("Failed to get a record from output_txs: ", result).c_str()));
ot.output_id = *(uint64_t *)k.mv_data;
ot.tx_hash = *(crypto::hash *)v.mv_data;
result = mdb_cursor_get(c_oixs, &k, &v, MDB_NEXT);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to get a record from output_indices: ", result).c_str()));
ot.local_index = *(uint64_t *)v.mv_data;
result = mdb_cursor_put(c_cur, (MDB_val *)&zerokval, &nv, MDB_APPENDDUP);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to put a record into output_txr: ", result).c_str()));
result = mdb_cursor_del(c_otxs, 0);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to delete a record from output_txs: ", result).c_str()));
result = mdb_cursor_del(c_oixs, 0);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to delete a record from output_indices: ", result).c_str()));
i++;
}
result = mdb_txn_begin(m_env, NULL, 0, txn);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to create a transaction for the db: ", result).c_str()));
result = mdb_drop(txn, indices, 1);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to delete output_indices from the db: ", result).c_str()));
result = mdb_drop(txn, o_otxs, 1);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to delete output_txs from the db: ", result).c_str()));
RENAME_DB("output_txr");
mdb_dbi_close(m_env, m_output_txs);
lmdb_db_open(txn, "output_txs", MDB_INTEGERKEY | MDB_DUPSORT | MDB_DUPFIXED, m_output_txs, "Failed to open db handle for output_txs");
mdb_set_dupsort(txn, m_output_txs, compare_uint64);
txn.commit();
} while(0);
LOG_PRINT_L0("Total number of txs: " << m_num_txs);
LOG_PRINT_L1("txs migration will update tx_indices, tx_outputs, and txs...");
do {
LOG_PRINT_L1("migrating tx_indices:");
/* The existing indices lack information that version 1 needs, so we just regenerate it
* all from the original Blocks.
*/
MDB_dbi o_tx_unlocks;
result = mdb_txn_begin(m_env, NULL, 0, txn);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to create a transaction for the db: ", result).c_str()));
result = mdb_dbi_open(txn, "tx_unlocks", 0, &o_tx_unlocks);
if (result == MDB_NOTFOUND) {
txn.abort();
LOG_PRINT_L1(" tx_indices already migrated");
break;
}
MDB_dbi o_tx_heights;
MDB_cursor *c_cur, *c_blocks, *c_props;
MDB_cursor *c_tx_unlocks, *c_tx_heights;
lmdb_db_open(txn, "tx_heights", 0, o_tx_heights, "Failed to open db handle for tx_heights");
txindex ti;
MDB_val_set(iv, ti);
unsigned int j;
blobdata bd;
block b;
uint64_t num_txs = m_num_txs;
i = 0;
ti.data.block_id = 0;
z = m_num_txs;
while(1) {
if (!(ti.data.block_id % 1000)) {
if (i) {
LOGIF(1) {
std::cout << i << " / " << z << " \r" << std::flush;
}
MDB_val_set(pk, "txblk");
MDB_val_set(pv, ti.data.block_id);
result = mdb_cursor_put(c_props, &pk, &pv, 0);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to update txblk property: ", result).c_str()));
txn.commit();
result = mdb_txn_begin(m_env, NULL, 0, txn);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to create a transaction for the db: ", result).c_str()));
}
result = mdb_cursor_open(txn, m_tx_indices, &c_cur);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to open a cursor for tx_indices: ", result).c_str()));
result = mdb_cursor_open(txn, m_properties, &c_props);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to open a cursor for properties: ", result).c_str()));
result = mdb_cursor_open(txn, o_tx_unlocks, &c_tx_unlocks);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to open a cursor for tx_unlocks: ", result).c_str()));
result = mdb_cursor_open(txn, o_tx_heights, &c_tx_heights);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to open a cursor for tx_heights: ", result).c_str()));
result = mdb_cursor_open(txn, m_blocks, &c_blocks);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to open a cursor for blocks: ", result).c_str()));
/* we don't care about the unlocks/heights content or cursor position.
* we just delete 1 record from each whenever we add a record.
*/
mdb_cursor_get(c_tx_unlocks, &k, &v, MDB_FIRST);
mdb_cursor_get(c_tx_heights, &k, &v, MDB_FIRST);
if (!i) {
MDB_stat ms;
mdb_stat(txn, m_tx_indices, &ms);
i = ms.ms_entries;
if (i) {
/* remember the ID of the last block we migrated */
MDB_val_set(pk, "txblk");
result = mdb_cursor_get(c_props, &pk, &v, MDB_SET);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to get a record from tx_indices: ", result).c_str()));
ti.data.block_id = *(uint64_t *)v.mv_data;
}
num_txs = i;
}
if (i) {
k.mv_data = (void *)&ti.data.block_id;
k.mv_size = sizeof(ti.data.block_id);
result = mdb_cursor_get(c_blocks, &k, &v, MDB_SET);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to get a record from blocks: ", result).c_str()));
}
}
result = mdb_cursor_get(c_blocks, &k, &v, MDB_NEXT);
if (result == MDB_NOTFOUND) {
MDB_val_set(pk, "txblk");
mdb_cursor_get(c_props, &pk, &v, MDB_SET);
mdb_cursor_del(c_props, 0);
txn.commit();
break;
} else if (result)
throw0(DB_ERROR(lmdb_error("Failed to get a record from blocks: ", result).c_str()));
bd.assign(reinterpret_cast<char*>(v.mv_data), v.mv_size);
if (!parse_and_validate_block_from_blob(bd, b))
throw0(DB_ERROR("Failed to parse block from blob retrieved from the db"));
ti.key = get_transaction_hash(b.miner_tx);
ti.data.tx_id = num_txs++;
ti.data.block_id = *(uint64_t *)k.mv_data;
ti.data.unlock_time = b.miner_tx.unlock_time;
i++;
result = mdb_cursor_put(c_cur, (MDB_val *)&zerokval, &iv, 0);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to put a record into tx_indices: ", result).c_str()));
result = mdb_cursor_del(c_tx_unlocks, 0);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to delete a record from tx_unlocks: ", result).c_str()));
result = mdb_cursor_del(c_tx_heights, 0);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to delete a record from tx_heights: ", result).c_str()));
for (j=0; j<b.tx_hashes.size(); j++) {
ti.key = b.tx_hashes[j];
ti.data.tx_id = num_txs++;
result = mdb_cursor_put(c_cur, (MDB_val *)&zerokval, &iv, 0);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to put a record into tx_indices: ", result).c_str()));
result = mdb_cursor_del(c_tx_unlocks, 0);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to delete a record from tx_unlocks: ", result).c_str()));
result = mdb_cursor_del(c_tx_heights, 0);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to delete a record from tx_heights: ", result).c_str()));
i++;
}
}
result = mdb_txn_begin(m_env, NULL, 0, txn);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to create a transaction for the db: ", result).c_str()));
result = mdb_drop(txn, o_tx_unlocks, 1);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to delete tx_unlocks from the db: ", result).c_str()));
result = mdb_drop(txn, o_tx_heights, 1);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to delete tx_heights from the db: ", result).c_str()));
txn.commit();
m_num_txs = num_txs;
} while(0);
do {
LOG_PRINT_L1("migrating txs and tx_outputs:");
unsigned int flags; unsigned int flags;
result = mdb_txn_begin(m_env, NULL, 0, txn); result = mdb_txn_begin(m_env, NULL, 0, txn);
@ -3529,30 +3190,26 @@ void BlockchainLMDB::migrate_0_1()
break; break;
} }
MDB_dbi o_txs, o_tx_outputs; MDB_dbi o_txs;
txindex *txp; blobdata bd;
MDB_val ik, iv; block b;
MDB_val hk;
o_txs = m_txs; o_txs = m_txs;
mdb_set_compare(txn, o_txs, compare_hash32);
lmdb_db_open(txn, "txr", MDB_INTEGERKEY | MDB_CREATE, m_txs, "Failed to open db handle for txr"); lmdb_db_open(txn, "txr", MDB_INTEGERKEY | MDB_CREATE, m_txs, "Failed to open db handle for txr");
o_tx_outputs = m_tx_outputs;
lmdb_db_open(txn, "tx_outputr", MDB_INTEGERKEY | MDB_CREATE, m_tx_outputs, "Failed to open db handle for tx_outputr");
mdb_set_dupsort(txn, o_tx_outputs, compare_uint64);
/* a hack to let us find amount index from global index */ txn.commit();
mdb_set_dupsort(txn, m_output_amounts, compare_amtidx);
{ MDB_cursor *c_blocks, *c_txs, *c_props, *c_cur;
MDB_stat ms;
mdb_stat(txn, m_tx_indices, &ms);
m_num_txs = ms.ms_entries;
}
MDB_cursor *c_cur, *c_curtxo, *c_txi;
MDB_cursor *c_txs, *c_tx_outputs;
MDB_cursor *c_oamts;
i = 0; i = 0;
z = m_num_txs; z = m_height;
hk.mv_size = sizeof(crypto::hash);
set_batch_transactions(true);
batch_start(1000);
txn.m_txn = m_write_txn->m_txn;
m_height = 0;
while(1) { while(1) {
if (!(i % 1000)) { if (!(i % 1000)) {
@ -3560,122 +3217,80 @@ void BlockchainLMDB::migrate_0_1()
LOGIF(1) { LOGIF(1) {
std::cout << i << " / " << z << " \r" << std::flush; std::cout << i << " / " << z << " \r" << std::flush;
} }
MDB_val_set(pk, "txblk");
MDB_val_set(pv, m_height);
result = mdb_cursor_put(c_props, &pk, &pv, 0);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to update txblk property: ", result).c_str()));
txn.commit(); txn.commit();
result = mdb_txn_begin(m_env, NULL, 0, txn); result = mdb_txn_begin(m_env, NULL, 0, txn);
if (result) if (result)
throw0(DB_ERROR(lmdb_error("Failed to create a transaction for the db: ", result).c_str())); throw0(DB_ERROR(lmdb_error("Failed to create a transaction for the db: ", result).c_str()));
m_write_txn->m_txn = txn.m_txn;
m_write_batch_txn->m_txn = txn.m_txn;
memset(&m_wcursors, 0, sizeof(m_wcursors));
} }
result = mdb_cursor_open(txn, m_txs, &c_cur); result = mdb_cursor_open(txn, m_blocks, &c_blocks);
if (result) if (result)
throw0(DB_ERROR(lmdb_error("Failed to open a cursor for txr: ", result).c_str())); throw0(DB_ERROR(lmdb_error("Failed to open a cursor for blocks: ", result).c_str()));
result = mdb_cursor_open(txn, m_properties, &c_props);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to open a cursor for properties: ", result).c_str()));
result = mdb_cursor_open(txn, o_txs, &c_txs); result = mdb_cursor_open(txn, o_txs, &c_txs);
if (result) if (result)
throw0(DB_ERROR(lmdb_error("Failed to open a cursor for txs: ", result).c_str())); throw0(DB_ERROR(lmdb_error("Failed to open a cursor for txs: ", result).c_str()));
result = mdb_cursor_open(txn, o_tx_outputs, &c_tx_outputs);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to open a cursor for tx_outputs: ", result).c_str()));
result = mdb_cursor_open(txn, m_tx_outputs, &c_curtxo);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to open a cursor for tx_outputr: ", result).c_str()));
result = mdb_cursor_open(txn, m_tx_indices, &c_txi);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to open a cursor for tx_indices: ", result).c_str()));
result = mdb_cursor_open(txn, m_output_amounts, &c_oamts);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to open a cursor for output_amounts: ", result).c_str()));
if (!i) { if (!i) {
MDB_stat ms; MDB_stat ms;
mdb_stat(txn, m_output_txs, &ms);
m_num_outputs = ms.ms_entries;
mdb_stat(txn, m_txs, &ms); mdb_stat(txn, m_txs, &ms);
i = ms.ms_entries; m_num_txs = i = ms.ms_entries;
if (i) {
m_num_txs = i;
MDB_val_set(pk, "txblk");
result = mdb_cursor_get(c_props, &pk, &k, MDB_SET);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to get a record from properties: ", result).c_str()));
m_height = *(uint64_t *)k.mv_data;
}
} }
if (i) { if (i) {
result = mdb_cursor_get(c_txs, &k, &v, MDB_FIRST); result = mdb_cursor_get(c_blocks, &k, &v, MDB_SET);
if (result) if (result)
throw0(DB_ERROR(lmdb_error("Failed to get a record from txs: ", result).c_str())); throw0(DB_ERROR(lmdb_error("Failed to get a record from blocks: ", result).c_str()));
result = mdb_cursor_get(c_txi, (MDB_val *)&zerokval, &k, MDB_GET_BOTH);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to get a record from tx_indices: ", result).c_str()));
result = mdb_cursor_get(c_txi, &k, &v, MDB_PREV);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to get a record from tx_indices: ", result).c_str()));
} }
} }
/* The new tx_indices and the old txs tables are both sorted using result = mdb_cursor_get(c_blocks, &k, &v, MDB_NEXT);
* compare_hash32 so we can walk thru them sequentially and they will
* stay in lock step with each other. Unfortunately, even though the
* old tx_outputs was keyed with the tx hash, it wasn't set to use
* the compare_hash32 function, so its records are in some other random
* hash order. This costs us the majority of CPU time when walking
* thru that table. Iterating with MDB_NEXT is cheap/free, but searching
* through a multi-million record table with 32byte keys is quite costly.
*/
result = mdb_cursor_get(c_txi, &k, &v, MDB_NEXT);
if (result == MDB_NOTFOUND) { if (result == MDB_NOTFOUND) {
txn.commit(); MDB_val_set(pk, "txblk");
mdb_cursor_get(c_props, &pk, &v, MDB_SET);
mdb_cursor_del(c_props, 0);
batch_stop();
break; break;
} else if (result) } else if (result)
throw0(DB_ERROR(lmdb_error("Failed to get a record from tx_indices: ", result).c_str())); throw0(DB_ERROR(lmdb_error("Failed to get a record from blocks: ", result).c_str()));
txp = (txindex *)v.mv_data;
k.mv_data = (void *)&txp->key;
k.mv_size = sizeof(txp->key);
ik.mv_data = (void *)&txp->data.tx_id;
ik.mv_size = sizeof(txp->data.tx_id);
result = mdb_cursor_get(c_txs, &k, &v, MDB_FIRST); bd.assign(reinterpret_cast<char*>(v.mv_data), v.mv_size);
if (result) if (!parse_and_validate_block_from_blob(bd, b))
throw0(DB_ERROR(lmdb_error("Failed to get a record from txs: ", result).c_str())); throw0(DB_ERROR("Failed to parse block from blob retrieved from the db"));
result = mdb_cursor_put(c_cur, &ik, &v, 0);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to put a record into txr: ", result).c_str()));
result = mdb_cursor_get(c_tx_outputs, &k, &iv, MDB_SET);
if (!result) {
blobdata bd;
bd.assign(reinterpret_cast<char*>(v.mv_data), v.mv_size);
add_transaction(null_hash, b.miner_tx);
for (unsigned int j = 0; j<b.tx_hashes.size(); j++) {
transaction tx; transaction tx;
hk.mv_data = &b.tx_hashes[j];
result = mdb_cursor_get(c_txs, &hk, &v, MDB_SET);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to get record from txs: ", result).c_str()));
bd.assign(reinterpret_cast<char*>(v.mv_data), v.mv_size);
if (!parse_and_validate_tx_from_blob(bd, tx)) if (!parse_and_validate_tx_from_blob(bd, tx))
throw0(DB_ERROR("Failed to parse tx from blob retrieved from the db")); throw0(DB_ERROR("Failed to parse tx from blob retrieved from the db"));
add_transaction(null_hash, tx, &b.tx_hashes[j]);
/* turn global output indices into amount output indices */ result = mdb_cursor_del(c_txs, 0);
std::vector<uint64_t> indices;
int j;
for (j=0;;j++) {
MDB_val_set(vk, tx.vout[j].amount);
result = mdb_cursor_get(c_oamts, &vk, &iv, MDB_GET_BOTH);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to get a record from output_amounts: ", result).c_str()));
outkey *ok = (outkey *)v.mv_data;
indices.push_back(ok->amount_index);
result = mdb_cursor_get(c_tx_outputs, &k, &iv, MDB_NEXT_DUP);
if (result == MDB_NOTFOUND)
break;
else if (result)
throw0(DB_ERROR(lmdb_error("Failed to get a record from tx_outputs: ", result).c_str()));
}
v.mv_data = (void *)indices.data();
v.mv_size = sizeof(uint64_t) * indices.size();
result = mdb_cursor_put(c_curtxo, &ik, &v, 0);
if (result) if (result)
throw0(DB_ERROR(lmdb_error("Failed to put a record into tx_outputr: ", result).c_str())); throw0(DB_ERROR(lmdb_error("Failed to get record from txs: ", result).c_str()));
result = mdb_cursor_del(c_tx_outputs, MDB_NODUPDATA); }
if (result)
throw0(DB_ERROR(lmdb_error("Failed to delete a record from tx_outputs: ", result).c_str()));
} else if (result == MDB_NOTFOUND) {
/* get_tx_amount_output_indices expects a record here, even if it's empty */
v.mv_size = 0;
v.mv_data = NULL;
result = mdb_cursor_put(c_curtxo, &ik, &v, 0);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to put a record into tx_outputr: ", result).c_str()));
} else
throw0(DB_ERROR(lmdb_error("Failed to get a record from tx_outputs: ", result).c_str()));
result = mdb_cursor_del(c_txs, 0);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to delete a record from txs: ", result).c_str()));
i++; i++;
m_height = i;
} }
result = mdb_txn_begin(m_env, NULL, 0, txn); result = mdb_txn_begin(m_env, NULL, 0, txn);
if (result) if (result)
@ -3683,103 +3298,12 @@ void BlockchainLMDB::migrate_0_1()
result = mdb_drop(txn, o_txs, 1); result = mdb_drop(txn, o_txs, 1);
if (result) if (result)
throw0(DB_ERROR(lmdb_error("Failed to delete txs from the db: ", result).c_str())); throw0(DB_ERROR(lmdb_error("Failed to delete txs from the db: ", result).c_str()));
result = mdb_drop(txn, o_tx_outputs, 1);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to delete tx_outputs from the db: ", result).c_str()));
RENAME_DB("txr"); RENAME_DB("txr");
RENAME_DB("tx_outputr");
mdb_dbi_close(m_env, m_txs); mdb_dbi_close(m_env, m_txs);
mdb_dbi_close(m_env, m_tx_outputs);
lmdb_db_open(txn, "txs", MDB_INTEGERKEY, m_txs, "Failed to open db handle for txs"); lmdb_db_open(txn, "txs", MDB_INTEGERKEY, m_txs, "Failed to open db handle for txs");
lmdb_db_open(txn, "tx_outputs", MDB_INTEGERKEY, m_tx_outputs, "Failed to open db handle for tx_outputs");
mdb_set_dupsort(txn, m_output_amounts, compare_uint64);
txn.commit();
} while(0);
do {
LOG_PRINT_L1("migrating spent_keys:");
MDB_dbi o_keys;
unsigned int flags;
result = mdb_txn_begin(m_env, NULL, 0, txn);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to create a transaction for the db: ", result).c_str()));
result = mdb_dbi_flags(txn, m_spent_keys, &flags);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to retrieve spent_keys flags: ", result).c_str()));
/* if the flags are what we expect, this table has already been migrated */
if ((flags & (MDB_INTEGERKEY|MDB_DUPSORT|MDB_DUPFIXED)) == (MDB_INTEGERKEY|MDB_DUPSORT|MDB_DUPFIXED)) {
txn.abort();
LOG_PRINT_L1(" spent_keys already migrated");
break;
}
/* the spent_keys table name is the same but the old version and new version
* have incompatible DB flags. Create a new table with the right flags.
*/
o_keys = m_spent_keys;
lmdb_db_open(txn, "spent_keyr", MDB_INTEGERKEY | MDB_CREATE | MDB_DUPSORT | MDB_DUPFIXED, m_spent_keys, "Failed to open db handle for spent_keyr");
mdb_set_dupsort(txn, m_spent_keys, compare_hash32);
MDB_cursor *c_old, *c_cur;
i = 0;
MDB_stat ms;
mdb_stat(txn, o_keys, &ms);
z = ms.ms_entries;
mdb_stat(txn, m_spent_keys, &ms);
z += ms.ms_entries;
while(1) {
if (!(i % 2000)) {
if (i) {
LOGIF(1) {
std::cout << i << " / " << z << " \r" << std::flush;
}
txn.commit();
result = mdb_txn_begin(m_env, NULL, 0, txn);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to create a transaction for the db: ", result).c_str()));
}
result = mdb_cursor_open(txn, m_spent_keys, &c_cur);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to open a cursor for spent_keyr: ", result).c_str()));
result = mdb_cursor_open(txn, o_keys, &c_old);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to open a cursor for spent_keys: ", result).c_str()));
if (!i)
i = ms.ms_entries;
}
result = mdb_cursor_get(c_old, &k, NULL, MDB_NEXT);
if (result == MDB_NOTFOUND) {
txn.commit();
break;
}
else if (result)
throw0(DB_ERROR(lmdb_error("Failed to get a record from spent_keys: ", result).c_str()));
result = mdb_cursor_put(c_cur, (MDB_val *)&zerokval, &k, MDB_APPENDDUP);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to put a record into spent_keyr: ", result).c_str()));
result = mdb_cursor_del(c_old, 0);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to delete a record from spent_keys: ", result).c_str()));
i++;
}
result = mdb_txn_begin(m_env, NULL, 0, txn);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to create a transaction for the db: ", result).c_str()));
/* Delete the old table */
result = mdb_drop(txn, o_keys, 1);
if (result)
throw0(DB_ERROR(lmdb_error("Failed to delete old spent_keys table: ", result).c_str()));
RENAME_DB("spent_keyr");
mdb_dbi_close(m_env, m_spent_keys);
lmdb_db_open(txn, "spent_keys", MDB_INTEGERKEY | MDB_DUPSORT | MDB_DUPFIXED, m_spent_keys, "Failed to open db handle for spent_keys");
mdb_set_dupsort(txn, m_spent_keys, compare_hash32);
txn.commit(); txn.commit();
} while(0); } while(0);