Merge branch 'master' into gfio

[fio.git] / profiles / tiobench.c

diff --git a/profiles/tiobench.c b/profiles/tiobench.c
index d239468f9f3272269089e08590c8ac69d52c4ddc..0279e1cafbec95d286f802889ec8045729a2c3cc 100644 (file)
--- a/profiles/tiobench.c
+++ b/profiles/tiobench.c
@@ -1,9 +1,18 @@
 #include "../fio.h"
 #include "../profile.h"
+#include "../parse.h"
+
+static unsigned long long size;
+static unsigned int loops = 1;
+static unsigned int bs = 4096;
+static unsigned int nthreads = 1;
+static char *dir;
+
+char sz_idx[80], bs_idx[80], loop_idx[80], dir_idx[80], t_idx[80];
 
 static const char *tb_opts[] = {
-       "buffered=0", "size=4*1024*$mb_memory", "bs=4k", "timeout=600",
-       "numjobs=4", "group_reporting", "thread", "overwrite=1",
+       "buffered=0", sz_idx, bs_idx, loop_idx, dir_idx, t_idx,
+       "timeout=600", "group_reporting", "thread", "overwrite=1",
        "filename=.fio.tio.1:.fio.tio.2:.fio.tio.3:.fio.tio.4",
        "name=seqwrite", "rw=write", "end_fsync=1",
        "name=randwrite", "stonewall", "rw=randwrite", "end_fsync=1",
@@ -11,15 +20,96 @@ static const char *tb_opts[] = {
        "name=randread", "stonewall", "rw=randread", NULL,
 };
 
+static struct fio_option options[] = {
+       {
+               .name   = "size",
+               .lname  = "Tiobench size",
+               .type   = FIO_OPT_STR_VAL,
+               .roff1  = &size,
+               .help   = "Size in MB",
+               .category = FIO_OPT_C_PROFILE,
+               .group  = FIO_OPT_G_TIOBENCH,
+       },
+       {
+               .name   = "block",
+               .lname  = "Tiobench block",
+               .type   = FIO_OPT_INT,
+               .roff1  = &bs,
+               .help   = "Block size in bytes",
+               .def    = "4k",
+               .category = FIO_OPT_C_PROFILE,
+               .group  = FIO_OPT_G_TIOBENCH,
+       },
+       {
+               .name   = "numruns",
+               .lname  = "Tiobench numruns",
+               .type   = FIO_OPT_INT,
+               .roff1  = &loops,
+               .help   = "Number of runs",
+               .category = FIO_OPT_C_PROFILE,
+               .group  = FIO_OPT_G_TIOBENCH,
+       },
+       {
+               .name   = "dir",
+               .lname  = "Tiobench directory",
+               .type   = FIO_OPT_STR_STORE,
+               .roff1  = &dir,
+               .help   = "Test directory",
+               .category = FIO_OPT_C_PROFILE,
+               .group  = FIO_OPT_G_TIOBENCH,
+       },
+       {
+               .name   = "threads",
+               .lname  = "Tiobench threads",
+               .type   = FIO_OPT_INT,
+               .roff1  = &nthreads,
+               .help   = "Number of Threads",
+               .category = FIO_OPT_C_PROFILE,
+               .group  = FIO_OPT_G_TIOBENCH,
+       },
+       {
+               .name   = NULL,
+       },
+};
+
+/*
+ * Fill our private options into the command line
+ */
+static void tb_prep_cmdline(void)
+{
+
+       /*
+        * tiobench uses size as MB, so multiply up
+        */
+       size *= 1024 * 1024ULL;
+       if (size)
+               sprintf(sz_idx, "size=%llu", size);
+       else
+               strcpy(sz_idx, "size=4*1024*$mb_memory");
+
+       sprintf(bs_idx, "bs=%u", bs);
+       sprintf(loop_idx, "loops=%u", loops);
+
+       if (dir)
+               sprintf(dir_idx, "directory=%s", dir);
+       else
+               sprintf(dir_idx, "directory=./");
+
+       sprintf(t_idx, "numjobs=%u", nthreads);
+}
+
 static struct profile_ops tiobench_profile = {
        .name           = "tiobench",
-       .version        = FIO_PROFILE_VERSION,
-       .def_ops        = tb_opts,
+       .desc           = "tiotest/tiobench benchmark",
+       .options        = options,
+       .prep_cmd       = tb_prep_cmdline,
+       .cmdline        = tb_opts,
 };
 
 static void fio_init tiobench_register(void)
 {
-       register_profile(&tiobench_profile);
+       if (register_profile(&tiobench_profile))
+               log_err("fio: failed to register profile 'tiobench'\n");
 }
 
 static void fio_exit tiobench_unregister(void)