Merge pull request #698

ee7a8b8 Get rid of lmdb_cur (Howard Chu)
This commit is contained in:
Riccardo Spagni 2016-03-05 23:06:26 +02:00
commit c3af15702f
No known key found for this signature in database
GPG key ID: 55432DF31CCD4FCD

View file

@ -65,45 +65,6 @@ inline void throw1(const T &e)
throw e; throw e;
} }
// cursor needs to be closed when it goes out of scope,
// this helps if the function using it throws
struct lmdb_cur
{
lmdb_cur(MDB_txn* txn, MDB_dbi dbi)
{
if (mdb_cursor_open(txn, dbi, &m_cur))
throw0(cryptonote::DB_ERROR("Error opening db cursor"));
done = false;
}
~lmdb_cur()
{
close();
}
operator MDB_cursor*()
{
return m_cur;
}
operator MDB_cursor**()
{
return &m_cur;
}
void close()
{
if (!done)
{
mdb_cursor_close(m_cur);
done = true;
}
}
private:
MDB_cursor* m_cur;
bool done;
};
template<typename T> template<typename T>
struct MDB_val_copy: public MDB_val struct MDB_val_copy: public MDB_val
{ {
@ -782,12 +743,12 @@ void BlockchainLMDB::remove_tx_outputs(const crypto::hash& tx_hash, const transa
{ {
LOG_PRINT_L3("BlockchainLMDB::" << __func__); LOG_PRINT_L3("BlockchainLMDB::" << __func__);
lmdb_cur cur(*m_write_txn, m_tx_outputs); mdb_txn_cursors *m_cursors = &m_wcursors;
MDB_val_copy<crypto::hash> k(tx_hash); MDB_val_copy<crypto::hash> k(tx_hash);
MDB_val v; MDB_val v;
CURSOR(tx_outputs)
auto result = mdb_cursor_get(cur, &k, &v, MDB_SET); auto result = mdb_cursor_get(m_cur_tx_outputs, &k, &v, MDB_SET);
if (result == MDB_NOTFOUND) if (result == MDB_NOTFOUND)
{ {
LOG_PRINT_L2("tx has no outputs, so no global output indices"); LOG_PRINT_L2("tx has no outputs, so no global output indices");
@ -799,9 +760,9 @@ void BlockchainLMDB::remove_tx_outputs(const crypto::hash& tx_hash, const transa
else else
{ {
mdb_size_t num_elems = 0; mdb_size_t num_elems = 0;
mdb_cursor_count(cur, &num_elems); mdb_cursor_count(m_cur_tx_outputs, &num_elems);
mdb_cursor_get(cur, &k, &v, MDB_LAST_DUP); mdb_cursor_get(m_cur_tx_outputs, &k, &v, MDB_LAST_DUP);
for (uint64_t i = num_elems; i > 0; --i) for (uint64_t i = num_elems; i > 0; --i)
{ {
@ -809,12 +770,10 @@ void BlockchainLMDB::remove_tx_outputs(const crypto::hash& tx_hash, const transa
remove_output(*(const uint64_t*)v.mv_data, tx_output.amount); remove_output(*(const uint64_t*)v.mv_data, tx_output.amount);
if (i > 1) if (i > 1)
{ {
mdb_cursor_get(cur, &k, &v, MDB_PREV_DUP); mdb_cursor_get(m_cur_tx_outputs, &k, &v, MDB_PREV_DUP);
} }
} }
} }
cur.close();
} }
// TODO: probably remove this function // TODO: probably remove this function
@ -870,29 +829,29 @@ void BlockchainLMDB::remove_amount_output_index(const uint64_t amount, const uin
{ {
LOG_PRINT_L3("BlockchainLMDB::" << __func__); LOG_PRINT_L3("BlockchainLMDB::" << __func__);
check_open(); check_open();
mdb_txn_cursors *m_cursors = &m_wcursors;
lmdb_cur cur(*m_write_txn, m_output_amounts); CURSOR(output_amounts);
MDB_val_copy<uint64_t> k(amount); MDB_val_copy<uint64_t> k(amount);
MDB_val v; MDB_val v;
auto result = mdb_cursor_get(cur, &k, &v, MDB_SET); auto result = mdb_cursor_get(m_cur_output_amounts, &k, &v, MDB_SET);
if (result == MDB_NOTFOUND) if (result == MDB_NOTFOUND)
throw1(OUTPUT_DNE("Attempting to get an output index by amount and amount index, but amount not found")); throw1(OUTPUT_DNE("Attempting to get an output index by amount and amount index, but amount not found"));
else if (result) else if (result)
throw0(DB_ERROR("DB error attempting to get an output")); throw0(DB_ERROR("DB error attempting to get an output"));
mdb_size_t num_elems = 0; mdb_size_t num_elems = 0;
mdb_cursor_count(cur, &num_elems); mdb_cursor_count(m_cur_output_amounts, &num_elems);
mdb_cursor_get(cur, &k, &v, MDB_LAST_DUP); mdb_cursor_get(m_cur_output_amounts, &k, &v, MDB_LAST_DUP);
uint64_t amount_output_index = 0; uint64_t amount_output_index = 0;
uint64_t goi = 0; uint64_t goi = 0;
bool found_index = false; bool found_index = false;
for (uint64_t i = num_elems; i > 0; --i) for (uint64_t i = num_elems; i > 0; --i)
{ {
mdb_cursor_get(cur, &k, &v, MDB_GET_CURRENT); mdb_cursor_get(m_cur_output_amounts, &k, &v, MDB_GET_CURRENT);
goi = *(const uint64_t *)v.mv_data; goi = *(const uint64_t *)v.mv_data;
if (goi == global_output_index) if (goi == global_output_index)
{ {
@ -901,23 +860,21 @@ void BlockchainLMDB::remove_amount_output_index(const uint64_t amount, const uin
break; break;
} }
if (i > 1) if (i > 1)
mdb_cursor_get(cur, &k, &v, MDB_PREV_DUP); mdb_cursor_get(m_cur_output_amounts, &k, &v, MDB_PREV_DUP);
} }
if (found_index) if (found_index)
{ {
// found the amount output index // found the amount output index
// now delete it // now delete it
result = mdb_cursor_del(cur, 0); result = mdb_cursor_del(m_cur_output_amounts, 0);
if (result) if (result)
throw0(DB_ERROR(std::string("Error deleting amount output index ").append(boost::lexical_cast<std::string>(amount_output_index)).c_str())); throw0(DB_ERROR(std::string("Error deleting amount output index ").append(boost::lexical_cast<std::string>(amount_output_index)).c_str()));
} }
else else
{ {
// not found // not found
cur.close();
throw1(OUTPUT_DNE("Failed to find amount output index")); throw1(OUTPUT_DNE("Failed to find amount output index"));
} }
cur.close();
} }
void BlockchainLMDB::add_spent_key(const crypto::key_image& k_image) void BlockchainLMDB::add_spent_key(const crypto::key_image& k_image)