Commit acb3af0d authored by John Esmet's avatar John Esmet Committed by Yoni Fogel

add check overwrites


git-svn-id: file:///svn/toku/tokudb@47686 c7de825b-a66e-492c-adef-691d508d4ae1
parent ff222fa0
...@@ -115,6 +115,7 @@ struct cli_args { ...@@ -115,6 +115,7 @@ struct cli_args {
bool warm_cache; // warm caches before running stress_table bool warm_cache; // warm caches before running stress_table
bool blackhole; // all message injects are no-ops. helps measure txn/logging/locktree overhead. bool blackhole; // all message injects are no-ops. helps measure txn/logging/locktree overhead.
bool prelocked_write; // use prelocked_write flag for insertions, avoiding the locktree bool prelocked_write; // use prelocked_write flag for insertions, avoiding the locktree
bool unique_checks; // use uniqueness checking during insert. makes it slow.
}; };
struct arg { struct arg {
...@@ -683,6 +684,13 @@ size_t_max(size_t a, size_t b) { ...@@ -683,6 +684,13 @@ size_t_max(size_t a, size_t b) {
return (a > b) ? a : b; return (a > b) ? a : b;
} }
static int get_put_flags(struct cli_args *args) {
int flags = 0;
flags |= args->prelocked_write ? DB_PRELOCKED_WRITE : 0;
flags |= args->unique_checks ? DB_NOOVERWRITE : 0;
return flags;
}
static int random_put_in_db(DB *db, DB_TXN *txn, ARG arg, void *stats_extra) { static int random_put_in_db(DB *db, DB_TXN *txn, ARG arg, void *stats_extra) {
int r = 0; int r = 0;
uint8_t rand_key_b[size_t_max(arg->cli->key_size, sizeof(uint64_t))]; uint8_t rand_key_b[size_t_max(arg->cli->key_size, sizeof(uint64_t))];
...@@ -704,7 +712,7 @@ static int random_put_in_db(DB *db, DB_TXN *txn, ARG arg, void *stats_extra) { ...@@ -704,7 +712,7 @@ static int random_put_in_db(DB *db, DB_TXN *txn, ARG arg, void *stats_extra) {
DBT key, val; DBT key, val;
dbt_init(&key, &rand_key_b, sizeof rand_key_b); dbt_init(&key, &rand_key_b, sizeof rand_key_b);
dbt_init(&val, valbuf, sizeof valbuf); dbt_init(&val, valbuf, sizeof valbuf);
int flags = arg->cli->prelocked_write ? DB_PRELOCKED_WRITE : 0; int flags = get_put_flags(arg->cli);
r = db->put(db, txn, &key, &val, flags); r = db->put(db, txn, &key, &val, flags);
if (r != 0) { if (r != 0) {
goto cleanup; goto cleanup;
...@@ -809,16 +817,18 @@ static int UU() serial_put_op(DB_TXN *txn, ARG arg, void *operation_extra, void ...@@ -809,16 +817,18 @@ static int UU() serial_put_op(DB_TXN *txn, ARG arg, void *operation_extra, void
uint64_t puts_to_increment = 0; uint64_t puts_to_increment = 0;
for (uint32_t i = 0; i < arg->cli->txn_size; ++i) { for (uint32_t i = 0; i < arg->cli->txn_size; ++i) {
rand_key_key[0] = extra->current++; rand_key_key[0] = extra->current++;
#if 0
if (arg->cli->interleave) { if (arg->cli->interleave) {
rand_key_i[3] = arg->thread_idx; rand_key_i[3] = arg->thread_idx;
} else { } else {
rand_key_i[0] = arg->thread_idx; rand_key_i[0] = arg->thread_idx;
} }
#endif
fill_zeroed_array(valbuf, arg->cli->val_size, arg->random_data, arg->cli->compressibility); fill_zeroed_array(valbuf, arg->cli->val_size, arg->random_data, arg->cli->compressibility);
DBT key, val; DBT key, val;
dbt_init(&key, &rand_key_b, sizeof rand_key_b); dbt_init(&key, &rand_key_b, sizeof rand_key_b);
dbt_init(&val, valbuf, sizeof valbuf); dbt_init(&val, valbuf, sizeof valbuf);
int flags = arg->cli->prelocked_write ? DB_PRELOCKED_WRITE : 0; int flags = get_put_flags(arg->cli);
r = db->put(db, txn, &key, &val, flags); r = db->put(db, txn, &key, &val, flags);
if (r != 0) { if (r != 0) {
goto cleanup; goto cleanup;
...@@ -1679,7 +1689,8 @@ static struct cli_args UU() get_default_args(void) { ...@@ -1679,7 +1689,8 @@ static struct cli_args UU() get_default_args(void) {
.single_txn = false, .single_txn = false,
.warm_cache = false, .warm_cache = false,
.blackhole = false, .blackhole = false,
.prelocked_write = false .prelocked_write = false,
.unique_checks = false,
}; };
return DEFAULT_ARGS; return DEFAULT_ARGS;
} }
...@@ -2053,6 +2064,7 @@ static inline void parse_stress_test_args (int argc, char *const argv[], struct ...@@ -2053,6 +2064,7 @@ static inline void parse_stress_test_args (int argc, char *const argv[], struct
BOOL_ARG("recover", do_recover), BOOL_ARG("recover", do_recover),
BOOL_ARG("blackhole", blackhole), BOOL_ARG("blackhole", blackhole),
BOOL_ARG("prelocked_write", prelocked_write), BOOL_ARG("prelocked_write", prelocked_write),
BOOL_ARG("unique_checks", unique_checks),
STRING_ARG("--envdir", env_args.envdir), STRING_ARG("--envdir", env_args.envdir),
LOCAL_STRING_ARG("--perf_format", perf_format_s, "human"), LOCAL_STRING_ARG("--perf_format", perf_format_s, "human"),
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment