0
0
Fork 0
mirror of https://github.com/matrix-construct/construct synced 2024-12-25 06:53:58 +01:00
construct/include/ircd/db/txn.h
2018-09-26 15:28:36 -07:00

150 lines
4.2 KiB
C++

// Matrix Construct
//
// Copyright (C) Matrix Construct Developers, Authors & Contributors
// Copyright (C) 2016-2018 Jason Volk <jason@zemos.net>
//
// Permission to use, copy, modify, and/or distribute this software for any
// purpose with or without fee is hereby granted, provided that the above
// copyright notice and this permission notice is present in all copies. The
// full license for this software is available in the LICENSE file.
#pragma once
#define HAVE_IRCD_DB_TXN_H
namespace ircd::db
{
struct txn;
bool for_each(const txn &, const std::function<bool (const delta &)> &);
void for_each(const txn &, const std::function<void (const delta &)> &);
std::string debug(const txn &);
using seq_closure_bool = std::function<bool (txn &, const uint64_t &)>;
using seq_closure = std::function<void (txn &, const uint64_t &)>;
bool for_each(database &d, const uint64_t &seq, const seq_closure_bool &);
void for_each(database &d, const uint64_t &seq, const seq_closure &);
void get(database &d, const uint64_t &seq, const seq_closure &);
}
struct ircd::db::txn
{
database *d {nullptr};
std::unique_ptr<rocksdb::WriteBatch> wb;
public:
struct opts;
struct checkpoint;
struct append;
struct handler;
using delta_closure = std::function<void (const delta &)>;
using delta_closure_bool = std::function<bool (const delta &)>;
using value_closure = std::function<void (const string_view &)>;
explicit operator const rocksdb::WriteBatch &() const;
explicit operator const database &() const;
explicit operator rocksdb::WriteBatch &();
explicit operator database &();
bool get(const op &, const string_view &col, const string_view &key, const value_closure &) const;
void at(const op &, const string_view &col, const string_view &key, const value_closure &) const;
bool has(const op &, const string_view &col, const string_view &key) const;
bool get(const op &, const string_view &col, const delta_closure &) const;
void at(const op &, const string_view &col, const delta_closure &) const;
bool has(const op &, const string_view &col) const;
bool has(const op &) const;
size_t bytes() const; // size of data in txn.
size_t size() const; // count of updates in txn.
// commit
void operator()(database &, const sopts & = {});
void operator()(const sopts & = {});
// clear
void clear();
txn() = default;
txn(database &);
txn(database &, const opts &);
txn(database &, std::unique_ptr<rocksdb::WriteBatch> &&);
~txn() noexcept;
};
struct ircd::db::txn::append
{
append(txn &, database &, const delta &);
append(txn &, column &, const column::delta &);
append(txn &, const cell::delta &);
append(txn &, const row::delta &);
append(txn &, const delta &);
append(txn &, const string_view &key, const json::iov &);
template<class... T> append(txn &, const string_view &key, const json::tuple<T...> &, const op & = op::SET);
template<class... T> append(txn &, const string_view &key, const json::tuple<T...> &, std::array<column, sizeof...(T)> &, const op & = op::SET);
};
struct ircd::db::txn::checkpoint
{
txn &t;
checkpoint(txn &);
~checkpoint() noexcept;
};
struct ircd::db::txn::opts
{
size_t reserve_bytes = 0;
size_t max_bytes = 0;
};
template<class... T>
ircd::db::txn::append::append(txn &txn,
const string_view &key,
const json::tuple<T...> &tuple,
const op &op)
{
for_each(tuple, [&txn, &key, &op]
(const auto &col, auto&& val)
{
if(!value_required(op) || defined(json::value(val))) append
{
txn, delta
{
op,
col,
key,
value_required(op)?
byte_view<string_view>{val}:
byte_view<string_view>{}
}
};
});
}
template<class... T>
ircd::db::txn::append::append(txn &txn,
const string_view &key,
const json::tuple<T...> &tuple,
std::array<column, sizeof...(T)> &col,
const op &op)
{
size_t i{0};
for_each(tuple, [&txn, &key, &col, &op, &i]
(const auto &, auto&& val)
{
if(!value_required(op) || defined(json::value(val))) append
{
txn, col.at(i), column::delta
{
op,
key,
value_required(op)?
byte_view<string_view>{val}:
byte_view<string_view>{}
}
};
++i;
});
}