MirrorHub/construct
0
0
Fork 0
mirror of https://github.com/matrix-construct/construct synced 2024-12-26 15:33:54 +01:00
Code Issues Releases Wiki Activity

ircd::db: Add features to txn.

Browse source
This commit is contained in:
Jason Volk 2017-09-18 20:40:13 -07:00
parent d7e9c9182a
commit fb026f58a7
2 changed files with 134 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
include/ircd/db/txn.h
View file

@ -27,6 +27,7 @@ namespace ircd::db
{
struct txn;
bool until(const txn &, const std::function<bool (const delta &)> &);
void for_each(const txn &, const std::function<void (const delta &)> &);
std::string debug(const txn &);
}
@ -47,7 +48,15 @@ class ircd::db::txn
explicit operator rocksdb::WriteBatch &();
explicit operator database &();
string_view get(const op &, const string_view &col, const string_view &key) const;
string_view at(const op &, const string_view &col, const string_view &key) const;
bool has(const op &, const string_view &col, const string_view &key) const;
delta get(const op &, const string_view &col) const;
delta at(const op &, const string_view &col) const;
bool has(const op &, const string_view &col) const;
bool has(const op &) const;
size_t bytes() const;
size_t size() const;

134
ircd/db.cc
View file

@ -1305,7 +1305,7 @@ struct ircd::db::txn::handler
using Slice = rocksdb::Slice;
const database &d;
const std::function<void (const delta &)> &cb;
const std::function<bool (const delta &)> &cb;
bool _continue {true};
Status callback(const delta &) noexcept;
@ -1324,7 +1324,7 @@ struct ircd::db::txn::handler
Status PutCF(const uint32_t cfid, const Slice &, const Slice &) noexcept override;
handler(const database &d,
const std::function<void (const delta &)> &cb)
const std::function<bool (const delta &)> &cb)
:d{d}
,cb{cb}
{}
@ -1340,14 +1340,29 @@ ircd::db::debug(const txn &t)
void
ircd::db::for_each(const txn &t,
const std::function<void (const delta &)> &closure)
{
const auto re{[&closure]
(const delta &delta)
{
closure(delta);
return true;
}};
const database &d(t);
const rocksdb::WriteBatch &wb{t};
txn::handler h{d, re};
wb.Iterate(&h);
}
bool
ircd::db::until(const txn &t,
const std::function<bool (const delta &)> &closure)
{
const database &d(t);
const rocksdb::WriteBatch &wb{t};
txn::handler h{d, closure};
throw_on_error
{
wb.Iterate(&h)
};
wb.Iterate(&h);
return h._continue;
}
///
@ -1446,20 +1461,21 @@ noexcept try
catch(const std::exception &e)
{
_continue = false;
return Status::Aborted(slice(string_view{e.what()}));
log::critical("txn::handler: cfid[%u]: %s", cfid, e.what());
std::terminate();
}
rocksdb::Status
ircd::db::txn::handler::callback(const delta &delta)
noexcept try
{
cb(delta);
_continue = cb(delta);
return Status::OK();
}
catch(const std::exception &e)
{
_continue = false;
return Status::Aborted(slice(string_view{e.what()}));
return Status::OK();
}
bool
@ -1549,6 +1565,106 @@ const
return false;
}
bool
ircd::db::txn::has(const op &op,
const string_view &col)
const
{
return !until(*this, [&op, &col]
(const auto &delta)
{
return std::get<0>(delta) == op &&
std::get<1>(delta) == col;
});
}
bool
ircd::db::txn::has(const op &op,
const string_view &col,
const string_view &key)
const
{
return !until(*this, [&op, &col, &key]
(const auto &delta)
{
return std::get<0>(delta) == op &&
std::get<1>(delta) == col &&
std::get<2>(delta) == key;
});
}
ircd::db::delta
ircd::db::txn::at(const op &op,
const string_view &col)
const
{
const auto ret(get(op, col));
if(unlikely(!std::get<2>(ret)))
throw not_found("db::txn::at(%s, %s): no matching delta in transaction",
reflect(op),
col);
return ret;
}
ircd::db::delta
ircd::db::txn::get(const op &op,
const string_view &col)
const
{
delta ret;
until(*this, [&ret, &op, &col]
(const delta &delta)
{
if(std::get<0>(delta) == op &&
std::get<1>(delta) == col)
{
ret = delta;
return false;
}
else return true;
});
return ret;
}
ircd::string_view
ircd::db::txn::at(const op &op,
const string_view &col,
const string_view &key)
const
{
const auto ret(get(op, col, key));
if(unlikely(!ret))
throw not_found("db::txn::at(%s, %s, %s): no matching delta in transaction",
reflect(op),
col,
key);
return ret;
}
ircd::string_view
ircd::db::txn::get(const op &op,
const string_view &col,
const string_view &key)
const
{
string_view ret;
until(*this, [&ret, &op, &col, &key]
(const delta &delta)
{
if(std::get<0>(delta) == op &&
std::get<1>(delta) == col &&
std::get<2>(delta) == key)
{
ret = std::get<3>(delta);
return false;
}
else return true;
});
return ret;
}
ircd::db::txn::operator
ircd::db::database &()
{