t_pebbles.cc

/* Copyright (c) 2017 Howard Chu @ Symas Corp. */

#include <sys/types.h>
#include <sys/stat.h>
#include <stdio.h>
#include <errno.h>
#include <stdlib.h>

#include <hyperleveldb/cache.h>
#include <hyperleveldb/db.h>
#include <hyperleveldb/filter_policy.h>
#include <hyperleveldb/write_batch.h>

#include "dbb.h"

// Number of bytes to buffer in memtable before compacting
// (initialized to default value by "main")
static int FLAGS_write_buffer_size = 0;

// Number of bytes to use as a cache of uncompressed data.
// Negative means use default settings.
static int64_t FLAGS_cache_size = -1;

// Maximum number of files to keep open at the same time (use default if == 0)
static int FLAGS_open_files = 0;

// Bloom filter bits per key.
// Negative means use default settings.
static int FLAGS_bloom_bits = -1;

using namespace leveldb;

static Cache *cache;
static const FilterPolicy *filter_policy;
static DB *db;
static WriteOptions write_options;

static void db_open(int dbflags) {
	Options options;
	options.create_if_missing = !FLAGS_use_existing_db;

	if (!FLAGS_write_buffer_size)
		FLAGS_write_buffer_size = Options().write_buffer_size;
	if (!FLAGS_open_files)
		FLAGS_open_files = Options().max_open_files;
	if (FLAGS_cache_size >= 0 && !cache)
		cache = NewLRUCache(FLAGS_cache_size);
	if (FLAGS_bloom_bits >= 0 && !filter_policy)
		filter_policy = NewBloomFilterPolicy(FLAGS_bloom_bits);

	options.block_cache = cache;
	options.write_buffer_size = FLAGS_write_buffer_size;
	options.max_open_files = FLAGS_open_files;
	options.filter_policy = filter_policy;
	options.compression = FLAGS_compression != 0 ? kSnappyCompression : kNoCompression;
	write_options = WriteOptions();
	if (dbflags & DBB_SYNC)
		write_options.sync = true;

	Status s = DB::Open(options, FLAGS_db, &db);
	if (!s.ok()) {
		fprintf(stderr, "open error: %s\n", s.ToString().c_str());
		exit(1);
	}
}

static void db_close() {
	delete filter_policy;
	delete cache;
	delete db;
	filter_policy = NULL;
	cache = NULL;
	db = NULL;
}

static void db_write(DBB_local *dl) {
	DBB_global *dg = dl->dl_global;

	if (dg->dg_num != FLAGS_num) {
		char msg[100];
		snprintf(msg, sizeof(msg), "(%ld ops)", dg->dg_num);
		DBB_message(dl, msg);
	}

	DBB_val dv;
	WriteBatch batch;
	Status s;
	int64_t bytes = 0;
	unsigned long i = 0;
	dv.dv_size = FLAGS_value_size;
	do {
		batch.Clear();
		for (int j = 0; j < dg->dg_batchsize; j++) {
			const uint64_t k = (dg->dg_order == DO_FORWARD) ? i+j : (DBB_random(dl->dl_rndctx) % FLAGS_num);
			char key[100];
			snprintf(key, sizeof(key), "%016lx", k);
			DBB_randstring(dl, &dv);
			batch.Put(key, Slice((const char *)dv.dv_data, dv.dv_size));
			bytes += FLAGS_value_size + FLAGS_key_size;
			DBB_opdone(dl);
		}
		s = db->Write(write_options, &batch);
		if (!s.ok()) {
			fprintf(stderr, "put error: %s\n", s.ToString().c_str());
			exit(1);
		}
		i += dg->dg_batchsize;
	} while (!DBB_done(dl));
	dl->dl_bytes += bytes;
}

static void db_read(DBB_local *dl) {
	DBB_global *dg = dl->dl_global;

	int64_t bytes = 0;
	if (dl->dl_order == DO_RANDOM) {
		ReadOptions options;
		std::string value;
		size_t read = 0;
		size_t found = 0;
		char key[100];
		do {
			const uint64_t k = DBB_random(dl->dl_rndctx) % FLAGS_num;
			snprintf(key, sizeof(key), "%016lx", k);
			read++;
			if (db->Get(options, key, &value).ok()) {
				bytes += FLAGS_key_size + value.size();
				found++;
			}
			DBB_opdone(dl);
		} while (!DBB_done(dl));
		char msg[100];
		snprintf(msg, sizeof(msg), "(%zd of %zd found)", found, read);
		DBB_message(dl, msg);
	} else {
		int i = 0;
		Iterator* iter = db->NewIterator(ReadOptions());
		void (Iterator::*seek)();
		void (Iterator::*next)();
		if (dl->dl_order == DO_FORWARD) {
			seek = &Iterator::SeekToFirst;
			next = &Iterator::Next;
		} else {
			seek = &Iterator::SeekToLast;
			next = &Iterator::Prev;
		}
		for ((iter->*seek)(); i < dg->dg_reads && iter->Valid(); (iter->*next)()) {
			bytes += iter->key().size() + iter->value().size();
			DBB_opdone(dl);
			++i;
		}
		delete iter;
	}
	dl->dl_bytes += bytes;
}

static char *db_verstr() {
	static char vstr[32];
	snprintf(vstr, sizeof(vstr), "%d.%d", kMajorVersion, kMinorVersion);
	return vstr;
}

static arg_desc db_opts[] = {
	{ "write_buffer_size", arg_int, &FLAGS_write_buffer_size },
	{ "cache_size", arg_long, &FLAGS_cache_size },
	{ "open_files", arg_int, &FLAGS_open_files },
	{ "bloom_bits", arg_int, &FLAGS_bloom_bits },
	{ NULL }
};

static DBB_backend db_pebbles = {
	"pebbles",
	"PebblesDB",
	db_opts,
	db_verstr,
	db_open,
	db_close,
	db_read,
	db_write
};

extern DBB_backend *dbb_backend;
DBB_backend *dbb_backend = &db_pebbles;