hcq/sysbench
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Added pareto distribution

Browse Source
This commit is contained in:
Vadim Tkachenko
2012-09-21 13:47:59 -07:00
parent adc885ac12
commit 27557f4c7e
2 changed files with 26 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
sysbench/sysbench.c
View File

@ -81,7 +81,8 @@ typedef enum
{
DIST_TYPE_UNIFORM,
DIST_TYPE_GAUSSIAN,
DIST_TYPE_SPECIAL
DIST_TYPE_SPECIAL,
DIST_TYPE_PARETO
} rand_dist_t;
/* Event queue data type for the tx-rate mode */
@ -99,6 +100,10 @@ static unsigned int rand_pct;
static unsigned int rand_res;
static int rand_seed; /* optional seed set on the command line */
/* parameters for Pareto distribution */
static double pareto_h; /* parameter h */
static double pareto_power; /* parameter pre-calculated by h */
/* Random seed used to generate unique random numbers */
static unsigned long long rnd_seed;
/* Mutex to protect random seed */
@ -131,7 +136,7 @@ sb_arg_t general_args[] =
{"help", "print help and exit", SB_ARG_TYPE_FLAG, NULL},
{"version", "print version and exit", SB_ARG_TYPE_FLAG, "off"},
{"rand-init", "initialize random number generator", SB_ARG_TYPE_FLAG, "off"},
{"rand-type", "random numbers distribution {uniform,gaussian,special}", SB_ARG_TYPE_STRING,
{"rand-type", "random numbers distribution {uniform,gaussian,special,pareto}", SB_ARG_TYPE_STRING,
"special"},
{"rand-spec-iter", "number of iterations used for numbers generation", SB_ARG_TYPE_INT, "12"},
{"rand-spec-pct", "percentage of values to be treated as 'special' (for special distribution)",
@ -139,6 +144,7 @@ sb_arg_t general_args[] =
{"rand-spec-res", "percentage of 'special' values to use (for special distribution)",
SB_ARG_TYPE_INT, "75"},
{"rand-seed", "seed for random number generator, ignored when 0", SB_ARG_TYPE_INT, "0"},
{"rand-pareto-h", "parameter h for pareto distibution", SB_ARG_TYPE_FLOAT, "0.2"},
{NULL, NULL, SB_ARG_TYPE_NULL, NULL}
};
@ -1020,6 +1026,11 @@ static int init(void)
rand_type = DIST_TYPE_SPECIAL;
rand_func = &sb_rand_special;
}
else if (!strcmp(s, "pareto"))
{
rand_type = DIST_TYPE_PARETO;
rand_func = &sb_rand_pareto;
}
else
{
log_text(LOG_FATAL, "Invalid random numbers distribution: %s.", s);
@ -1030,6 +1041,9 @@ static int init(void)
rand_pct = sb_get_value_int("rand-spec-pct");
rand_res = sb_get_value_int("rand-spec-res");
pareto_h = sb_get_value_float("rand-pareto-h");
pareto_power = log(pareto_h)/log(1.0-pareto_h);
sb_globals.tx_rate = sb_get_value_int("tx-rate");
sb_globals.report_interval = sb_get_value_int("report-interval");
@ -1271,6 +1285,15 @@ int sb_rand_special(int a, int b)
return a + res;
}
/* Pareto distribution */
int sb_rand_pareto(int a, int b)
{
double randf;
randf = (double) sb_rnd() / (double) SB_MAX_RND;
return a + (int)(b - a + 1) * pow(randf, pareto_power);
}
/* Generate unique random id */

1
sysbench/sysbench.h
View File

@ -227,6 +227,7 @@ int sb_rand(int, int);
int sb_rand_uniform(int, int);
int sb_rand_gaussian(int, int);
int sb_rand_special(int, int);
int sb_rand_pareto(int, int);
int sb_rand_uniq(int a, int b);
void sb_rand_str(const char *, char *);