0
0
Fork 0
mirror of https://github.com/matrix-construct/construct synced 2024-12-12 16:43:07 +01:00
construct/include/ircd/db/txn.h
Jason Volk b677eeb4b7 Allow various unsigned conversions.
construct: Fix comparison signedness warning.
2022-06-29 14:21:36 -07:00

150 lines
4 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;
using delta_closure = std::function<void (const delta &)>;
using delta_closure_bool = std::function<bool (const delta &)>;
using seq_closure = std::function<void (txn &, const uint64_t &)>;
using seq_closure_bool = std::function<bool (txn &, const uint64_t &)>;
bool for_each(const txn &, const delta_closure_bool &);
void for_each(const txn &, const delta_closure &);
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 &);
string_view debug(const mutable_buffer &out, const txn &, const long &fmt = 0);
string_view debug(const mutable_buffer &out, database &, const rocksdb::WriteBatch &, const long &fmt = 0);
}
struct ircd::db::txn
{
struct opts;
struct checkpoint;
struct append;
struct handler;
enum state :uint8_t;
using value_closure = std::function<void (const string_view &)>;
database *d {nullptr};
std::unique_ptr<rocksdb::WriteBatch> wb;
enum state state {0};
public:
explicit operator const rocksdb::WriteBatch &() const;
explicit operator const database &() const;
explicit operator rocksdb::WriteBatch &();
explicit operator database &();
template<class T = string_view> T val(const op &, const string_view &col, const string_view &key, T def = {}) const;
template<class T = string_view> T at(const op &, const string_view &col, const string_view &key) const;
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 & = {});
// reset
void clear();
auto release() noexcept;
txn() = default;
txn(database &);
txn(database &, const opts &);
txn(database &, std::unique_ptr<rocksdb::WriteBatch> &&);
~txn() noexcept;
};
enum ircd::db::txn::state
:uint8_t
{
BUILD = 0,
COMMIT = 1,
COMMITTED = 2,
};
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 &);
};
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>
T
ircd::db::txn::at(const op &op,
const string_view &col,
const string_view &key)
const
{
T ret;
at(op, col, key, [&ret](const string_view &val)
{
ret = byte_view<T>(val);
});
return ret;
}
template<class T>
T
ircd::db::txn::val(const op &op,
const string_view &col,
const string_view &key,
T ret)
const
{
get(op, col, key, value_closure{[&ret](const string_view &val)
{
ret = byte_view<T>(val);
}});
return ret;
}
inline auto
ircd::db::txn::release()
noexcept
{
return wb.release();
}