2018-01-18 03:14:13 +01:00
|
|
|
// 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
|
2018-02-04 03:22:01 +01:00
|
|
|
// copyright notice and this permission notice is present in all copies. The
|
|
|
|
// full license for this software is available in the LICENSE file.
|
2016-09-24 06:01:57 +02:00
|
|
|
|
2017-08-23 23:37:06 +02:00
|
|
|
#include <rocksdb/version.h>
|
2016-09-24 06:01:57 +02:00
|
|
|
#include <rocksdb/db.h>
|
2016-09-26 06:07:22 +02:00
|
|
|
#include <rocksdb/cache.h>
|
2017-03-31 00:57:08 +02:00
|
|
|
#include <rocksdb/comparator.h>
|
2017-03-23 22:58:24 +01:00
|
|
|
#include <rocksdb/merge_operator.h>
|
|
|
|
#include <rocksdb/perf_level.h>
|
2017-04-03 06:02:32 +02:00
|
|
|
#include <rocksdb/perf_context.h>
|
2018-05-24 02:03:09 +02:00
|
|
|
#include <rocksdb/iostats_context.h>
|
2017-03-23 22:58:24 +01:00
|
|
|
#include <rocksdb/listener.h>
|
|
|
|
#include <rocksdb/statistics.h>
|
2017-03-24 00:10:17 +01:00
|
|
|
#include <rocksdb/convenience.h>
|
|
|
|
#include <rocksdb/env.h>
|
2017-08-31 07:12:58 +02:00
|
|
|
#include <rocksdb/slice_transform.h>
|
2018-03-23 03:56:08 +01:00
|
|
|
#include <rocksdb/utilities/checkpoint.h>
|
2018-04-15 08:00:56 +02:00
|
|
|
#include <rocksdb/filter_policy.h>
|
|
|
|
#include <rocksdb/table.h>
|
2018-04-21 00:45:57 +02:00
|
|
|
#include <rocksdb/sst_file_manager.h>
|
2016-09-24 06:01:57 +02:00
|
|
|
|
2018-05-30 09:14:07 +02:00
|
|
|
// ircd::db interfaces requiring complete RocksDB (frontside).
|
2018-01-18 03:34:20 +01:00
|
|
|
#include <ircd/db/database/comparator.h>
|
|
|
|
#include <ircd/db/database/prefix_transform.h>
|
|
|
|
#include <ircd/db/database/mergeop.h>
|
|
|
|
#include <ircd/db/database/events.h>
|
|
|
|
#include <ircd/db/database/stats.h>
|
|
|
|
#include <ircd/db/database/logs.h>
|
|
|
|
#include <ircd/db/database/column.h>
|
2018-03-23 04:47:46 +01:00
|
|
|
#include <ircd/db/database/txn.h>
|
2018-05-30 09:14:07 +02:00
|
|
|
|
|
|
|
// RocksDB embedding environment callback interfaces (backside).
|
2018-01-19 02:59:22 +01:00
|
|
|
#include <ircd/db/database/env/env.h>
|
|
|
|
#include <ircd/db/database/env/writable_file.h>
|
|
|
|
#include <ircd/db/database/env/sequential_file.h>
|
|
|
|
#include <ircd/db/database/env/random_access_file.h>
|
|
|
|
#include <ircd/db/database/env/random_rw_file.h>
|
|
|
|
#include <ircd/db/database/env/directory.h>
|
|
|
|
#include <ircd/db/database/env/file_lock.h>
|
2018-06-09 22:40:21 +02:00
|
|
|
#include <ircd/db/database/env/state.h>
|
|
|
|
|
2018-08-16 13:48:21 +02:00
|
|
|
#define IRCD_DB_PORT
|
2018-01-18 03:34:20 +01:00
|
|
|
|
2018-05-30 09:14:07 +02:00
|
|
|
// RocksDB port linktime-overriding interfaces (experimental).
|
|
|
|
#ifdef IRCD_DB_PORT
|
|
|
|
#include <ircd/db/database/env/port.h>
|
|
|
|
#endif
|
|
|
|
|
|
|
|
// Internal utility interface for this definition file.
|
2018-01-24 00:39:44 +01:00
|
|
|
#include "db.h"
|
2017-03-24 02:36:49 +01:00
|
|
|
|
2018-05-30 09:14:07 +02:00
|
|
|
//
|
|
|
|
// Misc / General linkages
|
|
|
|
//
|
|
|
|
|
2018-01-24 00:39:44 +01:00
|
|
|
decltype(ircd::db::log)
|
|
|
|
ircd::db::log
|
2017-09-20 06:40:53 +02:00
|
|
|
{
|
|
|
|
// Dedicated logging facility for the database subsystem
|
|
|
|
"db", 'D'
|
|
|
|
};
|
|
|
|
|
2018-01-24 00:39:44 +01:00
|
|
|
decltype(ircd::db::rog)
|
|
|
|
ircd::db::rog
|
2017-09-30 08:07:55 +02:00
|
|
|
{
|
|
|
|
// Dedicated logging facility for rocksdb's log callbacks
|
2018-01-24 00:39:44 +01:00
|
|
|
"rdb", 'R'
|
|
|
|
};
|
2017-09-30 08:07:55 +02:00
|
|
|
|
2017-08-23 22:37:47 +02:00
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
//
|
|
|
|
// init
|
|
|
|
//
|
|
|
|
|
2017-09-08 11:09:34 +02:00
|
|
|
namespace ircd::db
|
|
|
|
{
|
2018-06-12 09:24:59 +02:00
|
|
|
static void init_compressions();
|
2017-09-08 11:09:34 +02:00
|
|
|
static void init_directory();
|
|
|
|
static void init_version();
|
|
|
|
}
|
2017-08-23 22:37:47 +02:00
|
|
|
|
2017-08-23 23:37:06 +02:00
|
|
|
static char ircd_db_version[64];
|
|
|
|
const char *const ircd::db::version(ircd_db_version);
|
|
|
|
|
2018-06-12 09:24:59 +02:00
|
|
|
ircd::db::init::init()
|
|
|
|
{
|
|
|
|
init_compressions();
|
|
|
|
init_directory();
|
|
|
|
}
|
|
|
|
|
|
|
|
ircd::db::init::~init()
|
|
|
|
noexcept
|
2017-08-23 23:37:06 +02:00
|
|
|
{
|
|
|
|
}
|
|
|
|
|
2018-06-12 09:24:59 +02:00
|
|
|
void
|
2017-08-23 22:37:47 +02:00
|
|
|
ircd::db::init_directory()
|
|
|
|
try
|
|
|
|
{
|
2018-04-03 02:18:02 +02:00
|
|
|
const auto dbdir
|
|
|
|
{
|
|
|
|
fs::get(fs::DB)
|
|
|
|
};
|
|
|
|
|
2017-08-23 22:37:47 +02:00
|
|
|
if(fs::mkdir(dbdir))
|
2018-08-16 10:27:43 +02:00
|
|
|
log::notice
|
|
|
|
{
|
|
|
|
log, "Created new database directory at `%s'", dbdir
|
|
|
|
};
|
2017-08-23 22:37:47 +02:00
|
|
|
else
|
2018-08-16 10:27:43 +02:00
|
|
|
log::info
|
|
|
|
{
|
|
|
|
log, "Using database directory at `%s'", dbdir
|
|
|
|
};
|
2017-08-23 22:37:47 +02:00
|
|
|
}
|
|
|
|
catch(const fs::error &e)
|
|
|
|
{
|
2018-08-16 10:27:43 +02:00
|
|
|
log::error
|
|
|
|
{
|
|
|
|
log, "Cannot start database system: %s", e.what()
|
|
|
|
};
|
|
|
|
|
2017-08-23 22:37:47 +02:00
|
|
|
if(ircd::debugmode)
|
|
|
|
throw;
|
|
|
|
}
|
|
|
|
|
2018-06-12 09:24:59 +02:00
|
|
|
void
|
|
|
|
ircd::db::init_compressions()
|
2017-08-23 22:37:47 +02:00
|
|
|
{
|
2018-06-12 09:24:59 +02:00
|
|
|
const auto compressions
|
|
|
|
{
|
|
|
|
rocksdb::GetSupportedCompressions()
|
|
|
|
};
|
|
|
|
|
|
|
|
if(compressions.empty())
|
|
|
|
log::warning
|
|
|
|
{
|
|
|
|
"No compression libraries have been linked with the DB."
|
|
|
|
" This is probably not what you want."
|
|
|
|
};
|
2017-08-23 22:37:47 +02:00
|
|
|
}
|
|
|
|
|
2018-06-12 09:24:59 +02:00
|
|
|
// Renders a version string from the defines included here.
|
|
|
|
__attribute__((constructor))
|
|
|
|
void
|
|
|
|
ircd::db::init_version()
|
2017-08-23 22:37:47 +02:00
|
|
|
{
|
2018-06-12 09:24:59 +02:00
|
|
|
snprintf(ircd_db_version, sizeof(ircd_db_version), "%d.%d.%d",
|
|
|
|
ROCKSDB_MAJOR,
|
|
|
|
ROCKSDB_MINOR,
|
|
|
|
ROCKSDB_PATCH);
|
2017-08-23 22:37:47 +02:00
|
|
|
}
|
|
|
|
|
2017-03-31 00:57:08 +02:00
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
//
|
|
|
|
// database
|
|
|
|
//
|
2016-09-24 06:01:57 +02:00
|
|
|
|
2017-03-31 00:57:08 +02:00
|
|
|
void
|
|
|
|
ircd::db::sync(database &d)
|
2016-09-24 06:01:57 +02:00
|
|
|
{
|
2017-04-03 06:02:32 +02:00
|
|
|
throw_on_error
|
|
|
|
{
|
|
|
|
d.d->SyncWAL()
|
|
|
|
};
|
2016-09-24 06:01:57 +02:00
|
|
|
}
|
|
|
|
|
2018-03-23 03:41:50 +01:00
|
|
|
/// Flushes all columns. Note that if blocking=true, blocking may occur for
|
|
|
|
/// each column individually.
|
|
|
|
void
|
|
|
|
ircd::db::flush(database &d,
|
2018-04-27 02:19:29 +02:00
|
|
|
const bool &sync)
|
|
|
|
{
|
|
|
|
throw_on_error
|
|
|
|
{
|
|
|
|
d.d->FlushWAL(sync)
|
|
|
|
};
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Moves memory structures to SST files for all columns. This doesn't
|
|
|
|
/// necessarily sort anything that wasn't previously sorted, but it may create
|
|
|
|
/// new SST files and shouldn't be confused with a typical fflush().
|
|
|
|
/// Note that if blocking=true, blocking may occur for each column individually.
|
|
|
|
void
|
|
|
|
ircd::db::sort(database &d,
|
|
|
|
const bool &blocking)
|
2018-03-23 03:41:50 +01:00
|
|
|
{
|
2018-04-20 23:12:57 +02:00
|
|
|
for(const auto &c : d.columns)
|
|
|
|
{
|
|
|
|
db::column column{*c};
|
2018-04-27 02:19:29 +02:00
|
|
|
db::sort(column, blocking);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
ircd::db::compact(database &d)
|
|
|
|
{
|
2018-05-11 01:41:34 +02:00
|
|
|
static const std::pair<string_view, string_view> range
|
|
|
|
{
|
|
|
|
{}, {}
|
|
|
|
};
|
|
|
|
|
2018-04-27 02:19:29 +02:00
|
|
|
for(const auto &c : d.columns)
|
|
|
|
{
|
|
|
|
db::column column{*c};
|
2018-05-11 01:41:34 +02:00
|
|
|
compact(column, range, -1);
|
|
|
|
compact(column, -1);
|
2018-04-20 23:12:57 +02:00
|
|
|
}
|
2018-03-23 03:41:50 +01:00
|
|
|
}
|
|
|
|
|
2018-05-23 01:52:34 +02:00
|
|
|
void
|
|
|
|
ircd::db::check(database &d)
|
|
|
|
{
|
|
|
|
assert(d.d);
|
|
|
|
throw_on_error
|
|
|
|
{
|
|
|
|
d.d->VerifyChecksum()
|
|
|
|
};
|
|
|
|
}
|
|
|
|
|
2018-03-23 03:56:08 +01:00
|
|
|
/// Writes a snapshot of this database to the directory specified. The
|
|
|
|
/// snapshot consists of hardlinks to the bulk data files of this db, but
|
|
|
|
/// copies the other stuff that usually gets corrupted. The directory can
|
|
|
|
/// then be opened as its own database either read-only or read-write.
|
|
|
|
/// Incremental backups and rollbacks can begin from this interface. Note
|
|
|
|
/// this may be an expensive blocking operation.
|
2018-05-25 05:01:52 +02:00
|
|
|
uint64_t
|
|
|
|
ircd::db::checkpoint(database &d)
|
2018-03-23 03:56:08 +01:00
|
|
|
{
|
2018-05-25 05:01:52 +02:00
|
|
|
if(!d.checkpointer)
|
2018-03-23 03:56:08 +01:00
|
|
|
throw error
|
|
|
|
{
|
2018-05-25 05:01:52 +02:00
|
|
|
"Checkpointing is not available for db(%p) '%s",
|
|
|
|
&d,
|
|
|
|
name(d)
|
2018-03-23 03:56:08 +01:00
|
|
|
};
|
|
|
|
|
2018-05-25 05:01:52 +02:00
|
|
|
const auto seqnum
|
|
|
|
{
|
|
|
|
sequence(d)
|
|
|
|
};
|
|
|
|
|
|
|
|
const std::string dir
|
|
|
|
{
|
|
|
|
db::path(name(d), seqnum)
|
|
|
|
};
|
|
|
|
|
2018-03-23 03:56:08 +01:00
|
|
|
throw_on_error
|
|
|
|
{
|
2018-05-25 05:01:52 +02:00
|
|
|
d.checkpointer->CreateCheckpoint(dir, 0)
|
2018-03-23 03:56:08 +01:00
|
|
|
};
|
|
|
|
|
2018-05-25 05:01:52 +02:00
|
|
|
log::debug
|
|
|
|
{
|
|
|
|
log, "'%s': Checkpoint at sequence %lu in `%s' complete",
|
|
|
|
name(d),
|
|
|
|
seqnum,
|
|
|
|
dir
|
|
|
|
};
|
|
|
|
|
|
|
|
return seqnum;
|
2018-03-23 03:56:08 +01:00
|
|
|
}
|
|
|
|
|
2018-04-03 20:26:56 +02:00
|
|
|
/// This wraps RocksDB's "File Deletions" which means after RocksDB
|
|
|
|
/// compresses some file it then destroys the uncompressed version;
|
|
|
|
/// setting this to false will disable that and retain both versions.
|
|
|
|
/// This is useful when a direct reference is being manually held by
|
|
|
|
/// us into the uncompressed version which must remain valid.
|
|
|
|
void
|
|
|
|
ircd::db::fdeletions(database &d,
|
2018-04-20 22:38:49 +02:00
|
|
|
const bool &enable,
|
|
|
|
const bool &force)
|
2018-04-03 20:26:56 +02:00
|
|
|
{
|
|
|
|
if(enable) throw_on_error
|
|
|
|
{
|
2018-04-20 22:38:49 +02:00
|
|
|
d.d->EnableFileDeletions(force)
|
2018-04-03 20:26:56 +02:00
|
|
|
};
|
|
|
|
else throw_on_error
|
|
|
|
{
|
|
|
|
d.d->DisableFileDeletions()
|
|
|
|
};
|
|
|
|
}
|
|
|
|
|
2018-04-20 23:10:59 +02:00
|
|
|
void
|
|
|
|
ircd::db::setopt(database &d,
|
|
|
|
const string_view &key,
|
|
|
|
const string_view &val)
|
|
|
|
{
|
|
|
|
const std::unordered_map<std::string, std::string> options
|
|
|
|
{
|
|
|
|
{ std::string{key}, std::string{val} }
|
|
|
|
};
|
|
|
|
|
|
|
|
throw_on_error
|
|
|
|
{
|
|
|
|
d.d->SetDBOptions(options)
|
|
|
|
};
|
|
|
|
}
|
|
|
|
|
2018-04-15 13:45:31 +02:00
|
|
|
uint64_t
|
|
|
|
ircd::db::ticker(const database &d,
|
|
|
|
const string_view &key)
|
|
|
|
{
|
|
|
|
return ticker(d, ticker_id(key));
|
|
|
|
}
|
|
|
|
|
|
|
|
uint64_t
|
|
|
|
ircd::db::ticker(const database &d,
|
|
|
|
const uint32_t &id)
|
|
|
|
{
|
|
|
|
return d.stats->getTickerCount(id);
|
|
|
|
}
|
|
|
|
|
2018-04-17 23:13:36 +02:00
|
|
|
size_t
|
|
|
|
ircd::db::bytes(const database &d)
|
|
|
|
{
|
|
|
|
return std::accumulate(begin(d.columns), end(d.columns), size_t(0), []
|
|
|
|
(auto ret, const auto &colptr)
|
|
|
|
{
|
|
|
|
db::column c{*colptr};
|
|
|
|
return ret += db::bytes(c);
|
|
|
|
});
|
|
|
|
}
|
|
|
|
|
|
|
|
size_t
|
|
|
|
ircd::db::file_count(const database &d)
|
|
|
|
{
|
|
|
|
return std::accumulate(begin(d.columns), end(d.columns), size_t(0), []
|
|
|
|
(auto ret, const auto &colptr)
|
|
|
|
{
|
|
|
|
db::column c{*colptr};
|
|
|
|
return ret += db::file_count(c);
|
|
|
|
});
|
|
|
|
}
|
|
|
|
|
2018-04-03 20:44:57 +02:00
|
|
|
/// Get the live file list for db; see overlord documentation.
|
|
|
|
std::vector<std::string>
|
|
|
|
ircd::db::files(const database &d)
|
|
|
|
{
|
|
|
|
uint64_t ignored;
|
|
|
|
return files(d, ignored);
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Get the live file list for database relative to the database's directory.
|
|
|
|
/// One of the files is a manifest file which is over-allocated and its used
|
|
|
|
/// size is returned in the integer passed to the `msz` argument.
|
|
|
|
///
|
|
|
|
/// This list may not be completely up to date. The reliable way to get the
|
|
|
|
/// most current list is to flush all columns first and ensure no database
|
|
|
|
/// activity took place between the flushing and this query.
|
|
|
|
std::vector<std::string>
|
|
|
|
ircd::db::files(const database &cd,
|
|
|
|
uint64_t &msz)
|
|
|
|
{
|
|
|
|
std::vector<std::string> ret;
|
|
|
|
auto &d(const_cast<database &>(cd));
|
|
|
|
throw_on_error
|
|
|
|
{
|
|
|
|
d.d->GetLiveFiles(ret, &msz, false)
|
|
|
|
};
|
|
|
|
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
2017-03-31 00:57:08 +02:00
|
|
|
uint64_t
|
2018-03-23 03:35:50 +01:00
|
|
|
ircd::db::sequence(const database &cd)
|
2016-09-24 06:01:57 +02:00
|
|
|
{
|
2018-03-23 03:35:50 +01:00
|
|
|
database &d(const_cast<database &>(cd));
|
2017-03-31 00:57:08 +02:00
|
|
|
return d.d->GetLatestSequenceNumber();
|
|
|
|
}
|
|
|
|
|
2018-05-15 01:17:43 +02:00
|
|
|
rocksdb::Cache *
|
|
|
|
ircd::db::cache(database &d)
|
|
|
|
{
|
|
|
|
return d.cache.get();
|
|
|
|
}
|
|
|
|
|
|
|
|
const rocksdb::Cache *
|
|
|
|
ircd::db::cache(const database &d)
|
|
|
|
{
|
|
|
|
return d.cache.get();
|
|
|
|
}
|
|
|
|
|
2017-03-31 00:57:08 +02:00
|
|
|
template<>
|
2018-04-03 19:58:16 +02:00
|
|
|
ircd::db::prop_int
|
2018-03-23 03:35:50 +01:00
|
|
|
ircd::db::property(const database &cd,
|
2017-03-31 00:57:08 +02:00
|
|
|
const string_view &name)
|
|
|
|
{
|
|
|
|
uint64_t ret;
|
2018-03-23 03:35:50 +01:00
|
|
|
database &d(const_cast<database &>(cd));
|
2017-03-31 00:57:08 +02:00
|
|
|
if(!d.d->GetAggregatedIntProperty(slice(name), &ret))
|
2018-04-13 03:57:03 +02:00
|
|
|
throw not_found
|
|
|
|
{
|
|
|
|
"property '%s' for all columns in '%s' not found or not an integer.",
|
|
|
|
name,
|
|
|
|
db::name(d)
|
|
|
|
};
|
2017-03-31 00:57:08 +02:00
|
|
|
|
|
|
|
return ret;
|
2016-09-24 06:01:57 +02:00
|
|
|
}
|
|
|
|
|
2017-04-03 06:02:32 +02:00
|
|
|
std::shared_ptr<ircd::db::database::column>
|
|
|
|
ircd::db::shared_from(database::column &column)
|
|
|
|
{
|
|
|
|
return column.shared_from_this();
|
|
|
|
}
|
|
|
|
|
|
|
|
std::shared_ptr<const ircd::db::database::column>
|
|
|
|
ircd::db::shared_from(const database::column &column)
|
|
|
|
{
|
|
|
|
return column.shared_from_this();
|
|
|
|
}
|
|
|
|
|
2018-05-23 02:01:01 +02:00
|
|
|
const std::string &
|
|
|
|
ircd::db::uuid(const database &d)
|
|
|
|
{
|
|
|
|
return d.uuid;
|
|
|
|
}
|
|
|
|
|
|
|
|
const std::string &
|
|
|
|
ircd::db::name(const database &d)
|
|
|
|
{
|
|
|
|
return d.name;
|
|
|
|
}
|
|
|
|
|
2017-03-24 02:36:49 +01:00
|
|
|
//
|
2017-03-31 00:57:08 +02:00
|
|
|
// database
|
2017-03-24 02:36:49 +01:00
|
|
|
//
|
2018-05-25 05:01:52 +02:00
|
|
|
|
2017-08-31 07:13:50 +02:00
|
|
|
namespace ircd::db
|
|
|
|
{
|
2018-04-27 01:22:16 +02:00
|
|
|
extern const database::description default_description;
|
2017-08-31 07:13:50 +02:00
|
|
|
}
|
2016-11-29 16:23:38 +01:00
|
|
|
|
2018-05-25 05:01:52 +02:00
|
|
|
// Instance list linkage
|
|
|
|
template<>
|
|
|
|
decltype(ircd::util::instance_list<ircd::db::database>::list)
|
|
|
|
ircd::util::instance_list<ircd::db::database>::list
|
|
|
|
{};
|
|
|
|
|
2018-04-27 01:22:16 +02:00
|
|
|
decltype(ircd::db::default_description)
|
|
|
|
ircd::db::default_description
|
|
|
|
{
|
|
|
|
/// Requirement of RocksDB going back to LevelDB. This column must
|
|
|
|
/// always exist in all descriptions and probably should be at idx[0].
|
|
|
|
{ "default" }
|
|
|
|
};
|
|
|
|
|
2018-05-25 05:01:52 +02:00
|
|
|
ircd::db::database &
|
|
|
|
ircd::db::database::get(column &column)
|
|
|
|
{
|
|
|
|
assert(column.d);
|
|
|
|
return *column.d;
|
|
|
|
}
|
|
|
|
|
|
|
|
const ircd::db::database &
|
|
|
|
ircd::db::database::get(const column &column)
|
|
|
|
{
|
|
|
|
assert(column.d);
|
|
|
|
return *column.d;
|
|
|
|
}
|
|
|
|
|
|
|
|
ircd::db::database &
|
|
|
|
ircd::db::database::get(const string_view &name)
|
|
|
|
{
|
|
|
|
const auto pair
|
|
|
|
{
|
|
|
|
namepoint(name)
|
|
|
|
};
|
|
|
|
|
|
|
|
return get(pair.first, pair.second);
|
|
|
|
}
|
|
|
|
|
|
|
|
ircd::db::database &
|
|
|
|
ircd::db::database::get(const string_view &name,
|
|
|
|
const uint64_t &checkpoint)
|
|
|
|
{
|
|
|
|
auto *const &d
|
|
|
|
{
|
|
|
|
get(std::nothrow, name, checkpoint)
|
|
|
|
};
|
|
|
|
|
|
|
|
if(likely(d))
|
|
|
|
return *d;
|
|
|
|
|
|
|
|
throw checkpoint == uint64_t(-1)?
|
|
|
|
std::out_of_range{"No database with that name exists"}:
|
|
|
|
std::out_of_range{"No database with that name at that checkpoint exists"};
|
|
|
|
}
|
|
|
|
|
|
|
|
ircd::db::database *
|
|
|
|
ircd::db::database::get(std::nothrow_t,
|
|
|
|
const string_view &name)
|
|
|
|
{
|
|
|
|
const auto pair
|
|
|
|
{
|
|
|
|
namepoint(name)
|
|
|
|
};
|
|
|
|
|
|
|
|
return get(std::nothrow, pair.first, pair.second);
|
|
|
|
}
|
|
|
|
|
|
|
|
ircd::db::database *
|
|
|
|
ircd::db::database::get(std::nothrow_t,
|
|
|
|
const string_view &name,
|
|
|
|
const uint64_t &checkpoint)
|
|
|
|
{
|
|
|
|
for(auto *const &d : list)
|
|
|
|
if(name == d->name)
|
|
|
|
if(checkpoint == uint64_t(-1) || checkpoint == d->checkpoint)
|
|
|
|
return d;
|
|
|
|
|
|
|
|
return nullptr;
|
|
|
|
}
|
|
|
|
|
|
|
|
//
|
|
|
|
// database::database
|
|
|
|
//
|
|
|
|
|
|
|
|
ircd::db::database::database(const string_view &name,
|
2017-08-23 22:37:47 +02:00
|
|
|
std::string optstr)
|
2017-08-23 23:37:06 +02:00
|
|
|
:database
|
|
|
|
{
|
2018-05-25 05:01:52 +02:00
|
|
|
name, std::move(optstr), default_description
|
|
|
|
}
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
ircd::db::database::database(const string_view &name,
|
|
|
|
std::string optstr,
|
|
|
|
description description)
|
|
|
|
:database
|
|
|
|
{
|
|
|
|
namepoint(name).first, namepoint(name).second, std::move(optstr), std::move(description)
|
2017-08-23 23:37:06 +02:00
|
|
|
}
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
2018-05-25 05:01:52 +02:00
|
|
|
ircd::db::database::database(const string_view &name,
|
|
|
|
const uint64_t &checkpoint,
|
2017-08-23 22:37:47 +02:00
|
|
|
std::string optstr,
|
2017-09-19 05:53:45 +02:00
|
|
|
description description)
|
2016-09-24 06:01:57 +02:00
|
|
|
try
|
2017-03-31 00:57:08 +02:00
|
|
|
:name
|
|
|
|
{
|
2018-05-25 05:01:52 +02:00
|
|
|
namepoint(name).first
|
2017-03-31 00:57:08 +02:00
|
|
|
}
|
2018-05-25 05:01:52 +02:00
|
|
|
,checkpoint
|
2017-03-31 00:57:08 +02:00
|
|
|
{
|
2018-05-25 05:01:52 +02:00
|
|
|
// a -1 may have been generated by the db::namepoint() util when the user
|
|
|
|
// supplied just a name without a checkpoint. In the context of database
|
|
|
|
// opening/creation -1 just defaults to 0.
|
|
|
|
checkpoint == uint64_t(-1)? 0 : checkpoint
|
2017-03-31 00:57:08 +02:00
|
|
|
}
|
2017-09-19 04:17:36 +02:00
|
|
|
,optstr
|
|
|
|
{
|
|
|
|
std::move(optstr)
|
|
|
|
}
|
2018-05-25 05:01:52 +02:00
|
|
|
,path
|
|
|
|
{
|
|
|
|
db::path(this->name, this->checkpoint)
|
|
|
|
}
|
2018-01-18 04:22:35 +01:00
|
|
|
,env
|
|
|
|
{
|
|
|
|
std::make_shared<struct env>(this)
|
|
|
|
}
|
2017-03-31 00:57:08 +02:00
|
|
|
,logs
|
|
|
|
{
|
|
|
|
std::make_shared<struct logs>(this)
|
|
|
|
}
|
|
|
|
,stats
|
|
|
|
{
|
|
|
|
std::make_shared<struct stats>(this)
|
|
|
|
}
|
|
|
|
,events
|
|
|
|
{
|
|
|
|
std::make_shared<struct events>(this)
|
|
|
|
}
|
|
|
|
,mergeop
|
|
|
|
{
|
|
|
|
std::make_shared<struct mergeop>(this)
|
|
|
|
}
|
2018-04-21 00:45:57 +02:00
|
|
|
,ssts
|
|
|
|
{
|
2018-07-05 01:47:11 +02:00
|
|
|
//rocksdb::NewSstFileManager(env.get(), logs, {}, 0, true, nullptr, 0.05)
|
2018-04-21 00:45:57 +02:00
|
|
|
}
|
2017-08-23 23:37:06 +02:00
|
|
|
,cache{[this]
|
|
|
|
() -> std::shared_ptr<rocksdb::Cache>
|
2017-03-31 00:57:08 +02:00
|
|
|
{
|
2018-04-15 08:00:56 +02:00
|
|
|
//TODO: conf
|
2018-05-15 03:05:43 +02:00
|
|
|
const auto lru_cache_size{16_MiB};
|
2017-04-03 06:02:32 +02:00
|
|
|
return rocksdb::NewLRUCache(lru_cache_size);
|
2017-03-24 02:36:49 +01:00
|
|
|
}()}
|
2017-11-16 02:29:54 +01:00
|
|
|
,descriptors
|
2017-03-24 02:36:49 +01:00
|
|
|
{
|
2017-11-16 02:29:54 +01:00
|
|
|
std::move(description)
|
|
|
|
}
|
|
|
|
,column_names{[this]
|
|
|
|
{
|
|
|
|
// Existing columns at path. If any are left the descriptor set did not
|
|
|
|
// describe all of the columns found in the database at path.
|
2018-04-03 02:18:02 +02:00
|
|
|
const auto opts
|
|
|
|
{
|
2018-04-09 20:51:36 +02:00
|
|
|
make_dbopts(this->optstr)
|
2018-04-03 02:18:02 +02:00
|
|
|
};
|
|
|
|
|
|
|
|
const auto required
|
|
|
|
{
|
|
|
|
db::column_names(path, opts)
|
|
|
|
};
|
|
|
|
|
|
|
|
std::set<string_view> existing
|
|
|
|
{
|
|
|
|
begin(required), end(required)
|
|
|
|
};
|
2017-08-23 22:37:47 +02:00
|
|
|
|
2017-11-16 02:29:54 +01:00
|
|
|
// The names of the columns extracted from the descriptor set
|
|
|
|
std::vector<string_view> ret(descriptors.size());
|
|
|
|
std::transform(begin(descriptors), end(descriptors), begin(ret), [&existing]
|
|
|
|
(const auto &descriptor) -> string_view
|
2017-03-31 00:57:08 +02:00
|
|
|
{
|
2017-09-19 04:17:36 +02:00
|
|
|
existing.erase(descriptor.name);
|
2017-11-16 02:29:54 +01:00
|
|
|
return descriptor.name;
|
|
|
|
});
|
2017-09-19 04:17:36 +02:00
|
|
|
|
|
|
|
for(const auto &remain : existing)
|
2018-04-03 02:18:02 +02:00
|
|
|
throw error
|
|
|
|
{
|
|
|
|
"Failed to describe existing column '%s' (and %zd others...)",
|
|
|
|
remain,
|
|
|
|
existing.size() - 1
|
|
|
|
};
|
|
|
|
|
2017-11-16 02:29:54 +01:00
|
|
|
return ret;
|
|
|
|
}()}
|
|
|
|
,column_index{[this]
|
|
|
|
{
|
|
|
|
decltype(this->column_index) ret;
|
|
|
|
for(const auto &descriptor : this->descriptors)
|
|
|
|
ret.emplace(descriptor.name, -1);
|
2017-09-19 04:17:36 +02:00
|
|
|
|
|
|
|
return ret;
|
|
|
|
}()}
|
2017-11-16 02:29:54 +01:00
|
|
|
,d{[this]
|
2017-09-19 04:17:36 +02:00
|
|
|
{
|
|
|
|
bool fsck{false};
|
|
|
|
bool read_only{false};
|
2018-04-03 02:18:02 +02:00
|
|
|
auto opts
|
|
|
|
{
|
2018-04-09 20:51:36 +02:00
|
|
|
make_dbopts(this->optstr, &this->optstr, &read_only, &fsck)
|
2018-04-03 02:18:02 +02:00
|
|
|
};
|
2017-03-31 00:57:08 +02:00
|
|
|
|
2017-04-03 06:02:32 +02:00
|
|
|
// Setup sundry
|
2017-03-31 00:57:08 +02:00
|
|
|
opts.create_if_missing = true;
|
|
|
|
opts.create_missing_column_families = true;
|
2017-04-03 06:02:32 +02:00
|
|
|
opts.max_file_opening_threads = 0;
|
2018-04-13 07:46:33 +02:00
|
|
|
opts.stats_dump_period_sec = 0;
|
|
|
|
opts.enable_thread_tracking = true;
|
2018-07-05 01:47:11 +02:00
|
|
|
opts.avoid_flush_during_recovery = true;
|
2018-04-21 00:48:29 +02:00
|
|
|
opts.delete_obsolete_files_period_micros = 0;
|
|
|
|
opts.max_background_jobs = 2;
|
|
|
|
opts.max_background_flushes = 1;
|
|
|
|
opts.max_background_compactions = 1;
|
|
|
|
opts.max_subcompactions = 1;
|
|
|
|
opts.max_open_files = -1; //ircd::info::rlimit_nofile / 4;
|
2018-04-09 21:01:58 +02:00
|
|
|
//opts.allow_concurrent_memtable_write = true;
|
|
|
|
//opts.enable_write_thread_adaptive_yield = false;
|
2017-08-23 22:37:47 +02:00
|
|
|
//opts.use_fsync = true;
|
2017-03-31 00:57:08 +02:00
|
|
|
|
2018-04-13 07:46:33 +02:00
|
|
|
#ifdef RB_DEBUG
|
|
|
|
opts.dump_malloc_stats = true;
|
|
|
|
#endif
|
|
|
|
|
2018-01-18 04:22:35 +01:00
|
|
|
// Setup env
|
|
|
|
opts.env = env.get();
|
|
|
|
|
2018-04-21 00:45:57 +02:00
|
|
|
// Setup SST file mgmt
|
|
|
|
opts.sst_file_manager = this->ssts;
|
|
|
|
|
2017-03-24 02:36:49 +01:00
|
|
|
// Setup logging
|
|
|
|
logs->SetInfoLogLevel(ircd::debugmode? rocksdb::DEBUG_LEVEL : rocksdb::WARN_LEVEL);
|
2017-03-31 00:57:08 +02:00
|
|
|
opts.info_log_level = logs->GetInfoLogLevel();
|
|
|
|
opts.info_log = logs;
|
2017-03-24 02:36:49 +01:00
|
|
|
|
2017-03-23 22:58:24 +01:00
|
|
|
// Setup event and statistics callbacks
|
2017-03-31 00:57:08 +02:00
|
|
|
opts.listeners.emplace_back(this->events);
|
2018-04-15 09:24:38 +02:00
|
|
|
|
|
|
|
// Setup histogram collecting
|
|
|
|
//this->stats->stats_level_ = rocksdb::kAll;
|
|
|
|
this->stats->stats_level_ = rocksdb::kExceptTimeForMutex;
|
|
|
|
opts.statistics = this->stats;
|
2017-03-23 22:58:24 +01:00
|
|
|
|
|
|
|
// Setup performance metric options
|
|
|
|
//rocksdb::SetPerfLevel(rocksdb::PerfLevel::kDisable);
|
|
|
|
|
2018-07-07 01:53:10 +02:00
|
|
|
// Default corruption tolerance is zero-tolerance; db fails to open with
|
|
|
|
// error by default to inform the user. The rest of the options are
|
|
|
|
// various relaxations for how to proceed.
|
2017-11-30 20:04:25 +01:00
|
|
|
opts.wal_recovery_mode = rocksdb::WALRecoveryMode::kAbsoluteConsistency;
|
2018-07-07 01:53:10 +02:00
|
|
|
|
|
|
|
// When corrupted after crash, the DB is rolled back before the first
|
|
|
|
// corruption and erases everything after it, giving a consistent
|
|
|
|
// state up at that point, though losing some recent data.
|
|
|
|
if(ircd::pitrecdb) //TODO: no global?
|
|
|
|
opts.wal_recovery_mode = rocksdb::WALRecoveryMode::kPointInTimeRecovery;
|
|
|
|
|
|
|
|
// Skipping corrupted records will create gaps in the DB timeline where the
|
|
|
|
// application (like a matrix timeline) cannot tolerate the unexpected gap.
|
2018-04-09 21:01:58 +02:00
|
|
|
//opts.wal_recovery_mode = rocksdb::WALRecoveryMode::kSkipAnyCorruptedRecords;
|
2017-04-03 06:02:32 +02:00
|
|
|
|
2018-07-07 01:53:10 +02:00
|
|
|
// Tolerating corrupted records is very last-ditch for getting the database to
|
|
|
|
// open in a catastrophe. We have no use for this option but should use it for
|
|
|
|
//TODO: emergency salvage-mode.
|
|
|
|
//opts.wal_recovery_mode = rocksdb::WALRecoveryMode::kTolerateCorruptedTailRecords;
|
|
|
|
|
2017-04-03 06:02:32 +02:00
|
|
|
// Setup cache
|
|
|
|
opts.row_cache = this->cache;
|
2017-03-24 02:36:49 +01:00
|
|
|
|
|
|
|
// Setup column families
|
2017-11-16 02:29:54 +01:00
|
|
|
for(const auto &desc : descriptors)
|
2017-03-31 00:57:08 +02:00
|
|
|
{
|
2017-09-19 04:17:36 +02:00
|
|
|
const auto c
|
|
|
|
{
|
|
|
|
std::make_shared<column>(this, desc)
|
|
|
|
};
|
2017-03-31 00:57:08 +02:00
|
|
|
|
2017-09-19 04:17:36 +02:00
|
|
|
columns.emplace_back(c);
|
|
|
|
}
|
2017-03-24 02:36:49 +01:00
|
|
|
|
2017-03-31 00:57:08 +02:00
|
|
|
std::vector<rocksdb::ColumnFamilyHandle *> handles;
|
|
|
|
std::vector<rocksdb::ColumnFamilyDescriptor> columns(this->columns.size());
|
|
|
|
std::transform(begin(this->columns), end(this->columns), begin(columns), []
|
2017-09-19 04:17:36 +02:00
|
|
|
(const auto &column)
|
2017-03-31 00:57:08 +02:00
|
|
|
{
|
2017-09-19 04:17:36 +02:00
|
|
|
return static_cast<const rocksdb::ColumnFamilyDescriptor &>(*column);
|
2017-03-31 00:57:08 +02:00
|
|
|
});
|
2017-08-23 22:37:47 +02:00
|
|
|
|
2018-04-09 21:01:58 +02:00
|
|
|
// NOTE: rocksdb sez RepairDB is broken; can't use now
|
2017-08-23 22:37:47 +02:00
|
|
|
if(fsck && fs::is_dir(path))
|
2017-04-03 06:02:32 +02:00
|
|
|
{
|
2018-05-29 10:09:15 +02:00
|
|
|
log::notice
|
|
|
|
{
|
|
|
|
log, "Checking database @ `%s' columns[%zu]", path, columns.size()
|
|
|
|
};
|
2017-08-23 22:37:47 +02:00
|
|
|
|
|
|
|
throw_on_error
|
|
|
|
{
|
|
|
|
rocksdb::RepairDB(path, opts, columns)
|
|
|
|
};
|
|
|
|
|
2018-05-29 10:09:15 +02:00
|
|
|
log::info
|
|
|
|
{
|
|
|
|
log, "Database @ `%s' check complete", path
|
|
|
|
};
|
2017-04-03 06:02:32 +02:00
|
|
|
}
|
2017-08-23 22:37:47 +02:00
|
|
|
|
2018-04-03 02:19:01 +02:00
|
|
|
// If the directory does not exist, though rocksdb will create it, we can
|
|
|
|
// avoid scaring the user with an error log message if we just do that..
|
|
|
|
if(opts.create_if_missing && !fs::is_dir(path))
|
|
|
|
fs::mkdir(path);
|
|
|
|
|
2017-04-03 06:02:32 +02:00
|
|
|
// Announce attempt before usual point where exceptions are thrown
|
2018-05-29 10:09:15 +02:00
|
|
|
log::info
|
|
|
|
{
|
|
|
|
log, "Opening database \"%s\" @ `%s' with %zu columns...",
|
|
|
|
this->name,
|
|
|
|
path,
|
|
|
|
columns.size()
|
|
|
|
};
|
2017-04-03 06:02:32 +02:00
|
|
|
|
2017-09-20 06:40:53 +02:00
|
|
|
// Open DB into ptr
|
|
|
|
rocksdb::DB *ptr;
|
2017-08-23 22:37:47 +02:00
|
|
|
if(read_only)
|
|
|
|
throw_on_error
|
|
|
|
{
|
|
|
|
rocksdb::DB::OpenForReadOnly(opts, path, columns, &handles, &ptr)
|
|
|
|
};
|
|
|
|
else
|
|
|
|
throw_on_error
|
|
|
|
{
|
|
|
|
rocksdb::DB::Open(opts, path, columns, &handles, &ptr)
|
|
|
|
};
|
2017-03-31 00:57:08 +02:00
|
|
|
|
2018-04-27 01:39:59 +02:00
|
|
|
std::unique_ptr<rocksdb::DB> ret
|
2017-09-19 04:17:36 +02:00
|
|
|
{
|
2018-04-27 01:39:59 +02:00
|
|
|
ptr
|
|
|
|
};
|
|
|
|
|
|
|
|
for(const auto &handle : handles)
|
2017-11-16 02:29:54 +01:00
|
|
|
{
|
2018-04-27 01:39:59 +02:00
|
|
|
this->columns.at(handle->GetID())->handle.reset(handle);
|
|
|
|
this->column_index.at(handle->GetName()) = handle->GetID();
|
2017-11-16 02:29:54 +01:00
|
|
|
}
|
2017-03-24 02:36:49 +01:00
|
|
|
|
2018-04-27 01:39:59 +02:00
|
|
|
for(size_t i(0); i < this->columns.size(); ++i)
|
|
|
|
if(db::id(*this->columns[i]) != i)
|
|
|
|
throw error
|
|
|
|
{
|
|
|
|
"Columns misaligned: expecting id[%zd] got id[%u] '%s'",
|
|
|
|
i,
|
|
|
|
db::id(*this->columns[i]),
|
|
|
|
db::name(*this->columns[i])
|
|
|
|
};
|
|
|
|
|
|
|
|
return ret;
|
2017-03-24 02:36:49 +01:00
|
|
|
}()}
|
2018-05-23 02:01:01 +02:00
|
|
|
,uuid{[this]
|
|
|
|
{
|
|
|
|
std::string ret;
|
|
|
|
throw_on_error
|
|
|
|
{
|
|
|
|
d->GetDbIdentity(ret)
|
|
|
|
};
|
|
|
|
|
|
|
|
return ret;
|
|
|
|
}()}
|
2018-05-25 05:01:52 +02:00
|
|
|
,checkpointer{[this]
|
2018-03-23 03:56:08 +01:00
|
|
|
{
|
|
|
|
rocksdb::Checkpoint *checkpointer{nullptr};
|
|
|
|
throw_on_error
|
|
|
|
{
|
|
|
|
rocksdb::Checkpoint::Create(this->d.get(), &checkpointer)
|
|
|
|
};
|
|
|
|
|
|
|
|
return checkpointer;
|
|
|
|
}()}
|
2017-03-24 02:36:49 +01:00
|
|
|
{
|
2018-05-31 21:47:37 +02:00
|
|
|
if(ircd::checkdb)
|
2018-04-27 01:48:35 +02:00
|
|
|
{
|
2018-05-31 21:47:37 +02:00
|
|
|
log::notice
|
|
|
|
{
|
|
|
|
log, "'%s': Verifying database integrity. This may take several minutes...",
|
|
|
|
this->name
|
|
|
|
};
|
|
|
|
|
|
|
|
check(*this);
|
|
|
|
}
|
2018-04-27 01:48:35 +02:00
|
|
|
|
2018-05-29 10:09:15 +02:00
|
|
|
log::info
|
|
|
|
{
|
|
|
|
log, "'%s': Opened database @ `%s' with %zu columns at sequence number %lu.",
|
|
|
|
this->name,
|
|
|
|
path,
|
|
|
|
columns.size(),
|
|
|
|
d->GetLatestSequenceNumber()
|
|
|
|
};
|
2017-03-31 00:57:08 +02:00
|
|
|
}
|
2018-07-07 01:53:10 +02:00
|
|
|
catch(const corruption &e)
|
|
|
|
{
|
|
|
|
throw corruption
|
|
|
|
{
|
|
|
|
"Corruption for '%s' (%s). Try restarting with the -pitrecdb command line option",
|
|
|
|
this->name,
|
|
|
|
e.what()
|
|
|
|
};
|
|
|
|
}
|
2017-03-31 00:57:08 +02:00
|
|
|
catch(const std::exception &e)
|
|
|
|
{
|
2018-04-03 02:18:02 +02:00
|
|
|
throw error
|
|
|
|
{
|
|
|
|
"Failed to open db '%s': %s",
|
|
|
|
this->name,
|
|
|
|
e.what()
|
|
|
|
};
|
2017-03-31 00:57:08 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
ircd::db::database::~database()
|
2018-04-21 01:22:01 +02:00
|
|
|
noexcept try
|
2017-03-31 00:57:08 +02:00
|
|
|
{
|
2018-05-29 10:09:15 +02:00
|
|
|
log::info
|
|
|
|
{
|
|
|
|
log, "'%s': closing database @ `%s'...",
|
|
|
|
name,
|
|
|
|
path
|
|
|
|
};
|
2018-04-13 07:43:42 +02:00
|
|
|
|
2018-02-06 03:52:31 +01:00
|
|
|
rocksdb::CancelAllBackgroundWork(d.get(), true); // true = blocking
|
2018-05-29 10:09:15 +02:00
|
|
|
log::debug
|
|
|
|
{
|
|
|
|
log, "'%s': background_errors: %lu; flushing...",
|
|
|
|
name,
|
|
|
|
property<uint64_t>(*this, rocksdb::DB::Properties::kBackgroundErrors)
|
|
|
|
};
|
2018-01-18 06:44:56 +01:00
|
|
|
|
2018-04-27 02:19:29 +02:00
|
|
|
flush(*this);
|
2018-05-25 05:01:52 +02:00
|
|
|
this->checkpointer.reset(nullptr);
|
2018-04-27 02:19:29 +02:00
|
|
|
this->columns.clear();
|
2018-05-29 10:09:15 +02:00
|
|
|
log::debug
|
|
|
|
{
|
|
|
|
log, "'%s': closed columns; synchronizing...",
|
|
|
|
name
|
|
|
|
};
|
2018-04-27 02:19:29 +02:00
|
|
|
|
2018-02-06 03:52:31 +01:00
|
|
|
sync(*this);
|
2018-05-29 10:09:15 +02:00
|
|
|
log::debug
|
|
|
|
{
|
|
|
|
log, "'%s': synchronized with hardware.",
|
|
|
|
name
|
|
|
|
};
|
2018-04-21 01:22:01 +02:00
|
|
|
|
|
|
|
const auto sequence
|
|
|
|
{
|
|
|
|
d->GetLatestSequenceNumber()
|
|
|
|
};
|
|
|
|
|
|
|
|
throw_on_error
|
|
|
|
{
|
|
|
|
d->Close()
|
|
|
|
};
|
|
|
|
|
2018-05-29 10:09:15 +02:00
|
|
|
log::info
|
|
|
|
{
|
|
|
|
log, "'%s': closed database @ `%s' at sequence number %lu.",
|
|
|
|
name,
|
|
|
|
path,
|
|
|
|
sequence
|
|
|
|
};
|
2018-04-21 01:22:01 +02:00
|
|
|
}
|
|
|
|
catch(const std::exception &e)
|
|
|
|
{
|
|
|
|
log::error
|
|
|
|
{
|
|
|
|
"'%s': Error closing database @ '%s' :%s",
|
|
|
|
name,
|
|
|
|
path,
|
|
|
|
e.what()
|
|
|
|
};
|
|
|
|
|
|
|
|
return;
|
2017-03-31 00:57:08 +02:00
|
|
|
}
|
|
|
|
|
2017-08-30 23:07:36 +02:00
|
|
|
void
|
|
|
|
ircd::db::database::operator()(const delta &delta)
|
|
|
|
{
|
|
|
|
operator()(sopts{}, delta);
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
ircd::db::database::operator()(const std::initializer_list<delta> &deltas)
|
|
|
|
{
|
|
|
|
operator()(sopts{}, deltas);
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
ircd::db::database::operator()(const delta *const &begin,
|
|
|
|
const delta *const &end)
|
|
|
|
{
|
|
|
|
operator()(sopts{}, begin, end);
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
ircd::db::database::operator()(const sopts &sopts,
|
|
|
|
const delta &delta)
|
|
|
|
{
|
|
|
|
operator()(sopts, &delta, &delta + 1);
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
ircd::db::database::operator()(const sopts &sopts,
|
|
|
|
const std::initializer_list<delta> &deltas)
|
|
|
|
{
|
|
|
|
operator()(sopts, std::begin(deltas), std::end(deltas));
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
ircd::db::database::operator()(const sopts &sopts,
|
|
|
|
const delta *const &begin,
|
|
|
|
const delta *const &end)
|
|
|
|
{
|
|
|
|
rocksdb::WriteBatch batch;
|
|
|
|
std::for_each(begin, end, [this, &batch]
|
|
|
|
(const delta &delta)
|
|
|
|
{
|
|
|
|
const auto &op(std::get<op>(delta));
|
|
|
|
const auto &col(std::get<1>(delta));
|
|
|
|
const auto &key(std::get<2>(delta));
|
|
|
|
const auto &val(std::get<3>(delta));
|
|
|
|
db::column column(operator[](col));
|
|
|
|
append(batch, column, db::column::delta
|
|
|
|
{
|
|
|
|
op,
|
|
|
|
key,
|
|
|
|
val
|
|
|
|
});
|
|
|
|
});
|
|
|
|
|
2017-09-08 11:14:17 +02:00
|
|
|
commit(*this, batch, sopts);
|
2017-08-30 23:07:36 +02:00
|
|
|
}
|
|
|
|
|
2017-03-31 00:57:08 +02:00
|
|
|
ircd::db::database::column &
|
|
|
|
ircd::db::database::operator[](const string_view &name)
|
2017-09-19 04:17:36 +02:00
|
|
|
{
|
2017-11-16 02:29:54 +01:00
|
|
|
const auto it{column_index.find(name)};
|
|
|
|
if(unlikely(it == std::end(column_index)))
|
2018-05-29 10:09:15 +02:00
|
|
|
throw schema_error
|
|
|
|
{
|
|
|
|
"'%s': column '%s' is not available or specified in schema",
|
|
|
|
this->name,
|
|
|
|
name
|
|
|
|
};
|
2017-09-19 04:17:36 +02:00
|
|
|
|
|
|
|
return operator[](it->second);
|
|
|
|
}
|
|
|
|
|
|
|
|
ircd::db::database::column &
|
|
|
|
ircd::db::database::operator[](const uint32_t &id)
|
2017-04-03 06:02:32 +02:00
|
|
|
try
|
2017-03-31 00:57:08 +02:00
|
|
|
{
|
2017-09-19 04:17:36 +02:00
|
|
|
return *columns.at(id);
|
2017-03-31 00:57:08 +02:00
|
|
|
}
|
2017-04-03 06:02:32 +02:00
|
|
|
catch(const std::out_of_range &e)
|
|
|
|
{
|
2018-05-29 10:09:15 +02:00
|
|
|
throw schema_error
|
|
|
|
{
|
|
|
|
"'%s': column id[%u] is not available or specified in schema",
|
|
|
|
this->name,
|
|
|
|
id
|
|
|
|
};
|
2017-04-03 06:02:32 +02:00
|
|
|
}
|
2017-03-31 00:57:08 +02:00
|
|
|
|
|
|
|
const ircd::db::database::column &
|
|
|
|
ircd::db::database::operator[](const string_view &name)
|
2017-09-19 04:17:36 +02:00
|
|
|
const
|
|
|
|
{
|
2017-11-16 02:29:54 +01:00
|
|
|
const auto it{column_index.find(name)};
|
|
|
|
if(unlikely(it == std::end(column_index)))
|
2018-05-29 10:09:15 +02:00
|
|
|
throw schema_error
|
|
|
|
{
|
|
|
|
"'%s': column '%s' is not available or specified in schema",
|
|
|
|
this->name,
|
|
|
|
name
|
|
|
|
};
|
2017-09-19 04:17:36 +02:00
|
|
|
|
|
|
|
return operator[](it->second);
|
|
|
|
}
|
|
|
|
|
|
|
|
const ircd::db::database::column &
|
|
|
|
ircd::db::database::operator[](const uint32_t &id)
|
2017-04-03 06:02:32 +02:00
|
|
|
const try
|
2017-03-31 00:57:08 +02:00
|
|
|
{
|
2017-09-19 04:17:36 +02:00
|
|
|
return *columns.at(id);
|
2017-03-31 00:57:08 +02:00
|
|
|
}
|
2017-04-03 06:02:32 +02:00
|
|
|
catch(const std::out_of_range &e)
|
|
|
|
{
|
2018-05-29 10:09:15 +02:00
|
|
|
throw schema_error
|
|
|
|
{
|
|
|
|
"'%s': column id[%u] is not available or specified in schema",
|
|
|
|
this->name,
|
|
|
|
id
|
|
|
|
};
|
2017-04-03 06:02:32 +02:00
|
|
|
}
|
2017-03-31 00:57:08 +02:00
|
|
|
|
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
//
|
|
|
|
// database::comparator
|
|
|
|
//
|
|
|
|
|
|
|
|
const char *
|
|
|
|
ircd::db::database::comparator::Name()
|
2018-01-18 09:49:23 +01:00
|
|
|
const noexcept
|
2017-03-31 00:57:08 +02:00
|
|
|
{
|
|
|
|
assert(!user.name.empty());
|
2017-09-23 03:48:35 +02:00
|
|
|
return user.name.data();
|
2017-03-31 00:57:08 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
bool
|
|
|
|
ircd::db::database::comparator::Equal(const Slice &a,
|
|
|
|
const Slice &b)
|
2018-01-18 09:49:23 +01:00
|
|
|
const noexcept
|
2017-03-31 00:57:08 +02:00
|
|
|
{
|
2018-04-10 01:46:13 +02:00
|
|
|
return user.equal?
|
|
|
|
user.equal(slice(a), slice(b)):
|
|
|
|
Compare(a, b) == 0;
|
2017-03-31 00:57:08 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
int
|
|
|
|
ircd::db::database::comparator::Compare(const Slice &a,
|
|
|
|
const Slice &b)
|
2018-01-18 09:49:23 +01:00
|
|
|
const noexcept
|
2017-03-31 00:57:08 +02:00
|
|
|
{
|
|
|
|
assert(bool(user.less));
|
2017-09-23 03:48:35 +02:00
|
|
|
const auto sa{slice(a)};
|
|
|
|
const auto sb{slice(b)};
|
|
|
|
return user.less(sa, sb)? -1: // less[Y], equal[?], greater[?]
|
|
|
|
user.equal && user.equal(sa, sb)? 0: // less[N], equal[Y], greater[?]
|
|
|
|
user.equal? 1: // less[N], equal[N], greater[Y]
|
|
|
|
user.less(sb, sa)? 1: // less[N], equal[?], greater[Y]
|
|
|
|
0; // less[N], equal[Y], greater[N]
|
2017-03-31 00:57:08 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
void
|
2018-05-22 04:52:18 +02:00
|
|
|
ircd::db::database::comparator::FindShortestSeparator(std::string *const key,
|
|
|
|
const Slice &limit)
|
2018-01-18 09:49:23 +01:00
|
|
|
const noexcept
|
2017-03-31 00:57:08 +02:00
|
|
|
{
|
2018-05-22 04:52:18 +02:00
|
|
|
assert(key != nullptr);
|
|
|
|
if(user.separator)
|
|
|
|
user.separator(*key, slice(limit));
|
2017-03-31 00:57:08 +02:00
|
|
|
}
|
2017-03-24 02:36:49 +01:00
|
|
|
|
2017-03-31 00:57:08 +02:00
|
|
|
void
|
2018-05-22 04:52:18 +02:00
|
|
|
ircd::db::database::comparator::FindShortSuccessor(std::string *const key)
|
2018-01-18 09:49:23 +01:00
|
|
|
const noexcept
|
2017-03-31 00:57:08 +02:00
|
|
|
{
|
2018-05-22 04:52:18 +02:00
|
|
|
assert(key != nullptr);
|
|
|
|
if(user.successor)
|
|
|
|
user.successor(*key);
|
2017-03-24 02:36:49 +01:00
|
|
|
}
|
2017-03-31 00:57:08 +02:00
|
|
|
|
2017-08-31 07:12:58 +02:00
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
//
|
|
|
|
// database::prefix_transform
|
|
|
|
//
|
|
|
|
|
|
|
|
const char *
|
|
|
|
ircd::db::database::prefix_transform::Name()
|
2018-01-18 09:49:23 +01:00
|
|
|
const noexcept
|
2017-08-31 07:12:58 +02:00
|
|
|
{
|
|
|
|
assert(!user.name.empty());
|
|
|
|
return user.name.c_str();
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Slice
|
|
|
|
ircd::db::database::prefix_transform::Transform(const Slice &key)
|
2018-01-18 09:49:23 +01:00
|
|
|
const noexcept
|
2017-08-31 07:12:58 +02:00
|
|
|
{
|
|
|
|
assert(bool(user.get));
|
2017-09-23 09:28:34 +02:00
|
|
|
return slice(user.get(slice(key)));
|
2017-08-31 07:12:58 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
bool
|
|
|
|
ircd::db::database::prefix_transform::InRange(const Slice &key)
|
2018-01-18 09:49:23 +01:00
|
|
|
const noexcept
|
2017-08-31 07:12:58 +02:00
|
|
|
{
|
|
|
|
return InDomain(key);
|
|
|
|
}
|
|
|
|
|
|
|
|
bool
|
|
|
|
ircd::db::database::prefix_transform::InDomain(const Slice &key)
|
2018-01-18 09:49:23 +01:00
|
|
|
const noexcept
|
2017-08-31 07:12:58 +02:00
|
|
|
{
|
|
|
|
assert(bool(user.has));
|
2017-09-23 09:28:34 +02:00
|
|
|
return user.has(slice(key));
|
2017-08-31 07:12:58 +02:00
|
|
|
}
|
|
|
|
|
2017-03-31 00:57:08 +02:00
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
//
|
|
|
|
// database::column
|
|
|
|
//
|
2017-03-24 02:36:49 +01:00
|
|
|
|
2017-09-19 04:17:36 +02:00
|
|
|
void
|
|
|
|
ircd::db::drop(database::column &c)
|
|
|
|
{
|
|
|
|
if(!c.handle)
|
|
|
|
return;
|
|
|
|
|
|
|
|
throw_on_error
|
|
|
|
{
|
|
|
|
c.d->d->DropColumnFamily(c.handle.get())
|
|
|
|
};
|
|
|
|
}
|
|
|
|
|
|
|
|
uint32_t
|
|
|
|
ircd::db::id(const database::column &c)
|
|
|
|
{
|
|
|
|
if(!c.handle)
|
|
|
|
return -1;
|
|
|
|
|
|
|
|
return c.handle->GetID();
|
|
|
|
}
|
|
|
|
|
|
|
|
const std::string &
|
|
|
|
ircd::db::name(const database::column &c)
|
|
|
|
{
|
|
|
|
return c.name;
|
|
|
|
}
|
|
|
|
|
2017-09-21 05:38:39 +02:00
|
|
|
const ircd::db::database::descriptor &
|
|
|
|
ircd::db::describe(const database::column &c)
|
|
|
|
{
|
|
|
|
return c.descriptor;
|
|
|
|
}
|
|
|
|
|
2017-09-19 04:17:36 +02:00
|
|
|
//
|
|
|
|
// database::column
|
|
|
|
//
|
|
|
|
|
|
|
|
ircd::db::database::column::column(database *const &d,
|
|
|
|
const database::descriptor &descriptor)
|
|
|
|
:rocksdb::ColumnFamilyDescriptor
|
|
|
|
(
|
|
|
|
descriptor.name, database::options{descriptor.options}
|
|
|
|
)
|
2017-03-31 00:57:08 +02:00
|
|
|
,d{d}
|
2017-09-08 10:33:41 +02:00
|
|
|
,key_type{descriptor.type.first}
|
|
|
|
,mapped_type{descriptor.type.second}
|
2017-09-19 04:17:36 +02:00
|
|
|
,descriptor{descriptor}
|
2017-09-08 10:33:41 +02:00
|
|
|
,cmp{d, this->descriptor.cmp}
|
|
|
|
,prefix{d, this->descriptor.prefix}
|
2017-04-03 06:02:32 +02:00
|
|
|
,handle
|
2017-03-24 02:36:49 +01:00
|
|
|
{
|
2017-04-03 06:02:32 +02:00
|
|
|
nullptr, [this](rocksdb::ColumnFamilyHandle *const handle)
|
|
|
|
{
|
|
|
|
if(handle)
|
|
|
|
this->d->d->DestroyColumnFamilyHandle(handle);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
{
|
2018-04-18 00:28:08 +02:00
|
|
|
// If possible, deduce comparator based on type given in descriptor
|
2017-09-08 10:33:41 +02:00
|
|
|
if(!this->descriptor.cmp.less)
|
2017-03-31 00:57:08 +02:00
|
|
|
{
|
|
|
|
if(key_type == typeid(string_view))
|
2017-09-23 03:48:35 +02:00
|
|
|
this->cmp.user = cmp_string_view{};
|
2017-03-31 00:57:08 +02:00
|
|
|
else if(key_type == typeid(int64_t))
|
2017-09-23 03:48:35 +02:00
|
|
|
this->cmp.user = cmp_int64_t{};
|
2018-04-18 00:28:08 +02:00
|
|
|
else if(key_type == typeid(uint64_t))
|
|
|
|
this->cmp.user = cmp_uint64_t{};
|
2017-03-31 00:57:08 +02:00
|
|
|
else
|
2018-04-18 00:28:08 +02:00
|
|
|
throw error
|
|
|
|
{
|
|
|
|
"column '%s' key type[%s] requires user supplied comparator",
|
|
|
|
this->name,
|
|
|
|
key_type.name()
|
|
|
|
};
|
2017-03-31 00:57:08 +02:00
|
|
|
}
|
|
|
|
|
2017-09-08 10:33:41 +02:00
|
|
|
// Set the key comparator
|
2017-03-31 00:57:08 +02:00
|
|
|
this->options.comparator = &this->cmp;
|
|
|
|
|
2017-08-31 07:12:58 +02:00
|
|
|
// Set the prefix extractor
|
|
|
|
if(this->prefix.user.get && this->prefix.user.has)
|
|
|
|
this->options.prefix_extractor = std::shared_ptr<const rocksdb::SliceTransform>
|
|
|
|
{
|
|
|
|
&this->prefix, [](const rocksdb::SliceTransform *) {}
|
|
|
|
};
|
2017-08-23 22:37:47 +02:00
|
|
|
|
2018-04-18 00:28:08 +02:00
|
|
|
//
|
|
|
|
// Table options
|
|
|
|
//
|
2017-03-31 00:57:08 +02:00
|
|
|
|
2018-04-18 00:28:08 +02:00
|
|
|
// Setup the cache for assets.
|
2018-04-15 08:00:56 +02:00
|
|
|
const auto &cache_size(this->descriptor.cache_size);
|
2018-04-18 00:28:08 +02:00
|
|
|
if(cache_size)
|
|
|
|
table_opts.block_cache = rocksdb::NewLRUCache(cache_size);
|
2018-04-15 08:00:56 +02:00
|
|
|
|
2018-04-18 00:28:08 +02:00
|
|
|
// Setup the cache for compressed assets.
|
2018-04-15 08:00:56 +02:00
|
|
|
const auto &cache_size_comp(this->descriptor.cache_size_comp);
|
2018-04-18 00:28:08 +02:00
|
|
|
if(cache_size_comp)
|
|
|
|
table_opts.block_cache_compressed = rocksdb::NewLRUCache(cache_size_comp);
|
2018-04-15 08:00:56 +02:00
|
|
|
|
2018-04-18 00:28:08 +02:00
|
|
|
// Setup the bloom filter.
|
2018-04-15 08:00:56 +02:00
|
|
|
const auto &bloom_bits(this->descriptor.bloom_bits);
|
|
|
|
if(bloom_bits)
|
|
|
|
table_opts.filter_policy.reset(rocksdb::NewBloomFilterPolicy(bloom_bits, false));
|
|
|
|
|
2018-04-18 00:28:08 +02:00
|
|
|
// Tickers::READ_AMP_TOTAL_READ_BYTES / Tickers::READ_AMP_ESTIMATE_USEFUL_BYTES
|
|
|
|
//table_opts.read_amp_bytes_per_bit = 8;
|
|
|
|
|
2018-04-15 08:00:56 +02:00
|
|
|
this->options.table_factory.reset(rocksdb::NewBlockBasedTableFactory(table_opts));
|
|
|
|
|
2018-04-18 00:28:08 +02:00
|
|
|
//
|
|
|
|
// Misc options
|
|
|
|
//
|
|
|
|
|
|
|
|
// Set the compaction style; we don't override this in the descriptor yet.
|
|
|
|
this->options.compaction_style = rocksdb::kCompactionStyleLevel;
|
|
|
|
|
|
|
|
// Set the compaction priority; this should probably be in the descriptor
|
|
|
|
// but this is currently selected for the general matrix workload.
|
2018-04-21 00:48:29 +02:00
|
|
|
this->options.compaction_pri = rocksdb::CompactionPri::kOldestSmallestSeqFirst;
|
2018-04-18 00:28:08 +02:00
|
|
|
|
|
|
|
// Set filter reductions for this column. This means we expect a key to exist.
|
|
|
|
this->options.optimize_filters_for_hits = this->descriptor.expect_queries_hit;
|
|
|
|
|
|
|
|
// Compression
|
2018-04-20 22:38:49 +02:00
|
|
|
//TODO: descriptor / conf / detect etc...
|
|
|
|
//this->options.compression = rocksdb::kSnappyCompression;
|
2017-09-23 03:48:35 +02:00
|
|
|
|
2018-04-20 22:38:49 +02:00
|
|
|
//TODO: descriptor / conf
|
|
|
|
this->options.num_levels = 8;
|
2018-04-21 00:48:29 +02:00
|
|
|
this->options.target_file_size_base = 64_MiB;
|
2018-04-20 22:38:49 +02:00
|
|
|
this->options.target_file_size_multiplier = 4; // size at level
|
|
|
|
this->options.level0_file_num_compaction_trigger = 2;
|
|
|
|
|
|
|
|
log.debug("schema '%s' column [%s => %s] cmp[%s] pfx[%s] lru:%zu:%zu bloom:%zu %s",
|
2017-09-23 03:48:35 +02:00
|
|
|
db::name(*d),
|
|
|
|
demangle(key_type.name()),
|
|
|
|
demangle(mapped_type.name()),
|
|
|
|
this->cmp.Name(),
|
|
|
|
this->options.prefix_extractor? this->prefix.Name() : "none",
|
2018-04-15 08:00:56 +02:00
|
|
|
cache_size,
|
|
|
|
cache_size_comp,
|
|
|
|
bloom_bits,
|
2017-09-23 03:48:35 +02:00
|
|
|
this->descriptor.name);
|
2017-03-24 02:36:49 +01:00
|
|
|
}
|
|
|
|
|
2017-03-31 00:57:08 +02:00
|
|
|
ircd::db::database::column::~column()
|
|
|
|
noexcept
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
ircd::db::database::column::operator
|
|
|
|
database &()
|
|
|
|
{
|
|
|
|
return *d;
|
|
|
|
}
|
|
|
|
|
|
|
|
ircd::db::database::column::operator
|
|
|
|
rocksdb::ColumnFamilyHandle *()
|
|
|
|
{
|
2017-04-03 06:02:32 +02:00
|
|
|
return handle.get();
|
2017-03-31 00:57:08 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
ircd::db::database::column::operator
|
|
|
|
const database &()
|
|
|
|
const
|
|
|
|
{
|
|
|
|
return *d;
|
|
|
|
}
|
|
|
|
|
|
|
|
ircd::db::database::column::operator
|
|
|
|
const rocksdb::ColumnFamilyHandle *()
|
|
|
|
const
|
|
|
|
{
|
2017-04-03 06:02:32 +02:00
|
|
|
return handle.get();
|
2017-03-31 00:57:08 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
//
|
|
|
|
// database::snapshot
|
|
|
|
//
|
|
|
|
|
|
|
|
uint64_t
|
|
|
|
ircd::db::sequence(const database::snapshot &s)
|
|
|
|
{
|
|
|
|
const rocksdb::Snapshot *const rs(s);
|
2017-09-23 11:52:33 +02:00
|
|
|
return sequence(rs);
|
|
|
|
}
|
|
|
|
|
|
|
|
uint64_t
|
|
|
|
ircd::db::sequence(const rocksdb::Snapshot *const &rs)
|
|
|
|
{
|
|
|
|
return likely(rs)? rs->GetSequenceNumber() : 0ULL;
|
2017-03-31 00:57:08 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
ircd::db::database::snapshot::snapshot(database &d)
|
2017-08-23 22:37:47 +02:00
|
|
|
:s
|
2017-03-31 00:57:08 +02:00
|
|
|
{
|
2017-08-23 22:37:47 +02:00
|
|
|
d.d->GetSnapshot(),
|
|
|
|
[dp(weak_from(d))](const rocksdb::Snapshot *const s)
|
|
|
|
{
|
|
|
|
if(!s)
|
|
|
|
return;
|
2017-03-31 00:57:08 +02:00
|
|
|
|
2017-08-23 22:37:47 +02:00
|
|
|
const auto d(dp.lock());
|
|
|
|
d->d->ReleaseSnapshot(s);
|
|
|
|
}
|
|
|
|
}
|
2017-03-31 00:57:08 +02:00
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
ircd::db::database::snapshot::~snapshot()
|
2017-03-24 02:36:49 +01:00
|
|
|
noexcept
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
2017-04-03 06:02:32 +02:00
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
//
|
|
|
|
// database::logs
|
|
|
|
//
|
|
|
|
|
2017-03-24 02:36:49 +01:00
|
|
|
static
|
|
|
|
ircd::log::facility
|
|
|
|
translate(const rocksdb::InfoLogLevel &level)
|
|
|
|
{
|
|
|
|
switch(level)
|
|
|
|
{
|
|
|
|
// Treat all infomational messages from rocksdb as debug here for now.
|
|
|
|
// We can clean them up and make better reports for our users eventually.
|
|
|
|
default:
|
|
|
|
case rocksdb::InfoLogLevel::DEBUG_LEVEL: return ircd::log::facility::DEBUG;
|
|
|
|
case rocksdb::InfoLogLevel::INFO_LEVEL: return ircd::log::facility::DEBUG;
|
|
|
|
|
|
|
|
case rocksdb::InfoLogLevel::WARN_LEVEL: return ircd::log::facility::WARNING;
|
|
|
|
case rocksdb::InfoLogLevel::ERROR_LEVEL: return ircd::log::facility::ERROR;
|
|
|
|
case rocksdb::InfoLogLevel::FATAL_LEVEL: return ircd::log::facility::CRITICAL;
|
|
|
|
case rocksdb::InfoLogLevel::HEADER_LEVEL: return ircd::log::facility::NOTICE;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
2017-03-31 00:57:08 +02:00
|
|
|
ircd::db::database::logs::Logv(const char *const fmt,
|
|
|
|
va_list ap)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2017-03-24 02:36:49 +01:00
|
|
|
{
|
|
|
|
Logv(rocksdb::InfoLogLevel::DEBUG_LEVEL, fmt, ap);
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
2017-03-31 00:57:08 +02:00
|
|
|
ircd::db::database::logs::LogHeader(const char *const fmt,
|
|
|
|
va_list ap)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2017-03-24 02:36:49 +01:00
|
|
|
{
|
|
|
|
Logv(rocksdb::InfoLogLevel::DEBUG_LEVEL, fmt, ap);
|
|
|
|
}
|
|
|
|
|
2017-10-01 12:06:48 +02:00
|
|
|
#pragma GCC diagnostic push
|
|
|
|
#pragma GCC diagnostic ignored "-Wsuggest-attribute=format"
|
2017-03-24 02:36:49 +01:00
|
|
|
void
|
2017-03-31 00:57:08 +02:00
|
|
|
ircd::db::database::logs::Logv(const rocksdb::InfoLogLevel level,
|
|
|
|
const char *const fmt,
|
|
|
|
va_list ap)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2017-03-24 02:36:49 +01:00
|
|
|
{
|
|
|
|
if(level < GetInfoLogLevel())
|
|
|
|
return;
|
|
|
|
|
2018-01-12 07:21:51 +01:00
|
|
|
thread_local char buf[1024]; const auto len
|
2017-03-24 02:36:49 +01:00
|
|
|
{
|
2017-08-23 23:37:06 +02:00
|
|
|
vsnprintf(buf, sizeof(buf), fmt, ap)
|
2017-03-24 02:36:49 +01:00
|
|
|
};
|
|
|
|
|
|
|
|
const auto str
|
|
|
|
{
|
|
|
|
// RocksDB adds annoying leading whitespace to attempt to right-justify things and idc
|
2017-12-26 12:00:52 +01:00
|
|
|
lstrip(string_view{buf, size_t(len)}, ' ')
|
2017-03-24 02:36:49 +01:00
|
|
|
};
|
|
|
|
|
2017-08-23 22:37:47 +02:00
|
|
|
// Skip the options for now
|
|
|
|
if(startswith(str, "Options"))
|
|
|
|
return;
|
|
|
|
|
2017-09-30 08:07:55 +02:00
|
|
|
rog(translate(level), "'%s': %s", d->name, str);
|
2017-03-24 02:36:49 +01:00
|
|
|
}
|
2017-10-01 12:06:48 +02:00
|
|
|
#pragma GCC diagnostic pop
|
2017-03-24 02:36:49 +01:00
|
|
|
|
2017-04-03 06:02:32 +02:00
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
//
|
|
|
|
// database::mergeop
|
|
|
|
//
|
|
|
|
|
2017-03-24 02:36:49 +01:00
|
|
|
const char *
|
2017-03-31 00:57:08 +02:00
|
|
|
ircd::db::database::mergeop::Name()
|
2018-01-18 09:49:23 +01:00
|
|
|
const noexcept
|
2017-03-24 02:36:49 +01:00
|
|
|
{
|
|
|
|
return "<unnamed>";
|
|
|
|
}
|
|
|
|
|
|
|
|
bool
|
2017-03-31 00:57:08 +02:00
|
|
|
ircd::db::database::mergeop::Merge(const rocksdb::Slice &_key,
|
|
|
|
const rocksdb::Slice *const _exist,
|
|
|
|
const rocksdb::Slice &_update,
|
|
|
|
std::string *const newval,
|
|
|
|
rocksdb::Logger *const)
|
2018-01-18 09:49:23 +01:00
|
|
|
const noexcept try
|
2017-03-24 02:36:49 +01:00
|
|
|
{
|
|
|
|
const string_view key
|
|
|
|
{
|
|
|
|
_key.data(), _key.size()
|
|
|
|
};
|
|
|
|
|
|
|
|
const string_view exist
|
|
|
|
{
|
|
|
|
_exist? string_view { _exist->data(), _exist->size() } : string_view{}
|
|
|
|
};
|
|
|
|
|
|
|
|
const string_view update
|
|
|
|
{
|
|
|
|
_update.data(), _update.size()
|
|
|
|
};
|
|
|
|
|
|
|
|
if(exist.empty())
|
|
|
|
{
|
|
|
|
*newval = std::string(update);
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
//XXX caching opportunity?
|
|
|
|
*newval = merger(key, {exist, update}); // call the user
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
catch(const std::bad_function_call &e)
|
|
|
|
{
|
|
|
|
log.critical("merge: missing merge operator (%s)", e);
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
catch(const std::exception &e)
|
|
|
|
{
|
|
|
|
log.error("merge: %s", e);
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
2017-04-03 06:02:32 +02:00
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
//
|
2018-05-24 02:03:09 +02:00
|
|
|
// db/stats.h
|
2017-04-03 06:02:32 +02:00
|
|
|
//
|
|
|
|
|
2018-05-24 01:57:49 +02:00
|
|
|
std::string
|
|
|
|
ircd::db::string(const rocksdb::IOStatsContext &ic,
|
|
|
|
const bool &all)
|
|
|
|
{
|
|
|
|
const bool exclude_zeros(!all);
|
|
|
|
return ic.ToString(exclude_zeros);
|
|
|
|
}
|
|
|
|
|
|
|
|
const rocksdb::IOStatsContext &
|
|
|
|
ircd::db::iostats_current()
|
|
|
|
{
|
|
|
|
const auto *const &ret
|
|
|
|
{
|
|
|
|
rocksdb::get_iostats_context()
|
|
|
|
};
|
|
|
|
|
|
|
|
if(unlikely(!ret))
|
|
|
|
throw error
|
|
|
|
{
|
|
|
|
"IO counters are not available on this thread."
|
|
|
|
};
|
|
|
|
|
|
|
|
return *ret;
|
|
|
|
}
|
|
|
|
|
2018-05-24 01:27:44 +02:00
|
|
|
std::string
|
|
|
|
ircd::db::string(const rocksdb::PerfContext &pc,
|
|
|
|
const bool &all)
|
|
|
|
{
|
|
|
|
const bool exclude_zeros(!all);
|
|
|
|
return pc.ToString(exclude_zeros);
|
|
|
|
}
|
|
|
|
|
|
|
|
const rocksdb::PerfContext &
|
|
|
|
ircd::db::perf_current()
|
2017-04-03 06:02:32 +02:00
|
|
|
{
|
2018-05-24 01:27:44 +02:00
|
|
|
const auto *const &ret
|
2018-04-09 21:55:22 +02:00
|
|
|
{
|
|
|
|
rocksdb::get_perf_context()
|
|
|
|
};
|
|
|
|
|
2018-05-24 01:27:44 +02:00
|
|
|
if(unlikely(!ret))
|
|
|
|
throw error
|
|
|
|
{
|
|
|
|
"Performance counters are not available on this thread."
|
|
|
|
};
|
|
|
|
|
|
|
|
return *ret;
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
ircd::db::perf_level(const uint &level)
|
|
|
|
{
|
|
|
|
if(level >= rocksdb::PerfLevel::kOutOfBounds)
|
|
|
|
throw error
|
|
|
|
{
|
|
|
|
"Perf level of '%u' is invalid; maximum is '%u'",
|
|
|
|
level,
|
|
|
|
uint(rocksdb::PerfLevel::kOutOfBounds)
|
|
|
|
};
|
|
|
|
|
|
|
|
rocksdb::SetPerfLevel(rocksdb::PerfLevel(level));
|
|
|
|
}
|
|
|
|
|
|
|
|
uint
|
|
|
|
ircd::db::perf_level()
|
|
|
|
{
|
|
|
|
return rocksdb::GetPerfLevel();
|
2017-04-03 06:02:32 +02:00
|
|
|
}
|
|
|
|
|
2018-04-15 13:45:31 +02:00
|
|
|
uint32_t
|
|
|
|
ircd::db::ticker_id(const string_view &key)
|
|
|
|
{
|
|
|
|
for(const auto &pair : rocksdb::TickersNameMap)
|
|
|
|
if(key == pair.second)
|
|
|
|
return pair.first;
|
|
|
|
|
|
|
|
throw std::out_of_range
|
|
|
|
{
|
|
|
|
"No ticker with that key"
|
|
|
|
};
|
|
|
|
}
|
|
|
|
|
|
|
|
ircd::string_view
|
|
|
|
ircd::db::ticker_id(const uint32_t &id)
|
|
|
|
{
|
|
|
|
for(const auto &pair : rocksdb::TickersNameMap)
|
|
|
|
if(id == pair.first)
|
|
|
|
return pair.second;
|
|
|
|
|
|
|
|
return {};
|
|
|
|
}
|
|
|
|
|
|
|
|
decltype(ircd::db::ticker_max)
|
|
|
|
ircd::db::ticker_max
|
|
|
|
{
|
|
|
|
rocksdb::TICKER_ENUM_MAX
|
|
|
|
};
|
|
|
|
|
|
|
|
uint32_t
|
|
|
|
ircd::db::histogram_id(const string_view &key)
|
|
|
|
{
|
|
|
|
for(const auto &pair : rocksdb::HistogramsNameMap)
|
|
|
|
if(key == pair.second)
|
|
|
|
return pair.first;
|
|
|
|
|
|
|
|
throw std::out_of_range
|
|
|
|
{
|
|
|
|
"No histogram with that key"
|
|
|
|
};
|
|
|
|
}
|
|
|
|
|
|
|
|
ircd::string_view
|
|
|
|
ircd::db::histogram_id(const uint32_t &id)
|
|
|
|
{
|
|
|
|
for(const auto &pair : rocksdb::HistogramsNameMap)
|
|
|
|
if(id == pair.first)
|
|
|
|
return pair.second;
|
|
|
|
|
|
|
|
return {};
|
|
|
|
}
|
|
|
|
|
|
|
|
decltype(ircd::db::histogram_max)
|
|
|
|
ircd::db::histogram_max
|
|
|
|
{
|
|
|
|
rocksdb::HISTOGRAM_ENUM_MAX
|
|
|
|
};
|
|
|
|
|
2018-05-24 02:03:09 +02:00
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
//
|
|
|
|
// database::stats (db/database/stats.h) internal
|
|
|
|
//
|
|
|
|
|
2017-07-18 19:11:00 +02:00
|
|
|
uint64_t
|
|
|
|
ircd::db::database::stats::getAndResetTickerCount(const uint32_t type)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2017-07-18 19:11:00 +02:00
|
|
|
{
|
|
|
|
const auto ret(getTickerCount(type));
|
|
|
|
setTickerCount(type, 0);
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
2017-03-24 02:36:49 +01:00
|
|
|
bool
|
2017-03-31 00:57:08 +02:00
|
|
|
ircd::db::database::stats::HistEnabledForType(const uint32_t type)
|
2018-01-18 09:49:23 +01:00
|
|
|
const noexcept
|
2017-03-24 02:36:49 +01:00
|
|
|
{
|
|
|
|
return type < histogram.size();
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
2017-03-31 00:57:08 +02:00
|
|
|
ircd::db::database::stats::measureTime(const uint32_t type,
|
|
|
|
const uint64_t time)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2017-03-24 02:36:49 +01:00
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
2017-03-31 00:57:08 +02:00
|
|
|
ircd::db::database::stats::histogramData(const uint32_t type,
|
|
|
|
rocksdb::HistogramData *const data)
|
2018-01-18 09:49:23 +01:00
|
|
|
const noexcept
|
2017-03-24 02:36:49 +01:00
|
|
|
{
|
|
|
|
assert(data);
|
|
|
|
|
|
|
|
const auto &median(data->median);
|
|
|
|
const auto &percentile95(data->percentile95);
|
|
|
|
const auto &percentile88(data->percentile99);
|
|
|
|
const auto &average(data->average);
|
|
|
|
const auto &standard_deviation(data->standard_deviation);
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
2017-03-31 00:57:08 +02:00
|
|
|
ircd::db::database::stats::recordTick(const uint32_t type,
|
|
|
|
const uint64_t count)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2017-03-24 02:36:49 +01:00
|
|
|
{
|
|
|
|
ticker.at(type) += count;
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
2017-03-31 00:57:08 +02:00
|
|
|
ircd::db::database::stats::setTickerCount(const uint32_t type,
|
|
|
|
const uint64_t count)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2017-03-24 02:36:49 +01:00
|
|
|
{
|
|
|
|
ticker.at(type) = count;
|
|
|
|
}
|
|
|
|
|
|
|
|
uint64_t
|
2017-03-31 00:57:08 +02:00
|
|
|
ircd::db::database::stats::getTickerCount(const uint32_t type)
|
2018-01-18 09:49:23 +01:00
|
|
|
const noexcept
|
2017-03-24 02:36:49 +01:00
|
|
|
{
|
|
|
|
return ticker.at(type);
|
|
|
|
}
|
|
|
|
|
2018-01-18 03:34:20 +01:00
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
//
|
|
|
|
// database::events
|
|
|
|
//
|
|
|
|
|
2017-03-24 02:36:49 +01:00
|
|
|
void
|
2017-03-31 00:57:08 +02:00
|
|
|
ircd::db::database::events::OnFlushCompleted(rocksdb::DB *const db,
|
|
|
|
const rocksdb::FlushJobInfo &info)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2017-03-24 02:36:49 +01:00
|
|
|
{
|
2017-09-30 08:07:55 +02:00
|
|
|
rog.debug("'%s' @%p: flushed: column[%s] path[%s] tid[%lu] job[%d] writes[slow:%d stop:%d]",
|
2017-03-31 00:57:08 +02:00
|
|
|
d->name,
|
2017-03-24 02:36:49 +01:00
|
|
|
db,
|
|
|
|
info.cf_name,
|
|
|
|
info.file_path,
|
|
|
|
info.thread_id,
|
|
|
|
info.job_id,
|
|
|
|
info.triggered_writes_slowdown,
|
|
|
|
info.triggered_writes_stop);
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
2017-03-31 00:57:08 +02:00
|
|
|
ircd::db::database::events::OnCompactionCompleted(rocksdb::DB *const db,
|
|
|
|
const rocksdb::CompactionJobInfo &info)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2017-03-24 02:36:49 +01:00
|
|
|
{
|
2017-09-30 08:07:55 +02:00
|
|
|
rog.debug("'%s' @%p: compacted: column[%s] status[%d] tid[%lu] job[%d]",
|
2017-03-31 00:57:08 +02:00
|
|
|
d->name,
|
2017-03-24 02:36:49 +01:00
|
|
|
db,
|
|
|
|
info.cf_name,
|
|
|
|
int(info.status.code()),
|
|
|
|
info.thread_id,
|
|
|
|
info.job_id);
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
2017-03-31 00:57:08 +02:00
|
|
|
ircd::db::database::events::OnTableFileDeleted(const rocksdb::TableFileDeletionInfo &info)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2017-03-24 02:36:49 +01:00
|
|
|
{
|
2017-09-30 08:07:55 +02:00
|
|
|
rog.debug("'%s': table file deleted: db[%s] path[%s] status[%d] job[%d]",
|
2017-03-31 00:57:08 +02:00
|
|
|
d->name,
|
2017-03-24 02:36:49 +01:00
|
|
|
info.db_name,
|
|
|
|
info.file_path,
|
|
|
|
int(info.status.code()),
|
|
|
|
info.job_id);
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
2017-03-31 00:57:08 +02:00
|
|
|
ircd::db::database::events::OnTableFileCreated(const rocksdb::TableFileCreationInfo &info)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2017-03-24 02:36:49 +01:00
|
|
|
{
|
2017-09-30 08:07:55 +02:00
|
|
|
rog.debug("'%s': table file created: db[%s] path[%s] status[%d] job[%d]",
|
2017-03-31 00:57:08 +02:00
|
|
|
d->name,
|
2017-03-24 02:36:49 +01:00
|
|
|
info.db_name,
|
|
|
|
info.file_path,
|
|
|
|
int(info.status.code()),
|
|
|
|
info.job_id);
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
2017-03-31 00:57:08 +02:00
|
|
|
ircd::db::database::events::OnTableFileCreationStarted(const rocksdb::TableFileCreationBriefInfo &info)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2017-03-24 02:36:49 +01:00
|
|
|
{
|
2017-09-30 08:07:55 +02:00
|
|
|
rog.debug("'%s': table file creating: db[%s] column[%s] path[%s] job[%d]",
|
2017-03-31 00:57:08 +02:00
|
|
|
d->name,
|
2017-03-24 02:36:49 +01:00
|
|
|
info.db_name,
|
|
|
|
info.cf_name,
|
|
|
|
info.file_path,
|
|
|
|
info.job_id);
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
2017-03-31 00:57:08 +02:00
|
|
|
ircd::db::database::events::OnMemTableSealed(const rocksdb::MemTableInfo &info)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2017-03-24 02:36:49 +01:00
|
|
|
{
|
2017-09-30 08:07:55 +02:00
|
|
|
rog.debug("'%s': memory table sealed: column[%s] entries[%lu] deletes[%lu]",
|
2017-03-31 00:57:08 +02:00
|
|
|
d->name,
|
2017-03-24 02:36:49 +01:00
|
|
|
info.cf_name,
|
|
|
|
info.num_entries,
|
|
|
|
info.num_deletes);
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
2017-03-31 00:57:08 +02:00
|
|
|
ircd::db::database::events::OnColumnFamilyHandleDeletionStarted(rocksdb::ColumnFamilyHandle *const h)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2017-03-24 02:36:49 +01:00
|
|
|
{
|
2017-09-30 08:07:55 +02:00
|
|
|
rog.debug("'%s': column[%s] handle closing @ %p",
|
2017-03-31 00:57:08 +02:00
|
|
|
d->name,
|
|
|
|
h->GetName(),
|
2017-03-24 02:36:49 +01:00
|
|
|
h);
|
|
|
|
}
|
|
|
|
|
2018-01-18 04:22:35 +01:00
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
//
|
|
|
|
// database::env
|
|
|
|
//
|
|
|
|
|
2018-01-18 05:17:00 +01:00
|
|
|
//
|
|
|
|
// env
|
|
|
|
//
|
|
|
|
|
2018-01-18 04:22:35 +01:00
|
|
|
ircd::db::database::env::env(database *const &d)
|
2018-06-09 22:40:21 +02:00
|
|
|
:d{*d},
|
|
|
|
st{std::make_unique<state>(d)}
|
2018-01-18 04:22:35 +01:00
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
ircd::db::database::env::~env()
|
|
|
|
noexcept
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
|
|
|
ircd::db::database::env::NewSequentialFile(const std::string& name,
|
|
|
|
std::unique_ptr<SequentialFile>* r,
|
|
|
|
const EnvOptions& options)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 04:22:35 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-01-18 04:22:35 +01:00
|
|
|
log.debug("'%s': new sequential file '%s' options:%p",
|
|
|
|
d.name,
|
|
|
|
name,
|
|
|
|
&options);
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
2018-01-18 04:22:35 +01:00
|
|
|
|
2018-01-18 06:24:32 +01:00
|
|
|
std::unique_ptr<SequentialFile> defaults;
|
|
|
|
const auto ret
|
|
|
|
{
|
|
|
|
this->defaults.NewSequentialFile(name, &defaults, options)
|
|
|
|
};
|
|
|
|
|
|
|
|
*r = std::make_unique<sequential_file>(&d, name, options, std::move(defaults));
|
|
|
|
return ret;
|
2018-01-18 04:22:35 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
|
|
|
ircd::db::database::env::NewRandomAccessFile(const std::string& name,
|
|
|
|
std::unique_ptr<RandomAccessFile>* r,
|
|
|
|
const EnvOptions& options)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 04:22:35 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-08-16 13:49:23 +02:00
|
|
|
log::debug
|
2018-01-18 06:24:32 +01:00
|
|
|
{
|
2018-08-16 13:49:23 +02:00
|
|
|
log, "'%s': new random access file '%s' options:%p",
|
|
|
|
d.name,
|
|
|
|
name,
|
|
|
|
&options
|
2018-01-18 06:24:32 +01:00
|
|
|
};
|
2018-08-16 13:49:23 +02:00
|
|
|
#endif
|
2018-01-18 06:24:32 +01:00
|
|
|
|
2018-08-16 13:49:23 +02:00
|
|
|
*r = std::make_unique<random_access_file>(&d, name, options);
|
|
|
|
return Status::OK();
|
2018-01-18 04:22:35 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
|
|
|
ircd::db::database::env::NewWritableFile(const std::string& name,
|
|
|
|
std::unique_ptr<WritableFile>* r,
|
|
|
|
const EnvOptions& options)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 04:22:35 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-01-18 04:22:35 +01:00
|
|
|
log.debug("'%s': new writable file '%s' options:%p",
|
|
|
|
d.name,
|
|
|
|
name,
|
|
|
|
&options);
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
2018-01-18 04:22:35 +01:00
|
|
|
|
2018-01-18 05:17:00 +01:00
|
|
|
std::unique_ptr<WritableFile> defaults;
|
|
|
|
const auto ret
|
|
|
|
{
|
|
|
|
this->defaults.NewWritableFile(name, &defaults, options)
|
|
|
|
};
|
|
|
|
|
|
|
|
*r = std::make_unique<writable_file>(&d, name, options, std::move(defaults));
|
|
|
|
return ret;
|
2018-01-18 04:22:35 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
2018-01-18 06:24:32 +01:00
|
|
|
ircd::db::database::env::NewRandomRWFile(const std::string& name,
|
|
|
|
std::unique_ptr<RandomRWFile>* result,
|
|
|
|
const EnvOptions& options)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 04:22:35 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-01-18 06:24:32 +01:00
|
|
|
log.debug("'%s': new random read/write file '%s' options:%p",
|
2018-01-18 04:22:35 +01:00
|
|
|
d.name,
|
|
|
|
name,
|
|
|
|
&options);
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
2018-01-18 04:22:35 +01:00
|
|
|
|
2018-01-18 06:24:32 +01:00
|
|
|
std::unique_ptr<RandomRWFile> defaults;
|
|
|
|
const auto ret
|
|
|
|
{
|
|
|
|
this->defaults.NewRandomRWFile(name, &defaults, options)
|
|
|
|
};
|
|
|
|
|
|
|
|
*result = std::make_unique<random_rw_file>(&d, name, options, std::move(defaults));
|
|
|
|
return ret;
|
2018-01-18 04:22:35 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
2018-01-18 06:24:32 +01:00
|
|
|
ircd::db::database::env::NewDirectory(const std::string& name,
|
|
|
|
std::unique_ptr<Directory>* result)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 04:22:35 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-01-18 06:24:32 +01:00
|
|
|
log.debug("'%s': new directory '%s'",
|
2018-01-18 04:22:35 +01:00
|
|
|
d.name,
|
2018-01-18 06:24:32 +01:00
|
|
|
name);
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
2018-01-18 04:22:35 +01:00
|
|
|
|
2018-01-18 06:24:32 +01:00
|
|
|
std::unique_ptr<Directory> defaults;
|
|
|
|
const auto ret
|
|
|
|
{
|
|
|
|
this->defaults.NewDirectory(name, &defaults)
|
|
|
|
};
|
|
|
|
|
|
|
|
*result = std::make_unique<directory>(&d, name, std::move(defaults));
|
|
|
|
return ret;
|
2018-01-18 04:22:35 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
2018-01-18 06:24:32 +01:00
|
|
|
ircd::db::database::env::ReopenWritableFile(const std::string& name,
|
|
|
|
std::unique_ptr<WritableFile>* result,
|
|
|
|
const EnvOptions& options)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 04:22:35 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-01-18 06:24:32 +01:00
|
|
|
log.debug("'%s': reopen writable file '%s' options:%p",
|
2018-01-18 04:22:35 +01:00
|
|
|
d.name,
|
|
|
|
name,
|
|
|
|
&options);
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
2018-01-18 04:22:35 +01:00
|
|
|
|
2018-01-18 06:24:32 +01:00
|
|
|
return defaults.ReopenWritableFile(name, result, options);
|
2018-01-18 04:22:35 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
2018-01-18 06:24:32 +01:00
|
|
|
ircd::db::database::env::ReuseWritableFile(const std::string& name,
|
|
|
|
const std::string& old_name,
|
|
|
|
std::unique_ptr<WritableFile>* r,
|
|
|
|
const EnvOptions& options)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 04:22:35 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-01-18 06:24:32 +01:00
|
|
|
log.debug("'%s': reuse writable file '%s' old '%s' options:%p",
|
2018-01-18 04:22:35 +01:00
|
|
|
d.name,
|
2018-01-18 06:24:32 +01:00
|
|
|
name,
|
|
|
|
old_name,
|
|
|
|
&options);
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
2018-01-18 04:22:35 +01:00
|
|
|
|
2018-01-18 06:24:32 +01:00
|
|
|
return defaults.ReuseWritableFile(name, old_name, r, options);
|
2018-01-18 04:22:35 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
|
|
|
ircd::db::database::env::FileExists(const std::string& f)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 04:22:35 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-01-18 04:22:35 +01:00
|
|
|
log.debug("'%s': file exists '%s'",
|
|
|
|
d.name,
|
|
|
|
f);
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
2018-01-18 04:22:35 +01:00
|
|
|
|
|
|
|
return defaults.FileExists(f);
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
|
|
|
ircd::db::database::env::GetChildren(const std::string& dir,
|
|
|
|
std::vector<std::string>* r)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 04:22:35 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-01-18 04:22:35 +01:00
|
|
|
log.debug("'%s': get children of directory '%s'",
|
|
|
|
d.name,
|
|
|
|
dir);
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
2018-01-18 04:22:35 +01:00
|
|
|
|
|
|
|
return defaults.GetChildren(dir, r);
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
|
|
|
ircd::db::database::env::GetChildrenFileAttributes(const std::string& dir,
|
|
|
|
std::vector<FileAttributes>* result)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 04:22:35 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-01-18 04:22:35 +01:00
|
|
|
log.debug("'%s': get children file attributes of directory '%s'",
|
|
|
|
d.name,
|
|
|
|
dir);
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
2018-01-18 04:22:35 +01:00
|
|
|
|
|
|
|
return defaults.GetChildrenFileAttributes(dir, result);
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
|
|
|
ircd::db::database::env::DeleteFile(const std::string& name)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 04:22:35 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-01-18 04:22:35 +01:00
|
|
|
log.debug("'%s': delete file '%s'",
|
|
|
|
d.name,
|
|
|
|
name);
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
2018-01-18 04:22:35 +01:00
|
|
|
|
|
|
|
return defaults.DeleteFile(name);
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
|
|
|
ircd::db::database::env::CreateDir(const std::string& name)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 04:22:35 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-01-18 04:22:35 +01:00
|
|
|
log.debug("'%s': create directory '%s'",
|
|
|
|
d.name,
|
2018-03-24 22:53:57 +01:00
|
|
|
name);
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
2018-01-18 04:22:35 +01:00
|
|
|
|
|
|
|
return defaults.CreateDir(name);
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
|
|
|
ircd::db::database::env::CreateDirIfMissing(const std::string& name)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 04:22:35 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-01-18 04:22:35 +01:00
|
|
|
log.debug("'%s': create directory if missing '%s'",
|
|
|
|
d.name,
|
|
|
|
name);
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
2018-01-18 04:22:35 +01:00
|
|
|
|
|
|
|
return defaults.CreateDirIfMissing(name);
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
|
|
|
ircd::db::database::env::DeleteDir(const std::string& name)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 04:22:35 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-01-18 04:22:35 +01:00
|
|
|
log.debug("'%s': delete directory '%s'",
|
|
|
|
d.name,
|
|
|
|
name);
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
2018-01-18 04:22:35 +01:00
|
|
|
|
|
|
|
return defaults.DeleteDir(name);
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
|
|
|
ircd::db::database::env::GetFileSize(const std::string& name,
|
|
|
|
uint64_t* s)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 04:22:35 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-01-18 04:22:35 +01:00
|
|
|
log.debug("'%s': get file size '%s'",
|
|
|
|
d.name,
|
|
|
|
name);
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
2018-01-18 04:22:35 +01:00
|
|
|
|
|
|
|
return defaults.GetFileSize(name, s);
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
|
|
|
ircd::db::database::env::GetFileModificationTime(const std::string& name,
|
|
|
|
uint64_t* file_mtime)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 04:22:35 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-01-18 04:22:35 +01:00
|
|
|
log.debug("'%s': get file mtime '%s'",
|
|
|
|
d.name,
|
|
|
|
name);
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
2018-01-18 04:22:35 +01:00
|
|
|
|
|
|
|
return defaults.GetFileModificationTime(name, file_mtime);
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
|
|
|
ircd::db::database::env::RenameFile(const std::string& s,
|
|
|
|
const std::string& t)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 04:22:35 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-01-18 04:22:35 +01:00
|
|
|
log.debug("'%s': rename file '%s' to '%s'",
|
|
|
|
d.name,
|
|
|
|
s,
|
|
|
|
t);
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
2018-01-18 04:22:35 +01:00
|
|
|
|
|
|
|
return defaults.RenameFile(s, t);
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
|
|
|
ircd::db::database::env::LinkFile(const std::string& s,
|
|
|
|
const std::string& t)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 04:22:35 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-01-18 04:22:35 +01:00
|
|
|
log.debug("'%s': link file '%s' to '%s'",
|
|
|
|
d.name,
|
|
|
|
s,
|
|
|
|
t);
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
2018-01-18 04:22:35 +01:00
|
|
|
|
|
|
|
return defaults.LinkFile(s, t);
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
|
|
|
ircd::db::database::env::LockFile(const std::string& name,
|
|
|
|
FileLock** l)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 04:22:35 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-01-18 04:22:35 +01:00
|
|
|
log.debug("'%s': lock file '%s'",
|
|
|
|
d.name,
|
|
|
|
name);
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
2018-01-18 04:22:35 +01:00
|
|
|
|
|
|
|
return defaults.LockFile(name, l);
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
|
|
|
ircd::db::database::env::UnlockFile(FileLock* l)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 04:22:35 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-01-18 04:22:35 +01:00
|
|
|
log.debug("'%s': unlock file lock:%p",
|
|
|
|
d.name,
|
|
|
|
l);
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
2018-01-18 04:22:35 +01:00
|
|
|
|
|
|
|
return defaults.UnlockFile(l);
|
|
|
|
}
|
|
|
|
|
2018-06-01 15:23:39 +02:00
|
|
|
rocksdb::Status
|
|
|
|
ircd::db::database::env::GetTestDirectory(std::string* path)
|
|
|
|
noexcept
|
|
|
|
{
|
|
|
|
return defaults.GetTestDirectory(path);
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
|
|
|
ircd::db::database::env::GetAbsolutePath(const std::string& db_path,
|
|
|
|
std::string* output_path)
|
|
|
|
noexcept
|
|
|
|
{
|
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
|
|
|
log.debug("'%s': get absolute path from '%s' ret:%p",
|
|
|
|
d.name,
|
|
|
|
db_path,
|
|
|
|
output_path);
|
|
|
|
#endif
|
|
|
|
|
|
|
|
return defaults.GetAbsolutePath(db_path, output_path);
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
|
|
|
ircd::db::database::env::NewLogger(const std::string& name,
|
|
|
|
std::shared_ptr<Logger>* result)
|
|
|
|
noexcept
|
|
|
|
{
|
2018-08-16 10:27:43 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
|
|
|
log::debug
|
|
|
|
{
|
|
|
|
log, "'%s': new logger '%s' result:%p",
|
|
|
|
d.name,
|
|
|
|
name,
|
|
|
|
(const void *)result
|
|
|
|
};
|
|
|
|
#endif
|
|
|
|
|
2018-06-01 15:23:39 +02:00
|
|
|
return defaults.NewLogger(name, result);
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
|
|
|
ircd::db::database::env::GetHostName(char* name,
|
|
|
|
uint64_t len)
|
|
|
|
noexcept
|
|
|
|
{
|
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
|
|
|
log.debug("'%s': get host name name:%p len:%lu",
|
|
|
|
d.name,
|
|
|
|
name,
|
|
|
|
len);
|
|
|
|
#endif
|
|
|
|
|
|
|
|
return defaults.GetHostName(name, len);
|
|
|
|
}
|
|
|
|
|
|
|
|
uint64_t
|
|
|
|
ircd::db::database::env::NowMicros()
|
|
|
|
noexcept
|
|
|
|
{
|
|
|
|
return defaults.NowMicros();
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
|
|
|
ircd::db::database::env::GetCurrentTime(int64_t* unix_time)
|
|
|
|
noexcept
|
|
|
|
{
|
|
|
|
return defaults.GetCurrentTime(unix_time);
|
|
|
|
}
|
|
|
|
|
|
|
|
std::string
|
|
|
|
ircd::db::database::env::TimeToString(uint64_t time)
|
|
|
|
noexcept
|
|
|
|
{
|
|
|
|
return defaults.TimeToString(time);
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
ircd::db::database::env::SleepForMicroseconds(int micros)
|
|
|
|
noexcept
|
|
|
|
{
|
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
|
|
|
log.debug("'%s': sleep for %d microseconds",
|
|
|
|
d.name,
|
|
|
|
micros);
|
|
|
|
#endif
|
|
|
|
|
|
|
|
defaults.SleepForMicroseconds(micros);
|
|
|
|
}
|
|
|
|
|
2018-01-18 04:22:35 +01:00
|
|
|
void
|
|
|
|
ircd::db::database::env::Schedule(void (*f)(void* arg),
|
|
|
|
void* a,
|
|
|
|
Priority prio,
|
|
|
|
void* tag,
|
|
|
|
void (*u)(void* arg))
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 04:22:35 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-08-16 13:48:21 +02:00
|
|
|
log::debug
|
|
|
|
{
|
|
|
|
log, "'%s': schedule func:%p a:%p tag:%p u:%p prio:%s",
|
|
|
|
d.name,
|
|
|
|
f,
|
|
|
|
a,
|
|
|
|
tag,
|
|
|
|
u,
|
|
|
|
reflect(prio)
|
|
|
|
};
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
2018-01-18 04:22:35 +01:00
|
|
|
|
2018-08-16 13:48:21 +02:00
|
|
|
assert(st);
|
|
|
|
auto &pool
|
|
|
|
{
|
|
|
|
st->pool.at(prio)
|
|
|
|
};
|
|
|
|
|
|
|
|
auto &tasks
|
|
|
|
{
|
|
|
|
st->tasks.at(prio)
|
|
|
|
};
|
|
|
|
|
|
|
|
tasks.emplace_back(state::task
|
|
|
|
{
|
|
|
|
f, u, a
|
|
|
|
});
|
|
|
|
|
|
|
|
pool([this, &tasks]
|
|
|
|
{
|
|
|
|
assert(this->st);
|
|
|
|
if(tasks.empty())
|
|
|
|
return;
|
|
|
|
|
|
|
|
const auto task
|
|
|
|
{
|
|
|
|
std::move(tasks.front())
|
|
|
|
};
|
|
|
|
|
|
|
|
tasks.pop_front();
|
|
|
|
|
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
|
|
|
log::debug
|
|
|
|
{
|
|
|
|
log, "'%s': func:%p arg:%p",
|
|
|
|
this->d.name,
|
|
|
|
task.func,
|
|
|
|
task.arg,
|
|
|
|
};
|
|
|
|
#endif
|
|
|
|
|
|
|
|
// Execute the task
|
|
|
|
task.func(task.arg);
|
|
|
|
});
|
2018-01-18 04:22:35 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
int
|
2018-08-16 13:48:21 +02:00
|
|
|
ircd::db::database::env::UnSchedule(void *const tag,
|
|
|
|
const Priority prio)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 04:22:35 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-08-16 13:48:21 +02:00
|
|
|
log::debug
|
|
|
|
{
|
|
|
|
log, "'%s': unschedule tag:%p prio:%s",
|
|
|
|
d.name,
|
|
|
|
tag,
|
|
|
|
reflect(prio)
|
|
|
|
};
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
2018-01-18 04:22:35 +01:00
|
|
|
|
2018-08-16 13:48:21 +02:00
|
|
|
assert(st);
|
|
|
|
auto &tasks
|
|
|
|
{
|
|
|
|
st->tasks.at(prio)
|
|
|
|
};
|
|
|
|
|
|
|
|
size_t i(0);
|
|
|
|
for(auto it(begin(tasks)); it != end(tasks); it = tasks.erase(it), ++i)
|
|
|
|
it->cancel(it->arg);
|
|
|
|
|
|
|
|
return i;
|
2018-01-18 04:22:35 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
ircd::db::database::env::StartThread(void (*f)(void*),
|
|
|
|
void* a)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 04:22:35 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-08-16 13:48:21 +02:00
|
|
|
log::debug
|
|
|
|
{
|
|
|
|
log, "'%s': start thread func:%p a:%p",
|
|
|
|
d.name,
|
|
|
|
f,
|
|
|
|
a
|
|
|
|
};
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
2018-01-18 04:22:35 +01:00
|
|
|
|
2018-08-16 13:48:21 +02:00
|
|
|
assert(0);
|
2018-01-18 04:22:35 +01:00
|
|
|
return defaults.StartThread(f, a);
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
ircd::db::database::env::WaitForJoin()
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 04:22:35 +01:00
|
|
|
{
|
2018-06-01 21:26:05 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
|
|
|
log::debug
|
|
|
|
{
|
|
|
|
log, "'%s': wait for all ctx to join",
|
|
|
|
d.name
|
|
|
|
};
|
|
|
|
#endif
|
|
|
|
|
2018-01-18 04:22:35 +01:00
|
|
|
return defaults.WaitForJoin();
|
|
|
|
}
|
|
|
|
|
|
|
|
unsigned int
|
2018-08-16 13:48:21 +02:00
|
|
|
ircd::db::database::env::GetThreadPoolQueueLen(Priority prio)
|
2018-01-18 09:49:23 +01:00
|
|
|
const noexcept
|
2018-01-18 04:22:35 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-08-16 13:48:21 +02:00
|
|
|
log::debug
|
|
|
|
{
|
|
|
|
log, "'%s': get thread pool queue len prio:%s",
|
|
|
|
d.name,
|
|
|
|
reflect(prio)
|
|
|
|
};
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
2018-01-18 04:22:35 +01:00
|
|
|
|
2018-08-16 13:48:21 +02:00
|
|
|
assert(st);
|
|
|
|
const auto &pool
|
|
|
|
{
|
|
|
|
st->pool.at(prio)
|
|
|
|
};
|
|
|
|
|
|
|
|
return pool.queued();
|
2018-01-18 04:22:35 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
ircd::db::database::env::SetBackgroundThreads(int num,
|
2018-08-16 13:48:21 +02:00
|
|
|
Priority prio)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 04:22:35 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-08-16 13:48:21 +02:00
|
|
|
log::debug
|
|
|
|
{
|
|
|
|
log, "'%s': set background threads num:%d prio:%s",
|
|
|
|
d.name,
|
|
|
|
num,
|
|
|
|
reflect(prio)
|
|
|
|
};
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
2018-01-18 04:22:35 +01:00
|
|
|
|
2018-08-16 13:48:21 +02:00
|
|
|
assert(st);
|
|
|
|
auto &pool
|
|
|
|
{
|
|
|
|
st->pool.at(prio)
|
|
|
|
};
|
|
|
|
|
|
|
|
const auto &size
|
|
|
|
{
|
|
|
|
ssize_t(pool.size())
|
|
|
|
};
|
|
|
|
|
|
|
|
if(size > num)
|
|
|
|
pool.del(size - num);
|
|
|
|
else if(size < num)
|
|
|
|
pool.add(num - size);
|
2018-01-18 04:22:35 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
ircd::db::database::env::IncBackgroundThreadsIfNeeded(int num,
|
2018-08-16 13:48:21 +02:00
|
|
|
Priority prio)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 04:22:35 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-08-16 13:48:21 +02:00
|
|
|
log::debug
|
|
|
|
{
|
|
|
|
log, "'%s': increase background threads num:%d prio:%s",
|
|
|
|
d.name,
|
|
|
|
num,
|
|
|
|
reflect(prio)
|
|
|
|
};
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
2018-01-18 04:22:35 +01:00
|
|
|
|
2018-08-16 13:48:21 +02:00
|
|
|
assert(st);
|
|
|
|
auto &pool
|
|
|
|
{
|
|
|
|
st->pool.at(prio)
|
|
|
|
};
|
|
|
|
|
|
|
|
pool.add(num);
|
2018-01-18 04:22:35 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
ircd::db::database::env::LowerThreadPoolIOPriority(Priority pool)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 04:22:35 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-01-18 04:22:35 +01:00
|
|
|
log.debug("'%s': lower thread pool priority prio:%s",
|
|
|
|
d.name,
|
|
|
|
reflect(pool));
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
2018-01-18 04:22:35 +01:00
|
|
|
|
|
|
|
defaults.LowerThreadPoolIOPriority(pool);
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
|
|
|
ircd::db::database::env::GetThreadList(std::vector<ThreadStatus>* thread_list)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 04:22:35 +01:00
|
|
|
{
|
2018-06-02 18:18:01 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
|
|
|
log::debug
|
|
|
|
{
|
|
|
|
log, "'%s': get thread list %p (%zu)",
|
|
|
|
d.name,
|
|
|
|
thread_list,
|
|
|
|
thread_list? thread_list->size() : 0UL
|
|
|
|
};
|
|
|
|
#endif
|
|
|
|
|
2018-01-18 04:22:35 +01:00
|
|
|
return defaults.GetThreadList(thread_list);
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::ThreadStatusUpdater*
|
|
|
|
ircd::db::database::env::GetThreadStatusUpdater()
|
2018-01-18 09:49:23 +01:00
|
|
|
const noexcept
|
2018-01-18 04:22:35 +01:00
|
|
|
{
|
2018-06-02 18:18:01 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
|
|
|
log::debug
|
|
|
|
{
|
|
|
|
log, "'%s': get thread status updater",
|
|
|
|
d.name,
|
|
|
|
};
|
|
|
|
#endif
|
|
|
|
|
2018-01-18 04:22:35 +01:00
|
|
|
return defaults.GetThreadStatusUpdater();
|
|
|
|
}
|
|
|
|
|
|
|
|
uint64_t
|
|
|
|
ircd::db::database::env::GetThreadID()
|
2018-01-18 09:49:23 +01:00
|
|
|
const noexcept
|
2018-01-18 04:22:35 +01:00
|
|
|
{
|
2018-06-02 18:18:01 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
|
|
|
log::debug
|
|
|
|
{
|
|
|
|
log, "'%s': get thread ID",
|
|
|
|
d.name,
|
|
|
|
};
|
|
|
|
#endif
|
|
|
|
|
2018-08-16 13:48:21 +02:00
|
|
|
return ctx::this_ctx::id();
|
2018-01-18 04:22:35 +01:00
|
|
|
}
|
|
|
|
|
2018-04-09 21:55:22 +02:00
|
|
|
int
|
2018-08-16 13:48:21 +02:00
|
|
|
ircd::db::database::env::GetBackgroundThreads(Priority prio)
|
2018-04-09 21:55:22 +02:00
|
|
|
noexcept
|
|
|
|
{
|
2018-06-02 18:18:01 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
|
|
|
log::debug
|
|
|
|
{
|
|
|
|
log, "'%s': get background threads prio:%s",
|
|
|
|
d.name,
|
2018-08-16 13:48:21 +02:00
|
|
|
reflect(prio)
|
2018-06-02 18:18:01 +02:00
|
|
|
};
|
|
|
|
#endif
|
|
|
|
|
2018-08-16 13:48:21 +02:00
|
|
|
assert(st);
|
|
|
|
const auto &pool
|
|
|
|
{
|
|
|
|
st->pool.at(prio)
|
|
|
|
};
|
|
|
|
|
|
|
|
return pool.size();
|
2018-04-09 21:55:22 +02:00
|
|
|
}
|
|
|
|
|
2018-01-18 05:17:00 +01:00
|
|
|
//
|
|
|
|
// writable_file
|
|
|
|
//
|
|
|
|
|
|
|
|
ircd::db::database::env::writable_file::writable_file(database *const &d,
|
|
|
|
const std::string &name,
|
|
|
|
const EnvOptions &opts,
|
|
|
|
std::unique_ptr<WritableFile> defaults)
|
|
|
|
:d{*d}
|
|
|
|
,defaults{std::move(defaults)}
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
ircd::db::database::env::writable_file::~writable_file()
|
|
|
|
noexcept
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
|
|
|
ircd::db::database::env::writable_file::Append(const Slice& s)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 05:17:00 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-01-18 05:17:00 +01:00
|
|
|
log.debug("'%s': wfile:%p append:%p bytes:%zu",
|
|
|
|
d.name,
|
|
|
|
this,
|
|
|
|
data(s),
|
|
|
|
size(s));
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
2018-01-18 05:17:00 +01:00
|
|
|
|
|
|
|
return defaults->Append(s);
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
|
|
|
ircd::db::database::env::writable_file::PositionedAppend(const Slice& s,
|
|
|
|
uint64_t offset)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 05:17:00 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-01-18 05:17:00 +01:00
|
|
|
log.debug("'%s': wfile:%p append:%p bytes:%zu offset:%lu",
|
|
|
|
d.name,
|
|
|
|
this,
|
|
|
|
data(s),
|
|
|
|
size(s),
|
|
|
|
offset);
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
2018-01-18 05:17:00 +01:00
|
|
|
|
|
|
|
return defaults->PositionedAppend(s, offset);
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
|
|
|
ircd::db::database::env::writable_file::Truncate(uint64_t size)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 05:17:00 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-01-18 05:17:00 +01:00
|
|
|
log.debug("'%s': wfile:%p truncate to %lu bytes",
|
|
|
|
d.name,
|
|
|
|
this,
|
|
|
|
size);
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
2018-01-18 05:17:00 +01:00
|
|
|
|
|
|
|
return defaults->Truncate(size);
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
|
|
|
ircd::db::database::env::writable_file::Close()
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 05:17:00 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-01-18 05:17:00 +01:00
|
|
|
log.debug("'%s': wfile:%p close",
|
|
|
|
d.name,
|
|
|
|
this);
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
|
|
|
|
2018-01-18 05:17:00 +01:00
|
|
|
return defaults->Close();
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
|
|
|
ircd::db::database::env::writable_file::Flush()
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 05:17:00 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-01-18 05:17:00 +01:00
|
|
|
log.debug("'%s': wfile:%p flush",
|
|
|
|
d.name,
|
|
|
|
this);
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
|
|
|
|
2018-01-18 05:17:00 +01:00
|
|
|
return defaults->Flush();
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
|
|
|
ircd::db::database::env::writable_file::Sync()
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 05:17:00 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-01-18 05:17:00 +01:00
|
|
|
log.debug("'%s': wfile:%p sync",
|
|
|
|
d.name,
|
|
|
|
this);
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
2018-01-18 05:17:00 +01:00
|
|
|
|
|
|
|
return defaults->Sync();
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
|
|
|
ircd::db::database::env::writable_file::Fsync()
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 05:17:00 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-01-18 05:17:00 +01:00
|
|
|
log.debug("'%s': wfile:%p fsync",
|
|
|
|
d.name,
|
|
|
|
this);
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
2018-01-18 05:17:00 +01:00
|
|
|
|
|
|
|
return defaults->Fsync();
|
|
|
|
}
|
|
|
|
|
|
|
|
bool
|
|
|
|
ircd::db::database::env::writable_file::IsSyncThreadSafe()
|
2018-01-18 09:49:23 +01:00
|
|
|
const noexcept
|
2018-01-18 05:17:00 +01:00
|
|
|
{
|
|
|
|
return defaults->IsSyncThreadSafe();
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
ircd::db::database::env::writable_file::SetIOPriority(Env::IOPriority prio)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 05:17:00 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-01-18 05:17:00 +01:00
|
|
|
log.debug("'%s': wfile:%p set IO prio to %s",
|
|
|
|
d.name,
|
|
|
|
this,
|
|
|
|
reflect(prio));
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
2018-01-18 05:17:00 +01:00
|
|
|
|
|
|
|
defaults->SetIOPriority(prio);
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Env::IOPriority
|
|
|
|
ircd::db::database::env::writable_file::GetIOPriority()
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 05:17:00 +01:00
|
|
|
{
|
|
|
|
return defaults->GetIOPriority();
|
|
|
|
}
|
|
|
|
|
|
|
|
uint64_t
|
|
|
|
ircd::db::database::env::writable_file::GetFileSize()
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 05:17:00 +01:00
|
|
|
{
|
|
|
|
return defaults->GetFileSize();
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
ircd::db::database::env::writable_file::GetPreallocationStatus(size_t* block_size,
|
|
|
|
size_t* last_allocated_block)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 05:17:00 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-01-18 05:17:00 +01:00
|
|
|
log.debug("'%s': wfile:%p get preallocation block_size:%p last_block:%p",
|
|
|
|
d.name,
|
|
|
|
this,
|
|
|
|
block_size,
|
|
|
|
last_allocated_block);
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
2018-01-18 05:17:00 +01:00
|
|
|
|
|
|
|
defaults->GetPreallocationStatus(block_size, last_allocated_block);
|
|
|
|
}
|
|
|
|
|
|
|
|
size_t
|
|
|
|
ircd::db::database::env::writable_file::GetUniqueId(char* id,
|
|
|
|
size_t max_size)
|
2018-01-18 09:49:23 +01:00
|
|
|
const noexcept
|
2018-01-18 05:17:00 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-01-18 05:17:00 +01:00
|
|
|
log.debug("'%s': wfile:%p get unique id:%p max_size:%zu",
|
|
|
|
d.name,
|
|
|
|
this,
|
|
|
|
id,
|
|
|
|
max_size);
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
2018-01-18 05:17:00 +01:00
|
|
|
|
|
|
|
return defaults->GetUniqueId(id, max_size);
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
|
|
|
ircd::db::database::env::writable_file::InvalidateCache(size_t offset,
|
|
|
|
size_t length)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 05:17:00 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-01-18 05:17:00 +01:00
|
|
|
log.debug("'%s': wfile:%p invalidate cache offset:%zu length:%zu",
|
|
|
|
d.name,
|
|
|
|
this,
|
|
|
|
offset,
|
|
|
|
length);
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
2018-01-18 05:17:00 +01:00
|
|
|
|
|
|
|
return defaults->InvalidateCache(offset, length);
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
ircd::db::database::env::writable_file::SetPreallocationBlockSize(size_t size)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 05:17:00 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-01-18 05:17:00 +01:00
|
|
|
log.debug("'%s': wfile:%p set preallocation block size:%zu",
|
|
|
|
d.name,
|
|
|
|
this,
|
|
|
|
size);
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
2018-01-18 05:17:00 +01:00
|
|
|
|
|
|
|
defaults->SetPreallocationBlockSize(size);
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
ircd::db::database::env::writable_file::PrepareWrite(size_t offset,
|
|
|
|
size_t length)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 05:17:00 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-01-18 05:17:00 +01:00
|
|
|
log.debug("'%s': wfile:%p prepare write offset:%zu length:%zu",
|
|
|
|
d.name,
|
|
|
|
this,
|
|
|
|
offset,
|
|
|
|
length);
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
|
|
|
|
2018-01-18 05:17:00 +01:00
|
|
|
defaults->PrepareWrite(offset, length);
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
|
|
|
ircd::db::database::env::writable_file::Allocate(uint64_t offset,
|
|
|
|
uint64_t length)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 05:17:00 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-01-18 05:17:00 +01:00
|
|
|
log.debug("'%s': wfile:%p allocate offset:%lu length:%lu",
|
|
|
|
d.name,
|
|
|
|
this,
|
|
|
|
offset,
|
|
|
|
length);
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
2018-01-18 05:17:00 +01:00
|
|
|
|
|
|
|
return defaults->Allocate(offset, length);
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
|
|
|
ircd::db::database::env::writable_file::RangeSync(uint64_t offset,
|
|
|
|
uint64_t length)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 05:17:00 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-01-18 05:17:00 +01:00
|
|
|
log.debug("'%s': wfile:%p range sync offset:%lu length:%lu",
|
|
|
|
d.name,
|
|
|
|
this,
|
|
|
|
offset,
|
|
|
|
length);
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
2018-01-18 05:17:00 +01:00
|
|
|
|
|
|
|
return defaults->RangeSync(offset, length);
|
|
|
|
}
|
|
|
|
|
2018-01-18 06:24:32 +01:00
|
|
|
//
|
|
|
|
// sequential_file
|
|
|
|
//
|
|
|
|
|
|
|
|
ircd::db::database::env::sequential_file::sequential_file(database *const &d,
|
|
|
|
const std::string &name,
|
|
|
|
const EnvOptions &opts,
|
|
|
|
std::unique_ptr<SequentialFile> defaults)
|
|
|
|
:d{*d}
|
|
|
|
,defaults{std::move(defaults)}
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
ircd::db::database::env::sequential_file::~sequential_file()
|
|
|
|
noexcept
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
|
|
|
ircd::db::database::env::sequential_file::Read(size_t length,
|
|
|
|
Slice *result,
|
|
|
|
char *scratch)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 06:24:32 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-01-18 06:24:32 +01:00
|
|
|
log.debug("'%s': seqfile:%p read:%p length:%zu scratch:%p",
|
|
|
|
d.name,
|
|
|
|
this,
|
|
|
|
result,
|
|
|
|
length,
|
|
|
|
scratch);
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
|
|
|
|
2018-01-18 06:24:32 +01:00
|
|
|
return defaults->Read(length, result, scratch);
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
|
|
|
ircd::db::database::env::sequential_file::PositionedRead(uint64_t offset,
|
|
|
|
size_t length,
|
|
|
|
Slice *result,
|
|
|
|
char *scratch)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 06:24:32 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-01-18 06:24:32 +01:00
|
|
|
log.debug("'%s': seqfile:%p read:%p length:%zu offset:%zu scratch:%p",
|
|
|
|
d.name,
|
|
|
|
this,
|
|
|
|
result,
|
|
|
|
length,
|
|
|
|
offset,
|
|
|
|
scratch);
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
|
|
|
|
2018-01-18 06:24:32 +01:00
|
|
|
return defaults->PositionedRead(offset, length, result, scratch);
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
|
|
|
ircd::db::database::env::sequential_file::Skip(uint64_t size)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 06:24:32 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-01-18 06:24:32 +01:00
|
|
|
log.debug("'%s': seqfile:%p skip:%zu",
|
|
|
|
d.name,
|
|
|
|
this,
|
|
|
|
size);
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
2018-01-18 06:24:32 +01:00
|
|
|
|
|
|
|
return defaults->Skip(size);
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
|
|
|
ircd::db::database::env::sequential_file::InvalidateCache(size_t offset,
|
|
|
|
size_t length)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 06:24:32 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-01-18 06:24:32 +01:00
|
|
|
log.debug("'%s': seqfile:%p invalidate cache offset:%zu length:%zu",
|
|
|
|
d.name,
|
|
|
|
this,
|
|
|
|
offset,
|
|
|
|
length);
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
2018-01-18 06:24:32 +01:00
|
|
|
|
|
|
|
return defaults->InvalidateCache(offset, length);
|
|
|
|
}
|
|
|
|
|
|
|
|
bool
|
|
|
|
ircd::db::database::env::sequential_file::use_direct_io()
|
2018-01-18 09:49:23 +01:00
|
|
|
const noexcept
|
2018-01-18 06:24:32 +01:00
|
|
|
{
|
|
|
|
return defaults->use_direct_io();
|
|
|
|
}
|
|
|
|
|
|
|
|
size_t
|
|
|
|
ircd::db::database::env::sequential_file::GetRequiredBufferAlignment()
|
2018-01-18 09:49:23 +01:00
|
|
|
const noexcept
|
2018-01-18 06:24:32 +01:00
|
|
|
{
|
|
|
|
return defaults->GetRequiredBufferAlignment();
|
|
|
|
}
|
|
|
|
|
|
|
|
//
|
|
|
|
// random_access_file
|
|
|
|
//
|
|
|
|
|
|
|
|
ircd::db::database::env::random_access_file::random_access_file(database *const &d,
|
|
|
|
const std::string &name,
|
2018-08-16 13:49:23 +02:00
|
|
|
const EnvOptions &opts)
|
|
|
|
:d
|
|
|
|
{
|
|
|
|
*d
|
|
|
|
}
|
|
|
|
,fd
|
|
|
|
{
|
|
|
|
name, std::ios_base::in | std::ios_base::out
|
|
|
|
}
|
2018-01-18 06:24:32 +01:00
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
ircd::db::database::env::random_access_file::~random_access_file()
|
|
|
|
noexcept
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
|
|
|
ircd::db::database::env::random_access_file::Prefetch(uint64_t offset,
|
|
|
|
size_t length)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 06:24:32 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-08-16 13:49:23 +02:00
|
|
|
log::debug
|
|
|
|
{
|
|
|
|
log, "'%s': rfile:%p prefetch offset:%zu length:%zu",
|
|
|
|
d.name,
|
|
|
|
this,
|
|
|
|
offset,
|
|
|
|
length
|
|
|
|
};
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
|
|
|
|
2018-08-16 13:49:23 +02:00
|
|
|
return Status::OK();
|
2018-01-18 06:24:32 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
|
|
|
ircd::db::database::env::random_access_file::Read(uint64_t offset,
|
|
|
|
size_t length,
|
|
|
|
Slice *result,
|
|
|
|
char *scratch)
|
2018-01-18 09:49:23 +01:00
|
|
|
const noexcept
|
2018-01-18 06:24:32 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-08-16 13:49:23 +02:00
|
|
|
log::debug
|
|
|
|
{
|
|
|
|
log, "'%s': rfile:%p read:%p offset:%zu length:%zu scratch:%p",
|
|
|
|
d.name,
|
|
|
|
this,
|
|
|
|
result,
|
|
|
|
offset,
|
|
|
|
length,
|
|
|
|
scratch
|
|
|
|
};
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
|
|
|
|
2018-08-16 13:49:23 +02:00
|
|
|
assert(result);
|
|
|
|
const mutable_buffer buf
|
|
|
|
{
|
|
|
|
scratch, length
|
|
|
|
};
|
|
|
|
|
|
|
|
const auto read
|
|
|
|
{
|
|
|
|
fs::read(fd, buf, offset)
|
|
|
|
};
|
|
|
|
|
|
|
|
*result = slice(read);
|
|
|
|
|
|
|
|
return Status::OK();
|
2018-01-18 06:24:32 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
|
|
|
ircd::db::database::env::random_access_file::InvalidateCache(size_t offset,
|
|
|
|
size_t length)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 06:24:32 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-08-16 13:49:23 +02:00
|
|
|
log::debug
|
|
|
|
{
|
|
|
|
log, "'%s': rfile:%p invalidate cache offset:%zu length:%zu",
|
|
|
|
d.name,
|
|
|
|
this,
|
|
|
|
offset,
|
|
|
|
length
|
|
|
|
};
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
2018-01-18 06:24:32 +01:00
|
|
|
|
2018-08-16 13:49:23 +02:00
|
|
|
return Status::OK();
|
2018-01-18 06:24:32 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
size_t
|
|
|
|
ircd::db::database::env::random_access_file::GetUniqueId(char* id,
|
|
|
|
size_t max_size)
|
2018-01-18 09:49:23 +01:00
|
|
|
const noexcept
|
2018-01-18 06:24:32 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-08-16 13:49:23 +02:00
|
|
|
log::debug
|
|
|
|
{
|
|
|
|
log, "'%s': rfile:%p get unique id:%p max_size:%zu",
|
|
|
|
d.name,
|
|
|
|
this,
|
|
|
|
id,
|
|
|
|
max_size
|
|
|
|
};
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
2018-01-18 06:24:32 +01:00
|
|
|
|
2018-08-16 13:49:23 +02:00
|
|
|
return 0;
|
2018-01-18 06:24:32 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
ircd::db::database::env::random_access_file::Hint(AccessPattern pattern)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 06:24:32 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-08-16 13:49:23 +02:00
|
|
|
log::debug
|
|
|
|
{
|
|
|
|
log, "'%s': rfile:%p hint %s",
|
|
|
|
d.name,
|
|
|
|
this,
|
|
|
|
reflect(pattern)
|
|
|
|
};
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
2018-01-18 06:24:32 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
bool
|
|
|
|
ircd::db::database::env::random_access_file::use_direct_io()
|
2018-01-18 09:49:23 +01:00
|
|
|
const noexcept
|
2018-01-18 06:24:32 +01:00
|
|
|
{
|
2018-08-16 13:49:23 +02:00
|
|
|
return false;
|
2018-01-18 06:24:32 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
size_t
|
|
|
|
ircd::db::database::env::random_access_file::GetRequiredBufferAlignment()
|
2018-01-18 09:49:23 +01:00
|
|
|
const noexcept
|
2018-01-18 06:24:32 +01:00
|
|
|
{
|
2018-08-16 13:49:23 +02:00
|
|
|
return 16;
|
2018-01-18 06:24:32 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
//
|
|
|
|
// random_rw_file
|
|
|
|
//
|
|
|
|
|
|
|
|
ircd::db::database::env::random_rw_file::random_rw_file(database *const &d,
|
|
|
|
const std::string &name,
|
|
|
|
const EnvOptions &opts,
|
|
|
|
std::unique_ptr<RandomRWFile> defaults)
|
|
|
|
:d{*d}
|
|
|
|
,defaults{std::move(defaults)}
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
ircd::db::database::env::random_rw_file::~random_rw_file()
|
|
|
|
noexcept
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
|
|
|
ircd::db::database::env::random_rw_file::Close()
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 06:24:32 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-01-18 06:24:32 +01:00
|
|
|
log.debug("'%s': rwfile:%p closec",
|
|
|
|
d.name,
|
|
|
|
this);
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
2018-01-18 06:24:32 +01:00
|
|
|
|
|
|
|
return defaults->Close();
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
|
|
|
ircd::db::database::env::random_rw_file::Fsync()
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 06:24:32 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-01-18 06:24:32 +01:00
|
|
|
log.debug("'%s': rwfile:%p fsync",
|
|
|
|
d.name,
|
|
|
|
this);
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
2018-01-18 06:24:32 +01:00
|
|
|
|
|
|
|
return defaults->Fsync();
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
|
|
|
ircd::db::database::env::random_rw_file::Sync()
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 06:24:32 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-01-18 06:24:32 +01:00
|
|
|
log.debug("'%s': rwfile:%p sync",
|
|
|
|
d.name,
|
|
|
|
this);
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
2018-01-18 06:24:32 +01:00
|
|
|
|
|
|
|
return defaults->Sync();
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
|
|
|
ircd::db::database::env::random_rw_file::Flush()
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 06:24:32 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-01-18 06:24:32 +01:00
|
|
|
log.debug("'%s': rwfile:%p flush",
|
|
|
|
d.name,
|
|
|
|
this);
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
2018-01-18 06:24:32 +01:00
|
|
|
|
|
|
|
return defaults->Flush();
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
|
|
|
ircd::db::database::env::random_rw_file::Read(uint64_t offset,
|
|
|
|
size_t length,
|
|
|
|
Slice *result,
|
|
|
|
char *scratch)
|
2018-01-18 09:49:23 +01:00
|
|
|
const noexcept
|
2018-01-18 06:24:32 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-01-18 06:24:32 +01:00
|
|
|
log.debug("'%s': rwfile:%p read:%p offset:%zu length:%zu scratch:%p",
|
|
|
|
d.name,
|
|
|
|
this,
|
|
|
|
result,
|
|
|
|
offset,
|
|
|
|
length,
|
|
|
|
scratch);
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
2018-01-18 06:24:32 +01:00
|
|
|
|
|
|
|
return defaults->Read(offset, length, result, scratch);
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
|
|
|
ircd::db::database::env::random_rw_file::Write(uint64_t offset,
|
|
|
|
const Slice &slice)
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 06:24:32 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-01-18 06:24:32 +01:00
|
|
|
log.debug("'%s': rwfile:%p write:%p offset:%zu length:%zu",
|
|
|
|
d.name,
|
|
|
|
this,
|
|
|
|
data(slice),
|
|
|
|
offset,
|
|
|
|
size(slice));
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
2018-01-18 06:24:32 +01:00
|
|
|
|
|
|
|
return defaults->Write(offset, slice);
|
|
|
|
}
|
|
|
|
|
|
|
|
bool
|
|
|
|
ircd::db::database::env::random_rw_file::use_direct_io()
|
2018-01-18 09:49:23 +01:00
|
|
|
const noexcept
|
2018-01-18 06:24:32 +01:00
|
|
|
{
|
|
|
|
return defaults->use_direct_io();
|
|
|
|
}
|
|
|
|
|
|
|
|
size_t
|
|
|
|
ircd::db::database::env::random_rw_file::GetRequiredBufferAlignment()
|
2018-01-18 09:49:23 +01:00
|
|
|
const noexcept
|
2018-01-18 06:24:32 +01:00
|
|
|
{
|
|
|
|
return defaults->GetRequiredBufferAlignment();
|
|
|
|
}
|
|
|
|
|
|
|
|
//
|
|
|
|
// directory
|
|
|
|
//
|
|
|
|
|
|
|
|
ircd::db::database::env::directory::directory(database *const &d,
|
|
|
|
const std::string &name,
|
|
|
|
std::unique_ptr<Directory> defaults)
|
|
|
|
:d{*d}
|
|
|
|
,defaults{std::move(defaults)}
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
ircd::db::database::env::directory::~directory()
|
|
|
|
noexcept
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
|
|
|
ircd::db::database::env::directory::Fsync()
|
2018-01-18 09:49:23 +01:00
|
|
|
noexcept
|
2018-01-18 06:24:32 +01:00
|
|
|
{
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_ENV
|
2018-01-18 06:24:32 +01:00
|
|
|
log.debug("'%s': directory:%p fsync",
|
|
|
|
d.name,
|
|
|
|
this);
|
2018-05-29 10:42:48 +02:00
|
|
|
#endif
|
2018-01-18 06:24:32 +01:00
|
|
|
|
|
|
|
return defaults->Fsync();
|
|
|
|
}
|
|
|
|
|
|
|
|
//
|
|
|
|
// file_lock
|
|
|
|
//
|
|
|
|
|
|
|
|
ircd::db::database::env::file_lock::file_lock(database *const &d)
|
|
|
|
:d{*d}
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
ircd::db::database::env::file_lock::~file_lock()
|
|
|
|
noexcept
|
|
|
|
{
|
|
|
|
}
|
2018-01-18 05:17:00 +01:00
|
|
|
|
2018-05-30 09:14:07 +02:00
|
|
|
//
|
|
|
|
// rocksdb::port (EXPERIMENTAL)
|
|
|
|
//
|
|
|
|
|
|
|
|
#ifdef IRCD_DB_PORT
|
|
|
|
|
|
|
|
//
|
|
|
|
// Mutex
|
|
|
|
//
|
|
|
|
|
|
|
|
rocksdb::port::Mutex::Mutex()
|
|
|
|
{
|
2018-06-01 20:31:46 +02:00
|
|
|
#ifdef RB_DEBUG_DB_PORT_
|
|
|
|
if(unlikely(!ctx::current || !is_main_thread()))
|
|
|
|
return;
|
|
|
|
|
|
|
|
log::debug
|
|
|
|
{
|
|
|
|
db::log, "mutex %lu %p CTOR", ctx::id(), this
|
|
|
|
};
|
|
|
|
#endif
|
2018-05-30 09:14:07 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::port::Mutex::Mutex(bool adaptive)
|
2018-06-01 20:31:46 +02:00
|
|
|
:Mutex{}
|
2018-05-30 09:14:07 +02:00
|
|
|
{
|
|
|
|
}
|
|
|
|
|
2018-06-01 20:31:46 +02:00
|
|
|
rocksdb::port::Mutex::~Mutex()
|
|
|
|
{
|
|
|
|
#ifdef RB_DEBUG_DB_PORT_
|
|
|
|
if(unlikely(!ctx::current || !is_main_thread()))
|
|
|
|
return;
|
|
|
|
|
|
|
|
log::debug
|
|
|
|
{
|
|
|
|
db::log, "mutex %lu %p DTOR", ctx::id(), this
|
|
|
|
};
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
|
2018-05-30 09:14:07 +02:00
|
|
|
void
|
|
|
|
rocksdb::port::Mutex::Lock()
|
|
|
|
{
|
2018-06-01 20:31:46 +02:00
|
|
|
if(unlikely(!is_main_thread() || !ctx::current))
|
|
|
|
return;
|
|
|
|
|
|
|
|
#ifdef RB_DEBUG_DB_PORT
|
|
|
|
log::debug
|
|
|
|
{
|
|
|
|
db::log, "mutex %lu %p LOCK", ctx::id(), this
|
|
|
|
};
|
|
|
|
#endif
|
|
|
|
|
2018-05-30 09:14:07 +02:00
|
|
|
mu.lock();
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
rocksdb::port::Mutex::Unlock()
|
|
|
|
{
|
2018-06-01 20:31:46 +02:00
|
|
|
if(unlikely(!is_main_thread() || !ctx::current))
|
|
|
|
return;
|
|
|
|
|
|
|
|
#ifdef RB_DEBUG_DB_PORT
|
|
|
|
log::debug
|
|
|
|
{
|
|
|
|
db::log, "mutex %lu %p UNLOCK", ctx::id(), this
|
|
|
|
};
|
|
|
|
#endif
|
|
|
|
|
|
|
|
assert(mu.locked());
|
2018-05-30 09:14:07 +02:00
|
|
|
mu.unlock();
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
rocksdb::port::Mutex::AssertHeld()
|
|
|
|
{
|
2018-06-01 20:31:46 +02:00
|
|
|
if(unlikely(!is_main_thread() || !ctx::current))
|
|
|
|
return;
|
|
|
|
|
|
|
|
assert(mu.locked());
|
2018-05-30 09:14:07 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
//
|
|
|
|
// RWMutex
|
|
|
|
//
|
|
|
|
|
|
|
|
rocksdb::port::RWMutex::RWMutex()
|
|
|
|
{
|
2018-06-01 20:31:46 +02:00
|
|
|
#ifdef RB_DEBUG_DB_PORT_
|
|
|
|
log::debug
|
|
|
|
{
|
|
|
|
db::log, "shared_mutex %lu %p CTOR", ctx::id(), this
|
|
|
|
};
|
|
|
|
#endif
|
2018-05-30 09:14:07 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::port::RWMutex::~RWMutex()
|
|
|
|
{
|
2018-06-01 20:31:46 +02:00
|
|
|
#ifdef RB_DEBUG_DB_PORT_
|
|
|
|
log::debug
|
|
|
|
{
|
|
|
|
db::log, "shared_mutex %lu %p DTOR", ctx::id(), this
|
|
|
|
};
|
|
|
|
#endif
|
2018-05-30 09:14:07 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
rocksdb::port::RWMutex::ReadLock()
|
|
|
|
{
|
2018-06-01 20:31:46 +02:00
|
|
|
#ifdef RB_DEBUG_DB_PORT
|
|
|
|
log::debug
|
|
|
|
{
|
|
|
|
db::log, "shared_mutex %lu %p LOCK SHARED", ctx::id(), this
|
|
|
|
};
|
|
|
|
#endif
|
|
|
|
|
|
|
|
assert_main_thread();
|
2018-05-30 09:14:07 +02:00
|
|
|
mu.lock_shared();
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
rocksdb::port::RWMutex::WriteLock()
|
|
|
|
{
|
2018-06-01 20:31:46 +02:00
|
|
|
#ifdef RB_DEBUG_DB_PORT
|
|
|
|
log::debug
|
|
|
|
{
|
|
|
|
db::log, "shared_mutex %lu %p LOCK", ctx::id(), this
|
|
|
|
};
|
|
|
|
#endif
|
|
|
|
|
|
|
|
assert_main_thread();
|
2018-05-30 09:14:07 +02:00
|
|
|
mu.lock();
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
rocksdb::port::RWMutex::ReadUnlock()
|
|
|
|
{
|
2018-06-01 20:31:46 +02:00
|
|
|
#ifdef RB_DEBUG_DB_PORT
|
|
|
|
log::debug
|
|
|
|
{
|
|
|
|
db::log, "shared_mutex %lu %p UNLOCK SHARED", ctx::id(), this
|
|
|
|
};
|
|
|
|
#endif
|
|
|
|
|
|
|
|
assert_main_thread();
|
2018-05-30 09:14:07 +02:00
|
|
|
mu.unlock_shared();
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
rocksdb::port::RWMutex::WriteUnlock()
|
|
|
|
{
|
2018-06-01 20:31:46 +02:00
|
|
|
#ifdef RB_DEBUG_DB_PORT
|
|
|
|
log::debug
|
|
|
|
{
|
|
|
|
db::log, "shared_mutex %lu %p UNLOCK", ctx::id(), this
|
|
|
|
};
|
|
|
|
#endif
|
|
|
|
|
|
|
|
assert_main_thread();
|
2018-05-30 09:14:07 +02:00
|
|
|
mu.unlock();
|
|
|
|
}
|
|
|
|
|
|
|
|
//
|
|
|
|
// CondVar
|
|
|
|
//
|
|
|
|
|
|
|
|
rocksdb::port::CondVar::CondVar(Mutex *mu)
|
|
|
|
:mu{mu}
|
|
|
|
{
|
2018-06-01 20:31:46 +02:00
|
|
|
#ifdef RB_DEBUG_DB_PORT_
|
|
|
|
log::debug
|
|
|
|
{
|
|
|
|
db::log, "cond %lu %p %p CTOR", ctx::id(), this, mu
|
|
|
|
};
|
|
|
|
#endif
|
2018-05-30 09:14:07 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::port::CondVar::~CondVar()
|
|
|
|
{
|
2018-06-01 20:31:46 +02:00
|
|
|
#ifdef RB_DEBUG_DB_PORT_
|
|
|
|
log::debug
|
|
|
|
{
|
|
|
|
db::log, "cond %lu %p %p DTOR", ctx::id(), this, mu
|
|
|
|
};
|
|
|
|
#endif
|
2018-05-30 09:14:07 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
rocksdb::port::CondVar::Wait()
|
|
|
|
{
|
2018-06-01 20:31:46 +02:00
|
|
|
#ifdef RB_DEBUG_DB_PORT
|
|
|
|
log::debug
|
|
|
|
{
|
|
|
|
db::log, "cond %lu %p %p WAIT", ctx::id(), this, mu
|
|
|
|
};
|
|
|
|
#endif
|
|
|
|
|
2018-05-30 09:14:07 +02:00
|
|
|
assert(mu);
|
2018-06-01 20:31:46 +02:00
|
|
|
assert_main_thread();
|
2018-08-16 13:48:21 +02:00
|
|
|
mu->AssertHeld();
|
2018-06-01 20:31:46 +02:00
|
|
|
std::unique_lock<decltype(mu->mu)> l
|
|
|
|
{
|
|
|
|
mu->mu, std::adopt_lock
|
|
|
|
};
|
|
|
|
|
|
|
|
cv.wait(l);
|
2018-05-30 09:14:07 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
// Returns true if timeout occurred
|
|
|
|
bool
|
|
|
|
rocksdb::port::CondVar::TimedWait(uint64_t abs_time_us)
|
|
|
|
{
|
2018-06-01 20:31:46 +02:00
|
|
|
#ifdef RB_DEBUG_DB_PORT
|
|
|
|
log::debug
|
|
|
|
{
|
|
|
|
db::log, "cond %lu %p %p WAIT_UNTIL %lu", ctx::id(), this, mu, abs_time_us
|
|
|
|
};
|
|
|
|
#endif
|
|
|
|
|
2018-05-30 09:14:07 +02:00
|
|
|
assert(mu);
|
2018-06-01 20:31:46 +02:00
|
|
|
assert_main_thread();
|
2018-08-16 13:48:21 +02:00
|
|
|
mu->AssertHeld();
|
2018-05-30 09:14:07 +02:00
|
|
|
const std::chrono::microseconds us(abs_time_us);
|
2018-06-01 20:31:46 +02:00
|
|
|
const std::chrono::steady_clock::time_point tp(us);
|
|
|
|
std::unique_lock<decltype(mu->mu)> l
|
|
|
|
{
|
|
|
|
mu->mu, std::adopt_lock
|
|
|
|
};
|
|
|
|
|
|
|
|
return cv.wait_until(l, tp) == std::cv_status::timeout;
|
2018-05-30 09:14:07 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
rocksdb::port::CondVar::Signal()
|
|
|
|
{
|
2018-06-01 20:31:46 +02:00
|
|
|
#ifdef RB_DEBUG_DB_PORT
|
|
|
|
log::debug
|
|
|
|
{
|
|
|
|
db::log, "cond %lu %p %p NOTIFY", ctx::id(), this, mu
|
|
|
|
};
|
|
|
|
#endif
|
|
|
|
|
|
|
|
assert_main_thread();
|
2018-05-30 09:14:07 +02:00
|
|
|
cv.notify_one();
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
rocksdb::port::CondVar::SignalAll()
|
|
|
|
{
|
2018-06-01 20:31:46 +02:00
|
|
|
#ifdef RB_DEBUG_DB_PORT
|
|
|
|
log::debug
|
|
|
|
{
|
|
|
|
db::log, "cond %lu %p %p BROADCAST", ctx::id(), this, mu
|
|
|
|
};
|
|
|
|
#endif
|
|
|
|
|
|
|
|
assert_main_thread();
|
2018-05-30 09:14:07 +02:00
|
|
|
cv.notify_all();
|
|
|
|
}
|
|
|
|
|
|
|
|
#endif // IRCD_DB_PORT
|
|
|
|
|
2018-06-09 22:40:21 +02:00
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
//
|
|
|
|
// db/database/env/state.h
|
|
|
|
//
|
|
|
|
|
2017-09-19 04:19:02 +02:00
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
//
|
2018-01-30 18:58:36 +01:00
|
|
|
// db/txn.h
|
2017-09-19 04:19:02 +02:00
|
|
|
//
|
|
|
|
|
2018-03-28 00:21:30 +02:00
|
|
|
void
|
|
|
|
ircd::db::get(database &d,
|
|
|
|
const uint64_t &seq,
|
|
|
|
const seq_closure &closure)
|
|
|
|
{
|
|
|
|
for_each(d, seq, seq_closure_bool{[&closure]
|
|
|
|
(txn &txn, const uint64_t &seq)
|
|
|
|
{
|
|
|
|
closure(txn, seq);
|
|
|
|
return false;
|
|
|
|
}});
|
|
|
|
}
|
|
|
|
|
2018-03-23 04:49:11 +01:00
|
|
|
void
|
|
|
|
ircd::db::for_each(database &d,
|
|
|
|
const uint64_t &seq,
|
|
|
|
const seq_closure &closure)
|
|
|
|
{
|
|
|
|
for_each(d, seq, seq_closure_bool{[&closure]
|
|
|
|
(txn &txn, const uint64_t &seq)
|
|
|
|
{
|
|
|
|
closure(txn, seq);
|
|
|
|
return true;
|
|
|
|
}});
|
|
|
|
}
|
|
|
|
|
|
|
|
bool
|
|
|
|
ircd::db::for_each(database &d,
|
|
|
|
const uint64_t &seq,
|
|
|
|
const seq_closure_bool &closure)
|
|
|
|
{
|
|
|
|
std::unique_ptr<rocksdb::TransactionLogIterator> tit;
|
|
|
|
throw_on_error
|
|
|
|
{
|
|
|
|
d.d->GetUpdatesSince(seq, &tit)
|
|
|
|
};
|
|
|
|
|
|
|
|
assert(bool(tit));
|
|
|
|
for(; tit->Valid(); tit->Next())
|
|
|
|
{
|
|
|
|
auto batchres
|
|
|
|
{
|
|
|
|
tit->GetBatch()
|
|
|
|
};
|
|
|
|
|
|
|
|
throw_on_error
|
|
|
|
{
|
|
|
|
tit->status()
|
|
|
|
};
|
|
|
|
|
|
|
|
db::txn txn
|
|
|
|
{
|
|
|
|
d, std::move(batchres.writeBatchPtr)
|
|
|
|
};
|
|
|
|
|
|
|
|
assert(bool(txn.wb));
|
|
|
|
if(!closure(txn, batchres.sequence))
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2017-09-19 04:19:02 +02:00
|
|
|
std::string
|
2018-01-30 18:58:36 +01:00
|
|
|
ircd::db::debug(const txn &t)
|
2017-09-19 04:19:02 +02:00
|
|
|
{
|
|
|
|
const rocksdb::WriteBatch &wb(t);
|
|
|
|
return db::debug(wb);
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
2018-01-30 18:58:36 +01:00
|
|
|
ircd::db::for_each(const txn &t,
|
2017-09-19 04:19:02 +02:00
|
|
|
const std::function<void (const delta &)> &closure)
|
2017-09-19 05:40:13 +02:00
|
|
|
{
|
|
|
|
const auto re{[&closure]
|
|
|
|
(const delta &delta)
|
|
|
|
{
|
|
|
|
closure(delta);
|
|
|
|
return true;
|
|
|
|
}};
|
|
|
|
|
|
|
|
const database &d(t);
|
|
|
|
const rocksdb::WriteBatch &wb{t};
|
2018-01-30 18:58:36 +01:00
|
|
|
txn::handler h{d, re};
|
2017-09-19 05:40:13 +02:00
|
|
|
wb.Iterate(&h);
|
|
|
|
}
|
|
|
|
|
|
|
|
bool
|
2018-04-17 00:16:51 +02:00
|
|
|
ircd::db::for_each(const txn &t,
|
|
|
|
const std::function<bool (const delta &)> &closure)
|
2017-09-19 04:19:02 +02:00
|
|
|
{
|
|
|
|
const database &d(t);
|
|
|
|
const rocksdb::WriteBatch &wb{t};
|
2018-01-30 18:58:36 +01:00
|
|
|
txn::handler h{d, closure};
|
2017-09-19 05:40:13 +02:00
|
|
|
wb.Iterate(&h);
|
|
|
|
return h._continue;
|
2017-09-19 04:19:02 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
///
|
2018-03-23 04:47:46 +01:00
|
|
|
/// handler (db/database/txn.h)
|
2017-09-19 04:19:02 +02:00
|
|
|
///
|
|
|
|
|
|
|
|
rocksdb::Status
|
2018-01-30 18:58:36 +01:00
|
|
|
ircd::db::txn::handler::PutCF(const uint32_t cfid,
|
2017-09-19 04:19:02 +02:00
|
|
|
const Slice &key,
|
|
|
|
const Slice &val)
|
|
|
|
noexcept
|
|
|
|
{
|
|
|
|
return callback(cfid, op::SET, key, val);
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
2018-01-30 18:58:36 +01:00
|
|
|
ircd::db::txn::handler::DeleteCF(const uint32_t cfid,
|
2017-09-19 04:19:02 +02:00
|
|
|
const Slice &key)
|
|
|
|
noexcept
|
|
|
|
{
|
|
|
|
return callback(cfid, op::DELETE, key, {});
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
2018-01-30 18:58:36 +01:00
|
|
|
ircd::db::txn::handler::DeleteRangeCF(const uint32_t cfid,
|
2017-09-19 04:19:02 +02:00
|
|
|
const Slice &begin,
|
|
|
|
const Slice &end)
|
|
|
|
noexcept
|
|
|
|
{
|
|
|
|
return callback(cfid, op::DELETE_RANGE, begin, end);
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
2018-01-30 18:58:36 +01:00
|
|
|
ircd::db::txn::handler::SingleDeleteCF(const uint32_t cfid,
|
2017-09-19 04:19:02 +02:00
|
|
|
const Slice &key)
|
|
|
|
noexcept
|
|
|
|
{
|
|
|
|
return callback(cfid, op::SINGLE_DELETE, key, {});
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
2018-01-30 18:58:36 +01:00
|
|
|
ircd::db::txn::handler::MergeCF(const uint32_t cfid,
|
2017-09-19 04:19:02 +02:00
|
|
|
const Slice &key,
|
|
|
|
const Slice &value)
|
|
|
|
noexcept
|
|
|
|
{
|
|
|
|
return callback(cfid, op::MERGE, key, value);
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
2018-01-30 18:58:36 +01:00
|
|
|
ircd::db::txn::handler::MarkBeginPrepare()
|
2017-09-19 04:19:02 +02:00
|
|
|
noexcept
|
|
|
|
{
|
|
|
|
return Status::OK();
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
2018-01-30 18:58:36 +01:00
|
|
|
ircd::db::txn::handler::MarkEndPrepare(const Slice &xid)
|
2017-09-19 04:19:02 +02:00
|
|
|
noexcept
|
|
|
|
{
|
|
|
|
return Status::OK();
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
2018-01-30 18:58:36 +01:00
|
|
|
ircd::db::txn::handler::MarkCommit(const Slice &xid)
|
2017-09-19 04:19:02 +02:00
|
|
|
noexcept
|
|
|
|
{
|
|
|
|
return Status::OK();
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
2018-01-30 18:58:36 +01:00
|
|
|
ircd::db::txn::handler::MarkRollback(const Slice &xid)
|
2017-09-19 04:19:02 +02:00
|
|
|
noexcept
|
|
|
|
{
|
|
|
|
return Status::OK();
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
2018-01-30 18:58:36 +01:00
|
|
|
ircd::db::txn::handler::callback(const uint32_t &cfid,
|
2017-09-19 04:19:02 +02:00
|
|
|
const op &op,
|
|
|
|
const Slice &a,
|
|
|
|
const Slice &b)
|
|
|
|
noexcept try
|
|
|
|
{
|
|
|
|
auto &c{d[cfid]};
|
|
|
|
const delta delta
|
|
|
|
{
|
|
|
|
op,
|
|
|
|
db::name(c),
|
|
|
|
slice(a),
|
|
|
|
slice(b)
|
|
|
|
};
|
|
|
|
|
|
|
|
return callback(delta);
|
|
|
|
}
|
|
|
|
catch(const std::exception &e)
|
|
|
|
{
|
|
|
|
_continue = false;
|
2018-05-29 10:09:15 +02:00
|
|
|
log::critical
|
|
|
|
{
|
|
|
|
"txn::handler: cfid[%u]: %s", cfid, e.what()
|
|
|
|
};
|
|
|
|
|
2017-11-26 01:20:42 +01:00
|
|
|
ircd::terminate();
|
2017-09-19 04:19:02 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status
|
2018-01-30 18:58:36 +01:00
|
|
|
ircd::db::txn::handler::callback(const delta &delta)
|
2017-09-19 04:19:02 +02:00
|
|
|
noexcept try
|
|
|
|
{
|
2017-09-19 05:40:13 +02:00
|
|
|
_continue = cb(delta);
|
2017-09-19 04:19:02 +02:00
|
|
|
return Status::OK();
|
|
|
|
}
|
|
|
|
catch(const std::exception &e)
|
|
|
|
{
|
|
|
|
_continue = false;
|
2017-09-19 05:40:13 +02:00
|
|
|
return Status::OK();
|
2017-09-19 04:19:02 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
bool
|
2018-01-30 18:58:36 +01:00
|
|
|
ircd::db::txn::handler::Continue()
|
2017-09-19 04:19:02 +02:00
|
|
|
noexcept
|
|
|
|
{
|
|
|
|
return _continue;
|
|
|
|
}
|
|
|
|
|
|
|
|
//
|
2018-01-30 18:58:36 +01:00
|
|
|
// txn
|
2017-09-19 04:19:02 +02:00
|
|
|
//
|
|
|
|
|
2018-01-30 18:58:36 +01:00
|
|
|
ircd::db::txn::txn(database &d)
|
2018-03-23 04:48:23 +01:00
|
|
|
:txn{d, opts{}}
|
2017-09-19 04:19:02 +02:00
|
|
|
{
|
|
|
|
}
|
|
|
|
|
2018-01-30 18:58:36 +01:00
|
|
|
ircd::db::txn::txn(database &d,
|
2017-09-19 04:19:02 +02:00
|
|
|
const opts &opts)
|
|
|
|
:d{&d}
|
|
|
|
,wb
|
|
|
|
{
|
|
|
|
std::make_unique<rocksdb::WriteBatch>(opts.reserve_bytes, opts.max_bytes)
|
|
|
|
}
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
2018-03-23 04:48:23 +01:00
|
|
|
ircd::db::txn::txn(database &d,
|
|
|
|
std::unique_ptr<rocksdb::WriteBatch> &&wb)
|
|
|
|
:d{&d}
|
|
|
|
,wb{std::move(wb)}
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
2018-01-30 18:58:36 +01:00
|
|
|
ircd::db::txn::~txn()
|
2017-09-19 04:19:02 +02:00
|
|
|
noexcept
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
2018-01-30 18:58:36 +01:00
|
|
|
ircd::db::txn::operator()(const sopts &opts)
|
2017-09-19 04:19:02 +02:00
|
|
|
{
|
|
|
|
assert(bool(d));
|
|
|
|
operator()(*d, opts);
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
2018-01-30 18:58:36 +01:00
|
|
|
ircd::db::txn::operator()(database &d,
|
2017-09-19 04:19:02 +02:00
|
|
|
const sopts &opts)
|
|
|
|
{
|
|
|
|
assert(bool(wb));
|
|
|
|
commit(d, *wb, opts);
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
2018-01-30 18:58:36 +01:00
|
|
|
ircd::db::txn::clear()
|
2017-09-19 04:19:02 +02:00
|
|
|
{
|
|
|
|
assert(bool(wb));
|
|
|
|
wb->Clear();
|
|
|
|
}
|
|
|
|
|
|
|
|
size_t
|
2018-01-30 18:58:36 +01:00
|
|
|
ircd::db::txn::size()
|
2017-09-19 04:19:02 +02:00
|
|
|
const
|
|
|
|
{
|
|
|
|
assert(bool(wb));
|
|
|
|
return wb->Count();
|
|
|
|
}
|
|
|
|
|
|
|
|
size_t
|
2018-01-30 18:58:36 +01:00
|
|
|
ircd::db::txn::bytes()
|
2017-09-19 04:19:02 +02:00
|
|
|
const
|
|
|
|
{
|
|
|
|
assert(bool(wb));
|
|
|
|
return wb->GetDataSize();
|
|
|
|
}
|
|
|
|
|
|
|
|
bool
|
2018-01-30 18:58:36 +01:00
|
|
|
ircd::db::txn::has(const op &op)
|
2017-09-19 04:19:02 +02:00
|
|
|
const
|
|
|
|
{
|
|
|
|
assert(bool(wb));
|
|
|
|
switch(op)
|
|
|
|
{
|
|
|
|
case op::GET: assert(0); return false;
|
|
|
|
case op::SET: return wb->HasPut();
|
|
|
|
case op::MERGE: return wb->HasMerge();
|
|
|
|
case op::DELETE: return wb->HasDelete();
|
|
|
|
case op::DELETE_RANGE: return wb->HasDeleteRange();
|
|
|
|
case op::SINGLE_DELETE: return wb->HasSingleDelete();
|
|
|
|
}
|
|
|
|
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
2017-09-19 05:40:13 +02:00
|
|
|
bool
|
2018-01-30 18:58:36 +01:00
|
|
|
ircd::db::txn::has(const op &op,
|
2017-09-19 05:40:13 +02:00
|
|
|
const string_view &col)
|
|
|
|
const
|
|
|
|
{
|
2018-04-17 00:16:51 +02:00
|
|
|
return !for_each(*this, delta_closure_bool{[&op, &col]
|
2017-09-19 05:40:13 +02:00
|
|
|
(const auto &delta)
|
|
|
|
{
|
2017-09-28 03:25:39 +02:00
|
|
|
return std::get<delta.OP>(delta) == op &&
|
|
|
|
std::get<delta.COL>(delta) == col;
|
2018-04-17 00:16:51 +02:00
|
|
|
}});
|
2017-09-19 05:40:13 +02:00
|
|
|
}
|
|
|
|
|
2018-04-17 00:16:51 +02:00
|
|
|
void
|
2018-01-30 18:58:36 +01:00
|
|
|
ircd::db::txn::at(const op &op,
|
2018-04-17 00:16:51 +02:00
|
|
|
const string_view &col,
|
|
|
|
const delta_closure &closure)
|
2017-09-19 05:40:13 +02:00
|
|
|
const
|
|
|
|
{
|
2018-04-17 00:16:51 +02:00
|
|
|
if(!get(op, col, closure))
|
|
|
|
throw not_found
|
|
|
|
{
|
|
|
|
"db::txn::at(%s, %s): no matching delta in transaction",
|
|
|
|
reflect(op),
|
|
|
|
col
|
|
|
|
};
|
2017-09-19 05:40:13 +02:00
|
|
|
}
|
|
|
|
|
2018-04-17 00:16:51 +02:00
|
|
|
bool
|
2018-01-30 18:58:36 +01:00
|
|
|
ircd::db::txn::get(const op &op,
|
2018-04-17 00:16:51 +02:00
|
|
|
const string_view &col,
|
|
|
|
const delta_closure &closure)
|
2017-09-19 05:40:13 +02:00
|
|
|
const
|
|
|
|
{
|
2018-04-17 00:16:51 +02:00
|
|
|
return !for_each(*this, delta_closure_bool{[&op, &col, &closure]
|
2017-09-19 05:40:13 +02:00
|
|
|
(const delta &delta)
|
|
|
|
{
|
2017-09-28 03:25:39 +02:00
|
|
|
if(std::get<delta.OP>(delta) == op &&
|
|
|
|
std::get<delta.COL>(delta) == col)
|
2017-09-19 05:40:13 +02:00
|
|
|
{
|
2018-04-17 00:16:51 +02:00
|
|
|
closure(delta);
|
|
|
|
return false;
|
2017-09-19 05:40:13 +02:00
|
|
|
}
|
2018-04-17 00:16:51 +02:00
|
|
|
else return true;
|
|
|
|
}});
|
|
|
|
}
|
2017-09-19 05:40:13 +02:00
|
|
|
|
2018-04-17 00:16:51 +02:00
|
|
|
bool
|
|
|
|
ircd::db::txn::has(const op &op,
|
|
|
|
const string_view &col,
|
|
|
|
const string_view &key)
|
|
|
|
const
|
|
|
|
{
|
|
|
|
return !for_each(*this, delta_closure_bool{[&op, &col, &key]
|
|
|
|
(const auto &delta)
|
|
|
|
{
|
|
|
|
return std::get<delta.OP>(delta) == op &&
|
|
|
|
std::get<delta.COL>(delta) == col &&
|
|
|
|
std::get<delta.KEY>(delta) == key;
|
|
|
|
}});
|
2017-09-19 05:40:13 +02:00
|
|
|
}
|
|
|
|
|
2018-04-17 00:16:51 +02:00
|
|
|
void
|
2018-01-30 18:58:36 +01:00
|
|
|
ircd::db::txn::at(const op &op,
|
2017-09-19 05:40:13 +02:00
|
|
|
const string_view &col,
|
2018-04-17 00:16:51 +02:00
|
|
|
const string_view &key,
|
|
|
|
const value_closure &closure)
|
2017-09-19 05:40:13 +02:00
|
|
|
const
|
|
|
|
{
|
2018-04-17 00:16:51 +02:00
|
|
|
if(!get(op, col, key, closure))
|
|
|
|
throw not_found
|
|
|
|
{
|
|
|
|
"db::txn::at(%s, %s, %s): no matching delta in transaction",
|
|
|
|
reflect(op),
|
|
|
|
col,
|
|
|
|
key
|
|
|
|
};
|
2017-09-19 05:40:13 +02:00
|
|
|
}
|
|
|
|
|
2018-04-17 00:16:51 +02:00
|
|
|
bool
|
2018-01-30 18:58:36 +01:00
|
|
|
ircd::db::txn::get(const op &op,
|
2017-09-19 05:40:13 +02:00
|
|
|
const string_view &col,
|
2018-04-17 00:16:51 +02:00
|
|
|
const string_view &key,
|
|
|
|
const value_closure &closure)
|
2017-09-19 05:40:13 +02:00
|
|
|
const
|
|
|
|
{
|
2018-04-17 00:16:51 +02:00
|
|
|
return !for_each(*this, delta_closure_bool{[&op, &col, &key, &closure]
|
2017-09-19 05:40:13 +02:00
|
|
|
(const delta &delta)
|
|
|
|
{
|
2017-09-28 03:25:39 +02:00
|
|
|
if(std::get<delta.OP>(delta) == op &&
|
|
|
|
std::get<delta.COL>(delta) == col &&
|
|
|
|
std::get<delta.KEY>(delta) == key)
|
2017-09-19 05:40:13 +02:00
|
|
|
{
|
2018-04-17 00:16:51 +02:00
|
|
|
closure(std::get<delta.VAL>(delta));
|
|
|
|
return false;
|
2017-09-19 05:40:13 +02:00
|
|
|
}
|
2018-04-17 00:16:51 +02:00
|
|
|
else return true;
|
|
|
|
}});
|
2017-09-19 05:40:13 +02:00
|
|
|
}
|
|
|
|
|
2018-01-30 18:58:36 +01:00
|
|
|
ircd::db::txn::operator
|
2017-09-19 04:19:02 +02:00
|
|
|
ircd::db::database &()
|
|
|
|
{
|
|
|
|
assert(bool(d));
|
|
|
|
return *d;
|
|
|
|
}
|
|
|
|
|
2018-01-30 18:58:36 +01:00
|
|
|
ircd::db::txn::operator
|
2017-09-19 04:19:02 +02:00
|
|
|
rocksdb::WriteBatch &()
|
|
|
|
{
|
|
|
|
assert(bool(wb));
|
|
|
|
return *wb;
|
|
|
|
}
|
|
|
|
|
2018-01-30 18:58:36 +01:00
|
|
|
ircd::db::txn::operator
|
2017-09-19 04:19:02 +02:00
|
|
|
const ircd::db::database &()
|
|
|
|
const
|
|
|
|
{
|
|
|
|
assert(bool(d));
|
|
|
|
return *d;
|
|
|
|
}
|
|
|
|
|
2018-01-30 18:58:36 +01:00
|
|
|
ircd::db::txn::operator
|
2017-09-19 04:19:02 +02:00
|
|
|
const rocksdb::WriteBatch &()
|
|
|
|
const
|
|
|
|
{
|
|
|
|
assert(bool(wb));
|
|
|
|
return *wb;
|
|
|
|
}
|
|
|
|
|
|
|
|
//
|
|
|
|
// Checkpoint
|
|
|
|
//
|
|
|
|
|
2018-01-30 18:58:36 +01:00
|
|
|
ircd::db::txn::checkpoint::checkpoint(txn &t)
|
2017-09-19 04:19:02 +02:00
|
|
|
:t{t}
|
|
|
|
{
|
|
|
|
assert(bool(t.wb));
|
|
|
|
t.wb->SetSavePoint();
|
|
|
|
}
|
|
|
|
|
2018-01-30 18:58:36 +01:00
|
|
|
ircd::db::txn::checkpoint::~checkpoint()
|
2017-09-19 04:19:02 +02:00
|
|
|
noexcept
|
|
|
|
{
|
|
|
|
if(likely(!std::uncaught_exception()))
|
|
|
|
throw_on_error { t.wb->PopSavePoint() };
|
|
|
|
else
|
|
|
|
throw_on_error { t.wb->RollbackToSavePoint() };
|
|
|
|
}
|
|
|
|
|
2018-01-30 18:58:36 +01:00
|
|
|
ircd::db::txn::append::append(txn &t,
|
2017-09-19 04:19:02 +02:00
|
|
|
const string_view &key,
|
|
|
|
const json::iov &iov)
|
|
|
|
{
|
|
|
|
std::for_each(std::begin(iov), std::end(iov), [&t, &key]
|
|
|
|
(const auto &member)
|
|
|
|
{
|
|
|
|
append
|
|
|
|
{
|
|
|
|
t, delta
|
|
|
|
{
|
|
|
|
member.first, // col
|
|
|
|
key, // key
|
|
|
|
member.second // val
|
|
|
|
}
|
|
|
|
};
|
|
|
|
});
|
|
|
|
}
|
|
|
|
|
2018-01-30 18:58:36 +01:00
|
|
|
ircd::db::txn::append::append(txn &t,
|
2017-09-19 04:19:02 +02:00
|
|
|
const delta &delta)
|
|
|
|
{
|
|
|
|
assert(bool(t.d));
|
|
|
|
append(t, *t.d, delta);
|
|
|
|
}
|
|
|
|
|
2018-01-30 18:58:36 +01:00
|
|
|
ircd::db::txn::append::append(txn &t,
|
2017-09-19 04:19:02 +02:00
|
|
|
const row::delta &delta)
|
|
|
|
{
|
|
|
|
assert(0);
|
|
|
|
}
|
|
|
|
|
2018-01-30 18:58:36 +01:00
|
|
|
ircd::db::txn::append::append(txn &t,
|
2017-09-19 04:19:02 +02:00
|
|
|
const cell::delta &delta)
|
|
|
|
{
|
|
|
|
db::append(*t.wb, delta);
|
|
|
|
}
|
|
|
|
|
2018-01-30 18:58:36 +01:00
|
|
|
ircd::db::txn::append::append(txn &t,
|
2017-09-19 04:19:02 +02:00
|
|
|
column &c,
|
|
|
|
const column::delta &delta)
|
|
|
|
{
|
|
|
|
db::append(*t.wb, c, delta);
|
|
|
|
}
|
|
|
|
|
2018-01-30 18:58:36 +01:00
|
|
|
ircd::db::txn::append::append(txn &t,
|
2017-09-19 04:19:02 +02:00
|
|
|
database &d,
|
|
|
|
const delta &delta)
|
|
|
|
{
|
|
|
|
db::column c{d[std::get<1>(delta)]};
|
|
|
|
db::append(*t.wb, c, db::column::delta
|
|
|
|
{
|
|
|
|
std::get<op>(delta),
|
|
|
|
std::get<2>(delta),
|
|
|
|
std::get<3>(delta)
|
|
|
|
});
|
|
|
|
}
|
|
|
|
|
2017-09-08 11:02:47 +02:00
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
//
|
|
|
|
// db/index.h
|
|
|
|
//
|
|
|
|
|
2017-09-22 05:08:11 +02:00
|
|
|
const ircd::db::gopts
|
|
|
|
ircd::db::index::applied_opts
|
2017-09-08 11:02:47 +02:00
|
|
|
{
|
2017-09-22 05:08:11 +02:00
|
|
|
{ get::PREFIX }
|
|
|
|
};
|
2017-09-08 11:02:47 +02:00
|
|
|
|
2017-09-22 05:08:11 +02:00
|
|
|
template<class pos>
|
|
|
|
bool
|
2017-09-22 12:31:44 +02:00
|
|
|
ircd::db::seek(index::const_iterator_base &it,
|
2018-05-25 11:34:44 +02:00
|
|
|
const pos &p)
|
2017-09-22 05:08:11 +02:00
|
|
|
{
|
2018-05-25 11:34:44 +02:00
|
|
|
it.opts |= index::applied_opts;
|
|
|
|
return seek(static_cast<column::const_iterator_base &>(it), p);
|
2017-09-22 05:08:11 +02:00
|
|
|
}
|
2018-05-25 11:34:44 +02:00
|
|
|
template bool ircd::db::seek<ircd::db::pos>(index::const_iterator_base &, const pos &);
|
|
|
|
template bool ircd::db::seek<ircd::string_view>(index::const_iterator_base &, const string_view &);
|
2017-09-08 11:02:47 +02:00
|
|
|
|
2017-09-22 05:08:11 +02:00
|
|
|
ircd::db::index::const_iterator
|
2017-09-22 12:31:44 +02:00
|
|
|
ircd::db::index::begin(const string_view &key,
|
|
|
|
gopts opts)
|
2017-09-22 05:08:11 +02:00
|
|
|
{
|
|
|
|
const_iterator ret
|
|
|
|
{
|
2018-05-25 11:34:44 +02:00
|
|
|
c, {}, std::move(opts)
|
2017-09-22 05:08:11 +02:00
|
|
|
};
|
|
|
|
|
2018-05-25 11:34:44 +02:00
|
|
|
seek(ret, key);
|
2017-09-22 05:08:11 +02:00
|
|
|
return ret;
|
2017-09-08 11:02:47 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
ircd::db::index::const_iterator
|
2017-09-22 12:31:44 +02:00
|
|
|
ircd::db::index::end(const string_view &key,
|
|
|
|
gopts opts)
|
2017-09-08 11:02:47 +02:00
|
|
|
{
|
2017-09-22 05:08:11 +02:00
|
|
|
const_iterator ret
|
|
|
|
{
|
2018-05-25 11:34:44 +02:00
|
|
|
c, {}, std::move(opts)
|
2017-09-22 05:08:11 +02:00
|
|
|
};
|
|
|
|
|
2018-05-25 11:34:44 +02:00
|
|
|
if(seek(ret, key))
|
|
|
|
seek(ret, pos::END);
|
2017-09-22 12:31:44 +02:00
|
|
|
|
2017-09-22 05:08:11 +02:00
|
|
|
return ret;
|
2017-09-08 11:02:47 +02:00
|
|
|
}
|
|
|
|
|
2017-09-24 04:47:36 +02:00
|
|
|
/// NOTE: RocksDB says they don't support reverse iteration over a prefix range
|
|
|
|
/// This means we have to forward scan to the end and then walk back! Reverse
|
|
|
|
/// iterations of an index shoud only be used for debugging and statistics! The
|
|
|
|
/// index should be ordered the way it will be primarily accessed using the
|
|
|
|
/// comparator. If it will be accessed in different directions, make another
|
|
|
|
/// index column.
|
2017-09-22 12:31:44 +02:00
|
|
|
ircd::db::index::const_reverse_iterator
|
|
|
|
ircd::db::index::rbegin(const string_view &key,
|
|
|
|
gopts opts)
|
2017-09-08 11:02:47 +02:00
|
|
|
{
|
2017-09-22 12:31:44 +02:00
|
|
|
const_reverse_iterator ret
|
|
|
|
{
|
2018-05-25 11:34:44 +02:00
|
|
|
c, {}, std::move(opts)
|
2017-09-22 12:31:44 +02:00
|
|
|
};
|
|
|
|
|
2018-05-25 11:34:44 +02:00
|
|
|
if(seek(ret, key))
|
2017-09-24 04:47:36 +02:00
|
|
|
{
|
2018-05-25 11:34:44 +02:00
|
|
|
while(seek(ret, pos::NEXT));
|
|
|
|
seek(ret, pos::PREV);
|
2017-09-24 04:47:36 +02:00
|
|
|
}
|
2017-09-22 12:31:44 +02:00
|
|
|
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
|
|
|
ircd::db::index::const_reverse_iterator
|
|
|
|
ircd::db::index::rend(const string_view &key,
|
|
|
|
gopts opts)
|
|
|
|
{
|
|
|
|
const_reverse_iterator ret
|
|
|
|
{
|
2018-05-25 11:34:44 +02:00
|
|
|
c, {}, std::move(opts)
|
2017-09-22 12:31:44 +02:00
|
|
|
};
|
|
|
|
|
2018-05-25 11:34:44 +02:00
|
|
|
if(seek(ret, key))
|
|
|
|
seek(ret, pos::END);
|
2017-09-22 12:31:44 +02:00
|
|
|
|
|
|
|
return ret;
|
2017-09-08 11:02:47 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
//
|
|
|
|
// const_iterator
|
|
|
|
//
|
|
|
|
|
2017-09-22 05:08:11 +02:00
|
|
|
ircd::db::index::const_iterator &
|
|
|
|
ircd::db::index::const_iterator::operator--()
|
2017-09-08 11:02:47 +02:00
|
|
|
{
|
2017-09-22 12:31:44 +02:00
|
|
|
if(likely(bool(*this)))
|
|
|
|
seek(*this, pos::PREV);
|
|
|
|
else
|
|
|
|
seek(*this, pos::BACK);
|
|
|
|
|
2017-09-22 05:08:11 +02:00
|
|
|
return *this;
|
|
|
|
}
|
|
|
|
|
|
|
|
ircd::db::index::const_iterator &
|
|
|
|
ircd::db::index::const_iterator::operator++()
|
|
|
|
{
|
2017-09-22 12:31:44 +02:00
|
|
|
if(likely(bool(*this)))
|
|
|
|
seek(*this, pos::NEXT);
|
|
|
|
else
|
|
|
|
seek(*this, pos::FRONT);
|
|
|
|
|
2017-09-22 05:08:11 +02:00
|
|
|
return *this;
|
2017-09-08 11:02:47 +02:00
|
|
|
}
|
|
|
|
|
2017-09-22 12:31:44 +02:00
|
|
|
ircd::db::index::const_reverse_iterator &
|
|
|
|
ircd::db::index::const_reverse_iterator::operator--()
|
|
|
|
{
|
|
|
|
if(likely(bool(*this)))
|
|
|
|
seek(*this, pos::NEXT);
|
|
|
|
else
|
|
|
|
seek(*this, pos::FRONT);
|
|
|
|
|
|
|
|
return *this;
|
|
|
|
}
|
|
|
|
|
|
|
|
ircd::db::index::const_reverse_iterator &
|
|
|
|
ircd::db::index::const_reverse_iterator::operator++()
|
|
|
|
{
|
|
|
|
if(likely(bool(*this)))
|
|
|
|
seek(*this, pos::PREV);
|
|
|
|
else
|
|
|
|
seek(*this, pos::BACK);
|
|
|
|
|
|
|
|
return *this;
|
|
|
|
}
|
|
|
|
|
|
|
|
const ircd::db::index::const_iterator_base::value_type &
|
|
|
|
ircd::db::index::const_iterator_base::operator*()
|
2017-09-08 11:02:47 +02:00
|
|
|
const
|
|
|
|
{
|
2017-09-22 05:08:11 +02:00
|
|
|
const auto &prefix
|
|
|
|
{
|
|
|
|
describe(*c).prefix
|
|
|
|
};
|
|
|
|
|
|
|
|
// Fetch the full value like a standard column first
|
2017-09-22 12:31:44 +02:00
|
|
|
column::const_iterator_base::operator*();
|
2017-09-22 05:08:11 +02:00
|
|
|
string_view &key{val.first};
|
|
|
|
|
|
|
|
// When there's no prefixing this index column is just
|
|
|
|
// like a normal column. Otherwise, we remove the prefix
|
|
|
|
// from the key the user will end up seeing.
|
2017-09-08 11:02:47 +02:00
|
|
|
if(prefix.has && prefix.has(key))
|
|
|
|
{
|
2017-09-22 05:08:11 +02:00
|
|
|
const auto &first(prefix.get(key));
|
|
|
|
const auto &second(key.substr(first.size()));
|
2017-09-08 11:02:47 +02:00
|
|
|
key = second;
|
|
|
|
}
|
|
|
|
|
2017-09-22 05:08:11 +02:00
|
|
|
return val;
|
2017-09-08 11:02:47 +02:00
|
|
|
}
|
|
|
|
|
2017-09-22 12:31:44 +02:00
|
|
|
const ircd::db::index::const_iterator_base::value_type *
|
|
|
|
ircd::db::index::const_iterator_base::operator->()
|
2017-09-08 11:02:47 +02:00
|
|
|
const
|
|
|
|
{
|
|
|
|
return &this->operator*();
|
|
|
|
}
|
|
|
|
|
2017-03-24 02:36:49 +01:00
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
//
|
2017-04-03 06:02:32 +02:00
|
|
|
// db/cell.h
|
2017-03-24 02:36:49 +01:00
|
|
|
//
|
|
|
|
|
2017-08-23 22:37:47 +02:00
|
|
|
uint64_t
|
|
|
|
ircd::db::sequence(const cell &c)
|
|
|
|
{
|
|
|
|
const database::snapshot &ss(c);
|
|
|
|
return sequence(database::snapshot(c));
|
|
|
|
}
|
|
|
|
|
|
|
|
const std::string &
|
|
|
|
ircd::db::name(const cell &c)
|
|
|
|
{
|
|
|
|
return name(c.c);
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
ircd::db::write(const cell::delta &delta,
|
|
|
|
const sopts &sopts)
|
|
|
|
{
|
|
|
|
write(&delta, &delta + 1, sopts);
|
|
|
|
}
|
|
|
|
|
2017-04-03 06:02:32 +02:00
|
|
|
void
|
|
|
|
ircd::db::write(const sopts &sopts,
|
|
|
|
const std::initializer_list<cell::delta> &deltas)
|
2017-03-24 02:36:49 +01:00
|
|
|
{
|
2017-04-03 06:02:32 +02:00
|
|
|
write(deltas, sopts);
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
ircd::db::write(const std::initializer_list<cell::delta> &deltas,
|
|
|
|
const sopts &sopts)
|
2017-03-24 02:36:49 +01:00
|
|
|
{
|
2017-08-23 22:37:47 +02:00
|
|
|
write(std::begin(deltas), std::end(deltas), sopts);
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
ircd::db::write(const cell::delta *const &begin,
|
|
|
|
const cell::delta *const &end,
|
|
|
|
const sopts &sopts)
|
|
|
|
{
|
|
|
|
if(begin == end)
|
2017-04-03 06:02:32 +02:00
|
|
|
return;
|
2017-03-31 00:57:08 +02:00
|
|
|
|
2017-08-23 22:37:47 +02:00
|
|
|
// Find the database through one of the cell's columns. cell::deltas
|
|
|
|
// may come from different columns so we do nothing else with this.
|
|
|
|
auto &front(*begin);
|
2017-08-29 03:44:27 +02:00
|
|
|
column &c(std::get<cell *>(front)->c);
|
2017-04-03 06:02:32 +02:00
|
|
|
database &d(c);
|
|
|
|
|
|
|
|
rocksdb::WriteBatch batch;
|
2017-08-23 22:37:47 +02:00
|
|
|
std::for_each(begin, end, [&batch]
|
|
|
|
(const cell::delta &delta)
|
|
|
|
{
|
2017-04-03 06:02:32 +02:00
|
|
|
append(batch, delta);
|
2017-08-23 22:37:47 +02:00
|
|
|
});
|
2017-04-03 06:02:32 +02:00
|
|
|
|
2017-09-08 11:14:17 +02:00
|
|
|
commit(d, batch, sopts);
|
2017-04-03 06:02:32 +02:00
|
|
|
}
|
|
|
|
|
2017-08-30 23:05:15 +02:00
|
|
|
template<class pos>
|
|
|
|
bool
|
|
|
|
ircd::db::seek(cell &c,
|
2018-02-06 00:15:02 +01:00
|
|
|
const pos &p,
|
|
|
|
gopts opts)
|
2017-08-30 23:05:15 +02:00
|
|
|
{
|
|
|
|
column &cc(c);
|
|
|
|
database::column &dc(cc);
|
|
|
|
|
2018-02-06 00:15:02 +01:00
|
|
|
if(!opts.snapshot)
|
|
|
|
opts.snapshot = c.ss;
|
|
|
|
|
2017-09-22 05:08:11 +02:00
|
|
|
const auto ropts(make_opts(opts));
|
2017-08-30 23:05:15 +02:00
|
|
|
return seek(dc, p, ropts, c.it);
|
|
|
|
}
|
2018-02-06 00:15:02 +01:00
|
|
|
template bool ircd::db::seek<ircd::db::pos>(cell &, const pos &, gopts);
|
|
|
|
template bool ircd::db::seek<ircd::string_view>(cell &, const string_view &, gopts);
|
2017-08-30 23:05:15 +02:00
|
|
|
|
2017-04-03 06:02:32 +02:00
|
|
|
// Linkage for incomplete rocksdb::Iterator
|
|
|
|
ircd::db::cell::cell()
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
2017-09-14 21:59:20 +02:00
|
|
|
ircd::db::cell::cell(database &d,
|
|
|
|
const string_view &colname,
|
|
|
|
gopts opts)
|
|
|
|
:cell
|
|
|
|
{
|
|
|
|
column(d[colname]), std::unique_ptr<rocksdb::Iterator>{}, std::move(opts)
|
|
|
|
}
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
2017-04-03 06:02:32 +02:00
|
|
|
ircd::db::cell::cell(database &d,
|
|
|
|
const string_view &colname,
|
|
|
|
const string_view &index,
|
|
|
|
gopts opts)
|
|
|
|
:cell
|
|
|
|
{
|
|
|
|
column(d[colname]), index, std::move(opts)
|
|
|
|
}
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
ircd::db::cell::cell(column column,
|
|
|
|
const string_view &index,
|
|
|
|
gopts opts)
|
|
|
|
:c{std::move(column)}
|
|
|
|
,ss{opts.snapshot}
|
2018-02-06 00:16:29 +01:00
|
|
|
,it{!index.empty()? seek(this->c, index, opts) : std::unique_ptr<rocksdb::Iterator>{}}
|
2017-04-03 06:02:32 +02:00
|
|
|
{
|
2017-09-14 21:59:20 +02:00
|
|
|
if(bool(this->it))
|
2017-09-16 20:48:09 +02:00
|
|
|
if(!valid_eq(*this->it, index))
|
2017-09-14 21:59:20 +02:00
|
|
|
this->it.reset();
|
2017-04-03 06:02:32 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
ircd::db::cell::cell(column column,
|
|
|
|
const string_view &index,
|
2017-08-23 22:37:47 +02:00
|
|
|
std::unique_ptr<rocksdb::Iterator> it,
|
|
|
|
gopts opts)
|
2017-04-03 06:02:32 +02:00
|
|
|
:c{std::move(column)}
|
2017-09-14 21:59:20 +02:00
|
|
|
,ss{opts.snapshot}
|
|
|
|
,it{std::move(it)}
|
|
|
|
{
|
2017-09-23 06:55:31 +02:00
|
|
|
if(index.empty())
|
|
|
|
return;
|
|
|
|
|
2018-02-06 00:15:02 +01:00
|
|
|
seek(*this, index, opts);
|
2017-09-16 20:48:09 +02:00
|
|
|
if(!valid_eq(*this->it, index))
|
2017-09-14 21:59:20 +02:00
|
|
|
this->it.reset();
|
|
|
|
}
|
|
|
|
|
|
|
|
ircd::db::cell::cell(column column,
|
|
|
|
std::unique_ptr<rocksdb::Iterator> it,
|
|
|
|
gopts opts)
|
|
|
|
:c{std::move(column)}
|
2017-08-23 22:37:47 +02:00
|
|
|
,ss{std::move(opts.snapshot)}
|
2017-04-03 06:02:32 +02:00
|
|
|
,it{std::move(it)}
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
// Linkage for incomplete rocksdb::Iterator
|
|
|
|
ircd::db::cell::cell(cell &&o)
|
2017-03-31 00:57:08 +02:00
|
|
|
noexcept
|
2017-04-03 06:02:32 +02:00
|
|
|
:c{std::move(o.c)}
|
|
|
|
,ss{std::move(o.ss)}
|
|
|
|
,it{std::move(o.it)}
|
2017-03-31 00:57:08 +02:00
|
|
|
{
|
|
|
|
}
|
|
|
|
|
2017-04-03 06:02:32 +02:00
|
|
|
// Linkage for incomplete rocksdb::Iterator
|
|
|
|
ircd::db::cell &
|
|
|
|
ircd::db::cell::operator=(cell &&o)
|
2017-03-31 00:57:08 +02:00
|
|
|
noexcept
|
|
|
|
{
|
2017-04-03 06:02:32 +02:00
|
|
|
c = std::move(o.c);
|
|
|
|
ss = std::move(o.ss);
|
|
|
|
it = std::move(o.it);
|
2017-03-31 00:57:08 +02:00
|
|
|
|
|
|
|
return *this;
|
|
|
|
}
|
|
|
|
|
2017-04-03 06:02:32 +02:00
|
|
|
// Linkage for incomplete rocksdb::Iterator
|
|
|
|
ircd::db::cell::~cell()
|
2017-03-31 00:57:08 +02:00
|
|
|
noexcept
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
2017-04-03 06:02:32 +02:00
|
|
|
bool
|
2017-09-14 21:59:20 +02:00
|
|
|
ircd::db::cell::load(const string_view &index,
|
|
|
|
gopts opts)
|
2016-09-24 06:01:57 +02:00
|
|
|
{
|
2017-04-03 06:02:32 +02:00
|
|
|
database &d(c);
|
2017-09-22 02:01:29 +02:00
|
|
|
if(valid(index) && !opts.snapshot && sequence(ss) == sequence(d))
|
2017-04-03 06:02:32 +02:00
|
|
|
return true;
|
2016-09-25 03:18:54 +02:00
|
|
|
|
2017-04-03 06:02:32 +02:00
|
|
|
if(bool(opts.snapshot))
|
|
|
|
{
|
|
|
|
this->it.reset();
|
|
|
|
this->ss = std::move(opts.snapshot);
|
|
|
|
}
|
2017-03-31 00:57:08 +02:00
|
|
|
|
2017-04-03 06:02:32 +02:00
|
|
|
database::column &c(this->c);
|
2017-08-30 23:05:15 +02:00
|
|
|
return seek(c, index, opts, this->it);
|
2017-04-03 06:02:32 +02:00
|
|
|
}
|
2017-03-24 02:36:49 +01:00
|
|
|
|
2017-04-03 06:02:32 +02:00
|
|
|
ircd::db::cell &
|
|
|
|
ircd::db::cell::operator=(const string_view &s)
|
2017-03-31 00:57:08 +02:00
|
|
|
{
|
2017-08-30 23:21:18 +02:00
|
|
|
write(c, key(), s);
|
2017-04-03 06:02:32 +02:00
|
|
|
return *this;
|
|
|
|
}
|
2017-03-31 00:57:08 +02:00
|
|
|
|
2017-08-23 22:37:47 +02:00
|
|
|
void
|
|
|
|
ircd::db::cell::operator()(const op &op,
|
|
|
|
const string_view &val,
|
|
|
|
const sopts &sopts)
|
|
|
|
{
|
|
|
|
write(cell::delta{op, *this, val}, sopts);
|
|
|
|
}
|
|
|
|
|
|
|
|
ircd::db::cell::operator
|
|
|
|
string_view()
|
2017-04-03 06:02:32 +02:00
|
|
|
{
|
|
|
|
return val();
|
2017-03-31 00:57:08 +02:00
|
|
|
}
|
|
|
|
|
2017-08-23 22:37:47 +02:00
|
|
|
ircd::db::cell::operator
|
|
|
|
string_view()
|
2017-04-03 06:02:32 +02:00
|
|
|
const
|
2017-03-14 19:39:26 +01:00
|
|
|
{
|
2017-04-03 06:02:32 +02:00
|
|
|
return val();
|
2017-03-14 19:39:26 +01:00
|
|
|
}
|
|
|
|
|
2017-04-03 06:02:32 +02:00
|
|
|
ircd::string_view
|
|
|
|
ircd::db::cell::val()
|
2017-03-14 19:39:26 +01:00
|
|
|
{
|
2017-04-03 06:02:32 +02:00
|
|
|
if(!valid())
|
|
|
|
load();
|
2017-03-31 00:57:08 +02:00
|
|
|
|
2017-04-03 06:02:32 +02:00
|
|
|
return likely(valid())? db::val(*it) : string_view{};
|
2017-03-14 19:39:26 +01:00
|
|
|
}
|
|
|
|
|
2017-04-03 06:02:32 +02:00
|
|
|
ircd::string_view
|
|
|
|
ircd::db::cell::key()
|
2017-03-14 19:39:26 +01:00
|
|
|
{
|
2017-04-03 06:02:32 +02:00
|
|
|
if(!valid())
|
|
|
|
load();
|
|
|
|
|
2017-09-14 21:59:20 +02:00
|
|
|
return likely(valid())? db::key(*it) : string_view{};
|
2017-03-14 19:39:26 +01:00
|
|
|
}
|
|
|
|
|
2017-04-03 06:02:32 +02:00
|
|
|
ircd::string_view
|
|
|
|
ircd::db::cell::val()
|
|
|
|
const
|
2016-09-24 06:01:57 +02:00
|
|
|
{
|
2017-04-03 06:02:32 +02:00
|
|
|
return likely(valid())? db::val(*it) : string_view{};
|
2017-03-24 02:36:49 +01:00
|
|
|
}
|
2017-03-14 19:39:26 +01:00
|
|
|
|
2017-04-03 06:02:32 +02:00
|
|
|
ircd::string_view
|
|
|
|
ircd::db::cell::key()
|
|
|
|
const
|
2017-03-31 00:57:08 +02:00
|
|
|
{
|
2017-09-14 21:59:20 +02:00
|
|
|
return likely(valid())? db::key(*it) : string_view{};
|
|
|
|
}
|
|
|
|
|
|
|
|
bool
|
2017-09-16 20:48:09 +02:00
|
|
|
ircd::db::cell::valid()
|
2017-09-14 21:59:20 +02:00
|
|
|
const
|
|
|
|
{
|
2017-09-16 20:48:09 +02:00
|
|
|
return bool(it) && db::valid(*it);
|
2017-04-03 06:02:32 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
bool
|
2017-09-16 20:48:09 +02:00
|
|
|
ircd::db::cell::valid(const string_view &s)
|
2017-04-03 06:02:32 +02:00
|
|
|
const
|
|
|
|
{
|
2017-09-16 20:48:09 +02:00
|
|
|
return bool(it) && db::valid_eq(*it, s);
|
|
|
|
}
|
|
|
|
|
|
|
|
bool
|
|
|
|
ircd::db::cell::valid_gt(const string_view &s)
|
|
|
|
const
|
|
|
|
{
|
|
|
|
return bool(it) && db::valid_gt(*it, s);
|
|
|
|
}
|
|
|
|
|
|
|
|
bool
|
|
|
|
ircd::db::cell::valid_lte(const string_view &s)
|
|
|
|
const
|
|
|
|
{
|
|
|
|
return bool(it) && db::valid_lte(*it, s);
|
2017-03-31 00:57:08 +02:00
|
|
|
}
|
|
|
|
|
2017-04-03 06:02:32 +02:00
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
2017-03-31 00:57:08 +02:00
|
|
|
//
|
2017-04-03 06:02:32 +02:00
|
|
|
// db/row.h
|
2017-03-31 00:57:08 +02:00
|
|
|
//
|
|
|
|
|
2017-08-23 22:37:47 +02:00
|
|
|
void
|
|
|
|
ircd::db::del(row &row,
|
|
|
|
const sopts &sopts)
|
|
|
|
{
|
|
|
|
write(row::delta{op::DELETE, row}, sopts);
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
ircd::db::write(const row::delta &delta,
|
|
|
|
const sopts &sopts)
|
|
|
|
{
|
|
|
|
write(&delta, &delta + 1, sopts);
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
ircd::db::write(const sopts &sopts,
|
|
|
|
const std::initializer_list<row::delta> &deltas)
|
|
|
|
{
|
|
|
|
write(deltas, sopts);
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
ircd::db::write(const std::initializer_list<row::delta> &deltas,
|
|
|
|
const sopts &sopts)
|
|
|
|
{
|
|
|
|
write(std::begin(deltas), std::end(deltas), sopts);
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
ircd::db::write(const row::delta *const &begin,
|
|
|
|
const row::delta *const &end,
|
|
|
|
const sopts &sopts)
|
|
|
|
{
|
|
|
|
// Count the total number of cells for this transaction.
|
|
|
|
const auto cells
|
|
|
|
{
|
|
|
|
std::accumulate(begin, end, size_t(0), []
|
|
|
|
(auto ret, const row::delta &delta)
|
|
|
|
{
|
2017-08-29 03:44:27 +02:00
|
|
|
const auto &row(std::get<row *>(delta));
|
|
|
|
return ret += row->size();
|
2017-08-23 22:37:47 +02:00
|
|
|
})
|
|
|
|
};
|
|
|
|
|
|
|
|
//TODO: allocator?
|
|
|
|
std::vector<cell::delta> deltas;
|
|
|
|
deltas.reserve(cells);
|
|
|
|
|
|
|
|
// Compose all of the cells from all of the rows into a single txn
|
|
|
|
std::for_each(begin, end, [&deltas]
|
|
|
|
(const auto &delta)
|
|
|
|
{
|
|
|
|
const auto &op(std::get<op>(delta));
|
2017-08-29 03:44:27 +02:00
|
|
|
const auto &row(std::get<row *>(delta));
|
|
|
|
std::for_each(std::begin(*row), std::end(*row), [&deltas, &op]
|
2017-08-23 22:37:47 +02:00
|
|
|
(auto &cell)
|
|
|
|
{
|
|
|
|
// For operations like DELETE which don't require a value in
|
|
|
|
// the delta, we can skip a potentially expensive load of the cell.
|
|
|
|
const auto value
|
|
|
|
{
|
|
|
|
value_required(op)? cell.val() : string_view{}
|
|
|
|
};
|
|
|
|
|
|
|
|
deltas.emplace_back(op, cell, value);
|
|
|
|
});
|
|
|
|
});
|
|
|
|
|
|
|
|
// Commitment
|
|
|
|
write(&deltas.front(), &deltas.front() + deltas.size(), sopts);
|
|
|
|
}
|
|
|
|
|
2017-08-30 23:05:15 +02:00
|
|
|
template<class pos>
|
2017-09-12 21:03:13 +02:00
|
|
|
size_t
|
2017-08-30 23:05:15 +02:00
|
|
|
ircd::db::seek(row &r,
|
|
|
|
const pos &p)
|
|
|
|
{
|
2017-10-17 09:47:30 +02:00
|
|
|
if(r.empty())
|
|
|
|
return 0;
|
|
|
|
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_SEEK
|
|
|
|
const ircd::timer timer;
|
|
|
|
#endif
|
|
|
|
|
2017-09-23 10:26:17 +02:00
|
|
|
const column &c(r[0]);
|
|
|
|
const database &d(c);
|
|
|
|
const auto ret
|
2017-08-30 23:05:15 +02:00
|
|
|
{
|
2017-09-29 06:19:02 +02:00
|
|
|
std::count_if(begin(r), end(r), [&p]
|
2017-09-23 10:26:17 +02:00
|
|
|
(auto &cell)
|
|
|
|
{
|
|
|
|
return seek(cell, p);
|
|
|
|
})
|
|
|
|
};
|
2018-05-29 10:42:48 +02:00
|
|
|
|
|
|
|
#ifdef RB_DEBUG_DB_SEEK
|
|
|
|
log::debug
|
|
|
|
{
|
|
|
|
log, "'%s' %lu:%lu '%s' row SEEK %zu of %zu in %ld$us",
|
|
|
|
name(d),
|
|
|
|
sequence(d),
|
|
|
|
sequence(r[0]),
|
|
|
|
name(c),
|
|
|
|
ret,
|
|
|
|
r.size(),
|
|
|
|
timer.at<microseconds>().count()
|
|
|
|
};
|
|
|
|
#endif
|
|
|
|
|
2017-09-23 10:26:17 +02:00
|
|
|
return ret;
|
2017-08-30 23:05:15 +02:00
|
|
|
}
|
2017-09-12 21:03:13 +02:00
|
|
|
template size_t ircd::db::seek<ircd::db::pos>(row &, const pos &);
|
|
|
|
template size_t ircd::db::seek<ircd::string_view>(row &, const string_view &);
|
2017-08-30 23:05:15 +02:00
|
|
|
|
2017-09-17 00:20:13 +02:00
|
|
|
//
|
|
|
|
// row
|
|
|
|
//
|
2017-08-30 23:05:15 +02:00
|
|
|
|
2017-04-03 06:02:32 +02:00
|
|
|
ircd::db::row::row(database &d,
|
|
|
|
const string_view &key,
|
2018-05-13 03:26:49 +02:00
|
|
|
const vector_view<const string_view> &colnames,
|
2017-09-29 06:19:02 +02:00
|
|
|
const vector_view<cell> &buf,
|
2017-08-23 22:37:47 +02:00
|
|
|
gopts opts)
|
2018-05-20 08:13:54 +02:00
|
|
|
:vector_view<cell>
|
|
|
|
{
|
|
|
|
buf.data(),
|
|
|
|
colnames.empty()?
|
|
|
|
d.columns.size():
|
|
|
|
colnames.size()
|
|
|
|
}
|
2017-03-24 02:36:49 +01:00
|
|
|
{
|
2017-04-03 06:02:32 +02:00
|
|
|
using std::end;
|
|
|
|
using std::begin;
|
|
|
|
using rocksdb::Iterator;
|
|
|
|
using rocksdb::ColumnFamilyHandle;
|
|
|
|
|
2017-08-23 22:37:47 +02:00
|
|
|
if(!opts.snapshot)
|
|
|
|
opts.snapshot = database::snapshot(d);
|
|
|
|
|
2017-04-03 06:02:32 +02:00
|
|
|
const rocksdb::ReadOptions options
|
|
|
|
{
|
|
|
|
make_opts(opts)
|
|
|
|
};
|
|
|
|
|
2018-05-20 08:13:54 +02:00
|
|
|
const size_t &column_count
|
2017-04-03 06:02:32 +02:00
|
|
|
{
|
2018-05-20 08:13:54 +02:00
|
|
|
vector_view<cell>::size()
|
2017-04-03 06:02:32 +02:00
|
|
|
};
|
|
|
|
|
2018-05-13 03:26:08 +02:00
|
|
|
database::column *colptr[column_count];
|
2017-04-03 06:02:32 +02:00
|
|
|
if(colnames.empty())
|
2018-05-13 03:26:08 +02:00
|
|
|
std::transform(begin(d.columns), end(d.columns), colptr, [&colnames]
|
2017-04-03 06:02:32 +02:00
|
|
|
(const auto &p)
|
|
|
|
{
|
2017-09-19 04:17:36 +02:00
|
|
|
return p.get();
|
2017-04-03 06:02:32 +02:00
|
|
|
});
|
|
|
|
else
|
2018-05-13 03:26:08 +02:00
|
|
|
std::transform(begin(colnames), end(colnames), colptr, [&d]
|
2017-04-03 06:02:32 +02:00
|
|
|
(const auto &name)
|
|
|
|
{
|
|
|
|
return &d[name];
|
|
|
|
});
|
|
|
|
|
2017-08-23 22:37:47 +02:00
|
|
|
//TODO: allocator
|
2018-05-13 03:26:08 +02:00
|
|
|
std::vector<ColumnFamilyHandle *> handles(column_count);
|
|
|
|
std::transform(colptr, colptr + column_count, begin(handles), []
|
2017-04-03 06:02:32 +02:00
|
|
|
(database::column *const &ptr)
|
|
|
|
{
|
|
|
|
return ptr->handle.get();
|
|
|
|
});
|
|
|
|
|
2017-08-23 22:37:47 +02:00
|
|
|
//TODO: does this block?
|
2017-04-03 06:02:32 +02:00
|
|
|
std::vector<Iterator *> iterators;
|
|
|
|
throw_on_error
|
|
|
|
{
|
|
|
|
d.d->NewIterators(options, handles, &iterators)
|
|
|
|
};
|
|
|
|
|
2018-05-13 03:26:08 +02:00
|
|
|
for(size_t i(0); i < this->size() && i < column_count; ++i)
|
2017-04-03 06:02:32 +02:00
|
|
|
{
|
|
|
|
std::unique_ptr<Iterator> it(iterators.at(i));
|
2018-05-13 03:26:08 +02:00
|
|
|
(*this)[i] = cell { *colptr[i], key, std::move(it), opts };
|
2017-04-03 06:02:32 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
ircd::db::row::operator()(const op &op,
|
|
|
|
const string_view &col,
|
|
|
|
const string_view &val,
|
|
|
|
const sopts &sopts)
|
|
|
|
{
|
|
|
|
write(cell::delta{op, (*this)[col], val}, sopts);
|
|
|
|
}
|
|
|
|
|
2017-09-22 02:00:48 +02:00
|
|
|
ircd::db::cell &
|
|
|
|
ircd::db::row::operator[](const string_view &column)
|
|
|
|
{
|
|
|
|
const auto it(find(column));
|
|
|
|
if(unlikely(it == end()))
|
2018-05-29 10:09:15 +02:00
|
|
|
throw schema_error
|
|
|
|
{
|
|
|
|
"column '%s' not specified in the descriptor schema", column
|
|
|
|
};
|
2017-09-22 02:00:48 +02:00
|
|
|
|
|
|
|
return *it;
|
|
|
|
}
|
|
|
|
|
|
|
|
const ircd::db::cell &
|
|
|
|
ircd::db::row::operator[](const string_view &column)
|
|
|
|
const
|
|
|
|
{
|
|
|
|
const auto it(find(column));
|
|
|
|
if(unlikely(it == end()))
|
2018-05-29 10:09:15 +02:00
|
|
|
throw schema_error
|
|
|
|
{
|
|
|
|
"column '%s' not specified in the descriptor schema", column
|
|
|
|
};
|
2017-09-22 02:00:48 +02:00
|
|
|
|
|
|
|
return *it;
|
|
|
|
}
|
|
|
|
|
2017-04-03 06:02:32 +02:00
|
|
|
ircd::db::row::iterator
|
|
|
|
ircd::db::row::find(const string_view &col)
|
|
|
|
{
|
2017-09-29 06:19:02 +02:00
|
|
|
return std::find_if(std::begin(*this), std::end(*this), [&col]
|
2017-04-03 06:02:32 +02:00
|
|
|
(const auto &cell)
|
|
|
|
{
|
|
|
|
return name(cell.c) == col;
|
|
|
|
});
|
|
|
|
}
|
|
|
|
|
|
|
|
ircd::db::row::const_iterator
|
|
|
|
ircd::db::row::find(const string_view &col)
|
|
|
|
const
|
|
|
|
{
|
2017-09-29 06:19:02 +02:00
|
|
|
return std::find_if(std::begin(*this), std::end(*this), [&col]
|
2017-04-03 06:02:32 +02:00
|
|
|
(const auto &cell)
|
|
|
|
{
|
|
|
|
return name(cell.c) == col;
|
|
|
|
});
|
|
|
|
}
|
|
|
|
|
2017-08-30 23:22:19 +02:00
|
|
|
bool
|
|
|
|
ircd::db::row::valid()
|
|
|
|
const
|
|
|
|
{
|
2017-09-29 06:19:02 +02:00
|
|
|
return std::any_of(std::begin(*this), std::end(*this), []
|
2017-08-30 23:22:19 +02:00
|
|
|
(const auto &cell)
|
|
|
|
{
|
|
|
|
return cell.valid();
|
|
|
|
});
|
|
|
|
}
|
|
|
|
|
2017-09-16 20:48:09 +02:00
|
|
|
bool
|
|
|
|
ircd::db::row::valid(const string_view &s)
|
|
|
|
const
|
|
|
|
{
|
2017-09-29 06:19:02 +02:00
|
|
|
return std::any_of(std::begin(*this), std::end(*this), [&s]
|
2017-09-16 20:48:09 +02:00
|
|
|
(const auto &cell)
|
|
|
|
{
|
|
|
|
return cell.valid(s);
|
|
|
|
});
|
|
|
|
}
|
|
|
|
|
2017-04-03 06:02:32 +02:00
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
//
|
|
|
|
// db/column.h
|
|
|
|
//
|
|
|
|
|
|
|
|
std::string
|
|
|
|
ircd::db::read(column &column,
|
|
|
|
const string_view &key,
|
|
|
|
const gopts &gopts)
|
|
|
|
{
|
|
|
|
std::string ret;
|
2018-05-05 00:55:21 +02:00
|
|
|
const auto closure([&ret]
|
2017-04-03 06:02:32 +02:00
|
|
|
(const string_view &src)
|
|
|
|
{
|
|
|
|
ret.assign(begin(src), end(src));
|
|
|
|
});
|
|
|
|
|
2018-05-05 00:55:21 +02:00
|
|
|
column(key, closure, gopts);
|
2017-04-03 06:02:32 +02:00
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
|
|
|
ircd::string_view
|
|
|
|
ircd::db::read(column &column,
|
|
|
|
const string_view &key,
|
2018-01-26 18:58:35 +01:00
|
|
|
const mutable_buffer &buf,
|
2017-04-03 06:02:32 +02:00
|
|
|
const gopts &gopts)
|
|
|
|
{
|
2018-05-05 00:55:21 +02:00
|
|
|
string_view ret;
|
|
|
|
const auto closure([&ret, &buf]
|
2017-04-03 06:02:32 +02:00
|
|
|
(const string_view &src)
|
|
|
|
{
|
2018-05-05 00:55:21 +02:00
|
|
|
ret = { data(buf), copy(buf, src) };
|
2017-04-03 06:02:32 +02:00
|
|
|
});
|
|
|
|
|
2018-05-05 00:55:21 +02:00
|
|
|
column(key, closure, gopts);
|
|
|
|
return ret;
|
2017-04-03 06:02:32 +02:00
|
|
|
}
|
|
|
|
|
2018-05-05 00:57:50 +02:00
|
|
|
std::string
|
|
|
|
ircd::db::read(column &column,
|
|
|
|
const string_view &key,
|
|
|
|
bool &found,
|
|
|
|
const gopts &gopts)
|
|
|
|
{
|
|
|
|
std::string ret;
|
|
|
|
const auto closure([&ret]
|
|
|
|
(const string_view &src)
|
|
|
|
{
|
|
|
|
ret.assign(begin(src), end(src));
|
|
|
|
});
|
|
|
|
|
|
|
|
found = column(key, std::nothrow, closure, gopts);
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
|
|
|
ircd::string_view
|
|
|
|
ircd::db::read(column &column,
|
|
|
|
const string_view &key,
|
|
|
|
bool &found,
|
|
|
|
const mutable_buffer &buf,
|
|
|
|
const gopts &gopts)
|
|
|
|
{
|
|
|
|
string_view ret;
|
|
|
|
const auto closure([&buf, &ret]
|
|
|
|
(const string_view &src)
|
|
|
|
{
|
|
|
|
ret = { data(buf), copy(buf, src) };
|
|
|
|
});
|
|
|
|
|
|
|
|
found = column(key, std::nothrow, closure, gopts);
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
2018-05-15 01:17:43 +02:00
|
|
|
rocksdb::Cache *
|
|
|
|
ircd::db::cache(column &column)
|
|
|
|
{
|
|
|
|
database::column &c(column);
|
|
|
|
return c.table_opts.block_cache.get();
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Cache *
|
|
|
|
ircd::db::cache_compressed(column &column)
|
|
|
|
{
|
|
|
|
database::column &c(column);
|
|
|
|
return c.table_opts.block_cache_compressed.get();
|
|
|
|
}
|
|
|
|
|
|
|
|
const rocksdb::Cache *
|
|
|
|
ircd::db::cache(const column &column)
|
|
|
|
{
|
|
|
|
const database::column &c(column);
|
|
|
|
return c.table_opts.block_cache.get();
|
|
|
|
}
|
|
|
|
|
|
|
|
const rocksdb::Cache *
|
|
|
|
ircd::db::cache_compressed(const column &column)
|
|
|
|
{
|
|
|
|
const database::column &c(column);
|
|
|
|
return c.table_opts.block_cache_compressed.get();
|
|
|
|
}
|
|
|
|
|
2017-04-03 06:02:32 +02:00
|
|
|
template<>
|
2018-04-03 19:58:16 +02:00
|
|
|
ircd::db::prop_str
|
2018-03-23 03:35:50 +01:00
|
|
|
ircd::db::property(const column &column,
|
2017-04-03 06:02:32 +02:00
|
|
|
const string_view &name)
|
|
|
|
{
|
|
|
|
std::string ret;
|
2018-03-23 03:35:50 +01:00
|
|
|
database::column &c(const_cast<db::column &>(column));
|
|
|
|
database &d(const_cast<db::column &>(column));
|
2018-04-13 03:57:03 +02:00
|
|
|
if(!d.d->GetProperty(c, slice(name), &ret))
|
|
|
|
throw not_found
|
|
|
|
{
|
|
|
|
"'property '%s' for column '%s' in '%s' not found.",
|
|
|
|
name,
|
|
|
|
db::name(column),
|
|
|
|
db::name(d)
|
|
|
|
};
|
|
|
|
|
2017-04-03 06:02:32 +02:00
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
|
|
|
template<>
|
2018-04-03 19:58:16 +02:00
|
|
|
ircd::db::prop_int
|
2018-03-23 03:35:50 +01:00
|
|
|
ircd::db::property(const column &column,
|
2017-04-03 06:02:32 +02:00
|
|
|
const string_view &name)
|
|
|
|
{
|
|
|
|
uint64_t ret;
|
2018-03-23 03:35:50 +01:00
|
|
|
database::column &c(const_cast<db::column &>(column));
|
|
|
|
database &d(const_cast<db::column &>(column));
|
2017-04-03 06:02:32 +02:00
|
|
|
if(!d.d->GetIntProperty(c, slice(name), &ret))
|
2018-04-13 03:57:03 +02:00
|
|
|
throw not_found
|
|
|
|
{
|
|
|
|
"property '%s' for column '%s' in '%s' not found or not an integer.",
|
|
|
|
name,
|
|
|
|
db::name(column),
|
|
|
|
db::name(d)
|
|
|
|
};
|
2017-04-03 06:02:32 +02:00
|
|
|
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
2018-04-03 20:05:03 +02:00
|
|
|
template<>
|
|
|
|
ircd::db::prop_map
|
|
|
|
ircd::db::property(const column &column,
|
|
|
|
const string_view &name)
|
|
|
|
{
|
2018-04-09 21:55:22 +02:00
|
|
|
std::map<std::string, std::string> ret;
|
2018-04-03 20:05:03 +02:00
|
|
|
database::column &c(const_cast<db::column &>(column));
|
|
|
|
database &d(const_cast<db::column &>(column));
|
2018-04-13 03:57:03 +02:00
|
|
|
if(!d.d->GetMapProperty(c, slice(name), &ret))
|
|
|
|
ret.emplace(std::string{name}, property<std::string>(column, name));
|
|
|
|
|
2018-04-03 20:05:03 +02:00
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
2017-04-03 06:02:32 +02:00
|
|
|
size_t
|
2018-03-23 03:35:50 +01:00
|
|
|
ircd::db::bytes(const column &column)
|
2017-04-03 06:02:32 +02:00
|
|
|
{
|
|
|
|
rocksdb::ColumnFamilyMetaData cfm;
|
2018-03-23 03:35:50 +01:00
|
|
|
database &d(const_cast<db::column &>(column));
|
|
|
|
database::column &c(const_cast<db::column &>(column));
|
2017-04-03 06:02:32 +02:00
|
|
|
assert(bool(c.handle));
|
|
|
|
d.d->GetColumnFamilyMetaData(c.handle.get(), &cfm);
|
|
|
|
return cfm.size;
|
|
|
|
}
|
|
|
|
|
|
|
|
size_t
|
2018-03-23 03:35:50 +01:00
|
|
|
ircd::db::file_count(const column &column)
|
2017-04-03 06:02:32 +02:00
|
|
|
{
|
|
|
|
rocksdb::ColumnFamilyMetaData cfm;
|
2018-03-23 03:35:50 +01:00
|
|
|
database &d(const_cast<db::column &>(column));
|
|
|
|
database::column &c(const_cast<db::column &>(column));
|
2017-04-03 06:02:32 +02:00
|
|
|
assert(bool(c.handle));
|
|
|
|
d.d->GetColumnFamilyMetaData(c.handle.get(), &cfm);
|
|
|
|
return cfm.file_count;
|
|
|
|
}
|
|
|
|
|
2017-09-19 04:17:36 +02:00
|
|
|
uint32_t
|
|
|
|
ircd::db::id(const column &column)
|
|
|
|
{
|
|
|
|
const database::column &c(column);
|
|
|
|
return id(c);
|
|
|
|
}
|
|
|
|
|
2017-04-03 06:02:32 +02:00
|
|
|
const std::string &
|
|
|
|
ircd::db::name(const column &column)
|
|
|
|
{
|
|
|
|
const database::column &c(column);
|
|
|
|
return name(c);
|
|
|
|
}
|
|
|
|
|
2017-09-21 05:38:39 +02:00
|
|
|
const ircd::db::database::descriptor &
|
|
|
|
ircd::db::describe(const column &column)
|
|
|
|
{
|
|
|
|
const database::column &c(column);
|
|
|
|
return describe(c);
|
|
|
|
}
|
|
|
|
|
2017-04-03 06:02:32 +02:00
|
|
|
void
|
2018-04-27 02:19:29 +02:00
|
|
|
ircd::db::sort(column &column,
|
|
|
|
const bool &blocking)
|
2017-04-03 06:02:32 +02:00
|
|
|
{
|
|
|
|
database::column &c(column);
|
2018-04-20 23:12:57 +02:00
|
|
|
database &d(*c.d);
|
|
|
|
rocksdb::FlushOptions opts;
|
|
|
|
opts.wait = blocking;
|
2018-04-27 02:19:29 +02:00
|
|
|
log.debug("'%s':'%s' @%lu FLUSH (sort)",
|
2018-04-20 23:12:57 +02:00
|
|
|
name(d),
|
|
|
|
name(c),
|
|
|
|
sequence(d));
|
|
|
|
|
|
|
|
throw_on_error
|
|
|
|
{
|
|
|
|
d.d->Flush(opts, c)
|
|
|
|
};
|
2016-09-24 06:01:57 +02:00
|
|
|
}
|
|
|
|
|
2018-04-15 11:42:57 +02:00
|
|
|
void
|
|
|
|
ircd::db::compact(column &column,
|
2018-05-11 01:41:34 +02:00
|
|
|
const int &level_)
|
2018-04-15 11:42:57 +02:00
|
|
|
{
|
|
|
|
database::column &c(column);
|
2018-04-20 23:12:57 +02:00
|
|
|
database &d(*c.d);
|
|
|
|
|
2018-04-21 01:48:07 +02:00
|
|
|
rocksdb::ColumnFamilyMetaData cfmd;
|
|
|
|
d.d->GetColumnFamilyMetaData(c, &cfmd);
|
|
|
|
for(const auto &level : cfmd.levels)
|
|
|
|
{
|
2018-05-11 01:41:34 +02:00
|
|
|
if(level_ != -1 && level.level != level_)
|
|
|
|
continue;
|
|
|
|
|
2018-04-21 01:48:07 +02:00
|
|
|
if(level.files.empty())
|
|
|
|
continue;
|
|
|
|
|
|
|
|
std::vector<std::string> files(level.files.size());
|
|
|
|
std::transform(level.files.begin(), level.files.end(), files.begin(), []
|
|
|
|
(auto &metadata)
|
|
|
|
{
|
|
|
|
return std::move(metadata.name);
|
|
|
|
});
|
|
|
|
|
2018-05-11 01:41:34 +02:00
|
|
|
log::debug
|
|
|
|
{
|
|
|
|
log, "'%s':'%s' COMPACT level:%d files:%zu size:%zu",
|
|
|
|
name(d),
|
|
|
|
name(c),
|
|
|
|
level.level,
|
|
|
|
level.files.size(),
|
|
|
|
level.size
|
|
|
|
};
|
2018-04-21 01:48:07 +02:00
|
|
|
|
|
|
|
rocksdb::CompactionOptions opts;
|
|
|
|
throw_on_error
|
|
|
|
{
|
|
|
|
d.d->CompactFiles(opts, c, files, level.level)
|
|
|
|
};
|
|
|
|
}
|
2018-04-15 11:42:57 +02:00
|
|
|
}
|
|
|
|
|
2018-05-11 01:41:34 +02:00
|
|
|
void
|
|
|
|
ircd::db::compact(column &column,
|
|
|
|
const std::pair<string_view, string_view> &range,
|
|
|
|
const int &to_level)
|
|
|
|
{
|
|
|
|
database::column &c(column);
|
|
|
|
database &d(*c.d);
|
|
|
|
|
|
|
|
const auto begin(slice(range.first));
|
|
|
|
const rocksdb::Slice *const b
|
|
|
|
{
|
|
|
|
empty(range.first)? nullptr : &begin
|
|
|
|
};
|
|
|
|
|
|
|
|
const auto end(slice(range.second));
|
|
|
|
const rocksdb::Slice *const e
|
|
|
|
{
|
|
|
|
empty(range.second)? nullptr : &end
|
|
|
|
};
|
|
|
|
|
|
|
|
log::debug
|
|
|
|
{
|
|
|
|
log, "'%s':'%s' @%lu COMPACT [%s, %s] to level %d",
|
|
|
|
name(d),
|
|
|
|
name(c),
|
|
|
|
sequence(d),
|
|
|
|
range.first,
|
|
|
|
range.second,
|
|
|
|
to_level
|
|
|
|
};
|
|
|
|
|
|
|
|
rocksdb::CompactRangeOptions opts;
|
|
|
|
opts.change_level = true;
|
|
|
|
opts.target_level = to_level;
|
|
|
|
opts.allow_write_stall = true;
|
|
|
|
throw_on_error
|
|
|
|
{
|
|
|
|
d.d->CompactRange(opts, c, b, e)
|
|
|
|
};
|
|
|
|
}
|
|
|
|
|
2018-04-20 23:10:59 +02:00
|
|
|
void
|
|
|
|
ircd::db::setopt(column &column,
|
|
|
|
const string_view &key,
|
|
|
|
const string_view &val)
|
|
|
|
{
|
|
|
|
const std::unordered_map<std::string, std::string> options
|
|
|
|
{
|
|
|
|
{ std::string{key}, std::string{val} }
|
|
|
|
};
|
|
|
|
|
|
|
|
database::column &c(column);
|
|
|
|
database &d(c);
|
|
|
|
throw_on_error
|
|
|
|
{
|
|
|
|
d.d->SetOptions(c, options)
|
|
|
|
};
|
|
|
|
}
|
|
|
|
|
2016-09-24 06:01:57 +02:00
|
|
|
void
|
2017-03-31 00:57:08 +02:00
|
|
|
ircd::db::del(column &column,
|
|
|
|
const string_view &key,
|
|
|
|
const sopts &sopts)
|
2016-09-24 06:01:57 +02:00
|
|
|
{
|
2017-03-31 00:57:08 +02:00
|
|
|
database &d(column);
|
|
|
|
database::column &c(column);
|
2017-09-25 03:02:51 +02:00
|
|
|
log.debug("'%s' %lu '%s' DELETE key(%zu B)",
|
2017-04-03 06:02:32 +02:00
|
|
|
name(d),
|
|
|
|
sequence(d),
|
2017-09-25 03:02:51 +02:00
|
|
|
name(c),
|
2017-04-03 06:02:32 +02:00
|
|
|
key.size());
|
2016-09-24 06:01:57 +02:00
|
|
|
|
2016-09-25 06:12:43 +02:00
|
|
|
auto opts(make_opts(sopts));
|
2017-04-03 06:02:32 +02:00
|
|
|
throw_on_error
|
|
|
|
{
|
|
|
|
d.d->Delete(opts, c, slice(key))
|
|
|
|
};
|
2016-09-25 06:12:43 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
void
|
2017-03-31 00:57:08 +02:00
|
|
|
ircd::db::write(column &column,
|
|
|
|
const string_view &key,
|
2018-04-30 16:18:06 +02:00
|
|
|
const const_buffer &val,
|
2016-09-25 06:12:43 +02:00
|
|
|
const sopts &sopts)
|
|
|
|
{
|
2017-03-31 00:57:08 +02:00
|
|
|
database &d(column);
|
|
|
|
database::column &c(column);
|
2017-09-25 03:02:51 +02:00
|
|
|
log.debug("'%s' %lu '%s' PUT key(%zu B) val(%zu B)",
|
2017-04-03 06:02:32 +02:00
|
|
|
name(d),
|
|
|
|
sequence(d),
|
2017-09-25 03:02:51 +02:00
|
|
|
name(c),
|
2018-04-30 16:18:06 +02:00
|
|
|
size(key),
|
|
|
|
size(val));
|
2016-09-25 03:18:54 +02:00
|
|
|
|
|
|
|
auto opts(make_opts(sopts));
|
2017-04-03 06:02:32 +02:00
|
|
|
throw_on_error
|
|
|
|
{
|
|
|
|
d.d->Put(opts, c, slice(key), slice(val))
|
|
|
|
};
|
2016-09-25 06:12:43 +02:00
|
|
|
}
|
|
|
|
|
2017-10-28 01:48:27 +02:00
|
|
|
///
|
|
|
|
/// TODO: As soon as rocksdb regressed from allowing a null value argument
|
|
|
|
/// to the query, we are forced to watch rocksdb allocate and copy the value
|
|
|
|
/// into our string `discard` for no reason at all. If this regression remains
|
|
|
|
/// it may be more efficient to use a regular iterator seek like everything
|
|
|
|
/// else. Otherwise if they fix this, the discard string should become a
|
|
|
|
/// nullptr instead.
|
2017-03-31 01:20:01 +02:00
|
|
|
bool
|
|
|
|
ircd::db::has(column &column,
|
|
|
|
const string_view &key,
|
|
|
|
const gopts &gopts)
|
|
|
|
{
|
|
|
|
database &d(column);
|
|
|
|
database::column &c(column);
|
|
|
|
|
|
|
|
const auto k(slice(key));
|
|
|
|
auto opts(make_opts(gopts));
|
|
|
|
|
|
|
|
// Perform queries which are stymied from any sysentry
|
|
|
|
opts.read_tier = NON_BLOCKING;
|
|
|
|
|
|
|
|
// Perform a co-RP query to the filtration
|
2018-04-09 06:57:02 +02:00
|
|
|
thread_local std::string discard;
|
2018-04-10 01:57:13 +02:00
|
|
|
bool ret
|
2018-03-14 01:32:13 +01:00
|
|
|
{
|
2018-04-09 06:57:02 +02:00
|
|
|
d.d->KeyMayExist(opts, c, k, &discard, nullptr)
|
2018-03-14 01:32:13 +01:00
|
|
|
};
|
2017-10-28 01:48:27 +02:00
|
|
|
|
2018-04-10 01:57:13 +02:00
|
|
|
if(ret)
|
2017-03-31 01:20:01 +02:00
|
|
|
{
|
2018-04-10 01:57:13 +02:00
|
|
|
const auto it
|
2017-03-31 01:20:01 +02:00
|
|
|
{
|
2018-04-10 01:57:13 +02:00
|
|
|
seek(column, key, gopts)
|
|
|
|
};
|
2017-03-31 01:20:01 +02:00
|
|
|
|
2018-04-10 01:57:13 +02:00
|
|
|
ret = valid_eq(*it, key);
|
2017-03-31 01:20:01 +02:00
|
|
|
}
|
2018-04-10 01:57:13 +02:00
|
|
|
|
|
|
|
return ret;
|
2017-03-31 01:20:01 +02:00
|
|
|
}
|
|
|
|
|
2017-08-31 07:13:50 +02:00
|
|
|
//
|
|
|
|
// column
|
|
|
|
//
|
|
|
|
|
|
|
|
ircd::db::column::column(database::column &c)
|
|
|
|
:c{&c}
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
ircd::db::column::column(database &d,
|
|
|
|
const string_view &column_name)
|
|
|
|
:c{&d[column_name]}
|
|
|
|
{}
|
|
|
|
|
2017-03-23 22:58:24 +01:00
|
|
|
void
|
2017-08-23 22:37:47 +02:00
|
|
|
ircd::db::column::operator()(const delta &delta,
|
2017-03-31 00:57:08 +02:00
|
|
|
const sopts &sopts)
|
2017-03-23 22:58:24 +01:00
|
|
|
{
|
2017-08-23 22:37:47 +02:00
|
|
|
operator()(&delta, &delta + 1, sopts);
|
2017-03-23 22:58:24 +01:00
|
|
|
}
|
|
|
|
|
2017-04-03 06:02:32 +02:00
|
|
|
void
|
|
|
|
ircd::db::column::operator()(const sopts &sopts,
|
|
|
|
const std::initializer_list<delta> &deltas)
|
|
|
|
{
|
|
|
|
operator()(deltas, sopts);
|
|
|
|
}
|
|
|
|
|
2017-03-21 07:28:20 +01:00
|
|
|
void
|
2017-03-31 00:57:08 +02:00
|
|
|
ircd::db::column::operator()(const std::initializer_list<delta> &deltas,
|
|
|
|
const sopts &sopts)
|
2017-03-21 07:28:20 +01:00
|
|
|
{
|
2017-08-23 22:37:47 +02:00
|
|
|
operator()(std::begin(deltas), std::end(deltas), sopts);
|
2017-03-21 07:28:20 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
void
|
2017-08-23 22:37:47 +02:00
|
|
|
ircd::db::column::operator()(const delta *const &begin,
|
|
|
|
const delta *const &end,
|
2017-03-31 00:57:08 +02:00
|
|
|
const sopts &sopts)
|
2017-03-21 07:28:20 +01:00
|
|
|
{
|
2017-08-23 22:37:47 +02:00
|
|
|
database &d(*this);
|
|
|
|
|
2017-03-21 07:28:20 +01:00
|
|
|
rocksdb::WriteBatch batch;
|
2017-08-23 22:37:47 +02:00
|
|
|
std::for_each(begin, end, [this, &batch]
|
|
|
|
(const delta &delta)
|
|
|
|
{
|
|
|
|
append(batch, *this, delta);
|
|
|
|
});
|
2017-03-21 07:28:20 +01:00
|
|
|
|
2017-09-08 11:14:17 +02:00
|
|
|
commit(d, batch, sopts);
|
2017-03-21 07:28:20 +01:00
|
|
|
}
|
|
|
|
|
2016-09-24 06:01:57 +02:00
|
|
|
void
|
2017-03-31 00:57:08 +02:00
|
|
|
ircd::db::column::operator()(const string_view &key,
|
|
|
|
const gopts &gopts,
|
|
|
|
const view_closure &func)
|
2016-11-29 16:23:38 +01:00
|
|
|
{
|
|
|
|
return operator()(key, func, gopts);
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
2017-03-31 00:57:08 +02:00
|
|
|
ircd::db::column::operator()(const string_view &key,
|
|
|
|
const view_closure &func,
|
|
|
|
const gopts &gopts)
|
2016-09-24 06:01:57 +02:00
|
|
|
{
|
2017-03-31 00:57:08 +02:00
|
|
|
const auto it(seek(*this, key, gopts));
|
2017-09-16 20:48:09 +02:00
|
|
|
valid_eq_or_throw(*it, key);
|
2017-04-03 06:02:32 +02:00
|
|
|
func(val(*it));
|
|
|
|
}
|
2016-09-24 06:01:57 +02:00
|
|
|
|
2018-01-27 19:07:08 +01:00
|
|
|
bool
|
|
|
|
ircd::db::column::operator()(const string_view &key,
|
|
|
|
const std::nothrow_t,
|
|
|
|
const gopts &gopts,
|
|
|
|
const view_closure &func)
|
|
|
|
{
|
|
|
|
return operator()(key, std::nothrow, func, gopts);
|
|
|
|
}
|
|
|
|
|
|
|
|
bool
|
|
|
|
ircd::db::column::operator()(const string_view &key,
|
|
|
|
const std::nothrow_t,
|
|
|
|
const view_closure &func,
|
|
|
|
const gopts &gopts)
|
|
|
|
{
|
|
|
|
const auto it(seek(*this, key, gopts));
|
|
|
|
if(!valid_eq(*it, key))
|
|
|
|
return false;
|
|
|
|
|
|
|
|
func(val(*it));
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2017-04-03 06:02:32 +02:00
|
|
|
ircd::db::cell
|
|
|
|
ircd::db::column::operator[](const string_view &key)
|
|
|
|
const
|
|
|
|
{
|
|
|
|
return { *this, key };
|
2016-09-24 06:01:57 +02:00
|
|
|
}
|
|
|
|
|
2017-09-08 10:33:41 +02:00
|
|
|
ircd::db::column::operator
|
|
|
|
const database::descriptor &()
|
|
|
|
const
|
|
|
|
{
|
|
|
|
return c->descriptor;
|
|
|
|
}
|
2017-04-03 06:02:32 +02:00
|
|
|
|
2017-03-23 22:58:24 +01:00
|
|
|
//
|
2017-03-31 00:57:08 +02:00
|
|
|
// column::const_iterator
|
2017-03-23 22:58:24 +01:00
|
|
|
//
|
|
|
|
|
2016-09-26 06:07:22 +02:00
|
|
|
namespace ircd {
|
|
|
|
namespace db {
|
2016-09-24 06:01:57 +02:00
|
|
|
|
2016-09-26 06:07:22 +02:00
|
|
|
} // namespace db
|
|
|
|
} // namespace ircd
|
|
|
|
|
2017-03-31 00:57:08 +02:00
|
|
|
ircd::db::column::const_iterator
|
2018-05-25 11:34:44 +02:00
|
|
|
ircd::db::column::end(gopts gopts)
|
2016-09-26 06:07:22 +02:00
|
|
|
{
|
2017-09-22 12:31:44 +02:00
|
|
|
const_iterator ret
|
|
|
|
{
|
2018-05-25 11:34:44 +02:00
|
|
|
c, {}, std::move(gopts)
|
2017-09-22 12:31:44 +02:00
|
|
|
};
|
|
|
|
|
2018-05-25 11:34:44 +02:00
|
|
|
seek(ret, pos::END);
|
2017-09-22 12:31:44 +02:00
|
|
|
return ret;
|
2016-09-24 06:01:57 +02:00
|
|
|
}
|
|
|
|
|
2017-03-31 00:57:08 +02:00
|
|
|
ircd::db::column::const_iterator
|
2018-05-25 11:34:44 +02:00
|
|
|
ircd::db::column::begin(gopts gopts)
|
2016-09-24 06:01:57 +02:00
|
|
|
{
|
2016-09-26 06:07:22 +02:00
|
|
|
const_iterator ret
|
2016-09-24 06:01:57 +02:00
|
|
|
{
|
2018-05-25 11:34:44 +02:00
|
|
|
c, {}, std::move(gopts)
|
2016-09-26 06:07:22 +02:00
|
|
|
};
|
2016-09-24 06:01:57 +02:00
|
|
|
|
2018-05-25 11:34:44 +02:00
|
|
|
seek(ret, pos::FRONT);
|
2017-09-22 12:31:44 +02:00
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
|
|
|
ircd::db::column::const_reverse_iterator
|
2018-05-25 11:34:44 +02:00
|
|
|
ircd::db::column::rend(gopts gopts)
|
2017-09-22 12:31:44 +02:00
|
|
|
{
|
|
|
|
const_reverse_iterator ret
|
|
|
|
{
|
2018-05-25 11:34:44 +02:00
|
|
|
c, {}, std::move(gopts)
|
2017-09-22 12:31:44 +02:00
|
|
|
};
|
|
|
|
|
2018-05-25 11:34:44 +02:00
|
|
|
seek(ret, pos::END);
|
2017-09-22 12:31:44 +02:00
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
|
|
|
ircd::db::column::const_reverse_iterator
|
2018-05-25 11:34:44 +02:00
|
|
|
ircd::db::column::rbegin(gopts gopts)
|
2017-09-22 12:31:44 +02:00
|
|
|
{
|
|
|
|
const_reverse_iterator ret
|
|
|
|
{
|
2018-05-25 11:34:44 +02:00
|
|
|
c, {}, std::move(gopts)
|
2017-09-22 12:31:44 +02:00
|
|
|
};
|
|
|
|
|
2018-05-25 11:34:44 +02:00
|
|
|
seek(ret, pos::BACK);
|
2017-09-22 12:31:44 +02:00
|
|
|
return ret;
|
2016-09-26 06:07:22 +02:00
|
|
|
}
|
|
|
|
|
2017-03-31 00:57:08 +02:00
|
|
|
ircd::db::column::const_iterator
|
|
|
|
ircd::db::column::upper_bound(const string_view &key,
|
2018-05-25 11:34:44 +02:00
|
|
|
gopts gopts)
|
2016-09-26 06:07:22 +02:00
|
|
|
{
|
2018-05-25 11:34:44 +02:00
|
|
|
auto it(lower_bound(key, std::move(gopts)));
|
2017-03-31 00:57:08 +02:00
|
|
|
if(it && it.it->key().compare(slice(key)) == 0)
|
2016-09-26 06:07:22 +02:00
|
|
|
++it;
|
|
|
|
|
2017-09-22 12:31:44 +02:00
|
|
|
return it;
|
2016-09-26 06:07:22 +02:00
|
|
|
}
|
|
|
|
|
2017-03-31 00:57:08 +02:00
|
|
|
ircd::db::column::const_iterator
|
|
|
|
ircd::db::column::find(const string_view &key,
|
2018-05-25 11:34:44 +02:00
|
|
|
gopts gopts)
|
2016-09-26 06:07:22 +02:00
|
|
|
{
|
2017-03-31 00:57:08 +02:00
|
|
|
auto it(lower_bound(key, gopts));
|
|
|
|
if(!it || it.it->key().compare(slice(key)) != 0)
|
2017-09-22 05:08:11 +02:00
|
|
|
return end(gopts);
|
2017-03-31 00:57:08 +02:00
|
|
|
|
|
|
|
return it;
|
|
|
|
}
|
2016-09-26 06:07:22 +02:00
|
|
|
|
2017-03-31 00:57:08 +02:00
|
|
|
ircd::db::column::const_iterator
|
|
|
|
ircd::db::column::lower_bound(const string_view &key,
|
2018-05-25 11:34:44 +02:00
|
|
|
gopts gopts)
|
2017-03-31 00:57:08 +02:00
|
|
|
{
|
2016-09-26 06:07:22 +02:00
|
|
|
const_iterator ret
|
2016-09-24 06:01:57 +02:00
|
|
|
{
|
2018-05-25 11:34:44 +02:00
|
|
|
c, {}, std::move(gopts)
|
2016-09-26 06:07:22 +02:00
|
|
|
};
|
|
|
|
|
2018-05-25 11:34:44 +02:00
|
|
|
seek(ret, key);
|
2017-09-22 12:31:44 +02:00
|
|
|
return ret;
|
2016-09-26 06:07:22 +02:00
|
|
|
}
|
|
|
|
|
2017-09-22 12:31:44 +02:00
|
|
|
ircd::db::column::const_iterator &
|
|
|
|
ircd::db::column::const_iterator::operator--()
|
|
|
|
{
|
|
|
|
if(likely(bool(*this)))
|
|
|
|
seek(*this, pos::PREV);
|
|
|
|
else
|
|
|
|
seek(*this, pos::BACK);
|
|
|
|
|
|
|
|
return *this;
|
|
|
|
}
|
|
|
|
|
|
|
|
ircd::db::column::const_iterator &
|
|
|
|
ircd::db::column::const_iterator::operator++()
|
|
|
|
{
|
|
|
|
if(likely(bool(*this)))
|
|
|
|
seek(*this, pos::NEXT);
|
|
|
|
else
|
|
|
|
seek(*this, pos::FRONT);
|
|
|
|
|
|
|
|
return *this;
|
|
|
|
}
|
|
|
|
|
|
|
|
ircd::db::column::const_reverse_iterator &
|
|
|
|
ircd::db::column::const_reverse_iterator::operator--()
|
|
|
|
{
|
|
|
|
if(likely(bool(*this)))
|
|
|
|
seek(*this, pos::NEXT);
|
|
|
|
else
|
|
|
|
seek(*this, pos::FRONT);
|
|
|
|
|
|
|
|
return *this;
|
|
|
|
}
|
|
|
|
|
|
|
|
ircd::db::column::const_reverse_iterator &
|
|
|
|
ircd::db::column::const_reverse_iterator::operator++()
|
|
|
|
{
|
|
|
|
if(likely(bool(*this)))
|
|
|
|
seek(*this, pos::PREV);
|
|
|
|
else
|
|
|
|
seek(*this, pos::BACK);
|
|
|
|
|
|
|
|
return *this;
|
|
|
|
}
|
|
|
|
|
|
|
|
ircd::db::column::const_iterator_base::const_iterator_base(const_iterator_base &&o)
|
2017-03-31 00:57:08 +02:00
|
|
|
noexcept
|
2017-09-08 10:33:41 +02:00
|
|
|
:c{std::move(o.c)}
|
2018-05-25 11:34:44 +02:00
|
|
|
,opts{std::move(o.opts)}
|
2017-03-31 00:57:08 +02:00
|
|
|
,it{std::move(o.it)}
|
|
|
|
,val{std::move(o.val)}
|
2016-09-26 06:07:22 +02:00
|
|
|
{
|
|
|
|
}
|
2016-09-24 06:01:57 +02:00
|
|
|
|
2017-09-22 12:31:44 +02:00
|
|
|
ircd::db::column::const_iterator_base &
|
|
|
|
ircd::db::column::const_iterator_base::operator=(const_iterator_base &&o)
|
2017-03-31 00:57:08 +02:00
|
|
|
noexcept
|
2016-09-26 06:07:22 +02:00
|
|
|
{
|
2017-03-31 00:57:08 +02:00
|
|
|
c = std::move(o.c);
|
2018-05-25 11:34:44 +02:00
|
|
|
opts = std::move(o.opts);
|
2017-03-31 00:57:08 +02:00
|
|
|
it = std::move(o.it);
|
|
|
|
val = std::move(o.val);
|
|
|
|
return *this;
|
2016-09-26 06:07:22 +02:00
|
|
|
}
|
|
|
|
|
2017-09-22 12:31:44 +02:00
|
|
|
// linkage for incmplete rocksdb::Iterator
|
|
|
|
ircd::db::column::const_iterator_base::const_iterator_base()
|
2017-04-03 06:02:32 +02:00
|
|
|
{
|
|
|
|
}
|
|
|
|
|
2017-09-22 12:31:44 +02:00
|
|
|
// linkage for incmplete rocksdb::Iterator
|
|
|
|
ircd::db::column::const_iterator_base::~const_iterator_base()
|
|
|
|
noexcept
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
ircd::db::column::const_iterator_base::const_iterator_base(database::column *const &c,
|
2018-05-25 11:34:44 +02:00
|
|
|
std::unique_ptr<rocksdb::Iterator> &&it,
|
|
|
|
gopts opts)
|
2017-09-08 10:33:41 +02:00
|
|
|
:c{c}
|
2018-05-25 11:34:44 +02:00
|
|
|
,opts{std::move(opts)}
|
2017-03-31 00:57:08 +02:00
|
|
|
,it{std::move(it)}
|
2016-09-26 06:07:22 +02:00
|
|
|
{
|
|
|
|
}
|
|
|
|
|
2017-09-22 12:31:44 +02:00
|
|
|
const ircd::db::column::const_iterator_base::value_type &
|
|
|
|
ircd::db::column::const_iterator_base::operator*()
|
2016-09-26 06:07:22 +02:00
|
|
|
const
|
|
|
|
{
|
2017-09-16 20:48:09 +02:00
|
|
|
assert(it && valid(*it));
|
2017-04-03 06:02:32 +02:00
|
|
|
val.first = db::key(*it);
|
|
|
|
val.second = db::val(*it);
|
2016-09-26 06:07:22 +02:00
|
|
|
return val;
|
|
|
|
}
|
|
|
|
|
2017-09-22 12:31:44 +02:00
|
|
|
const ircd::db::column::const_iterator_base::value_type *
|
|
|
|
ircd::db::column::const_iterator_base::operator->()
|
2016-09-26 06:07:22 +02:00
|
|
|
const
|
|
|
|
{
|
|
|
|
return &operator*();
|
|
|
|
}
|
|
|
|
|
|
|
|
bool
|
2017-09-22 12:31:44 +02:00
|
|
|
ircd::db::column::const_iterator_base::operator!()
|
2016-09-26 06:07:22 +02:00
|
|
|
const
|
|
|
|
{
|
2017-09-16 20:48:09 +02:00
|
|
|
return !static_cast<bool>(*this);
|
2016-09-26 06:07:22 +02:00
|
|
|
}
|
|
|
|
|
2017-09-22 12:31:44 +02:00
|
|
|
ircd::db::column::const_iterator_base::operator bool()
|
2016-09-26 06:07:22 +02:00
|
|
|
const
|
|
|
|
{
|
2017-09-16 20:48:09 +02:00
|
|
|
if(!it)
|
|
|
|
return false;
|
|
|
|
|
|
|
|
if(!valid(*it))
|
|
|
|
return false;
|
|
|
|
|
|
|
|
return true;
|
2016-09-26 06:07:22 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
bool
|
2017-09-22 12:31:44 +02:00
|
|
|
ircd::db::operator!=(const column::const_iterator_base &a, const column::const_iterator_base &b)
|
2016-09-26 06:07:22 +02:00
|
|
|
{
|
2017-04-03 06:02:32 +02:00
|
|
|
return !(a == b);
|
2016-09-26 06:07:22 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
bool
|
2017-09-22 12:31:44 +02:00
|
|
|
ircd::db::operator==(const column::const_iterator_base &a, const column::const_iterator_base &b)
|
2016-09-26 06:07:22 +02:00
|
|
|
{
|
2017-04-03 06:02:32 +02:00
|
|
|
if(a && b)
|
2016-09-26 06:07:22 +02:00
|
|
|
{
|
2017-04-03 06:02:32 +02:00
|
|
|
const auto &ak(a.it->key());
|
|
|
|
const auto &bk(b.it->key());
|
|
|
|
return ak.compare(bk) == 0;
|
2016-09-26 06:07:22 +02:00
|
|
|
}
|
|
|
|
|
2017-04-03 06:02:32 +02:00
|
|
|
if(!a && !b)
|
2016-09-26 06:07:22 +02:00
|
|
|
return true;
|
|
|
|
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool
|
2017-09-22 12:31:44 +02:00
|
|
|
ircd::db::operator>(const column::const_iterator_base &a, const column::const_iterator_base &b)
|
2016-09-26 06:07:22 +02:00
|
|
|
{
|
2017-04-03 06:02:32 +02:00
|
|
|
if(a && b)
|
2016-09-26 06:07:22 +02:00
|
|
|
{
|
2017-04-03 06:02:32 +02:00
|
|
|
const auto &ak(a.it->key());
|
|
|
|
const auto &bk(b.it->key());
|
|
|
|
return ak.compare(bk) == 1;
|
2016-09-26 06:07:22 +02:00
|
|
|
}
|
|
|
|
|
2017-04-03 06:02:32 +02:00
|
|
|
if(!a && b)
|
2016-09-26 06:07:22 +02:00
|
|
|
return true;
|
|
|
|
|
2017-04-03 06:02:32 +02:00
|
|
|
if(!a && !b)
|
2016-09-26 06:07:22 +02:00
|
|
|
return false;
|
|
|
|
|
2017-04-03 06:02:32 +02:00
|
|
|
assert(!a && b);
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool
|
2017-09-22 12:31:44 +02:00
|
|
|
ircd::db::operator<(const column::const_iterator_base &a, const column::const_iterator_base &b)
|
2017-04-03 06:02:32 +02:00
|
|
|
{
|
|
|
|
if(a && b)
|
|
|
|
{
|
|
|
|
const auto &ak(a.it->key());
|
|
|
|
const auto &bk(b.it->key());
|
|
|
|
return ak.compare(bk) == -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
if(!a && b)
|
|
|
|
return false;
|
|
|
|
|
|
|
|
if(!a && !b)
|
|
|
|
return false;
|
|
|
|
|
|
|
|
assert(a && !b);
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
template<class pos>
|
2017-08-30 23:05:15 +02:00
|
|
|
bool
|
2017-09-22 12:31:44 +02:00
|
|
|
ircd::db::seek(column::const_iterator_base &it,
|
2018-05-25 11:34:44 +02:00
|
|
|
const pos &p)
|
2017-04-03 06:02:32 +02:00
|
|
|
{
|
|
|
|
database::column &c(it);
|
2017-09-22 05:08:11 +02:00
|
|
|
const auto ropts
|
2017-04-03 06:02:32 +02:00
|
|
|
{
|
2018-05-25 11:34:44 +02:00
|
|
|
make_opts(it.opts)
|
2017-04-03 06:02:32 +02:00
|
|
|
};
|
|
|
|
|
2017-09-22 05:08:11 +02:00
|
|
|
return seek(c, p, ropts, it.it);
|
2017-04-03 06:02:32 +02:00
|
|
|
}
|
2018-05-25 11:34:44 +02:00
|
|
|
template bool ircd::db::seek<ircd::db::pos>(column::const_iterator_base &, const pos &);
|
|
|
|
template bool ircd::db::seek<ircd::string_view>(column::const_iterator_base &, const string_view &);
|
2017-04-03 06:02:32 +02:00
|
|
|
|
2017-09-14 20:30:06 +02:00
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
//
|
|
|
|
// merge.h
|
|
|
|
//
|
|
|
|
|
|
|
|
std::string
|
|
|
|
ircd::db::merge_operator(const string_view &key,
|
|
|
|
const std::pair<string_view, string_view> &delta)
|
|
|
|
{
|
|
|
|
//ircd::json::index index{delta.first};
|
|
|
|
//index += delta.second;
|
|
|
|
//return index;
|
|
|
|
assert(0);
|
|
|
|
return {};
|
|
|
|
}
|
|
|
|
|
2017-09-08 11:14:17 +02:00
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
//
|
|
|
|
// writebatch
|
|
|
|
//
|
|
|
|
|
|
|
|
void
|
|
|
|
ircd::db::append(rocksdb::WriteBatch &batch,
|
|
|
|
const cell::delta &delta)
|
|
|
|
{
|
|
|
|
auto &column(std::get<cell *>(delta)->c);
|
|
|
|
append(batch, column, column::delta
|
|
|
|
{
|
|
|
|
std::get<op>(delta),
|
2017-09-14 21:59:20 +02:00
|
|
|
std::get<cell *>(delta)->key(),
|
2017-09-08 11:14:17 +02:00
|
|
|
std::get<string_view>(delta)
|
|
|
|
});
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
ircd::db::append(rocksdb::WriteBatch &batch,
|
|
|
|
column &column,
|
|
|
|
const column::delta &delta)
|
|
|
|
{
|
|
|
|
database::column &c(column);
|
|
|
|
|
|
|
|
const auto k(slice(std::get<1>(delta)));
|
|
|
|
const auto v(slice(std::get<2>(delta)));
|
|
|
|
switch(std::get<0>(delta))
|
|
|
|
{
|
|
|
|
case op::GET: assert(0); break;
|
|
|
|
case op::SET: batch.Put(c, k, v); break;
|
|
|
|
case op::MERGE: batch.Merge(c, k, v); break;
|
|
|
|
case op::DELETE: batch.Delete(c, k); break;
|
|
|
|
case op::DELETE_RANGE: batch.DeleteRange(c, k, v); break;
|
|
|
|
case op::SINGLE_DELETE: batch.SingleDelete(c, k); break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
ircd::db::commit(database &d,
|
|
|
|
rocksdb::WriteBatch &batch,
|
|
|
|
const sopts &sopts)
|
|
|
|
{
|
|
|
|
const auto opts(make_opts(sopts));
|
|
|
|
commit(d, batch, opts);
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
ircd::db::commit(database &d,
|
|
|
|
rocksdb::WriteBatch &batch,
|
|
|
|
const rocksdb::WriteOptions &opts)
|
|
|
|
{
|
2018-03-14 01:32:13 +01:00
|
|
|
#ifdef RB_DEBUG
|
|
|
|
ircd::timer timer;
|
|
|
|
#endif
|
2017-09-08 11:14:17 +02:00
|
|
|
|
|
|
|
throw_on_error
|
|
|
|
{
|
|
|
|
d.d->Write(opts, &batch)
|
|
|
|
};
|
2018-03-14 01:32:13 +01:00
|
|
|
|
|
|
|
#ifdef RB_DEBUG
|
2018-05-29 10:09:15 +02:00
|
|
|
log::debug
|
|
|
|
{
|
|
|
|
log, "'%s' %lu COMMIT %s in %ld$us",
|
|
|
|
d.name,
|
|
|
|
sequence(d),
|
|
|
|
debug(batch),
|
|
|
|
timer.at<microseconds>().count()
|
|
|
|
};
|
2018-03-14 01:32:13 +01:00
|
|
|
#endif
|
2017-09-08 11:14:17 +02:00
|
|
|
}
|
|
|
|
|
2017-09-19 04:17:36 +02:00
|
|
|
std::string
|
|
|
|
ircd::db::debug(const rocksdb::WriteBatch &batch)
|
|
|
|
{
|
2018-05-29 10:09:15 +02:00
|
|
|
return ircd::string(512, [&batch]
|
|
|
|
(const mutable_buffer &ret)
|
|
|
|
{
|
|
|
|
return snprintf(data(ret), size(ret)+1,
|
|
|
|
"%d deltas; size: %zuB :%s%s%s%s%s%s%s%s%s",
|
|
|
|
batch.Count(),
|
|
|
|
batch.GetDataSize(),
|
|
|
|
batch.HasPut()? " PUT" : "",
|
|
|
|
batch.HasDelete()? " DELETE" : "",
|
|
|
|
batch.HasSingleDelete()? " SINGLE_DELETE" : "",
|
|
|
|
batch.HasDeleteRange()? " DELETE_RANGE" : "",
|
|
|
|
batch.HasMerge()? " MERGE" : "",
|
|
|
|
batch.HasBeginPrepare()? " BEGIN_PREPARE" : "",
|
|
|
|
batch.HasEndPrepare()? " END_PREPARE" : "",
|
|
|
|
batch.HasCommit()? " COMMIT" : "",
|
|
|
|
batch.HasRollback()? " ROLLBACK" : "");
|
|
|
|
});
|
2017-09-19 04:17:36 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
bool
|
|
|
|
ircd::db::has(const rocksdb::WriteBatch &wb,
|
|
|
|
const op &op)
|
2017-09-08 11:14:17 +02:00
|
|
|
{
|
2017-09-19 04:17:36 +02:00
|
|
|
switch(op)
|
|
|
|
{
|
|
|
|
case op::GET: assert(0); return false;
|
|
|
|
case op::SET: return wb.HasPut();
|
|
|
|
case op::MERGE: return wb.HasMerge();
|
|
|
|
case op::DELETE: return wb.HasDelete();
|
|
|
|
case op::DELETE_RANGE: return wb.HasDeleteRange();
|
|
|
|
case op::SINGLE_DELETE: return wb.HasSingleDelete();
|
|
|
|
}
|
2017-09-08 11:14:17 +02:00
|
|
|
|
2017-09-19 04:17:36 +02:00
|
|
|
return false;
|
2017-09-08 11:14:17 +02:00
|
|
|
}
|
|
|
|
|
2017-04-03 06:02:32 +02:00
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
//
|
|
|
|
// seek
|
|
|
|
//
|
|
|
|
|
2017-09-08 11:14:17 +02:00
|
|
|
namespace ircd::db
|
|
|
|
{
|
|
|
|
static rocksdb::Iterator &_seek_(rocksdb::Iterator &, const pos &);
|
2017-09-24 04:47:36 +02:00
|
|
|
static rocksdb::Iterator &_seek_(rocksdb::Iterator &, const string_view &);
|
|
|
|
static rocksdb::Iterator &_seek_lower_(rocksdb::Iterator &, const string_view &);
|
|
|
|
static rocksdb::Iterator &_seek_upper_(rocksdb::Iterator &, const string_view &);
|
2017-09-23 10:13:26 +02:00
|
|
|
bool _seek(database::column &, const pos &, const rocksdb::ReadOptions &, std::unique_ptr<rocksdb::Iterator> &it);
|
|
|
|
bool _seek(database::column &, const string_view &, const rocksdb::ReadOptions &, std::unique_ptr<rocksdb::Iterator> &it);
|
2017-09-08 11:14:17 +02:00
|
|
|
}
|
|
|
|
|
2017-04-03 06:02:32 +02:00
|
|
|
std::unique_ptr<rocksdb::Iterator>
|
|
|
|
ircd::db::seek(column &column,
|
|
|
|
const string_view &key,
|
|
|
|
const gopts &opts)
|
|
|
|
{
|
|
|
|
database &d(column);
|
|
|
|
database::column &c(column);
|
2017-08-23 22:37:47 +02:00
|
|
|
|
|
|
|
std::unique_ptr<rocksdb::Iterator> ret;
|
2017-04-03 06:02:32 +02:00
|
|
|
seek(c, key, opts, ret);
|
|
|
|
return std::move(ret);
|
|
|
|
}
|
|
|
|
|
|
|
|
template<class pos>
|
2017-08-30 23:05:15 +02:00
|
|
|
bool
|
2017-04-03 06:02:32 +02:00
|
|
|
ircd::db::seek(database::column &c,
|
|
|
|
const pos &p,
|
|
|
|
const gopts &gopts,
|
|
|
|
std::unique_ptr<rocksdb::Iterator> &it)
|
|
|
|
{
|
|
|
|
auto opts
|
|
|
|
{
|
|
|
|
make_opts(gopts)
|
|
|
|
};
|
|
|
|
|
2017-08-30 23:05:15 +02:00
|
|
|
return seek(c, p, opts, it);
|
2017-04-03 06:02:32 +02:00
|
|
|
}
|
|
|
|
|
2017-09-23 10:13:26 +02:00
|
|
|
template<class pos>
|
|
|
|
bool
|
|
|
|
ircd::db::seek(database::column &c,
|
2018-04-27 02:52:24 +02:00
|
|
|
const pos &p,
|
|
|
|
const rocksdb::ReadOptions &opts,
|
|
|
|
std::unique_ptr<rocksdb::Iterator> &it)
|
2017-09-23 10:13:26 +02:00
|
|
|
{
|
|
|
|
if(!it)
|
|
|
|
{
|
|
|
|
database &d(*c.d);
|
|
|
|
rocksdb::ColumnFamilyHandle *const &cf(c);
|
|
|
|
it.reset(d.d->NewIterator(opts, cf));
|
|
|
|
}
|
|
|
|
|
|
|
|
return _seek(c, p, opts, it);
|
|
|
|
}
|
|
|
|
|
2017-08-30 23:05:15 +02:00
|
|
|
bool
|
2017-09-23 10:13:26 +02:00
|
|
|
ircd::db::_seek(database::column &c,
|
|
|
|
const string_view &p,
|
|
|
|
const rocksdb::ReadOptions &opts,
|
|
|
|
std::unique_ptr<rocksdb::Iterator> &it)
|
2017-04-03 06:02:32 +02:00
|
|
|
{
|
|
|
|
database &d(*c.d);
|
2017-09-23 10:13:26 +02:00
|
|
|
const ircd::timer timer;
|
2017-08-19 00:13:15 +02:00
|
|
|
|
|
|
|
_seek_(*it, p);
|
|
|
|
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_SEEK
|
|
|
|
log::debug
|
|
|
|
{
|
2018-08-16 14:06:26 +02:00
|
|
|
log, "'%s' %lu:%lu SEEK %s in %ld$us '%s'",
|
2018-05-29 10:42:48 +02:00
|
|
|
name(d),
|
|
|
|
sequence(d),
|
|
|
|
sequence(opts.snapshot),
|
2018-08-16 14:06:26 +02:00
|
|
|
it->status().ToString(),
|
2018-05-29 10:42:48 +02:00
|
|
|
timer.at<microseconds>().count(),
|
|
|
|
name(c)
|
|
|
|
};
|
|
|
|
#endif
|
|
|
|
|
2018-08-16 14:06:26 +02:00
|
|
|
return valid(*it);
|
2017-09-23 10:13:26 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
bool
|
|
|
|
ircd::db::_seek(database::column &c,
|
|
|
|
const pos &p,
|
|
|
|
const rocksdb::ReadOptions &opts,
|
|
|
|
std::unique_ptr<rocksdb::Iterator> &it)
|
|
|
|
{
|
|
|
|
database &d(*c.d);
|
|
|
|
const ircd::timer timer;
|
|
|
|
const bool valid_it
|
|
|
|
{
|
|
|
|
valid(*it)
|
|
|
|
};
|
|
|
|
|
|
|
|
_seek_(*it, p);
|
|
|
|
|
2018-05-29 10:42:48 +02:00
|
|
|
#ifdef RB_DEBUG_DB_SEEK
|
|
|
|
log::debug
|
|
|
|
{
|
2018-08-16 14:06:26 +02:00
|
|
|
log, "'%s' %lu:%lu SEEK[%s] %s -> %s in %ld$us '%s'",
|
2018-05-29 10:42:48 +02:00
|
|
|
name(d),
|
|
|
|
sequence(d),
|
|
|
|
sequence(opts.snapshot),
|
|
|
|
reflect(p),
|
|
|
|
valid_it? "VALID" : "INVALID",
|
2018-08-16 14:06:26 +02:00
|
|
|
it->status().ToString(),
|
2018-05-29 10:42:48 +02:00
|
|
|
timer.at<microseconds>().count(),
|
|
|
|
name(c)
|
|
|
|
};
|
|
|
|
#endif
|
|
|
|
|
2018-08-16 14:06:26 +02:00
|
|
|
return valid(*it);
|
2016-09-26 06:07:22 +02:00
|
|
|
}
|
|
|
|
|
2017-09-24 04:47:36 +02:00
|
|
|
/// Seek to entry NOT GREATER THAN key. That is, equal to or less than key
|
|
|
|
rocksdb::Iterator &
|
|
|
|
ircd::db::_seek_lower_(rocksdb::Iterator &it,
|
|
|
|
const string_view &sv)
|
|
|
|
{
|
|
|
|
it.SeekForPrev(slice(sv));
|
|
|
|
return it;
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Seek to entry NOT LESS THAN key. That is, equal to or greater than key
|
|
|
|
rocksdb::Iterator &
|
|
|
|
ircd::db::_seek_upper_(rocksdb::Iterator &it,
|
|
|
|
const string_view &sv)
|
|
|
|
{
|
|
|
|
it.Seek(slice(sv));
|
|
|
|
return it;
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Defaults to _seek_upper_ because it has better support from RocksDB.
|
|
|
|
rocksdb::Iterator &
|
|
|
|
ircd::db::_seek_(rocksdb::Iterator &it,
|
|
|
|
const string_view &sv)
|
|
|
|
{
|
|
|
|
return _seek_upper_(it, sv);
|
|
|
|
}
|
|
|
|
|
2017-09-08 11:14:17 +02:00
|
|
|
rocksdb::Iterator &
|
2017-04-03 06:02:32 +02:00
|
|
|
ircd::db::_seek_(rocksdb::Iterator &it,
|
|
|
|
const pos &p)
|
2016-09-26 06:07:22 +02:00
|
|
|
{
|
2017-04-03 06:02:32 +02:00
|
|
|
switch(p)
|
2016-09-26 06:07:22 +02:00
|
|
|
{
|
2017-04-03 06:02:32 +02:00
|
|
|
case pos::NEXT: it.Next(); break;
|
|
|
|
case pos::PREV: it.Prev(); break;
|
|
|
|
case pos::FRONT: it.SeekToFirst(); break;
|
|
|
|
case pos::BACK: it.SeekToLast(); break;
|
|
|
|
default:
|
|
|
|
case pos::END:
|
|
|
|
{
|
|
|
|
it.SeekToLast();
|
|
|
|
if(it.Valid())
|
|
|
|
it.Next();
|
2016-09-26 06:07:22 +02:00
|
|
|
|
2017-04-03 06:02:32 +02:00
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
2017-09-08 11:14:17 +02:00
|
|
|
|
|
|
|
return it;
|
2016-09-26 06:07:22 +02:00
|
|
|
}
|
|
|
|
|
2018-05-15 01:17:43 +02:00
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
//
|
|
|
|
// cache.h
|
|
|
|
//
|
|
|
|
|
2018-05-15 01:53:28 +02:00
|
|
|
void
|
|
|
|
ircd::db::for_each(rocksdb::Cache *const &cache,
|
|
|
|
const cache_closure &closure)
|
|
|
|
{
|
|
|
|
if(cache)
|
|
|
|
for_each(*cache, closure);
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
ircd::db::for_each(rocksdb::Cache &cache,
|
|
|
|
const cache_closure &closure)
|
|
|
|
{
|
|
|
|
thread_local rocksdb::Cache *_cache;
|
|
|
|
_cache = &cache;
|
|
|
|
|
|
|
|
thread_local const cache_closure *_closure;
|
|
|
|
_closure = &closure;
|
|
|
|
|
|
|
|
cache.ApplyToAllCacheEntries([]
|
|
|
|
(void *const data, const size_t charge)
|
|
|
|
{
|
|
|
|
assert(_cache);
|
|
|
|
assert(_closure);
|
|
|
|
auto *const &handle
|
|
|
|
{
|
|
|
|
reinterpret_cast<rocksdb::Cache::Handle *>(data)
|
|
|
|
};
|
|
|
|
|
|
|
|
const void *const &value
|
|
|
|
{
|
|
|
|
_cache->Value(handle)
|
|
|
|
};
|
|
|
|
|
|
|
|
assert(value);
|
|
|
|
const auto &s
|
|
|
|
{
|
|
|
|
*reinterpret_cast<const rocksdb::Slice *>(value)
|
|
|
|
};
|
|
|
|
|
|
|
|
(*_closure)(slice(s), charge);
|
|
|
|
},
|
|
|
|
false);
|
|
|
|
}
|
|
|
|
|
2018-05-24 04:04:18 +02:00
|
|
|
void
|
|
|
|
ircd::db::clear(rocksdb::Cache *const &cache)
|
|
|
|
{
|
|
|
|
if(cache)
|
|
|
|
clear(*cache);
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
ircd::db::clear(rocksdb::Cache &cache)
|
|
|
|
{
|
|
|
|
cache.EraseUnRefEntries();
|
|
|
|
}
|
|
|
|
|
2018-05-15 02:01:33 +02:00
|
|
|
void
|
|
|
|
ircd::db::remove(rocksdb::Cache *const &cache,
|
|
|
|
const string_view &key)
|
|
|
|
{
|
|
|
|
if(cache)
|
|
|
|
remove(*cache, key);
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
ircd::db::remove(rocksdb::Cache &cache,
|
|
|
|
const string_view &key)
|
|
|
|
{
|
|
|
|
cache.Erase(slice(key));
|
|
|
|
}
|
|
|
|
|
2018-05-15 01:36:48 +02:00
|
|
|
bool
|
|
|
|
ircd::db::exists(rocksdb::Cache *const &cache,
|
|
|
|
const string_view &key)
|
|
|
|
{
|
|
|
|
return cache? exists(*cache, key) : false;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool
|
|
|
|
ircd::db::exists(rocksdb::Cache &cache,
|
|
|
|
const string_view &key)
|
|
|
|
{
|
|
|
|
const custom_ptr<rocksdb::Cache::Handle> handle
|
|
|
|
{
|
|
|
|
cache.Lookup(slice(key)), [&cache](auto *const &handle)
|
|
|
|
{
|
|
|
|
cache.Release(handle);
|
|
|
|
}
|
|
|
|
};
|
|
|
|
|
|
|
|
return bool(handle);
|
|
|
|
}
|
|
|
|
|
2018-05-15 01:17:43 +02:00
|
|
|
size_t
|
|
|
|
ircd::db::usage(const rocksdb::Cache *const &cache)
|
|
|
|
{
|
|
|
|
return cache? usage(*cache) : 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
size_t
|
|
|
|
ircd::db::usage(const rocksdb::Cache &cache)
|
|
|
|
{
|
|
|
|
return cache.GetUsage();
|
|
|
|
}
|
|
|
|
|
|
|
|
bool
|
|
|
|
ircd::db::capacity(rocksdb::Cache *const &cache,
|
|
|
|
const size_t &cap)
|
|
|
|
{
|
|
|
|
if(!cache)
|
|
|
|
return false;
|
|
|
|
|
|
|
|
capacity(*cache, cap);
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
ircd::db::capacity(rocksdb::Cache &cache,
|
|
|
|
const size_t &cap)
|
|
|
|
{
|
|
|
|
cache.SetCapacity(cap);
|
|
|
|
}
|
|
|
|
|
|
|
|
size_t
|
|
|
|
ircd::db::capacity(const rocksdb::Cache *const &cache)
|
|
|
|
{
|
|
|
|
return cache? capacity(*cache): 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
size_t
|
|
|
|
ircd::db::capacity(const rocksdb::Cache &cache)
|
|
|
|
{
|
|
|
|
return cache.GetCapacity();
|
|
|
|
}
|
|
|
|
|
2017-03-24 02:36:49 +01:00
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
//
|
|
|
|
// Misc
|
|
|
|
//
|
|
|
|
|
|
|
|
std::vector<std::string>
|
|
|
|
ircd::db::column_names(const std::string &path,
|
2017-03-31 00:57:08 +02:00
|
|
|
const std::string &options)
|
2017-03-24 02:36:49 +01:00
|
|
|
{
|
2017-09-19 04:17:36 +02:00
|
|
|
return column_names(path, database::options{options});
|
2017-03-24 02:36:49 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
std::vector<std::string>
|
|
|
|
ircd::db::column_names(const std::string &path,
|
|
|
|
const rocksdb::DBOptions &opts)
|
|
|
|
try
|
|
|
|
{
|
|
|
|
std::vector<std::string> ret;
|
2017-04-03 06:02:32 +02:00
|
|
|
throw_on_error
|
|
|
|
{
|
|
|
|
rocksdb::DB::ListColumnFamilies(opts, path, &ret)
|
|
|
|
};
|
|
|
|
|
2017-03-24 02:36:49 +01:00
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
catch(const io_error &e)
|
|
|
|
{
|
|
|
|
return // No database found at path. Assume fresh.
|
|
|
|
{
|
|
|
|
{ rocksdb::kDefaultColumnFamilyName }
|
|
|
|
};
|
|
|
|
}
|
|
|
|
|
2017-03-31 00:57:08 +02:00
|
|
|
ircd::db::database::options::options(const database &d)
|
|
|
|
:options{d.d->GetDBOptions()}
|
2016-09-25 03:18:54 +02:00
|
|
|
{
|
2017-03-31 00:57:08 +02:00
|
|
|
}
|
2016-09-25 03:18:54 +02:00
|
|
|
|
2017-03-31 00:57:08 +02:00
|
|
|
ircd::db::database::options::options(const database::column &c)
|
2017-04-03 06:02:32 +02:00
|
|
|
:options
|
2017-03-31 00:57:08 +02:00
|
|
|
{
|
2017-04-03 06:02:32 +02:00
|
|
|
rocksdb::ColumnFamilyOptions
|
|
|
|
{
|
|
|
|
c.d->d->GetOptions(c.handle.get())
|
|
|
|
}
|
|
|
|
}{}
|
2016-09-25 03:18:54 +02:00
|
|
|
|
2017-03-31 00:57:08 +02:00
|
|
|
ircd::db::database::options::options(const rocksdb::DBOptions &opts)
|
|
|
|
{
|
2017-04-03 06:02:32 +02:00
|
|
|
throw_on_error
|
|
|
|
{
|
|
|
|
rocksdb::GetStringFromDBOptions(this, opts)
|
|
|
|
};
|
2017-03-31 00:57:08 +02:00
|
|
|
}
|
2016-09-25 03:18:54 +02:00
|
|
|
|
2017-03-31 00:57:08 +02:00
|
|
|
ircd::db::database::options::options(const rocksdb::ColumnFamilyOptions &opts)
|
|
|
|
{
|
2017-04-03 06:02:32 +02:00
|
|
|
throw_on_error
|
|
|
|
{
|
|
|
|
rocksdb::GetStringFromColumnFamilyOptions(this, opts)
|
|
|
|
};
|
2017-03-31 00:57:08 +02:00
|
|
|
}
|
2016-09-25 03:18:54 +02:00
|
|
|
|
2017-03-31 00:57:08 +02:00
|
|
|
ircd::db::database::options::operator rocksdb::PlainTableOptions()
|
|
|
|
const
|
|
|
|
{
|
|
|
|
rocksdb::PlainTableOptions ret;
|
2017-04-03 06:02:32 +02:00
|
|
|
throw_on_error
|
|
|
|
{
|
|
|
|
rocksdb::GetPlainTableOptionsFromString(ret, *this, &ret)
|
|
|
|
};
|
|
|
|
|
2017-03-31 00:57:08 +02:00
|
|
|
return ret;
|
|
|
|
}
|
2016-09-25 03:18:54 +02:00
|
|
|
|
2017-03-31 00:57:08 +02:00
|
|
|
ircd::db::database::options::operator rocksdb::BlockBasedTableOptions()
|
|
|
|
const
|
|
|
|
{
|
|
|
|
rocksdb::BlockBasedTableOptions ret;
|
2017-04-03 06:02:32 +02:00
|
|
|
throw_on_error
|
|
|
|
{
|
|
|
|
rocksdb::GetBlockBasedTableOptionsFromString(ret, *this, &ret)
|
|
|
|
};
|
|
|
|
|
2017-03-31 00:57:08 +02:00
|
|
|
return ret;
|
|
|
|
}
|
2016-09-25 03:18:54 +02:00
|
|
|
|
2017-03-31 00:57:08 +02:00
|
|
|
ircd::db::database::options::operator rocksdb::ColumnFamilyOptions()
|
|
|
|
const
|
|
|
|
{
|
|
|
|
rocksdb::ColumnFamilyOptions ret;
|
2017-04-03 06:02:32 +02:00
|
|
|
throw_on_error
|
|
|
|
{
|
|
|
|
rocksdb::GetColumnFamilyOptionsFromString(ret, *this, &ret)
|
|
|
|
};
|
|
|
|
|
2017-03-31 00:57:08 +02:00
|
|
|
return ret;
|
|
|
|
}
|
2016-09-25 03:18:54 +02:00
|
|
|
|
2017-03-31 00:57:08 +02:00
|
|
|
ircd::db::database::options::operator rocksdb::DBOptions()
|
|
|
|
const
|
|
|
|
{
|
|
|
|
rocksdb::DBOptions ret;
|
2017-04-03 06:02:32 +02:00
|
|
|
throw_on_error
|
|
|
|
{
|
|
|
|
rocksdb::GetDBOptionsFromString(ret, *this, &ret)
|
|
|
|
};
|
|
|
|
|
2017-03-31 00:57:08 +02:00
|
|
|
return ret;
|
|
|
|
}
|
2016-09-25 03:18:54 +02:00
|
|
|
|
2017-03-31 00:57:08 +02:00
|
|
|
ircd::db::database::options::operator rocksdb::Options()
|
|
|
|
const
|
|
|
|
{
|
|
|
|
rocksdb::Options ret;
|
2017-04-03 06:02:32 +02:00
|
|
|
throw_on_error
|
|
|
|
{
|
|
|
|
rocksdb::GetOptionsFromString(ret, *this, &ret)
|
|
|
|
};
|
|
|
|
|
2017-03-31 00:57:08 +02:00
|
|
|
return ret;
|
|
|
|
}
|
2016-09-25 03:18:54 +02:00
|
|
|
|
2017-03-31 00:57:08 +02:00
|
|
|
ircd::db::database::options::map::map(const options &o)
|
|
|
|
{
|
2017-04-03 06:02:32 +02:00
|
|
|
throw_on_error
|
|
|
|
{
|
|
|
|
rocksdb::StringToMap(o, this)
|
|
|
|
};
|
2017-03-31 00:57:08 +02:00
|
|
|
}
|
2016-09-25 03:18:54 +02:00
|
|
|
|
2017-03-31 00:57:08 +02:00
|
|
|
ircd::db::database::options::map::operator rocksdb::PlainTableOptions()
|
|
|
|
const
|
|
|
|
{
|
|
|
|
rocksdb::PlainTableOptions ret;
|
2017-04-03 06:02:32 +02:00
|
|
|
throw_on_error
|
|
|
|
{
|
|
|
|
rocksdb::GetPlainTableOptionsFromMap(ret, *this, &ret)
|
|
|
|
};
|
|
|
|
|
2017-03-31 00:57:08 +02:00
|
|
|
return ret;
|
|
|
|
}
|
2016-09-25 03:18:54 +02:00
|
|
|
|
2017-03-31 00:57:08 +02:00
|
|
|
ircd::db::database::options::map::operator rocksdb::BlockBasedTableOptions()
|
|
|
|
const
|
|
|
|
{
|
|
|
|
rocksdb::BlockBasedTableOptions ret;
|
2017-04-03 06:02:32 +02:00
|
|
|
throw_on_error
|
|
|
|
{
|
|
|
|
rocksdb::GetBlockBasedTableOptionsFromMap(ret, *this, &ret)
|
|
|
|
};
|
|
|
|
|
2017-03-31 00:57:08 +02:00
|
|
|
return ret;
|
|
|
|
}
|
2016-09-25 03:18:54 +02:00
|
|
|
|
2017-03-31 00:57:08 +02:00
|
|
|
ircd::db::database::options::map::operator rocksdb::ColumnFamilyOptions()
|
|
|
|
const
|
|
|
|
{
|
|
|
|
rocksdb::ColumnFamilyOptions ret;
|
2017-04-03 06:02:32 +02:00
|
|
|
throw_on_error
|
|
|
|
{
|
|
|
|
rocksdb::GetColumnFamilyOptionsFromMap(ret, *this, &ret)
|
|
|
|
};
|
|
|
|
|
2017-03-31 00:57:08 +02:00
|
|
|
return ret;
|
|
|
|
}
|
2016-09-25 03:18:54 +02:00
|
|
|
|
2017-03-31 00:57:08 +02:00
|
|
|
ircd::db::database::options::map::operator rocksdb::DBOptions()
|
|
|
|
const
|
|
|
|
{
|
|
|
|
rocksdb::DBOptions ret;
|
2017-04-03 06:02:32 +02:00
|
|
|
throw_on_error
|
|
|
|
{
|
|
|
|
rocksdb::GetDBOptionsFromMap(ret, *this, &ret)
|
|
|
|
};
|
|
|
|
|
2016-09-25 03:18:54 +02:00
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
2017-09-16 18:41:10 +02:00
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
//
|
|
|
|
// Misc
|
|
|
|
//
|
|
|
|
|
2017-09-19 04:17:36 +02:00
|
|
|
rocksdb::DBOptions
|
2018-04-09 20:51:36 +02:00
|
|
|
ircd::db::make_dbopts(std::string optstr,
|
|
|
|
std::string *const &out,
|
2017-09-19 04:17:36 +02:00
|
|
|
bool *const read_only,
|
|
|
|
bool *const fsck)
|
|
|
|
{
|
|
|
|
// RocksDB doesn't parse a read_only option, so we allow that to be added
|
|
|
|
// to open the database as read_only and then remove that from the string.
|
|
|
|
if(read_only)
|
|
|
|
*read_only = optstr_find_and_remove(optstr, "read_only=true;"s);
|
2018-04-09 20:51:36 +02:00
|
|
|
else
|
|
|
|
optstr_find_and_remove(optstr, "read_only=true;"s);
|
2017-09-19 04:17:36 +02:00
|
|
|
|
|
|
|
// We also allow the user to specify fsck=true to run a repair operation on
|
|
|
|
// the db. This may be expensive to do by default every startup.
|
|
|
|
if(fsck)
|
|
|
|
*fsck = optstr_find_and_remove(optstr, "fsck=true;"s);
|
2018-04-09 20:51:36 +02:00
|
|
|
else
|
|
|
|
optstr_find_and_remove(optstr, "fsck=true;"s);
|
2017-09-19 04:17:36 +02:00
|
|
|
|
|
|
|
// Generate RocksDB options from string
|
|
|
|
rocksdb::DBOptions opts
|
|
|
|
{
|
|
|
|
database::options(optstr)
|
|
|
|
};
|
|
|
|
|
2018-04-09 20:51:36 +02:00
|
|
|
if(out)
|
|
|
|
*out = std::move(optstr);
|
|
|
|
|
2017-09-19 04:17:36 +02:00
|
|
|
return opts;
|
|
|
|
}
|
|
|
|
|
2017-08-23 22:37:47 +02:00
|
|
|
bool
|
|
|
|
ircd::db::optstr_find_and_remove(std::string &optstr,
|
|
|
|
const std::string &what)
|
|
|
|
{
|
|
|
|
const auto pos(optstr.find(what));
|
|
|
|
if(pos == std::string::npos)
|
|
|
|
return false;
|
|
|
|
|
|
|
|
optstr.erase(pos, what.size());
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2016-09-25 03:18:54 +02:00
|
|
|
rocksdb::ReadOptions
|
2017-09-22 05:08:11 +02:00
|
|
|
ircd::db::make_opts(const gopts &opts)
|
2016-09-25 03:18:54 +02:00
|
|
|
{
|
|
|
|
rocksdb::ReadOptions ret;
|
2017-09-23 10:13:26 +02:00
|
|
|
assert(ret.fill_cache);
|
2018-08-16 14:06:26 +02:00
|
|
|
ret.read_tier = BLOCKING;
|
2016-09-25 03:18:54 +02:00
|
|
|
|
2017-09-22 05:08:11 +02:00
|
|
|
// slice* for exclusive upper bound. when prefixes are used this value must
|
|
|
|
// have the same prefix because ordering is not guaranteed between prefixes
|
2018-04-14 07:08:57 +02:00
|
|
|
ret.iterate_lower_bound = opts.lower_bound;
|
|
|
|
ret.iterate_upper_bound = opts.upper_bound;
|
2016-09-25 03:18:54 +02:00
|
|
|
|
2017-09-22 05:08:11 +02:00
|
|
|
ret += opts;
|
|
|
|
return ret;
|
|
|
|
}
|
2017-08-31 07:12:58 +02:00
|
|
|
|
2018-05-24 03:43:47 +02:00
|
|
|
ircd::conf::item<bool>
|
|
|
|
read_checksum
|
|
|
|
{
|
|
|
|
{ "name", "ircd.db.read.checksum" },
|
|
|
|
{ "default", false }
|
|
|
|
};
|
|
|
|
|
2017-09-22 05:08:11 +02:00
|
|
|
rocksdb::ReadOptions &
|
|
|
|
ircd::db::operator+=(rocksdb::ReadOptions &ret,
|
|
|
|
const gopts &opts)
|
|
|
|
{
|
2018-04-14 07:08:57 +02:00
|
|
|
ret.iter_start_seqnum = opts.seqnum;
|
|
|
|
ret.readahead_size = opts.readahead;
|
|
|
|
|
2017-09-22 05:08:11 +02:00
|
|
|
if(opts.snapshot && !test(opts, get::NO_SNAPSHOT))
|
|
|
|
ret.snapshot = opts.snapshot;
|
2016-09-25 03:18:54 +02:00
|
|
|
|
2017-09-22 05:08:11 +02:00
|
|
|
ret.pin_data = test(opts, get::PIN);
|
|
|
|
ret.fill_cache |= test(opts, get::CACHE);
|
|
|
|
ret.fill_cache &= !test(opts, get::NO_CACHE);
|
|
|
|
ret.tailing = test(opts, get::NO_SNAPSHOT);
|
|
|
|
ret.prefix_same_as_start = test(opts, get::PREFIX);
|
2017-09-24 04:47:36 +02:00
|
|
|
ret.total_order_seek = test(opts, get::ORDERED);
|
2018-05-24 03:43:47 +02:00
|
|
|
ret.verify_checksums = bool(read_checksum);
|
|
|
|
ret.verify_checksums |= test(opts, get::CHECKSUM);
|
|
|
|
ret.verify_checksums &= !test(opts, get::NO_CHECKSUM);
|
2016-09-25 03:18:54 +02:00
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::WriteOptions
|
2017-03-31 00:57:08 +02:00
|
|
|
ircd::db::make_opts(const sopts &opts)
|
2016-09-25 03:18:54 +02:00
|
|
|
{
|
|
|
|
rocksdb::WriteOptions ret;
|
2017-11-30 20:04:25 +01:00
|
|
|
//ret.no_slowdown = true; // read_tier = NON_BLOCKING for writes
|
2017-09-22 05:08:11 +02:00
|
|
|
ret += opts;
|
|
|
|
return ret;
|
|
|
|
}
|
2016-09-25 03:18:54 +02:00
|
|
|
|
2017-09-22 05:08:11 +02:00
|
|
|
rocksdb::WriteOptions &
|
|
|
|
ircd::db::operator+=(rocksdb::WriteOptions &ret,
|
|
|
|
const sopts &opts)
|
|
|
|
{
|
|
|
|
ret.sync = test(opts, set::FSYNC);
|
|
|
|
ret.disableWAL = test(opts, set::NO_JOURNAL);
|
|
|
|
ret.ignore_missing_column_families = test(opts, set::MISSING_COLUMNS);
|
2016-09-25 03:18:54 +02:00
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
2017-03-31 00:57:08 +02:00
|
|
|
void
|
2017-09-16 20:48:09 +02:00
|
|
|
ircd::db::valid_eq_or_throw(const rocksdb::Iterator &it,
|
|
|
|
const string_view &sv)
|
2016-09-26 06:07:22 +02:00
|
|
|
{
|
2018-04-18 05:07:21 +02:00
|
|
|
assert(!empty(sv));
|
2017-09-16 20:48:09 +02:00
|
|
|
if(!valid_eq(it, sv))
|
|
|
|
{
|
|
|
|
throw_on_error(it.status());
|
2018-04-18 05:07:21 +02:00
|
|
|
throw not_found{};
|
2017-09-16 20:48:09 +02:00
|
|
|
}
|
2017-03-31 00:57:08 +02:00
|
|
|
}
|
|
|
|
|
2016-09-26 06:07:22 +02:00
|
|
|
void
|
2017-03-31 00:57:08 +02:00
|
|
|
ircd::db::valid_or_throw(const rocksdb::Iterator &it)
|
2016-09-26 06:07:22 +02:00
|
|
|
{
|
|
|
|
if(!valid(it))
|
|
|
|
{
|
|
|
|
throw_on_error(it.status());
|
2018-04-18 05:07:21 +02:00
|
|
|
throw not_found{};
|
2017-03-20 12:25:01 +01:00
|
|
|
//assert(0); // status == ok + !Valid() == ???
|
2016-09-26 06:07:22 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2017-09-16 20:48:09 +02:00
|
|
|
bool
|
|
|
|
ircd::db::valid_lte(const rocksdb::Iterator &it,
|
|
|
|
const string_view &sv)
|
|
|
|
{
|
|
|
|
return valid(it, [&sv](const auto &it)
|
|
|
|
{
|
|
|
|
return it.key().compare(slice(sv)) <= 0;
|
|
|
|
});
|
|
|
|
}
|
|
|
|
|
|
|
|
bool
|
|
|
|
ircd::db::valid_gt(const rocksdb::Iterator &it,
|
|
|
|
const string_view &sv)
|
|
|
|
{
|
|
|
|
return valid(it, [&sv](const auto &it)
|
|
|
|
{
|
|
|
|
return it.key().compare(slice(sv)) > 0;
|
|
|
|
});
|
|
|
|
}
|
|
|
|
|
|
|
|
bool
|
|
|
|
ircd::db::valid_eq(const rocksdb::Iterator &it,
|
|
|
|
const string_view &sv)
|
|
|
|
{
|
|
|
|
return valid(it, [&sv](const auto &it)
|
|
|
|
{
|
|
|
|
return it.key().compare(slice(sv)) == 0;
|
|
|
|
});
|
|
|
|
}
|
|
|
|
|
|
|
|
bool
|
|
|
|
ircd::db::valid(const rocksdb::Iterator &it,
|
|
|
|
const valid_proffer &proffer)
|
|
|
|
{
|
|
|
|
return valid(it)? proffer(it) : false;
|
|
|
|
}
|
|
|
|
|
2016-09-26 06:07:22 +02:00
|
|
|
bool
|
2017-03-31 00:57:08 +02:00
|
|
|
ircd::db::operator!(const rocksdb::Iterator &it)
|
|
|
|
{
|
2017-09-08 11:14:17 +02:00
|
|
|
return !valid(it);
|
2017-03-31 00:57:08 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
bool
|
|
|
|
ircd::db::valid(const rocksdb::Iterator &it)
|
2016-09-26 06:07:22 +02:00
|
|
|
{
|
2017-09-08 11:14:17 +02:00
|
|
|
switch(it.status().code())
|
|
|
|
{
|
|
|
|
using rocksdb::Status;
|
|
|
|
|
|
|
|
case Status::kOk: break;
|
|
|
|
case Status::kNotFound: break;
|
|
|
|
case Status::kIncomplete: break;
|
|
|
|
default:
|
|
|
|
throw_on_error(it.status());
|
|
|
|
__builtin_unreachable();
|
|
|
|
}
|
|
|
|
|
2016-09-26 06:07:22 +02:00
|
|
|
return it.Valid();
|
2016-09-25 03:18:54 +02:00
|
|
|
}
|
|
|
|
|
2017-03-31 00:57:08 +02:00
|
|
|
ircd::db::throw_on_error::throw_on_error(const rocksdb::Status &s)
|
2016-09-24 06:01:57 +02:00
|
|
|
{
|
|
|
|
using rocksdb::Status;
|
|
|
|
|
|
|
|
switch(s.code())
|
|
|
|
{
|
|
|
|
case Status::kOk: return;
|
2017-03-31 00:57:08 +02:00
|
|
|
case Status::kNotFound: throw not_found("%s", s.ToString());
|
|
|
|
case Status::kCorruption: throw corruption("%s", s.ToString());
|
|
|
|
case Status::kNotSupported: throw not_supported("%s", s.ToString());
|
|
|
|
case Status::kInvalidArgument: throw invalid_argument("%s", s.ToString());
|
|
|
|
case Status::kIOError: throw io_error("%s", s.ToString());
|
|
|
|
case Status::kMergeInProgress: throw merge_in_progress("%s", s.ToString());
|
|
|
|
case Status::kIncomplete: throw incomplete("%s", s.ToString());
|
|
|
|
case Status::kShutdownInProgress: throw shutdown_in_progress("%s", s.ToString());
|
|
|
|
case Status::kTimedOut: throw timed_out("%s", s.ToString());
|
|
|
|
case Status::kAborted: throw aborted("%s", s.ToString());
|
|
|
|
case Status::kBusy: throw busy("%s", s.ToString());
|
|
|
|
case Status::kExpired: throw expired("%s", s.ToString());
|
|
|
|
case Status::kTryAgain: throw try_again("%s", s.ToString());
|
2016-09-24 06:01:57 +02:00
|
|
|
default:
|
2017-03-31 00:57:08 +02:00
|
|
|
throw error("code[%d] %s", s.code(), s.ToString());
|
2016-09-24 06:01:57 +02:00
|
|
|
}
|
|
|
|
}
|
2016-09-25 06:12:43 +02:00
|
|
|
|
2017-03-31 00:57:08 +02:00
|
|
|
std::vector<std::string>
|
|
|
|
ircd::db::available()
|
|
|
|
{
|
2018-05-25 05:01:52 +02:00
|
|
|
const auto prefix
|
|
|
|
{
|
|
|
|
fs::get(fs::DB)
|
|
|
|
};
|
|
|
|
|
|
|
|
const auto dirs
|
|
|
|
{
|
|
|
|
fs::ls(prefix)
|
|
|
|
};
|
|
|
|
|
|
|
|
std::vector<std::string> ret;
|
|
|
|
for(const auto &dir : dirs)
|
|
|
|
{
|
|
|
|
const auto name
|
|
|
|
{
|
|
|
|
lstrip(dir, prefix)
|
|
|
|
};
|
|
|
|
|
|
|
|
const auto checkpoints
|
|
|
|
{
|
|
|
|
fs::ls(dir)
|
|
|
|
};
|
|
|
|
|
|
|
|
for(const auto cpdir : checkpoints) try
|
|
|
|
{
|
|
|
|
const auto checkpoint
|
|
|
|
{
|
|
|
|
lstrip(lstrip(cpdir, dir), '/') //TODO: x-platform
|
|
|
|
};
|
|
|
|
|
|
|
|
auto path
|
|
|
|
{
|
|
|
|
db::path(name, lex_cast<uint64_t>(checkpoint))
|
|
|
|
};
|
|
|
|
|
|
|
|
ret.emplace_back(std::move(path));
|
|
|
|
}
|
|
|
|
catch(const bad_lex_cast &e)
|
|
|
|
{
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
|
|
|
std::string
|
|
|
|
ircd::db::path(const string_view &name)
|
|
|
|
{
|
|
|
|
const auto pair
|
|
|
|
{
|
|
|
|
namepoint(name)
|
|
|
|
};
|
|
|
|
|
|
|
|
return path(pair.first, pair.second);
|
|
|
|
}
|
|
|
|
|
|
|
|
std::string
|
|
|
|
ircd::db::path(const string_view &name,
|
|
|
|
const uint64_t &checkpoint)
|
|
|
|
{
|
|
|
|
const auto prefix
|
|
|
|
{
|
|
|
|
fs::get(fs::DB)
|
|
|
|
};
|
|
|
|
|
|
|
|
return fs::make_path(
|
|
|
|
{
|
|
|
|
prefix, name, lex_cast(checkpoint)
|
|
|
|
});
|
|
|
|
}
|
|
|
|
|
|
|
|
std::pair<ircd::string_view, uint64_t>
|
|
|
|
ircd::db::namepoint(const string_view &name_)
|
|
|
|
{
|
|
|
|
const auto s
|
|
|
|
{
|
|
|
|
split(name_, ':')
|
|
|
|
};
|
|
|
|
|
|
|
|
return
|
|
|
|
{
|
|
|
|
s.first,
|
|
|
|
s.second? lex_cast<uint64_t>(s.second) : uint64_t(-1)
|
|
|
|
};
|
2017-03-31 00:57:08 +02:00
|
|
|
}
|
|
|
|
|
2016-09-26 06:07:22 +02:00
|
|
|
std::string
|
2018-05-25 05:01:52 +02:00
|
|
|
ircd::db::namepoint(const string_view &name,
|
|
|
|
const uint64_t &checkpoint)
|
2017-03-31 00:57:08 +02:00
|
|
|
{
|
2018-05-25 05:01:52 +02:00
|
|
|
return std::string{name} + ':' + std::string{lex_cast(checkpoint)};
|
2017-03-31 00:57:08 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
std::pair<ircd::string_view, ircd::string_view>
|
|
|
|
ircd::db::operator*(const rocksdb::Iterator &it)
|
|
|
|
{
|
2017-04-03 06:02:32 +02:00
|
|
|
return { key(it), val(it) };
|
|
|
|
}
|
|
|
|
|
|
|
|
ircd::string_view
|
|
|
|
ircd::db::key(const rocksdb::Iterator &it)
|
|
|
|
{
|
|
|
|
return slice(it.key());
|
|
|
|
}
|
|
|
|
|
|
|
|
ircd::string_view
|
|
|
|
ircd::db::val(const rocksdb::Iterator &it)
|
|
|
|
{
|
|
|
|
return slice(it.value());
|
2017-03-31 00:57:08 +02:00
|
|
|
}
|
|
|
|
|
2018-01-18 05:28:34 +01:00
|
|
|
const char *
|
|
|
|
ircd::db::data(const rocksdb::Slice &slice)
|
|
|
|
{
|
|
|
|
return slice.data();
|
|
|
|
}
|
|
|
|
|
2018-01-18 05:15:13 +01:00
|
|
|
size_t
|
|
|
|
ircd::db::size(const rocksdb::Slice &slice)
|
|
|
|
{
|
|
|
|
return slice.size();
|
|
|
|
}
|
|
|
|
|
2017-03-31 00:57:08 +02:00
|
|
|
rocksdb::Slice
|
|
|
|
ircd::db::slice(const string_view &sv)
|
2016-09-26 06:07:22 +02:00
|
|
|
{
|
2017-03-31 00:57:08 +02:00
|
|
|
return { sv.data(), sv.size() };
|
2016-09-26 06:07:22 +02:00
|
|
|
}
|
2017-03-23 22:58:24 +01:00
|
|
|
|
2017-04-03 06:02:32 +02:00
|
|
|
ircd::string_view
|
|
|
|
ircd::db::slice(const rocksdb::Slice &sk)
|
|
|
|
{
|
|
|
|
return { sk.data(), sk.size() };
|
|
|
|
}
|
|
|
|
|
2017-03-23 22:58:24 +01:00
|
|
|
const std::string &
|
|
|
|
ircd::db::reflect(const rocksdb::Tickers &type)
|
|
|
|
{
|
|
|
|
const auto &names(rocksdb::TickersNameMap);
|
|
|
|
const auto it(std::find_if(begin(names), end(names), [&type]
|
|
|
|
(const auto &pair)
|
|
|
|
{
|
|
|
|
return pair.first == type;
|
|
|
|
}));
|
|
|
|
|
|
|
|
static const auto empty{"<ticker>?????"s};
|
|
|
|
return it != end(names)? it->second : empty;
|
|
|
|
}
|
|
|
|
|
|
|
|
const std::string &
|
|
|
|
ircd::db::reflect(const rocksdb::Histograms &type)
|
|
|
|
{
|
|
|
|
const auto &names(rocksdb::HistogramsNameMap);
|
|
|
|
const auto it(std::find_if(begin(names), end(names), [&type]
|
|
|
|
(const auto &pair)
|
|
|
|
{
|
|
|
|
return pair.first == type;
|
|
|
|
}));
|
|
|
|
|
|
|
|
static const auto empty{"<histogram>?????"s};
|
|
|
|
return it != end(names)? it->second : empty;
|
|
|
|
}
|
2017-08-23 22:37:47 +02:00
|
|
|
|
2017-09-16 20:51:52 +02:00
|
|
|
ircd::string_view
|
2017-08-30 23:06:34 +02:00
|
|
|
ircd::db::reflect(const pos &pos)
|
|
|
|
{
|
|
|
|
switch(pos)
|
|
|
|
{
|
2017-09-19 04:17:36 +02:00
|
|
|
case pos::NEXT: return "NEXT";
|
|
|
|
case pos::PREV: return "PREV";
|
|
|
|
case pos::FRONT: return "FRONT";
|
|
|
|
case pos::BACK: return "BACK";
|
|
|
|
case pos::END: return "END";
|
|
|
|
}
|
|
|
|
|
|
|
|
return "?????";
|
|
|
|
}
|
|
|
|
|
|
|
|
ircd::string_view
|
|
|
|
ircd::db::reflect(const op &op)
|
|
|
|
{
|
|
|
|
switch(op)
|
|
|
|
{
|
|
|
|
case op::GET: return "GET";
|
|
|
|
case op::SET: return "SET";
|
|
|
|
case op::MERGE: return "MERGE";
|
|
|
|
case op::DELETE_RANGE: return "DELETE_RANGE";
|
|
|
|
case op::DELETE: return "DELETE";
|
|
|
|
case op::SINGLE_DELETE: return "SINGLE_DELETE";
|
2017-08-30 23:06:34 +02:00
|
|
|
}
|
|
|
|
|
2017-09-19 04:17:36 +02:00
|
|
|
return "?????";
|
2017-08-30 23:06:34 +02:00
|
|
|
}
|
|
|
|
|
2018-01-18 04:22:35 +01:00
|
|
|
ircd::string_view
|
|
|
|
ircd::db::reflect(const rocksdb::Env::Priority &p)
|
|
|
|
{
|
|
|
|
switch(p)
|
|
|
|
{
|
2018-04-09 21:55:22 +02:00
|
|
|
case rocksdb::Env::Priority::BOTTOM: return "BOTTOM"_sv;
|
2018-01-18 04:22:35 +01:00
|
|
|
case rocksdb::Env::Priority::LOW: return "LOW"_sv;
|
|
|
|
case rocksdb::Env::Priority::HIGH: return "HIGH"_sv;
|
2018-08-16 11:04:16 +02:00
|
|
|
case rocksdb::Env::Priority::TOTAL: assert(0); break;
|
2018-01-18 04:22:35 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
return "????"_sv;
|
|
|
|
}
|
|
|
|
|
|
|
|
ircd::string_view
|
|
|
|
ircd::db::reflect(const rocksdb::Env::IOPriority &p)
|
|
|
|
{
|
|
|
|
switch(p)
|
|
|
|
{
|
|
|
|
case rocksdb::Env::IOPriority::IO_LOW: return "IO_LOW"_sv;
|
|
|
|
case rocksdb::Env::IOPriority::IO_HIGH: return "IO_HIGH"_sv;
|
2018-08-16 11:04:16 +02:00
|
|
|
case rocksdb::Env::IOPriority::IO_TOTAL: assert(0); break;
|
2018-01-18 04:22:35 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
return "IO_????"_sv;
|
|
|
|
}
|
|
|
|
|
2018-01-18 06:24:32 +01:00
|
|
|
ircd::string_view
|
|
|
|
ircd::db::reflect(const rocksdb::RandomAccessFile::AccessPattern &p)
|
|
|
|
{
|
|
|
|
switch(p)
|
|
|
|
{
|
|
|
|
case rocksdb::RandomAccessFile::AccessPattern::NORMAL: return "NORMAL"_sv;
|
|
|
|
case rocksdb::RandomAccessFile::AccessPattern::RANDOM: return "RANDOM"_sv;
|
|
|
|
case rocksdb::RandomAccessFile::AccessPattern::SEQUENTIAL: return "SEQUENTIAL"_sv;
|
|
|
|
case rocksdb::RandomAccessFile::AccessPattern::WILLNEED: return "WILLNEED"_sv;
|
|
|
|
case rocksdb::RandomAccessFile::AccessPattern::DONTNEED: return "DONTNEED"_sv;
|
|
|
|
}
|
|
|
|
|
|
|
|
return "??????"_sv;
|
|
|
|
}
|
|
|
|
|
2017-08-23 22:37:47 +02:00
|
|
|
bool
|
|
|
|
ircd::db::value_required(const op &op)
|
|
|
|
{
|
|
|
|
switch(op)
|
|
|
|
{
|
|
|
|
case op::SET:
|
|
|
|
case op::MERGE:
|
|
|
|
case op::DELETE_RANGE:
|
|
|
|
return true;
|
|
|
|
|
|
|
|
case op::GET:
|
|
|
|
case op::DELETE:
|
|
|
|
case op::SINGLE_DELETE:
|
|
|
|
return false;
|
|
|
|
}
|
2017-08-30 23:21:18 +02:00
|
|
|
|
|
|
|
assert(0);
|
|
|
|
return false;
|
2017-08-23 22:37:47 +02:00
|
|
|
}
|