Add benchmark for isc_{mem,mempool}_{get,put} operations

diff --git a/lib/isc/tests/mem_test.c b/lib/isc/tests/mem_test.c
index 59b710312ea3629d6af07f2445d9499e27549a08..eea3eed7cd2087ff1939dd33b389487eed3ed374 100644 (file)
--- a/lib/isc/tests/mem_test.c
+++ b/lib/isc/tests/mem_test.c
@@ -24,10 +24,14 @@
 
 #include <isc/file.h>
 #include <isc/mem.h>
+#include <isc/mutex.h>
+#include <isc/os.h>
 #include <isc/print.h>
 #include <isc/result.h>
 #include <isc/stdio.h>
 #include <isc/util.h>
+#include <isc/thread.h>
+#include <isc/time.h>
 
 #include "../mem_p.h"
 
@@ -422,6 +426,129 @@ isc_mem_traceflag_test(void **state) {
 }
 #endif
 
+#define ITERS 512
+#define NUM_ITEMS 1024 //768
+#define ITEM_SIZE 65534
+
+static void *
+mem_thread(void *arg) {
+       void *items[NUM_ITEMS];
+       size_t size = *((size_t *)arg);
+
+       for (int i = 0; i < ITERS; i++) {
+               for (int j = 0; j < NUM_ITEMS; j++) {
+                       items[j] = isc_mem_get(mctx, size);
+               }
+               for (int j = 0; j < NUM_ITEMS; j++) {
+                       isc_mem_put(mctx, items[j], size);
+               }
+       }
+
+       return (NULL);
+}
+
+static void
+isc_mem_benchmark(void **state) {
+       int nthreads = ISC_MAX(ISC_MIN(isc_os_ncpus(), 32), 1);
+       isc_thread_t threads[32];
+       isc_time_t ts1, ts2;
+       double t;
+       isc_result_t result;
+       size_t size = ITEM_SIZE;
+
+       UNUSED(state);
+
+       result = isc_time_now(&ts1);
+       assert_int_equal(result, ISC_R_SUCCESS);
+
+       for (int i = 0; i < nthreads; i++) {
+               result = isc_thread_create(mem_thread, &size, &threads[i]);
+               assert_int_equal(result, ISC_R_SUCCESS);
+               size = size / 2;
+       }
+       for (int i = 0; i < nthreads; i++) {
+               result = isc_thread_join(threads[i], NULL);
+               assert_int_equal(result, ISC_R_SUCCESS);
+       }
+
+       result = isc_time_now(&ts2);
+       assert_int_equal(result, ISC_R_SUCCESS);
+
+       t = isc_time_microdiff(&ts2, &ts1);
+
+       printf("[ TIME     ] isc_mem_benchmark: "
+              "%d isc_mem_{get,put} calls, %f seconds, %f calls/second\n",
+              nthreads * ITERS * NUM_ITEMS, t / 1000000.0,
+              (nthreads * ITERS * NUM_ITEMS) / (t / 1000000.0));
+}
+
+static void *
+mempool_thread(void *arg) {
+       isc_mempool_t *mp = (isc_mempool_t *)arg;
+       void *items[NUM_ITEMS];
+
+       for (int i = 0; i < ITERS; i++) {
+               for (int j = 0; j < NUM_ITEMS; j++) {
+                       items[j] = isc_mempool_get(mp);
+               }
+               for (int j = 0; j < NUM_ITEMS; j++) {
+                       isc_mempool_put(mp, items[j]);
+               }
+       }
+
+       return (NULL);
+}
+
+static void
+isc_mempool_benchmark(void **state) {
+       int nthreads = ISC_MAX(ISC_MIN(isc_os_ncpus(), 32), 1);
+       isc_thread_t threads[32];
+       isc_time_t ts1, ts2;
+       double t;
+       isc_result_t result;
+       size_t size = ITEM_SIZE;
+       isc_mempool_t *mp = NULL;
+       isc_mutex_t mplock;
+
+       isc_mutex_init(&mplock);
+
+       result = isc_mempool_create(mctx, ITEM_SIZE, &mp);
+       assert_int_equal(result, ISC_R_SUCCESS);
+
+       isc_mempool_associatelock(mp, &mplock);
+
+       isc_mempool_setfreemax(mp, 32768);
+       isc_mempool_setfillcount(mp, ISC_MAX(NUM_ITEMS / nthreads, 1));
+
+       UNUSED(state);
+
+       result = isc_time_now(&ts1);
+       assert_int_equal(result, ISC_R_SUCCESS);
+
+       for (int i = 0; i < nthreads; i++) {
+               result = isc_thread_create(mempool_thread, mp, &threads[i]);
+               assert_int_equal(result, ISC_R_SUCCESS);
+               size = size / 2;
+       }
+       for (int i = 0; i < nthreads; i++) {
+               result = isc_thread_join(threads[i], NULL);
+               assert_int_equal(result, ISC_R_SUCCESS);
+       }
+
+       result = isc_time_now(&ts2);
+       assert_int_equal(result, ISC_R_SUCCESS);
+
+       t = isc_time_microdiff(&ts2, &ts1);
+
+       printf("[ TIME     ] isc_mempool_benchmark: "
+              "%d isc_mempool_{get,put} calls, %f seconds, %f calls/second\n",
+              nthreads * ITERS * NUM_ITEMS, t / 1000000.0,
+              (nthreads * ITERS * NUM_ITEMS) / (t / 1000000.0));
+
+       isc_mempool_destroy(&mp);
+       isc_mutex_destroy(&mplock);
+}
+
 /*
  * Main
  */
@@ -444,6 +571,10 @@ main(void) {
                cmocka_unit_test_setup_teardown(isc_mem_traceflag_test,
                                _setup, _teardown),
 #endif
+               cmocka_unit_test_setup_teardown(isc_mem_benchmark,
+                                               _setup, _teardown),
+               cmocka_unit_test_setup_teardown(isc_mempool_benchmark,
+                                               _setup, _teardown),
        };
 
        return (cmocka_run_group_tests(tests, NULL, NULL));