Squashed 'src/nudb/' content from commit 00adc6a

git-subtree-dir: src/nudb
git-subtree-split: 00adc6a4f16679a376f40c967f77dfa544c179c1

test/callgrind_test.cpp

@@ -0,0 +1,92 @@
+//
+// Copyright (c) 2015-2016 Vinnie Falco (vinnie dot falco at gmail dot com)
+//
+// Distributed under the Boost Software License, Version 1.0. (See accompanying
+// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
+//
+
+#include <nudb/test/test_store.hpp>
+#include <beast/unit_test/suite.hpp>
+#include <cmath>
+#include <cstdlib>
+#include <cstring>
+#include <memory>
+#include <utility>
+
+namespace nudb {
+namespace test {
+
+// This test is designed for callgrind runs to find hotspots
+
+class callgrind_test : public beast::unit_test::suite
+{
+public:
+    // Creates and opens a database, performs a bunch
+    // of inserts, then alternates fetching all the keys
+    // with keys not present.
+    //
+    void
+    testCallgrind(std::size_t N)
+    {
+        using key_type = std::uint64_t;
+        std::size_t const blockSize = 4096;
+        float const loadFactor = 0.5;
+
+        error_code ec;
+        test_store ts{sizeof(key_type), blockSize, loadFactor};
+        ts.create(ec);
+        if(! BEAST_EXPECTS(! ec, ec.message()))
+            return;
+        ts.open(ec);
+        if(! BEAST_EXPECTS(! ec, ec.message()))
+            return;
+        for(std::size_t i = 0; i < N; ++i)
+        {
+            auto const item = ts[i];
+            ts.db.insert(item.key, item.data, item.size, ec);
+            if(! BEAST_EXPECTS(! ec, ec.message()))
+                return;
+        }
+        Buffer b;
+        for(std::size_t i = 0; i < N * 2; ++i)
+        {
+            if(! (i%2))
+            {
+                auto const item = ts[i/2];
+                ts.db.fetch(item.key, b, ec);
+                if(! BEAST_EXPECTS(! ec, ec.message()))
+                    return;
+                if(! BEAST_EXPECT(b.size() == item.size))
+                    return;
+                if(! BEAST_EXPECT(std::memcmp(b.data(),
+                        item.data, item.size) == 0))
+                    return;
+            }
+            else
+            {
+                auto const item = ts[N + i/2];
+                ts.db.fetch(item.key, b, ec);
+                if(! BEAST_EXPECTS(ec ==
+                        error::key_not_found, ec.message()))
+                    return;
+                ec = {};
+            }
+        }
+        ts.close(ec);
+        if(! BEAST_EXPECTS(! ec, ec.message()))
+            return;
+    }
+
+    void run()
+    {
+        // higher numbers, more pain
+        std::size_t constexpr N = 100000;
+
+        testCallgrind(N);
+    }
+};
+
+BEAST_DEFINE_TESTSUITE(callgrind, test, nudb);
+
+} // test
+} // nudb