X-Git-Url: http://git.ozlabs.org/?p=ccan;a=blobdiff_plain;f=ccan%2Fntdb%2Ftools%2Fspeed.c;fp=ccan%2Fntdb%2Ftools%2Fspeed.c;h=0000000000000000000000000000000000000000;hp=a82932186d4568f536ba023911469fcb90124e1f;hb=1aab74723e837a0fd8091e264a325bb2cdcdd0fa;hpb=979071e8587d4819a7f78613b68d29e222e5db63;ds=sidebyside diff --git a/ccan/ntdb/tools/speed.c b/ccan/ntdb/tools/speed.c deleted file mode 100644 index a8293218..00000000 --- a/ccan/ntdb/tools/speed.c +++ /dev/null @@ -1,439 +0,0 @@ -/* Simple speed test for NTDB */ -#include -#include -#include -#include -#include -#include -#include -#include -#include -#include "ntdb.h" - -/* Nanoseconds per operation */ -static size_t normalize(const struct timeval *start, - const struct timeval *stop, - unsigned int num) -{ - struct timeval diff; - - timersub(stop, start, &diff); - - /* Floating point is more accurate here. */ - return (double)(diff.tv_sec * 1000000 + diff.tv_usec) - / num * 1000; -} - -static size_t file_size(void) -{ - struct stat st; - - if (stat("/tmp/speed.ntdb", &st) != 0) - return -1; - return st.st_size; -} - -static int count_record(struct ntdb_context *ntdb, - NTDB_DATA key, NTDB_DATA data, void *p) -{ - int *total = p; - *total += *(int *)data.dptr; - return 0; -} - -static void dump_and_clear_stats(struct ntdb_context **ntdb, - int flags, - union ntdb_attribute *attr) -{ - union ntdb_attribute stats; - enum NTDB_ERROR ecode; - - stats.base.attr = NTDB_ATTRIBUTE_STATS; - stats.stats.size = sizeof(stats.stats); - ecode = ntdb_get_attribute(*ntdb, &stats); - if (ecode != NTDB_SUCCESS) - errx(1, "Getting stats: %s", ntdb_errorstr(ecode)); - - printf("allocs = %llu\n", - (unsigned long long)stats.stats.allocs); - printf(" alloc_subhash = %llu\n", - (unsigned long long)stats.stats.alloc_subhash); - printf(" alloc_chain = %llu\n", - (unsigned long long)stats.stats.alloc_chain); - printf(" alloc_bucket_exact = %llu\n", - (unsigned long long)stats.stats.alloc_bucket_exact); - printf(" alloc_bucket_max = %llu\n", - (unsigned long long)stats.stats.alloc_bucket_max); - printf(" alloc_leftover = %llu\n", - (unsigned long long)stats.stats.alloc_leftover); - printf(" alloc_coalesce_tried = %llu\n", - (unsigned long long)stats.stats.alloc_coalesce_tried); - printf(" alloc_coalesce_iterate_clash = %llu\n", - (unsigned long long)stats.stats.alloc_coalesce_iterate_clash); - printf(" alloc_coalesce_lockfail = %llu\n", - (unsigned long long)stats.stats.alloc_coalesce_lockfail); - printf(" alloc_coalesce_race = %llu\n", - (unsigned long long)stats.stats.alloc_coalesce_race); - printf(" alloc_coalesce_succeeded = %llu\n", - (unsigned long long)stats.stats.alloc_coalesce_succeeded); - printf(" alloc_coalesce_num_merged = %llu\n", - (unsigned long long)stats.stats.alloc_coalesce_num_merged); - printf("compares = %llu\n", - (unsigned long long)stats.stats.compares); - printf(" compare_wrong_offsetbits = %llu\n", - (unsigned long long)stats.stats.compare_wrong_offsetbits); - printf(" compare_wrong_keylen = %llu\n", - (unsigned long long)stats.stats.compare_wrong_keylen); - printf(" compare_wrong_rechash = %llu\n", - (unsigned long long)stats.stats.compare_wrong_rechash); - printf(" compare_wrong_keycmp = %llu\n", - (unsigned long long)stats.stats.compare_wrong_keycmp); - printf("transactions = %llu\n", - (unsigned long long)stats.stats.transactions); - printf(" transaction_cancel = %llu\n", - (unsigned long long)stats.stats.transaction_cancel); - printf(" transaction_nest = %llu\n", - (unsigned long long)stats.stats.transaction_nest); - printf(" transaction_expand_file = %llu\n", - (unsigned long long)stats.stats.transaction_expand_file); - printf(" transaction_read_direct = %llu\n", - (unsigned long long)stats.stats.transaction_read_direct); - printf(" transaction_read_direct_fail = %llu\n", - (unsigned long long)stats.stats.transaction_read_direct_fail); - printf(" transaction_write_direct = %llu\n", - (unsigned long long)stats.stats.transaction_write_direct); - printf(" transaction_write_direct_fail = %llu\n", - (unsigned long long)stats.stats.transaction_write_direct_fail); - printf("expands = %llu\n", - (unsigned long long)stats.stats.expands); - printf("frees = %llu\n", - (unsigned long long)stats.stats.frees); - printf("locks = %llu\n", - (unsigned long long)stats.stats.locks); - printf(" lock_lowlevel = %llu\n", - (unsigned long long)stats.stats.lock_lowlevel); - printf(" lock_nonblock = %llu\n", - (unsigned long long)stats.stats.lock_nonblock); - printf(" lock_nonblock_fail = %llu\n", - (unsigned long long)stats.stats.lock_nonblock_fail); - - /* Now clear. */ - ntdb_close(*ntdb); - *ntdb = ntdb_open("/tmp/speed.ntdb", flags, O_RDWR, 0, attr); -} - -static void ntdb_log(struct ntdb_context *ntdb, - enum ntdb_log_level level, - enum NTDB_ERROR ecode, - const char *message, - void *data) -{ - fprintf(stderr, "ntdb:%s:%s:%s\n", - ntdb_name(ntdb), ntdb_errorstr(ecode), message); -} - -int main(int argc, char *argv[]) -{ - unsigned int i, j, num = 1000, stage = 0, stopat = -1; - int flags = NTDB_DEFAULT; - bool transaction = false, summary = false; - NTDB_DATA key, data; - struct ntdb_context *ntdb; - struct timeval start, stop; - union ntdb_attribute seed, log; - bool do_stats = false; - enum NTDB_ERROR ecode; - - /* Try to keep benchmarks even. */ - seed.base.attr = NTDB_ATTRIBUTE_SEED; - seed.base.next = NULL; - seed.seed.seed = 0; - - log.base.attr = NTDB_ATTRIBUTE_LOG; - log.base.next = &seed; - log.log.fn = ntdb_log; - - if (argv[1] && strcmp(argv[1], "--internal") == 0) { - flags = NTDB_INTERNAL; - argc--; - argv++; - } - if (argv[1] && strcmp(argv[1], "--transaction") == 0) { - transaction = true; - argc--; - argv++; - } - if (argv[1] && strcmp(argv[1], "--no-sync") == 0) { - flags |= NTDB_NOSYNC; - argc--; - argv++; - } - if (argv[1] && strcmp(argv[1], "--summary") == 0) { - summary = true; - argc--; - argv++; - } - if (argv[1] && strcmp(argv[1], "--stats") == 0) { - do_stats = true; - argc--; - argv++; - } - - ntdb = ntdb_open("/tmp/speed.ntdb", flags, O_RDWR|O_CREAT|O_TRUNC, - 0600, &log); - if (!ntdb) - err(1, "Opening /tmp/speed.ntdb"); - - key.dptr = (void *)&i; - key.dsize = sizeof(i); - data = key; - - if (argv[1]) { - num = atoi(argv[1]); - argv++; - argc--; - } - - if (argv[1]) { - stopat = atoi(argv[1]); - argv++; - argc--; - } - - /* Add 1000 records. */ - printf("Adding %u records: ", num); fflush(stdout); - if (transaction && (ecode = ntdb_transaction_start(ntdb))) - errx(1, "starting transaction: %s", ntdb_errorstr(ecode)); - gettimeofday(&start, NULL); - for (i = 0; i < num; i++) - if ((ecode = ntdb_store(ntdb, key, data, NTDB_INSERT)) != 0) - errx(1, "Inserting key %u in ntdb: %s", - i, ntdb_errorstr(ecode)); - gettimeofday(&stop, NULL); - if (transaction && (ecode = ntdb_transaction_commit(ntdb))) - errx(1, "committing transaction: %s", ntdb_errorstr(ecode)); - printf(" %zu ns (%zu bytes)\n", - normalize(&start, &stop, num), file_size()); - - if (ntdb_check(ntdb, NULL, NULL)) - errx(1, "ntdb_check failed!"); - if (summary) { - char *sumstr = NULL; - ntdb_summary(ntdb, NTDB_SUMMARY_HISTOGRAMS, &sumstr); - printf("%s\n", sumstr); - free(sumstr); - } - if (do_stats) - dump_and_clear_stats(&ntdb, flags, &log); - - if (++stage == stopat) - exit(0); - - /* Finding 1000 records. */ - printf("Finding %u records: ", num); fflush(stdout); - if (transaction && (ecode = ntdb_transaction_start(ntdb))) - errx(1, "starting transaction: %s", ntdb_errorstr(ecode)); - gettimeofday(&start, NULL); - for (i = 0; i < num; i++) { - NTDB_DATA dbuf; - if ((ecode = ntdb_fetch(ntdb, key, &dbuf)) != NTDB_SUCCESS - || *(int *)dbuf.dptr != i) { - errx(1, "Fetching key %u in ntdb gave %u", - i, ecode ? ecode : *(int *)dbuf.dptr); - } - } - gettimeofday(&stop, NULL); - if (transaction && (ecode = ntdb_transaction_commit(ntdb))) - errx(1, "committing transaction: %s", ntdb_errorstr(ecode)); - printf(" %zu ns (%zu bytes)\n", - normalize(&start, &stop, num), file_size()); - if (ntdb_check(ntdb, NULL, NULL)) - errx(1, "ntdb_check failed!"); - if (summary) { - char *sumstr = NULL; - ntdb_summary(ntdb, NTDB_SUMMARY_HISTOGRAMS, &sumstr); - printf("%s\n", sumstr); - free(sumstr); - } - if (do_stats) - dump_and_clear_stats(&ntdb, flags, &log); - if (++stage == stopat) - exit(0); - - /* Missing 1000 records. */ - printf("Missing %u records: ", num); fflush(stdout); - if (transaction && (ecode = ntdb_transaction_start(ntdb))) - errx(1, "starting transaction: %s", ntdb_errorstr(ecode)); - gettimeofday(&start, NULL); - for (i = num; i < num*2; i++) { - NTDB_DATA dbuf; - ecode = ntdb_fetch(ntdb, key, &dbuf); - if (ecode != NTDB_ERR_NOEXIST) - errx(1, "Fetching key %u in ntdb gave %s", - i, ntdb_errorstr(ecode)); - } - gettimeofday(&stop, NULL); - if (transaction && (ecode = ntdb_transaction_commit(ntdb))) - errx(1, "committing transaction: %s", ntdb_errorstr(ecode)); - printf(" %zu ns (%zu bytes)\n", - normalize(&start, &stop, num), file_size()); - if (ntdb_check(ntdb, NULL, NULL)) - errx(1, "ntdb_check failed!"); - if (summary) { - char *sumstr = NULL; - ntdb_summary(ntdb, NTDB_SUMMARY_HISTOGRAMS, &sumstr); - printf("%s\n", sumstr); - free(sumstr); - } - if (do_stats) - dump_and_clear_stats(&ntdb, flags, &log); - if (++stage == stopat) - exit(0); - - /* Traverse 1000 records. */ - printf("Traversing %u records: ", num); fflush(stdout); - if (transaction && (ecode = ntdb_transaction_start(ntdb))) - errx(1, "starting transaction: %s", ntdb_errorstr(ecode)); - i = 0; - gettimeofday(&start, NULL); - if (ntdb_traverse(ntdb, count_record, &i) != num) - errx(1, "Traverse returned wrong number of records"); - if (i != (num - 1) * (num / 2)) - errx(1, "Traverse tallied to %u", i); - gettimeofday(&stop, NULL); - if (transaction && (ecode = ntdb_transaction_commit(ntdb))) - errx(1, "committing transaction: %s", ntdb_errorstr(ecode)); - printf(" %zu ns (%zu bytes)\n", - normalize(&start, &stop, num), file_size()); - if (ntdb_check(ntdb, NULL, NULL)) - errx(1, "ntdb_check failed!"); - if (summary) { - char *sumstr = NULL; - ntdb_summary(ntdb, NTDB_SUMMARY_HISTOGRAMS, &sumstr); - printf("%s\n", sumstr); - free(sumstr); - } - if (do_stats) - dump_and_clear_stats(&ntdb, flags, &log); - if (++stage == stopat) - exit(0); - - /* Delete 1000 records (not in order). */ - printf("Deleting %u records: ", num); fflush(stdout); - if (transaction && (ecode = ntdb_transaction_start(ntdb))) - errx(1, "starting transaction: %s", ntdb_errorstr(ecode)); - gettimeofday(&start, NULL); - for (j = 0; j < num; j++) { - i = (j + 100003) % num; - if ((ecode = ntdb_delete(ntdb, key)) != NTDB_SUCCESS) - errx(1, "Deleting key %u in ntdb: %s", - i, ntdb_errorstr(ecode)); - } - gettimeofday(&stop, NULL); - if (transaction && (ecode = ntdb_transaction_commit(ntdb))) - errx(1, "committing transaction: %s", ntdb_errorstr(ecode)); - printf(" %zu ns (%zu bytes)\n", - normalize(&start, &stop, num), file_size()); - if (ntdb_check(ntdb, NULL, NULL)) - errx(1, "ntdb_check failed!"); - if (summary) { - char *sumstr = NULL; - ntdb_summary(ntdb, NTDB_SUMMARY_HISTOGRAMS, &sumstr); - printf("%s\n", sumstr); - free(sumstr); - } - if (do_stats) - dump_and_clear_stats(&ntdb, flags, &log); - if (++stage == stopat) - exit(0); - - /* Re-add 1000 records (not in order). */ - printf("Re-adding %u records: ", num); fflush(stdout); - if (transaction && (ecode = ntdb_transaction_start(ntdb))) - errx(1, "starting transaction: %s", ntdb_errorstr(ecode)); - gettimeofday(&start, NULL); - for (j = 0; j < num; j++) { - i = (j + 100003) % num; - if ((ecode = ntdb_store(ntdb, key, data, NTDB_INSERT)) != 0) - errx(1, "Inserting key %u in ntdb: %s", - i, ntdb_errorstr(ecode)); - } - gettimeofday(&stop, NULL); - if (transaction && (ecode = ntdb_transaction_commit(ntdb))) - errx(1, "committing transaction: %s", ntdb_errorstr(ecode)); - printf(" %zu ns (%zu bytes)\n", - normalize(&start, &stop, num), file_size()); - if (ntdb_check(ntdb, NULL, NULL)) - errx(1, "ntdb_check failed!"); - if (summary) { - char *sumstr = NULL; - ntdb_summary(ntdb, NTDB_SUMMARY_HISTOGRAMS, &sumstr); - printf("%s\n", sumstr); - free(sumstr); - } - if (do_stats) - dump_and_clear_stats(&ntdb, flags, &log); - if (++stage == stopat) - exit(0); - - /* Append 1000 records. */ - if (transaction && (ecode = ntdb_transaction_start(ntdb))) - errx(1, "starting transaction: %s", ntdb_errorstr(ecode)); - printf("Appending %u records: ", num); fflush(stdout); - gettimeofday(&start, NULL); - for (i = 0; i < num; i++) - if ((ecode = ntdb_append(ntdb, key, data)) != NTDB_SUCCESS) - errx(1, "Appending key %u in ntdb: %s", - i, ntdb_errorstr(ecode)); - gettimeofday(&stop, NULL); - if (transaction && (ecode = ntdb_transaction_commit(ntdb))) - errx(1, "committing transaction: %s", ntdb_errorstr(ecode)); - printf(" %zu ns (%zu bytes)\n", - normalize(&start, &stop, num), file_size()); - if (ntdb_check(ntdb, NULL, NULL)) - errx(1, "ntdb_check failed!"); - if (summary) { - char *sumstr = NULL; - ntdb_summary(ntdb, NTDB_SUMMARY_HISTOGRAMS, &sumstr); - printf("%s\n", sumstr); - free(sumstr); - } - if (++stage == stopat) - exit(0); - - /* Churn 1000 records: not in order! */ - if (transaction && (ecode = ntdb_transaction_start(ntdb))) - errx(1, "starting transaction: %s", ntdb_errorstr(ecode)); - printf("Churning %u records: ", num); fflush(stdout); - gettimeofday(&start, NULL); - for (j = 0; j < num; j++) { - i = (j + 1000019) % num; - if ((ecode = ntdb_delete(ntdb, key)) != NTDB_SUCCESS) - errx(1, "Deleting key %u in ntdb: %s", - i, ntdb_errorstr(ecode)); - i += num; - if ((ecode = ntdb_store(ntdb, key, data, NTDB_INSERT)) != 0) - errx(1, "Inserting key %u in ntdb: %s", - i, ntdb_errorstr(ecode)); - } - gettimeofday(&stop, NULL); - if (transaction && (ecode = ntdb_transaction_commit(ntdb))) - errx(1, "committing transaction: %s", ntdb_errorstr(ecode)); - printf(" %zu ns (%zu bytes)\n", - normalize(&start, &stop, num), file_size()); - - if (ntdb_check(ntdb, NULL, NULL)) - errx(1, "ntdb_check failed!"); - if (summary) { - char *sumstr = NULL; - ntdb_summary(ntdb, NTDB_SUMMARY_HISTOGRAMS, &sumstr); - printf("%s\n", sumstr); - free(sumstr); - } - if (do_stats) - dump_and_clear_stats(&ntdb, flags, &log); - if (++stage == stopat) - exit(0); - - return 0; -}