mirror/mariadb
1
0
Fork 0
mirror of https://github.com/MariaDB/server.git synced 2025-01-22 06:44:16 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

refs #5702 fix generate row callback

Browse source 
git-svn-id: file:///svn/toku/tokudb@50314 c7de825b-a66e-492c-adef-691d508d4ae1
This commit is contained in:
John Esmet 2013-04-17 00:01:18 -04:00 • committed by Yoni Fogel
parent d341ca5098
commit 84d2b6350f
1 changed files with 33 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

46
src/tests/perf_iibench.cc
View file

@ -28,13 +28,17 @@ static int UU() iibench_put_op(DB_TXN *txn, ARG arg, void *operation_extra, void
DB **dbs = arg->dbp;
DB_ENV *env = arg->env;
DBT mult_key_dbt[num_dbs];
DBT mult_put_dbt[num_dbs];
DBT mult_val_dbt[num_dbs];
uint32_t mult_put_flags[num_dbs];
memset(mult_key_dbt, 0, sizeof(mult_key_dbt));
memset(mult_put_dbt, 0, sizeof(mult_put_dbt));
memset(mult_put_flags, 0, sizeof(mult_put_dbt));
// The first DB is a unique autoincrement index.
mult_put_flags[0] = DB_NOOVERWRITE;
memset(mult_val_dbt, 0, sizeof(mult_val_dbt));
// The first index is unique with serial autoincrement keys.
// The rest are have keys generated with this thread's random data.
for (int i = 0; i < num_dbs; i++) {
mult_put_flags[i] = i == 0 ? DB_NOOVERWRITE : 0;
dbs[i]->app_private = i == 0 ? nullptr : arg->random_data;
}
int r = 0;
uint8_t valbuf[arg->cli->val_size];
@ -42,22 +46,22 @@ static int UU() iibench_put_op(DB_TXN *txn, ARG arg, void *operation_extra, void
uint64_t puts_to_increment = 0;
for (uint32_t i = 0; i < arg->cli->txn_size; ++i) {
fill_zeroed_array(valbuf, arg->cli->val_size,
arg->random_data, arg->cli->compressibility);
struct iibench_op_extra *CAST_FROM_VOIDP(info, operation_extra);
uint64_t key = i == 0 ? next_autoincrement(info) : randu64(arg->random_data);
fill_zeroed_array(valbuf, arg->cli->val_size, arg->random_data, arg->cli->compressibility);
DBT key_dbt, val_dbt;
dbt_init(&key_dbt, &key, sizeof key);
dbt_init(&val_dbt, valbuf, sizeof valbuf);
uint64_t pk = next_autoincrement(info);
dbt_init(&mult_key_dbt[0], &pk, sizeof pk);
dbt_init(&mult_val_dbt[0], valbuf, sizeof valbuf);
r = env->put_multiple(
env,
dbs[0], // source db.
txn,
&key_dbt, // source db key
&val_dbt, // source db value
&mult_key_dbt[0], // source db key
&mult_val_dbt[0], // source db value
num_dbs, // total number of dbs
dbs, // array of dbs
mult_key_dbt, // array of keys
mult_put_dbt, // array of values
mult_val_dbt, // array of values
mult_put_flags // array of flags
);
if (r != 0) {
@ -93,10 +97,25 @@ stress_table(DB_ENV* env, DB** UU(dbp), struct cli_args *cli_args) {
run_workers(myargs, num_threads, cli_args->num_seconds, crash_at_end, cli_args);
}
static int iibench_generate_row_for_put(DB *dest_db, DB *src_db, DBT *dest_key, DBT *dest_val, const DBT *UU(src_key), const DBT *src_val) {
invariant(src_db != dest_db);
invariant(dest_db->app_private != nullptr);
struct random_data *CAST_FROM_VOIDP(r_data, dest_db->app_private);
uint64_t key = randu64(r_data);
DBT key_dbt = { .data = &key, .size = sizeof(key) };
*dest_key = key_dbt;
*dest_val = *src_val;
return 0;
}
int
test_main(int argc, char *const argv[]) {
struct cli_args args = get_default_args_for_perf();
args.num_elements = 0; // want to start with empty DBs
// In MySQL, iibench has one primary key and three secondaries and does 1k inserts per txn.
args.num_DBs = 4;
args.txn_size = 1000;
args.key_size = 8;
args.val_size = 8;
parse_stress_test_args(argc, argv, &args);
// when there are multiple threads, its valid for two of them to
// generate the same key and one of them fail with DB_LOCK_NOTGRANTED
@ -105,6 +124,7 @@ test_main(int argc, char *const argv[]) {
}
// TODO: Use a more complex, more expensive comparison function that
// unpacks keys and walks a descriptor to satisfy the comparison.
args.env_args.generate_put_callback = iibench_generate_row_for_put;
stress_test_main_with_cmp(&args, stress_uint64_dbt_cmp);
return 0;
}