Add batch support to BlockchainLMDB::get_output_key
This allows blockchain_import to work with batch and verify modes enabled (the default).
This commit is contained in:
parent
94ea3e8ed2
commit
71793ef43f
|
@ -2294,15 +2294,20 @@ void BlockchainLMDB::get_output_key(const uint64_t &amount, const std::vector<ui
|
||||||
if (global_indices.size() > 0)
|
if (global_indices.size() > 0)
|
||||||
{
|
{
|
||||||
mdb_txn_safe txn;
|
mdb_txn_safe txn;
|
||||||
|
mdb_txn_safe* txn_ptr = &txn;
|
||||||
|
if (m_batch_active)
|
||||||
|
txn_ptr = m_write_txn;
|
||||||
|
else
|
||||||
|
{
|
||||||
if (mdb_txn_begin(m_env, NULL, MDB_RDONLY, txn))
|
if (mdb_txn_begin(m_env, NULL, MDB_RDONLY, txn))
|
||||||
throw0(DB_ERROR("Failed to create a transaction for the db"));
|
throw0(DB_ERROR("Failed to create a transaction for the db"));
|
||||||
|
}
|
||||||
for (const uint64_t &index : global_indices)
|
for (const uint64_t &index : global_indices)
|
||||||
{
|
{
|
||||||
MDB_val_copy<uint64_t> k(index);
|
MDB_val_copy<uint64_t> k(index);
|
||||||
MDB_val v;
|
MDB_val v;
|
||||||
|
|
||||||
auto get_result = mdb_get(txn, m_output_keys, &k, &v);
|
auto get_result = mdb_get(*txn_ptr, m_output_keys, &k, &v);
|
||||||
if (get_result != 0)
|
if (get_result != 0)
|
||||||
throw0(DB_ERROR("Attempting to get output pubkey by global index, but key does not exist"));
|
throw0(DB_ERROR("Attempting to get output pubkey by global index, but key does not exist"));
|
||||||
else if (get_result)
|
else if (get_result)
|
||||||
|
@ -2312,6 +2317,7 @@ void BlockchainLMDB::get_output_key(const uint64_t &amount, const std::vector<ui
|
||||||
outputs.push_back(data);
|
outputs.push_back(data);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if (!m_batch_active)
|
||||||
txn.commit();
|
txn.commit();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue