+/* Simple speed tests for jmap. */
+#include <ccan/jmap/jmap_type.h>
+#include <ccan/jmap/jmap.c>
+#include <stdio.h>
+#include <stdlib.h>
+#include <time.h>
+#include <unistd.h>
+#include <sys/time.h>
+
+struct object {
+ /* Some contents. Doubles as consistency check. */
+ struct object *self;
+};
+
+/* 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;
+}
+
+JMAP_DEFINE_UINTIDX_TYPE(struct object, obj);
+
+int main(int argc, char *argv[])
+{
+ struct object *objs;
+ size_t i, j, num;
+ struct timeval start, stop;
+ struct jmap_obj *jmap;
+
+ num = argv[1] ? atoi(argv[1]) : 1000000;
+ objs = calloc(num, sizeof(objs[0]));
+
+ for (i = 0; i < num; i++) {
+ objs[i].self = &objs[i];
+ }
+
+ jmap = jmap_obj_new();
+
+ printf("Initial insert: ");
+ fflush(stdout);
+ gettimeofday(&start, NULL);
+ for (i = 0; i < num; i++)
+ jmap_obj_add(jmap, i, objs[i].self);
+ gettimeofday(&stop, NULL);
+ printf(" %zu ns\n", normalize(&start, &stop, num));
+
+ printf("Initial lookup (match): ");
+ fflush(stdout);
+ gettimeofday(&start, NULL);
+ for (i = 0; i < num; i++)
+ if (jmap_obj_get(jmap, i)->self != objs[i].self)
+ abort();
+ gettimeofday(&stop, NULL);
+ printf(" %zu ns\n", normalize(&start, &stop, num));
+
+ printf("Initial lookup (miss): ");
+ fflush(stdout);
+ gettimeofday(&start, NULL);
+ for (i = 0; i < num; i++)
+ if (jmap_obj_get(jmap, i+num))
+ abort();
+ gettimeofday(&stop, NULL);
+ printf(" %zu ns\n", normalize(&start, &stop, num));
+
+ /* Lookups in order are very cache-friendly for judy; try random */
+ printf("Initial lookup (random): ");
+ fflush(stdout);
+ gettimeofday(&start, NULL);
+ for (i = 0, j = 0; i < num; i++, j = (j + 10007) % num)
+ if (jmap_obj_get(jmap, j)->self != &objs[j])
+ abort();
+ gettimeofday(&stop, NULL);
+ printf(" %zu ns\n", normalize(&start, &stop, num));
+
+ printf("Initial delete all: ");
+ fflush(stdout);
+ gettimeofday(&start, NULL);
+ for (i = 0; i < num; i++)
+ jmap_obj_del(jmap, i);
+ gettimeofday(&stop, NULL);
+ printf(" %zu ns\n", normalize(&start, &stop, num));
+
+ printf("Initial re-inserting: ");
+ fflush(stdout);
+ gettimeofday(&start, NULL);
+ for (i = 0; i < num; i++)
+ jmap_obj_add(jmap, i, objs[i].self);
+ gettimeofday(&stop, NULL);
+ printf(" %zu ns\n", normalize(&start, &stop, num));
+
+ printf("Deleting first half: ");
+ fflush(stdout);
+ gettimeofday(&start, NULL);
+ for (i = 0; i < num; i+=2)
+ jmap_obj_del(jmap, i);
+ gettimeofday(&stop, NULL);
+ printf(" %zu ns\n", normalize(&start, &stop, num));
+
+ printf("Adding (a different) half: ");
+ fflush(stdout);
+
+ gettimeofday(&start, NULL);
+ for (i = 0; i < num; i+=2)
+ jmap_obj_add(jmap, num+i, objs[i].self);
+ gettimeofday(&stop, NULL);
+ printf(" %zu ns\n", normalize(&start, &stop, num));
+
+ printf("Lookup after half-change (match): ");
+ fflush(stdout);
+ gettimeofday(&start, NULL);
+ for (i = 1; i < num; i+=2)
+ if (jmap_obj_get(jmap, i)->self != objs[i].self)
+ abort();
+ for (i = 0; i < num; i+=2)
+ if (jmap_obj_get(jmap, i+num)->self != objs[i].self)
+ abort();
+ gettimeofday(&stop, NULL);
+ printf(" %zu ns\n", normalize(&start, &stop, num));
+
+ printf("Lookup after half-change(miss): ");
+ fflush(stdout);
+ gettimeofday(&start, NULL);
+ for (i = 0; i < num; i++)
+ if (jmap_obj_get(jmap, i+num*2))
+ abort();
+ gettimeofday(&stop, NULL);
+ printf(" %zu ns\n", normalize(&start, &stop, num));
+
+ /* Hashtables with delete markers can fill with markers over time.
+ * so do some changes to see how it operates in long-term. */
+ printf("Details: churning first time\n");
+ for (i = 1; i < num; i+=2) {
+ if (!jmap_obj_del(jmap, i))
+ abort();
+ jmap_obj_add(jmap, i, objs[i].self);
+ }
+ for (i = 0; i < num; i+=2) {
+ if (!jmap_obj_del(jmap, i+num))
+ abort();
+ jmap_obj_add(jmap, i, objs[i].self);
+ }
+ for (i = 1; i < 5; i++) {
+ printf("Churning %s time: ",
+ i == 1 ? "second"
+ : i == 2 ? "third"
+ : i == 3 ? "fourth"
+ : "fifth");
+ fflush(stdout);
+ gettimeofday(&start, NULL);
+ for (j = 0; j < num; j++) {
+ if (!jmap_obj_del(jmap, num*(i-1)+j))
+ abort();
+ jmap_obj_add(jmap, num*i+j, &objs[j]);
+ }
+ gettimeofday(&stop, NULL);
+ printf(" %zu ns\n", normalize(&start, &stop, num));
+ }
+
+ /* Spread out the keys more to try to make it harder. */
+ printf("Details: reinserting with spread\n");
+ for (i = 0; i < num; i++) {
+ if (!jmap_obj_del(jmap, num*4 + i))
+ abort();
+ jmap_obj_add(jmap, num * 5 + i * 9, objs[i].self);
+ }
+
+ if (jmap_obj_popcount(jmap, 0, -1) != num)
+ abort();
+
+ printf("Lookup after churn & spread (match): ");
+ fflush(stdout);
+ gettimeofday(&start, NULL);
+ for (i = 0; i < num; i++)
+ if (jmap_obj_get(jmap, num * 5 + i * 9)->self != objs[i].self) {
+ printf("i =%u\n", i);
+ abort();
+ }
+ gettimeofday(&stop, NULL);
+ printf(" %zu ns\n", normalize(&start, &stop, num));
+
+ printf("Lookup after churn & spread (miss): ");
+ fflush(stdout);
+ gettimeofday(&start, NULL);
+ for (i = 0; i < num; i++)
+ if (jmap_obj_get(jmap, num * 6 + i * 9))
+ abort();
+ gettimeofday(&stop, NULL);
+ printf(" %zu ns\n", normalize(&start, &stop, num));
+
+ printf("Lookup after churn & spread (random): ");
+ fflush(stdout);
+ gettimeofday(&start, NULL);
+ for (i = 0, j = 0; i < num; i++, j = (j + 10007) % num)
+ if (jmap_obj_get(jmap, num * 5 + j * 9)->self != &objs[j])
+ abort();
+ gettimeofday(&stop, NULL);
+ printf(" %zu ns\n", normalize(&start, &stop, num));
+
+ printf("Lookup after churn & spread (half-random): ");
+ fflush(stdout);
+ gettimeofday(&start, NULL);
+ for (i = 0, j = 0; i < num/2; i++, j = (j + 10007) % num) {
+ if (jmap_obj_get(jmap, num * 5 + j * 9)->self != &objs[j])
+ abort();
+ if (jmap_obj_get(jmap, num * 5 + (j + 1) * 9)->self != &objs[j+1])
+ abort();
+ }
+ gettimeofday(&stop, NULL);
+ printf(" %zu ns\n", normalize(&start, &stop, num));
+
+ printf("Deleting half after churn & spread: ");
+ fflush(stdout);
+ gettimeofday(&start, NULL);
+ for (i = 0; i < num; i+=2)
+ jmap_obj_del(jmap, num * 5 + i * 9);
+ gettimeofday(&stop, NULL);
+ printf(" %zu ns\n", normalize(&start, &stop, num));
+
+ printf("Adding (a different) half after churn & spread: ");
+ fflush(stdout);
+
+ gettimeofday(&start, NULL);
+ for (i = 0; i < num; i+=2)
+ jmap_obj_add(jmap, num * 6 + i * 9, objs[i].self);
+ gettimeofday(&stop, NULL);
+ printf(" %zu ns\n", normalize(&start, &stop, num));
+
+ return 0;
+}