MB-20231: Add unit test for mempool perf measurement

Change-Id: I9b87bd55256918284cc15b0df543f7988de6efb6
Reviewed-on: http://review.couchbase.org/66323
Reviewed-by: Chiyoung Seo <[email protected]>
Tested-by: buildbot <[email protected]>

Author: hisundar

tests/unit/CMakeLists.txt

@@ -10,6 +10,20 @@ add_executable(hash_test
 target_link_libraries(hash_test ${PTHREAD_LIB} ${LIBM} ${MALLOC_LIBRARIES}
                       ${PLATFORM_LIBRARY} ${LIBRT})
 
+add_executable(mempool_test
+               mempool_test.cc
+               ${ROOT_SRC}/memory_pool.cc
+               ${PROJECT_SOURCE_DIR}/${BREAKPAD_SRC}
+               ${GETTIMEOFDAY_VS}
+               ${ROOT_UTILS}/time_utils.cc)
+
+target_link_libraries(mempool_test ${PTHREAD_LIB} ${LIBM}
+                      ${ASYNC_IO_LIB} ${MALLOC_LIBRARIES}
+                      ${PLATFORM_LIBRARY} ${LIBRT} ${CRYPTO_LIB}
+                      ${DL_LIBRARIES} ${BREAKPAD_LIBRARIES})
+set_target_properties(mempool_test PROPERTIES COMPILE_FLAGS "${CB_GNU_CXX11_OPTION}")
+
+
 add_executable(bcache_test
                bcache_test.cc
                ${ROOT_SRC}/avltree.cc
@@ -183,6 +197,7 @@ target_link_libraries(btree_kv_test ${LIBM} ${MALLOC_LIBRARIES}
 
 # add test target
 add_test(hash_test hash_test)
+add_test(mempool_test mempool_test)
 add_test(bcache_test bcache_test)
 add_test(filemgr_test filemgr_test)
 add_test(btreeblock_test btreeblock_test)

tests/unit/mempool_test.cc

@@ -0,0 +1,115 @@
+/* -*- Mode: C++; tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*- */
+/*
+ *     Copyright 2016 Couchbase, Inc
+ *
+ *   Licensed under the Apache License, Version 2.0 (the "License");
+ *   you may not use this file except in compliance with the License.
+ *   You may obtain a copy of the License at
+ *
+ *       http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *   Unless required by applicable law or agreed to in writing, software
+ *   distributed under the License is distributed on an "AS IS" BASIS,
+ *   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ *   See the License for the specific language governing permissions and
+ *   limitations under the License.
+ */
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+
+#include "test.h"
+#include "memory_pool.h"
+
+struct worker_args{
+    int num_runs;
+    int num_bins;
+    size_t bin_size;
+    MemoryPool *mp;
+};
+
+void *basic_tester(void *args_)
+{
+    TEST_INIT();
+    struct worker_args *args = (struct worker_args *)args_;
+    size_t bin_size = args->bin_size;
+    for (int i = args->num_runs; i; --i) {
+        uint8_t *buf;
+        const int idx = args->mp->fetchBlock(&buf);
+        TEST_CHK(idx != -1);
+        for (int j = 100; j; --j) {
+            buf[rand() % (bin_size - 1)] = 'X';
+        }
+        args->mp->returnBlock(idx);
+    }
+    return NULL;
+}
+
+void basic_test(int iterations, int num_bins, size_t bin_size)
+{
+    TEST_INIT();
+    struct worker_args mpool;
+    struct timeval ts_begin, ts_cur, ts_gap;
+    gettimeofday(&ts_begin, NULL);
+
+    mpool.num_runs = iterations;
+    mpool.num_bins = num_bins;
+    mpool.bin_size = bin_size;
+    mpool.mp = new MemoryPool(mpool.num_bins, mpool.bin_size);
+    void *ret = basic_tester(&mpool);
+    TEST_CHK(!ret);
+    delete mpool.mp;
+
+    gettimeofday(&ts_cur, NULL);
+    ts_gap = _utime_gap(ts_begin, ts_cur);
+    char res[128];
+    sprintf(res, "basic test with %d runs of %d bins x %" _F64 "bytes"
+            " in %ld us",
+            iterations, num_bins, uint64_t(bin_size),
+            ts_gap.tv_sec*1000000 + ts_gap.tv_usec);
+    TEST_RESULT(res);
+}
+
+void multi_thread_test(int num_threads, int iterations, int num_bins,
+                       size_t bin_size)
+{
+    TEST_INIT();
+    thread_t *tid = alca(thread_t, num_threads);
+    struct timeval ts_begin, ts_cur, ts_gap;
+    gettimeofday(&ts_begin, NULL);
+
+    struct worker_args mpool;
+    mpool.num_runs = iterations;
+    mpool.num_bins = num_bins;
+    mpool.bin_size = bin_size;
+    mpool.mp = new MemoryPool(mpool.num_bins, mpool.bin_size);
+
+    for (int i = num_threads - 1; i; --i) {
+        thread_create(&tid[i], basic_tester, &mpool);
+    }
+
+    for (int i = num_threads - 1; i; --i) {
+        void *ret;
+        thread_join(tid[i], &ret);
+        TEST_CHK(!ret);
+    }
+
+    delete mpool.mp;
+
+    gettimeofday(&ts_cur, NULL);
+    ts_gap = _utime_gap(ts_begin, ts_cur);
+    char res[128];
+    sprintf(res, "multi-thread test with %d threads each with "
+                 "%d runs of %d bins x %" _F64 "bytes in %ld usec",
+                 num_threads, iterations, num_bins, uint64_t(bin_size),
+                 ts_gap.tv_sec*1000000 + ts_gap.tv_usec);
+    TEST_RESULT(res);
+}
+
+int main()
+{
+    basic_test(10000, 8, 10485760); //1000 runs of 8 x 10MB buffers
+    multi_thread_test(8, 10000, 8, 10485760); // repeat with 8 threads
+    return 0;
+}