445 lines
13 KiB
C++
445 lines
13 KiB
C++
// Copyright (c) Vitaliy Filippov, 2019+
|
|
// License: VNPL-1.1 (see README.md for details)
|
|
|
|
// FIO engine to test Blockstore
|
|
//
|
|
// Initialize storage for tests:
|
|
//
|
|
// dd if=/dev/zero of=test_data.bin bs=1M count=1024
|
|
//
|
|
// Random write:
|
|
//
|
|
// [LD_PRELOAD=libasan.so.8] \
|
|
// fio -name=test -thread -ioengine=../build/src/blockstore/libfio_vitastor_blk.so \
|
|
// -bs=4k -direct=1 -rw=randwrite -iodepth=16 -size=900M -loops=10 \
|
|
// -bs_config='{"data_device":"./test_data.bin","meta_offset":0,"journal_offset":16777216,"data_offset":33554432,"disable_data_fsync":true,"meta_format":3,"immediate_commit":"all","log_level":100,"journal_no_same_sector_overwrites":true,"journal_sector_buffer_count":1024}'
|
|
//
|
|
// Linear write:
|
|
//
|
|
// fio -thread -ioengine=./libfio_blockstore.so -name=test -bs=128k -direct=1 -fsync=32 -iodepth=32 -rw=write \
|
|
// -bs_config='{"data_device":"./test_data.bin"}' -size=1000M
|
|
//
|
|
// Random read (run with -iodepth=32 or -iodepth=1):
|
|
//
|
|
// fio -thread -ioengine=./libfio_blockstore.so -name=test -bs=4k -direct=1 -iodepth=32 -rw=randread \
|
|
// -bs_config='{"data_device":"./test_data.bin"}' -size=1000M
|
|
|
|
#include "blockstore.h"
|
|
#include "epoll_manager.h"
|
|
#include "malloc_or_die.h"
|
|
#include "json11/json11.hpp"
|
|
#include "../util/robin_hood.h"
|
|
#include "fio_headers.h"
|
|
|
|
struct bs_data
|
|
{
|
|
blockstore_i *bs;
|
|
epoll_manager_t *epmgr;
|
|
ring_loop_t *ringloop;
|
|
/* The list of completed io_u structs. */
|
|
std::vector<io_u*> completed;
|
|
robin_hood::unordered_flat_map<uint64_t, int> inflight_oids;
|
|
std::vector<io_u*> postponed;
|
|
int op_n = 0, inflight = 0;
|
|
bool ec = false;
|
|
bool imm = true;
|
|
bool last_sync = false;
|
|
bool trace = false;
|
|
uint8_t *bitmap = NULL;
|
|
uint32_t block_size = 0;
|
|
};
|
|
|
|
struct bs_options
|
|
{
|
|
int __pad;
|
|
char *json_config = NULL;
|
|
int ec = 0;
|
|
int trace = 0;
|
|
};
|
|
|
|
static struct fio_option options[] = {
|
|
{
|
|
.name = "bs_config",
|
|
.lname = "JSON config for Blockstore",
|
|
.type = FIO_OPT_STR_STORE,
|
|
.off1 = offsetof(struct bs_options, json_config),
|
|
.help = "JSON config for Blockstore",
|
|
.category = FIO_OPT_C_ENGINE,
|
|
.group = FIO_OPT_G_FILENAME,
|
|
},
|
|
{
|
|
.name = "ec",
|
|
.lname = "Use EC write method",
|
|
.type = FIO_OPT_BOOL,
|
|
.off1 = offsetof(struct bs_options, ec),
|
|
.help = "Use EC write method",
|
|
.def = "0",
|
|
.category = FIO_OPT_C_ENGINE,
|
|
.group = FIO_OPT_G_FILENAME,
|
|
},
|
|
{
|
|
.name = "bs_trace",
|
|
.lname = "trace",
|
|
.type = FIO_OPT_BOOL,
|
|
.off1 = offsetof(struct bs_options, trace),
|
|
.help = "Trace operations",
|
|
.def = "0",
|
|
.category = FIO_OPT_C_ENGINE,
|
|
.group = FIO_OPT_G_FILENAME,
|
|
},
|
|
{
|
|
.name = NULL,
|
|
},
|
|
};
|
|
|
|
static int bs_setup(struct thread_data *td)
|
|
{
|
|
bs_options *o = (bs_options*)td->eo;
|
|
bs_data *bsd;
|
|
//fio_file *f;
|
|
//int r;
|
|
//int64_t size;
|
|
|
|
bsd = new bs_data;
|
|
if (!bsd)
|
|
{
|
|
td_verror(td, errno, "calloc");
|
|
return 1;
|
|
}
|
|
td->io_ops_data = bsd;
|
|
bsd->ec = o->ec;
|
|
|
|
if (!td->files_index)
|
|
{
|
|
add_file(td, "blockstore", 0, 0);
|
|
td->o.nr_files = td->o.nr_files ? : 1;
|
|
td->o.open_files++;
|
|
}
|
|
|
|
bsd->trace = o->trace ? true : false;
|
|
|
|
//f = td->files[0];
|
|
//f->real_file_size = size;
|
|
return 0;
|
|
}
|
|
|
|
static void bs_cleanup(struct thread_data *td)
|
|
{
|
|
bs_data *bsd = (bs_data*)td->io_ops_data;
|
|
if (bsd)
|
|
{
|
|
while (1)
|
|
{
|
|
do
|
|
{
|
|
bsd->ringloop->loop();
|
|
if (bsd->bs->is_safe_to_stop())
|
|
goto safe;
|
|
} while (bsd->ringloop->has_work());
|
|
bsd->ringloop->wait();
|
|
}
|
|
safe:
|
|
delete bsd->bs;
|
|
delete bsd->epmgr;
|
|
delete bsd->ringloop;
|
|
free(bsd->bitmap);
|
|
delete bsd;
|
|
}
|
|
}
|
|
|
|
/* Connect to the server from each thread. */
|
|
static int bs_init(struct thread_data *td)
|
|
{
|
|
bs_options *o = (bs_options*)td->eo;
|
|
bs_data *bsd = (bs_data*)td->io_ops_data;
|
|
|
|
blockstore_config_t config;
|
|
if (o->json_config)
|
|
{
|
|
std::string json_err;
|
|
auto json_cfg = json11::Json::parse(o->json_config, json_err);
|
|
for (auto p: json_cfg.object_items())
|
|
{
|
|
if (p.second.is_string())
|
|
config[p.first] = p.second.string_value();
|
|
else
|
|
config[p.first] = p.second.dump();
|
|
}
|
|
}
|
|
bsd->bitmap = (uint8_t*)malloc_or_die(MAX_DATA_BLOCK_SIZE/512/8);
|
|
memset(bsd->bitmap, 0, MAX_DATA_BLOCK_SIZE/512/8);
|
|
bsd->ringloop = new ring_loop_t(RINGLOOP_DEFAULT_SIZE);
|
|
bsd->epmgr = new epoll_manager_t(bsd->ringloop);
|
|
bsd->bs = blockstore_i::create(config, bsd->ringloop, bsd->epmgr->tfd);
|
|
bsd->block_size = bsd->bs->get_block_size();
|
|
bsd->imm = config.find("immediate_commit") == config.end() ||
|
|
config["immediate_commit"] == "all";
|
|
while (1)
|
|
{
|
|
bsd->ringloop->loop();
|
|
if (bsd->bs->is_started())
|
|
break;
|
|
bsd->ringloop->wait();
|
|
}
|
|
|
|
log_info("fio: blockstore initialized\n");
|
|
return 0;
|
|
}
|
|
|
|
static enum fio_q_status _bs_queue(struct thread_data *td, struct io_u *io, bool force);
|
|
|
|
static void _bs_retry(struct bs_data *bsd, uint64_t offset)
|
|
{
|
|
// Retry postponed ops
|
|
auto inflight_it = bsd->inflight_oids.find(offset / bsd->block_size);
|
|
assert(inflight_it != bsd->inflight_oids.end());
|
|
inflight_it->second--;
|
|
if (inflight_it->second > 0)
|
|
{
|
|
for (size_t i = 0; i < bsd->postponed.size(); i++)
|
|
{
|
|
auto oio = bsd->postponed[i];
|
|
if (oio->offset/bsd->block_size == offset/bsd->block_size)
|
|
{
|
|
bsd->postponed.erase(bsd->postponed.begin()+i);
|
|
_bs_queue((thread_data*)oio->engine_data, oio, true);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
else
|
|
bsd->inflight_oids.erase(inflight_it);
|
|
}
|
|
|
|
/* Begin read or write request. */
|
|
static enum fio_q_status _bs_queue(struct thread_data *td, struct io_u *io, bool force)
|
|
{
|
|
bs_data *bsd = (bs_data*)td->io_ops_data;
|
|
if (io->ddir == DDIR_SYNC && bsd->last_sync)
|
|
{
|
|
return FIO_Q_COMPLETED;
|
|
}
|
|
|
|
fio_ro_check(td, io);
|
|
|
|
io->engine_data = td;
|
|
|
|
if (io->ddir == DDIR_WRITE || io->ddir == DDIR_READ)
|
|
assert(io->xfer_buflen <= bsd->block_size);
|
|
|
|
uint64_t stripe = io->offset / bsd->block_size;
|
|
if (!force && io->ddir == DDIR_WRITE)
|
|
{
|
|
auto & inflight = bsd->inflight_oids[stripe];
|
|
inflight++;
|
|
if (inflight > 1)
|
|
{
|
|
bsd->postponed.push_back(io);
|
|
return FIO_Q_QUEUED;
|
|
}
|
|
}
|
|
|
|
blockstore_op_t *op = new blockstore_op_t;
|
|
op->callback = NULL;
|
|
|
|
switch (io->ddir)
|
|
{
|
|
case DDIR_READ:
|
|
op->opcode = BS_OP_READ;
|
|
op->buf = (uint8_t*)io->xfer_buf;
|
|
op->oid = {
|
|
.inode = 1,
|
|
.stripe = stripe,
|
|
};
|
|
op->version = UINT64_MAX; // last unstable
|
|
op->offset = io->offset % bsd->block_size;
|
|
op->len = io->xfer_buflen;
|
|
op->bitmap = bsd->bitmap;
|
|
op->callback = [io, n = bsd->op_n](blockstore_op_t *op)
|
|
{
|
|
io->error = op->retval < 0 ? -op->retval : 0;
|
|
bs_data *bsd = ((bs_data*)((thread_data*)io->engine_data)->io_ops_data);
|
|
bsd->inflight--;
|
|
bsd->completed.push_back(io);
|
|
if (bsd->trace)
|
|
printf("--- OP_READ %zx n=%d retval=%d\n", (size_t)op, n, op->retval);
|
|
delete op;
|
|
};
|
|
break;
|
|
case DDIR_WRITE:
|
|
op->opcode = bsd->ec ? BS_OP_WRITE : BS_OP_WRITE_STABLE;
|
|
op->buf = (uint8_t*)io->xfer_buf;
|
|
op->oid = {
|
|
.inode = 1,
|
|
.stripe = stripe,
|
|
};
|
|
op->version = 0; // assign automatically
|
|
op->offset = io->offset % bsd->block_size;
|
|
op->len = io->xfer_buflen;
|
|
op->bitmap = bsd->bitmap;
|
|
if (bsd->ec)
|
|
{
|
|
op->callback = [io, n = bsd->op_n](blockstore_op_t *op)
|
|
{
|
|
bs_data *bsd = ((bs_data*)((thread_data*)io->engine_data)->io_ops_data);
|
|
if (bsd->trace)
|
|
printf("--- OP_WRITE %zx n=%d retval=%d\n", (size_t)op, n, op->retval);
|
|
if (op->retval < 0)
|
|
{
|
|
io->error = op->retval < 0 ? -op->retval : 0;
|
|
bsd->inflight--;
|
|
bsd->completed.push_back(io);
|
|
_bs_retry(bsd, io->offset);
|
|
delete op;
|
|
}
|
|
else
|
|
{
|
|
auto stab_op = new blockstore_op_t;
|
|
stab_op->opcode = BS_OP_STABLE;
|
|
stab_op->buf = (uint8_t*)malloc_or_die(sizeof(obj_ver_id));
|
|
obj_ver_id *ver = (obj_ver_id *)stab_op->buf;
|
|
ver[0].oid = op->oid;
|
|
ver[0].version = op->version;
|
|
stab_op->len = 1;
|
|
stab_op->callback = [io, n](blockstore_op_t *op)
|
|
{
|
|
bs_data *bsd = ((bs_data*)((thread_data*)io->engine_data)->io_ops_data);
|
|
if (bsd->trace)
|
|
printf("--- OP_STABLE %zx n=%d retval=%d\n", (size_t)op, n, op->retval);
|
|
io->error = op->retval < 0 ? -op->retval : 0;
|
|
bsd->inflight--;
|
|
bsd->completed.push_back(io);
|
|
_bs_retry(bsd, io->offset);
|
|
delete op;
|
|
};
|
|
bsd->bs->enqueue_op(stab_op);
|
|
delete op;
|
|
}
|
|
};
|
|
}
|
|
else
|
|
{
|
|
op->callback = [io, n = bsd->op_n](blockstore_op_t *op)
|
|
{
|
|
bs_data *bsd = ((bs_data*)((thread_data*)io->engine_data)->io_ops_data);
|
|
if (bsd->trace)
|
|
printf("--- OP_WRITE_STABLE %zx n=%d retval=%d\n", (size_t)op, n, op->retval);
|
|
io->error = op->retval < 0 ? -op->retval : 0;
|
|
bsd->inflight--;
|
|
bsd->completed.push_back(io);
|
|
_bs_retry(bsd, io->offset);
|
|
delete op;
|
|
};
|
|
}
|
|
bsd->last_sync = false;
|
|
break;
|
|
case DDIR_SYNC:
|
|
op->opcode = BS_OP_SYNC;
|
|
op->callback = [io, n = bsd->op_n](blockstore_op_t *op)
|
|
{
|
|
bs_data *bsd = ((bs_data*)((thread_data*)io->engine_data)->io_ops_data);
|
|
io->error = op->retval < 0 ? -op->retval : 0;
|
|
bsd->completed.push_back(io);
|
|
bsd->inflight--;
|
|
if (bsd->trace)
|
|
printf("--- OP_SYNC %zx n=%d retval=%d\n", (size_t)op, n, op->retval);
|
|
delete op;
|
|
};
|
|
bsd->last_sync = true;
|
|
break;
|
|
default:
|
|
io->error = EINVAL;
|
|
delete op;
|
|
return FIO_Q_COMPLETED;
|
|
}
|
|
|
|
if (bsd->trace)
|
|
printf("+++ %s %zx n=%d\n", op->opcode == BS_OP_READ ? "OP_READ" : (op->opcode == BS_OP_WRITE_STABLE ? "OP_WRITE" : "OP_SYNC"), (size_t)op, bsd->op_n);
|
|
io->error = 0;
|
|
bsd->inflight++;
|
|
bsd->bs->enqueue_op(op);
|
|
bsd->op_n++;
|
|
|
|
if (io->error != 0)
|
|
return FIO_Q_COMPLETED;
|
|
return FIO_Q_QUEUED;
|
|
}
|
|
|
|
static enum fio_q_status bs_queue(struct thread_data *td, struct io_u *io)
|
|
{
|
|
return _bs_queue(td, io, false);
|
|
}
|
|
|
|
static int bs_getevents(struct thread_data *td, unsigned int min, unsigned int max, const struct timespec *t)
|
|
{
|
|
bs_data *bsd = (bs_data*)td->io_ops_data;
|
|
// FIXME timeout
|
|
while (true)
|
|
{
|
|
bsd->ringloop->loop();
|
|
if (bsd->completed.size() >= min)
|
|
break;
|
|
bsd->ringloop->wait();
|
|
}
|
|
return bsd->completed.size();
|
|
}
|
|
|
|
static struct io_u *bs_event(struct thread_data *td, int event)
|
|
{
|
|
bs_data *bsd = (bs_data*)td->io_ops_data;
|
|
if (bsd->completed.size() == 0)
|
|
return NULL;
|
|
/* FIXME We ignore the event number and assume fio calls us exactly once for [0..nr_events-1] */
|
|
struct io_u *ev = bsd->completed.back();
|
|
bsd->completed.pop_back();
|
|
return ev;
|
|
}
|
|
|
|
static int bs_io_u_init(struct thread_data *td, struct io_u *io)
|
|
{
|
|
io->engine_data = NULL;
|
|
return 0;
|
|
}
|
|
|
|
static void bs_io_u_free(struct thread_data *td, struct io_u *io)
|
|
{
|
|
}
|
|
|
|
static int bs_open_file(struct thread_data *td, struct fio_file *f)
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
static int bs_invalidate(struct thread_data *td, struct fio_file *f)
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
struct ioengine_ops __attribute__((visibility("default"))) ioengine = {
|
|
.name = "vitastor_blockstore",
|
|
.version = FIO_IOOPS_VERSION,
|
|
.flags = FIO_MEMALIGN | FIO_DISKLESSIO | FIO_NOEXTEND,
|
|
.setup = bs_setup,
|
|
.init = bs_init,
|
|
.queue = bs_queue,
|
|
.getevents = bs_getevents,
|
|
.event = bs_event,
|
|
.cleanup = bs_cleanup,
|
|
.open_file = bs_open_file,
|
|
.invalidate = bs_invalidate,
|
|
.io_u_init = bs_io_u_init,
|
|
.io_u_free = bs_io_u_free,
|
|
.option_struct_size = sizeof(struct bs_options),
|
|
.options = options,
|
|
};
|
|
|
|
static void fio_init fio_bs_register(void)
|
|
{
|
|
register_ioengine(&ioengine);
|
|
}
|
|
|
|
static void fio_exit fio_bs_unregister(void)
|
|
{
|
|
unregister_ioengine(&ioengine);
|
|
}
|