mirror of
https://github.com/matrix-construct/construct
synced 2024-11-16 15:00:51 +01:00
150 lines
4 KiB
C++
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();
|
|
}
|