15 #include "hashtable.h"
17 //#define TEST_STRING_KEYS
19 static int num_threads_;
20 static volatile int start_, stop_, load_;
22 static int get_range_, put_range_;
23 static size_t num_keys_;
24 static double load_time_;
27 #define OP_SELECT_RANGE (1ULL << 20)
29 void *worker (void *arg) {
32 // Wait for all the worker threads to be ready.
33 (void)SYNC_ADD(&load_, -1);
37 int n = num_keys_ / 2 / num_threads_;
38 for (int i = 0; i < n; ++i) {
39 map_key_t key = (nbd_rand() & (num_keys_ - 1)) + 1;
40 map_set(map_, key, key);
43 // Wait for all the worker threads to be done loading.
44 (void)SYNC_ADD(&start_, -1);
50 map_key_t key = (nbd_rand() & (num_keys_ - 1)) + 1;
51 map_key_t x = nbd_rand() & (OP_SELECT_RANGE - 1);
57 #ifdef TEST_STRING_KEYS
58 ASSERT(val == DOES_NOT_EXIST || ns_cmp((nstring_t *)key, (nstring_t *)val) == 0);
60 ASSERT(val == DOES_NOT_EXIST || key == val);
62 } else if (x < put_range_) {
63 map_add(map_, key, key);
65 map_remove(map_, key);
73 uint64_t run_test (void) {
74 load_ = num_threads_ + 1;
75 start_ = num_threads_ + 1;
79 pthread_t thread[MAX_NUM_THREADS];
80 for (int i = 0; i < num_threads_; ++i) {
81 int rc = pthread_create(thread + i, NULL, worker, (void*)(size_t)i);
82 if (rc != 0) { perror("pthread_create"); exit(rc); }
85 do { /* nothing */ } while (load_ != 1);
88 struct timeval tv1, tv2;
89 gettimeofday(&tv1, NULL);
91 do { /* nothing */ } while (start_ != 1);
93 gettimeofday(&tv2, NULL);
94 load_time_ = (double)(1000000*(tv2.tv_sec - tv1.tv_sec) + tv2.tv_usec - tv1.tv_usec) / 1000000;
101 for (int i = 0; i < num_threads_; ++i) {
103 pthread_join(thread[i], &count);
104 ops += (size_t)count;
109 int main (int argc, char **argv) {
110 char* program_name = argv[0];
113 fprintf(stderr, "Usage: %s num_threads\n", program_name);
118 if (num_threads_ > MAX_NUM_THREADS) { num_threads_ = MAX_NUM_THREADS; }
122 num_threads_ = strtol(argv[1], NULL, 10);
124 fprintf(stderr, "%s: Invalid argument for number of threads\n", program_name);
127 if (num_threads_ <= 0) {
128 fprintf(stderr, "%s: Number of threads must be at least 1\n", program_name);
132 if (num_threads_ > MAX_NUM_THREADS) {
133 fprintf(stderr, "%s: Number of threads cannot be more than %d\n", program_name, MAX_NUM_THREADS);
137 int table_scale = 12;
139 table_scale = strtol(argv[2], NULL, 10);
141 fprintf(stderr, "%s: Invalid argument for the scale of the collection\n", program_name);
144 table_scale = strtol(argv[2], NULL, 10);
145 if (table_scale < 0 || table_scale > 36) {
146 fprintf(stderr, "%s: The scale of the collection must be between 0 and 36\n", program_name);
153 get_range_ = (int)((double)OP_SELECT_RANGE / 100 * read_ratio);
154 put_range_ = get_range_ + (int)(((double)OP_SELECT_RANGE - get_range_) / 100 * put_ratio);
157 static const map_impl_t *map_types[] = { &MAP_IMPL_HT };
158 for (int i = 0; i < sizeof(map_types)/sizeof(*map_types); ++i) {
159 #ifdef TEST_STRING_KEYS
160 map_ = map_alloc(map_types[i], &DATATYPE_NSTRING);
162 map_ = map_alloc(map_types[i], NULL);
165 num_keys_ = 1ULL << table_scale;
167 duration_ = 1 + table_scale/4;
168 double mops_per_sec = (double)run_test() / 1000000.0 / duration_;
170 printf("Threads:%-2d Size:2^%-2d load time:%-4.2f Mops/s:%-4.2f per-thread:%-4.2f ",
171 num_threads_, table_scale, load_time_, mops_per_sec, mops_per_sec/num_threads_);
172 map_print(map_, FALSE);