int main(int argc, char* argv[])
{
long int count;
srand(time(NULL));
if(argc != 3) {
fprintf(stderr, "Usage: bench <op: write | read> <count>\n");
exit(1);
}
if (strcmp(argv[1], "write") == 0) {
count = atoi(argv[2]);
_print_header(count);
_print_environment();
_write_test(count);
} else if (strcmp(argv[1], "read") == 0) {
count = atoi(argv[2]);
_print_header(count);
_print_environment();
_read_test(count);
} else if (strcmp(argv[1], "readone") == 0) {
_readone_test(argv[2]);
} else {
fprintf(stderr, "Usage: bench <op: write | read> <count>\n");
exit(1);
}
return 1;
}
void _print_header()
{
_print_environment();
double raw_size = (double)((KEY_SIZE + VAL_SIZE) * FLAGS_num) / 1048576.0;
printf("Keys: %d bytes each\n", KEY_SIZE);
printf("Values: %d bytes each\n", VAL_SIZE);
printf("Entries: %" PRIu64 "\n", FLAGS_num);
printf("RawSize: %.1f MB (estimated)\n", raw_size);
_print_warnings();
fprintf(stdout, LINE1);
}
