summaryrefslogtreecommitdiffstats
path: root/src/rocksdb/env/io_posix_test.cc
diff options
context:
space:
mode:
authorDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-21 11:54:28 +0000
committerDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-21 11:54:28 +0000
commite6918187568dbd01842d8d1d2c808ce16a894239 (patch)
tree64f88b554b444a49f656b6c656111a145cbbaa28 /src/rocksdb/env/io_posix_test.cc
parentInitial commit. (diff)
downloadceph-e6918187568dbd01842d8d1d2c808ce16a894239.tar.xz
ceph-e6918187568dbd01842d8d1d2c808ce16a894239.zip
Adding upstream version 18.2.2.upstream/18.2.2
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'src/rocksdb/env/io_posix_test.cc')
-rw-r--r--src/rocksdb/env/io_posix_test.cc141
1 files changed, 141 insertions, 0 deletions
diff --git a/src/rocksdb/env/io_posix_test.cc b/src/rocksdb/env/io_posix_test.cc
new file mode 100644
index 000000000..81ce50587
--- /dev/null
+++ b/src/rocksdb/env/io_posix_test.cc
@@ -0,0 +1,141 @@
+// Copyright (c) 2020-present, Facebook, Inc. All rights reserved.
+// This source code is licensed under both the GPLv2 (found in the
+// COPYING file in the root directory) and Apache 2.0 License
+// (found in the LICENSE.Apache file in the root directory).
+
+#include "test_util/testharness.h"
+
+#ifdef ROCKSDB_LIB_IO_POSIX
+#include "env/io_posix.h"
+
+namespace ROCKSDB_NAMESPACE {
+
+#ifdef OS_LINUX
+class LogicalBlockSizeCacheTest : public testing::Test {};
+
+// Tests the caching behavior.
+TEST_F(LogicalBlockSizeCacheTest, Cache) {
+ int ncall = 0;
+ auto get_fd_block_size = [&](int fd) {
+ ncall++;
+ return fd;
+ };
+ std::map<std::string, int> dir_fds{
+ {"/", 0},
+ {"/db", 1},
+ {"/db1", 2},
+ {"/db2", 3},
+ };
+ auto get_dir_block_size = [&](const std::string& dir, size_t* size) {
+ ncall++;
+ *size = dir_fds[dir];
+ return Status::OK();
+ };
+ LogicalBlockSizeCache cache(get_fd_block_size, get_dir_block_size);
+ ASSERT_EQ(0, ncall);
+ ASSERT_EQ(0, cache.Size());
+
+ ASSERT_EQ(6, cache.GetLogicalBlockSize("/sst", 6));
+ ASSERT_EQ(1, ncall);
+ ASSERT_EQ(7, cache.GetLogicalBlockSize("/db/sst1", 7));
+ ASSERT_EQ(2, ncall);
+ ASSERT_EQ(8, cache.GetLogicalBlockSize("/db/sst2", 8));
+ ASSERT_EQ(3, ncall);
+
+ ASSERT_OK(cache.RefAndCacheLogicalBlockSize({"/", "/db1/", "/db2"}));
+ ASSERT_EQ(3, cache.Size());
+ ASSERT_TRUE(cache.Contains("/"));
+ ASSERT_TRUE(cache.Contains("/db1"));
+ ASSERT_TRUE(cache.Contains("/db2"));
+ ASSERT_EQ(6, ncall);
+ // Block size for / is cached.
+ ASSERT_EQ(0, cache.GetLogicalBlockSize("/sst", 6));
+ ASSERT_EQ(6, ncall);
+ // No cached size for /db.
+ ASSERT_EQ(7, cache.GetLogicalBlockSize("/db/sst1", 7));
+ ASSERT_EQ(7, ncall);
+ ASSERT_EQ(8, cache.GetLogicalBlockSize("/db/sst2", 8));
+ ASSERT_EQ(8, ncall);
+ // Block size for /db1 is cached.
+ ASSERT_EQ(2, cache.GetLogicalBlockSize("/db1/sst1", 4));
+ ASSERT_EQ(8, ncall);
+ ASSERT_EQ(2, cache.GetLogicalBlockSize("/db1/sst2", 5));
+ ASSERT_EQ(8, ncall);
+ // Block size for /db2 is cached.
+ ASSERT_EQ(3, cache.GetLogicalBlockSize("/db2/sst1", 6));
+ ASSERT_EQ(8, ncall);
+ ASSERT_EQ(3, cache.GetLogicalBlockSize("/db2/sst2", 7));
+ ASSERT_EQ(8, ncall);
+
+ ASSERT_OK(cache.RefAndCacheLogicalBlockSize({"/db"}));
+ ASSERT_EQ(4, cache.Size());
+ ASSERT_TRUE(cache.Contains("/"));
+ ASSERT_TRUE(cache.Contains("/db1"));
+ ASSERT_TRUE(cache.Contains("/db2"));
+ ASSERT_TRUE(cache.Contains("/db"));
+
+ ASSERT_EQ(9, ncall);
+ // Block size for /db is cached.
+ ASSERT_EQ(1, cache.GetLogicalBlockSize("/db/sst1", 7));
+ ASSERT_EQ(9, ncall);
+ ASSERT_EQ(1, cache.GetLogicalBlockSize("/db/sst2", 8));
+ ASSERT_EQ(9, ncall);
+}
+
+// Tests the reference counting behavior.
+TEST_F(LogicalBlockSizeCacheTest, Ref) {
+ int ncall = 0;
+ auto get_fd_block_size = [&](int fd) {
+ ncall++;
+ return fd;
+ };
+ std::map<std::string, int> dir_fds{
+ {"/db", 0},
+ };
+ auto get_dir_block_size = [&](const std::string& dir, size_t* size) {
+ ncall++;
+ *size = dir_fds[dir];
+ return Status::OK();
+ };
+ LogicalBlockSizeCache cache(get_fd_block_size, get_dir_block_size);
+
+ ASSERT_EQ(0, ncall);
+
+ ASSERT_EQ(1, cache.GetLogicalBlockSize("/db/sst0", 1));
+ ASSERT_EQ(1, ncall);
+
+ ASSERT_OK(cache.RefAndCacheLogicalBlockSize({"/db"}));
+ ASSERT_EQ(2, ncall);
+ ASSERT_EQ(1, cache.GetRefCount("/db"));
+ // Block size for /db is cached. Ref count = 1.
+ ASSERT_EQ(0, cache.GetLogicalBlockSize("/db/sst1", 1));
+ ASSERT_EQ(2, ncall);
+
+ // Ref count = 2, but won't recompute the cached buffer size.
+ ASSERT_OK(cache.RefAndCacheLogicalBlockSize({"/db"}));
+ ASSERT_EQ(2, cache.GetRefCount("/db"));
+ ASSERT_EQ(2, ncall);
+
+ // Ref count = 1.
+ cache.UnrefAndTryRemoveCachedLogicalBlockSize({"/db"});
+ ASSERT_EQ(1, cache.GetRefCount("/db"));
+ // Block size for /db is still cached.
+ ASSERT_EQ(0, cache.GetLogicalBlockSize("/db/sst2", 1));
+ ASSERT_EQ(2, ncall);
+
+ // Ref count = 0 and cached buffer size for /db is removed.
+ cache.UnrefAndTryRemoveCachedLogicalBlockSize({"/db"});
+ ASSERT_EQ(0, cache.Size());
+ ASSERT_EQ(1, cache.GetLogicalBlockSize("/db/sst0", 1));
+ ASSERT_EQ(3, ncall);
+}
+#endif
+
+} // namespace ROCKSDB_NAMESPACE
+#endif
+
+int main(int argc, char** argv) {
+ ROCKSDB_NAMESPACE::port::InstallStackTraceHandler();
+ ::testing::InitGoogleTest(&argc, argv);
+ return RUN_ALL_TESTS();
+}