2022-02-24 00:06:27 +00:00
|
|
|
// Copyright (c) 2011-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).
|
|
|
|
|
2022-04-11 20:28:33 +00:00
|
|
|
#include "cache/compressed_secondary_cache.h"
|
2022-02-24 00:06:27 +00:00
|
|
|
|
|
|
|
#include <algorithm>
|
|
|
|
#include <cstdint>
|
|
|
|
|
|
|
|
#include "memory/jemalloc_nodump_allocator.h"
|
|
|
|
#include "memory/memory_allocator.h"
|
2022-06-10 19:23:10 +00:00
|
|
|
#include "rocksdb/convenience.h"
|
|
|
|
#include "rocksdb/secondary_cache.h"
|
2022-02-24 00:06:27 +00:00
|
|
|
#include "test_util/testharness.h"
|
|
|
|
#include "test_util/testutil.h"
|
|
|
|
#include "util/compression.h"
|
|
|
|
#include "util/random.h"
|
|
|
|
|
|
|
|
namespace ROCKSDB_NAMESPACE {
|
|
|
|
|
2022-04-11 20:28:33 +00:00
|
|
|
class CompressedSecondaryCacheTest : public testing::Test {
|
2022-02-24 00:06:27 +00:00
|
|
|
public:
|
2022-04-11 20:28:33 +00:00
|
|
|
CompressedSecondaryCacheTest() : fail_create_(false) {}
|
|
|
|
~CompressedSecondaryCacheTest() {}
|
2022-02-24 00:06:27 +00:00
|
|
|
|
|
|
|
protected:
|
|
|
|
class TestItem {
|
|
|
|
public:
|
|
|
|
TestItem(const char* buf, size_t size) : buf_(new char[size]), size_(size) {
|
|
|
|
memcpy(buf_.get(), buf, size);
|
|
|
|
}
|
|
|
|
~TestItem() {}
|
|
|
|
|
|
|
|
char* Buf() { return buf_.get(); }
|
|
|
|
size_t Size() { return size_; }
|
|
|
|
|
|
|
|
private:
|
|
|
|
std::unique_ptr<char[]> buf_;
|
|
|
|
size_t size_;
|
|
|
|
};
|
|
|
|
|
|
|
|
static size_t SizeCallback(void* obj) {
|
|
|
|
return reinterpret_cast<TestItem*>(obj)->Size();
|
|
|
|
}
|
|
|
|
|
|
|
|
static Status SaveToCallback(void* from_obj, size_t from_offset,
|
|
|
|
size_t length, void* out) {
|
|
|
|
TestItem* item = reinterpret_cast<TestItem*>(from_obj);
|
|
|
|
const char* buf = item->Buf();
|
|
|
|
EXPECT_EQ(length, item->Size());
|
|
|
|
EXPECT_EQ(from_offset, 0);
|
|
|
|
memcpy(out, buf, length);
|
|
|
|
return Status::OK();
|
|
|
|
}
|
|
|
|
|
|
|
|
static void DeletionCallback(const Slice& /*key*/, void* obj) {
|
|
|
|
delete reinterpret_cast<TestItem*>(obj);
|
|
|
|
obj = nullptr;
|
|
|
|
}
|
|
|
|
|
|
|
|
static Cache::CacheItemHelper helper_;
|
|
|
|
|
|
|
|
static Status SaveToCallbackFail(void* /*obj*/, size_t /*offset*/,
|
|
|
|
size_t /*size*/, void* /*out*/) {
|
|
|
|
return Status::NotSupported();
|
|
|
|
}
|
|
|
|
|
|
|
|
static Cache::CacheItemHelper helper_fail_;
|
|
|
|
|
|
|
|
Cache::CreateCallback test_item_creator = [&](const void* buf, size_t size,
|
|
|
|
void** out_obj,
|
|
|
|
size_t* charge) -> Status {
|
|
|
|
if (fail_create_) {
|
|
|
|
return Status::NotSupported();
|
|
|
|
}
|
|
|
|
*out_obj = reinterpret_cast<void*>(new TestItem((char*)buf, size));
|
|
|
|
*charge = size;
|
|
|
|
return Status::OK();
|
|
|
|
};
|
|
|
|
|
|
|
|
void SetFailCreate(bool fail) { fail_create_ = fail; }
|
|
|
|
|
2022-06-10 19:23:10 +00:00
|
|
|
void BasicTestHelper(std::shared_ptr<SecondaryCache> sec_cache) {
|
2022-04-11 20:28:33 +00:00
|
|
|
bool is_in_sec_cache{true};
|
2022-02-24 00:06:27 +00:00
|
|
|
// Lookup an non-existent key.
|
|
|
|
std::unique_ptr<SecondaryCacheResultHandle> handle0 =
|
2022-04-11 20:28:33 +00:00
|
|
|
sec_cache->Lookup("k0", test_item_creator, true, is_in_sec_cache);
|
2022-02-24 00:06:27 +00:00
|
|
|
ASSERT_EQ(handle0, nullptr);
|
|
|
|
|
|
|
|
Random rnd(301);
|
|
|
|
// Insert and Lookup the first item.
|
|
|
|
std::string str1;
|
|
|
|
test::CompressibleString(&rnd, 0.25, 1000, &str1);
|
|
|
|
TestItem item1(str1.data(), str1.length());
|
2022-04-11 20:28:33 +00:00
|
|
|
ASSERT_OK(sec_cache->Insert("k1", &item1,
|
|
|
|
&CompressedSecondaryCacheTest::helper_));
|
|
|
|
|
2022-02-24 00:06:27 +00:00
|
|
|
std::unique_ptr<SecondaryCacheResultHandle> handle1 =
|
2022-04-11 20:28:33 +00:00
|
|
|
sec_cache->Lookup("k1", test_item_creator, true, is_in_sec_cache);
|
2022-02-24 00:06:27 +00:00
|
|
|
ASSERT_NE(handle1, nullptr);
|
2022-04-11 20:28:33 +00:00
|
|
|
ASSERT_FALSE(is_in_sec_cache);
|
|
|
|
|
2022-02-24 00:06:27 +00:00
|
|
|
std::unique_ptr<TestItem> val1 =
|
|
|
|
std::unique_ptr<TestItem>(static_cast<TestItem*>(handle1->Value()));
|
|
|
|
ASSERT_NE(val1, nullptr);
|
|
|
|
ASSERT_EQ(memcmp(val1->Buf(), item1.Buf(), item1.Size()), 0);
|
|
|
|
|
2022-04-11 20:28:33 +00:00
|
|
|
// Lookup the first item again.
|
|
|
|
std::unique_ptr<SecondaryCacheResultHandle> handle1_1 =
|
|
|
|
sec_cache->Lookup("k1", test_item_creator, true, is_in_sec_cache);
|
|
|
|
ASSERT_EQ(handle1_1, nullptr);
|
|
|
|
|
2022-02-24 00:06:27 +00:00
|
|
|
// Insert and Lookup the second item.
|
|
|
|
std::string str2;
|
|
|
|
test::CompressibleString(&rnd, 0.5, 1000, &str2);
|
|
|
|
TestItem item2(str2.data(), str2.length());
|
2022-04-11 20:28:33 +00:00
|
|
|
ASSERT_OK(sec_cache->Insert("k2", &item2,
|
|
|
|
&CompressedSecondaryCacheTest::helper_));
|
2022-02-24 00:06:27 +00:00
|
|
|
std::unique_ptr<SecondaryCacheResultHandle> handle2 =
|
2022-04-11 20:28:33 +00:00
|
|
|
sec_cache->Lookup("k2", test_item_creator, true, is_in_sec_cache);
|
2022-02-24 00:06:27 +00:00
|
|
|
ASSERT_NE(handle2, nullptr);
|
|
|
|
std::unique_ptr<TestItem> val2 =
|
|
|
|
std::unique_ptr<TestItem>(static_cast<TestItem*>(handle2->Value()));
|
|
|
|
ASSERT_NE(val2, nullptr);
|
|
|
|
ASSERT_EQ(memcmp(val2->Buf(), item2.Buf(), item2.Size()), 0);
|
|
|
|
|
|
|
|
std::vector<SecondaryCacheResultHandle*> handles = {handle1.get(),
|
|
|
|
handle2.get()};
|
2022-04-11 20:28:33 +00:00
|
|
|
sec_cache->WaitAll(handles);
|
2022-02-24 00:06:27 +00:00
|
|
|
|
2022-04-11 20:28:33 +00:00
|
|
|
sec_cache.reset();
|
2022-02-24 00:06:27 +00:00
|
|
|
}
|
|
|
|
|
2022-06-10 19:23:10 +00:00
|
|
|
void BasicTest(bool sec_cache_is_compressed, bool use_jemalloc) {
|
|
|
|
CompressedSecondaryCacheOptions opts;
|
|
|
|
opts.capacity = 2048;
|
|
|
|
opts.num_shard_bits = 0;
|
|
|
|
opts.metadata_charge_policy = kDontChargeCacheMetadata;
|
|
|
|
|
|
|
|
if (sec_cache_is_compressed) {
|
|
|
|
if (!LZ4_Supported()) {
|
|
|
|
ROCKSDB_GTEST_SKIP("This test requires LZ4 support.");
|
|
|
|
opts.compression_type = CompressionType::kNoCompression;
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
opts.compression_type = CompressionType::kNoCompression;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (use_jemalloc) {
|
|
|
|
JemallocAllocatorOptions jopts;
|
|
|
|
std::shared_ptr<MemoryAllocator> allocator;
|
|
|
|
std::string msg;
|
|
|
|
if (JemallocNodumpAllocator::IsSupported(&msg)) {
|
|
|
|
Status s = NewJemallocNodumpAllocator(jopts, &allocator);
|
|
|
|
if (s.ok()) {
|
|
|
|
opts.memory_allocator = allocator;
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
ROCKSDB_GTEST_BYPASS("JEMALLOC not supported");
|
|
|
|
}
|
|
|
|
}
|
|
|
|
std::shared_ptr<SecondaryCache> sec_cache =
|
|
|
|
NewCompressedSecondaryCache(opts);
|
|
|
|
}
|
|
|
|
|
2022-02-24 00:06:27 +00:00
|
|
|
void FailsTest(bool sec_cache_is_compressed) {
|
2022-04-11 20:28:33 +00:00
|
|
|
CompressedSecondaryCacheOptions secondary_cache_opts;
|
2022-02-24 00:06:27 +00:00
|
|
|
if (sec_cache_is_compressed) {
|
|
|
|
if (!LZ4_Supported()) {
|
|
|
|
ROCKSDB_GTEST_SKIP("This test requires LZ4 support.");
|
|
|
|
secondary_cache_opts.compression_type = CompressionType::kNoCompression;
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
secondary_cache_opts.compression_type = CompressionType::kNoCompression;
|
|
|
|
}
|
|
|
|
|
|
|
|
secondary_cache_opts.capacity = 1100;
|
|
|
|
secondary_cache_opts.num_shard_bits = 0;
|
|
|
|
secondary_cache_opts.metadata_charge_policy = kDontChargeCacheMetadata;
|
2022-04-11 20:28:33 +00:00
|
|
|
std::shared_ptr<SecondaryCache> sec_cache =
|
|
|
|
NewCompressedSecondaryCache(secondary_cache_opts);
|
2022-02-24 00:06:27 +00:00
|
|
|
|
|
|
|
// Insert and Lookup the first item.
|
|
|
|
Random rnd(301);
|
|
|
|
std::string str1(rnd.RandomString(1000));
|
|
|
|
TestItem item1(str1.data(), str1.length());
|
2022-04-11 20:28:33 +00:00
|
|
|
ASSERT_OK(sec_cache->Insert("k1", &item1,
|
|
|
|
&CompressedSecondaryCacheTest::helper_));
|
2022-02-24 00:06:27 +00:00
|
|
|
|
|
|
|
// Insert and Lookup the second item.
|
|
|
|
std::string str2(rnd.RandomString(200));
|
|
|
|
TestItem item2(str2.data(), str2.length());
|
|
|
|
// k1 is evicted.
|
2022-04-11 20:28:33 +00:00
|
|
|
ASSERT_OK(sec_cache->Insert("k2", &item2,
|
|
|
|
&CompressedSecondaryCacheTest::helper_));
|
|
|
|
bool is_in_sec_cache{false};
|
2022-02-24 00:06:27 +00:00
|
|
|
std::unique_ptr<SecondaryCacheResultHandle> handle1_1 =
|
2022-04-11 20:28:33 +00:00
|
|
|
sec_cache->Lookup("k1", test_item_creator, true, is_in_sec_cache);
|
2022-02-24 00:06:27 +00:00
|
|
|
ASSERT_EQ(handle1_1, nullptr);
|
|
|
|
std::unique_ptr<SecondaryCacheResultHandle> handle2 =
|
2022-04-11 20:28:33 +00:00
|
|
|
sec_cache->Lookup("k2", test_item_creator, true, is_in_sec_cache);
|
2022-02-24 00:06:27 +00:00
|
|
|
ASSERT_NE(handle2, nullptr);
|
|
|
|
std::unique_ptr<TestItem> val2 =
|
|
|
|
std::unique_ptr<TestItem>(static_cast<TestItem*>(handle2->Value()));
|
|
|
|
ASSERT_NE(val2, nullptr);
|
|
|
|
ASSERT_EQ(memcmp(val2->Buf(), item2.Buf(), item2.Size()), 0);
|
|
|
|
|
|
|
|
// Create Fails.
|
|
|
|
SetFailCreate(true);
|
|
|
|
std::unique_ptr<SecondaryCacheResultHandle> handle2_1 =
|
2022-04-11 20:28:33 +00:00
|
|
|
sec_cache->Lookup("k2", test_item_creator, true, is_in_sec_cache);
|
2022-02-24 00:06:27 +00:00
|
|
|
ASSERT_EQ(handle2_1, nullptr);
|
|
|
|
|
|
|
|
// Save Fails.
|
|
|
|
std::string str3 = rnd.RandomString(10);
|
|
|
|
TestItem item3(str3.data(), str3.length());
|
2022-04-11 20:28:33 +00:00
|
|
|
ASSERT_NOK(sec_cache->Insert("k3", &item3,
|
|
|
|
&CompressedSecondaryCacheTest::helper_fail_));
|
2022-02-24 00:06:27 +00:00
|
|
|
|
2022-04-11 20:28:33 +00:00
|
|
|
sec_cache.reset();
|
2022-02-24 00:06:27 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
void BasicIntegrationTest(bool sec_cache_is_compressed) {
|
2022-04-11 20:28:33 +00:00
|
|
|
CompressedSecondaryCacheOptions secondary_cache_opts;
|
2022-02-24 00:06:27 +00:00
|
|
|
|
|
|
|
if (sec_cache_is_compressed) {
|
|
|
|
if (!LZ4_Supported()) {
|
|
|
|
ROCKSDB_GTEST_SKIP("This test requires LZ4 support.");
|
|
|
|
secondary_cache_opts.compression_type = CompressionType::kNoCompression;
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
secondary_cache_opts.compression_type = CompressionType::kNoCompression;
|
|
|
|
}
|
|
|
|
|
|
|
|
secondary_cache_opts.capacity = 2300;
|
|
|
|
secondary_cache_opts.num_shard_bits = 0;
|
|
|
|
secondary_cache_opts.metadata_charge_policy = kDontChargeCacheMetadata;
|
|
|
|
std::shared_ptr<SecondaryCache> secondary_cache =
|
2022-04-11 20:28:33 +00:00
|
|
|
NewCompressedSecondaryCache(secondary_cache_opts);
|
2022-02-24 00:06:27 +00:00
|
|
|
LRUCacheOptions lru_cache_opts(1024, 0, false, 0.5, nullptr,
|
|
|
|
kDefaultToAdaptiveMutex,
|
|
|
|
kDontChargeCacheMetadata);
|
|
|
|
lru_cache_opts.secondary_cache = secondary_cache;
|
|
|
|
std::shared_ptr<Cache> cache = NewLRUCache(lru_cache_opts);
|
|
|
|
std::shared_ptr<Statistics> stats = CreateDBStatistics();
|
|
|
|
|
|
|
|
Random rnd(301);
|
|
|
|
|
|
|
|
std::string str1 = rnd.RandomString(1010);
|
|
|
|
std::string str1_clone{str1};
|
|
|
|
TestItem* item1 = new TestItem(str1.data(), str1.length());
|
2022-04-11 20:28:33 +00:00
|
|
|
ASSERT_OK(cache->Insert("k1", item1, &CompressedSecondaryCacheTest::helper_,
|
2022-02-24 00:06:27 +00:00
|
|
|
str1.length()));
|
|
|
|
|
|
|
|
std::string str2 = rnd.RandomString(1020);
|
|
|
|
TestItem* item2 = new TestItem(str2.data(), str2.length());
|
|
|
|
// After Insert, lru cache contains k2 and secondary cache contains k1.
|
2022-04-11 20:28:33 +00:00
|
|
|
ASSERT_OK(cache->Insert("k2", item2, &CompressedSecondaryCacheTest::helper_,
|
2022-02-24 00:06:27 +00:00
|
|
|
str2.length()));
|
|
|
|
|
|
|
|
std::string str3 = rnd.RandomString(1020);
|
|
|
|
TestItem* item3 = new TestItem(str3.data(), str3.length());
|
|
|
|
// After Insert, lru cache contains k3 and secondary cache contains k1 and
|
|
|
|
// k2
|
2022-04-11 20:28:33 +00:00
|
|
|
ASSERT_OK(cache->Insert("k3", item3, &CompressedSecondaryCacheTest::helper_,
|
2022-02-24 00:06:27 +00:00
|
|
|
str3.length()));
|
|
|
|
|
|
|
|
Cache::Handle* handle;
|
2022-04-11 20:28:33 +00:00
|
|
|
handle = cache->Lookup("k3", &CompressedSecondaryCacheTest::helper_,
|
|
|
|
test_item_creator, Cache::Priority::LOW, true,
|
|
|
|
stats.get());
|
2022-02-24 00:06:27 +00:00
|
|
|
ASSERT_NE(handle, nullptr);
|
|
|
|
TestItem* val3 = static_cast<TestItem*>(cache->Value(handle));
|
|
|
|
ASSERT_NE(val3, nullptr);
|
|
|
|
ASSERT_EQ(memcmp(val3->Buf(), item3->Buf(), item3->Size()), 0);
|
|
|
|
cache->Release(handle);
|
|
|
|
|
|
|
|
// Lookup an non-existent key.
|
2022-04-11 20:28:33 +00:00
|
|
|
handle = cache->Lookup("k0", &CompressedSecondaryCacheTest::helper_,
|
|
|
|
test_item_creator, Cache::Priority::LOW, true,
|
|
|
|
stats.get());
|
2022-02-24 00:06:27 +00:00
|
|
|
ASSERT_EQ(handle, nullptr);
|
|
|
|
|
2022-04-11 20:28:33 +00:00
|
|
|
// This Lookup should promote k1 and erase k1 from the secondary cache,
|
|
|
|
// then k3 is demoted. So k2 and k3 are in the secondary cache.
|
|
|
|
handle = cache->Lookup("k1", &CompressedSecondaryCacheTest::helper_,
|
|
|
|
test_item_creator, Cache::Priority::LOW, true,
|
|
|
|
stats.get());
|
|
|
|
|
2022-02-24 00:06:27 +00:00
|
|
|
ASSERT_NE(handle, nullptr);
|
|
|
|
TestItem* val1_1 = static_cast<TestItem*>(cache->Value(handle));
|
|
|
|
ASSERT_NE(val1_1, nullptr);
|
|
|
|
ASSERT_EQ(memcmp(val1_1->Buf(), str1_clone.data(), str1_clone.size()), 0);
|
|
|
|
cache->Release(handle);
|
|
|
|
|
2022-04-11 20:28:33 +00:00
|
|
|
handle = cache->Lookup("k2", &CompressedSecondaryCacheTest::helper_,
|
|
|
|
test_item_creator, Cache::Priority::LOW, true,
|
|
|
|
stats.get());
|
|
|
|
ASSERT_NE(handle, nullptr);
|
|
|
|
cache->Release(handle);
|
2022-02-24 00:06:27 +00:00
|
|
|
|
|
|
|
cache.reset();
|
|
|
|
secondary_cache.reset();
|
|
|
|
}
|
|
|
|
|
|
|
|
void BasicIntegrationFailTest(bool sec_cache_is_compressed) {
|
2022-04-11 20:28:33 +00:00
|
|
|
CompressedSecondaryCacheOptions secondary_cache_opts;
|
2022-02-24 00:06:27 +00:00
|
|
|
|
|
|
|
if (sec_cache_is_compressed) {
|
|
|
|
if (!LZ4_Supported()) {
|
|
|
|
ROCKSDB_GTEST_SKIP("This test requires LZ4 support.");
|
|
|
|
secondary_cache_opts.compression_type = CompressionType::kNoCompression;
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
secondary_cache_opts.compression_type = CompressionType::kNoCompression;
|
|
|
|
}
|
|
|
|
|
|
|
|
secondary_cache_opts.capacity = 2048;
|
|
|
|
secondary_cache_opts.num_shard_bits = 0;
|
|
|
|
secondary_cache_opts.metadata_charge_policy = kDontChargeCacheMetadata;
|
|
|
|
std::shared_ptr<SecondaryCache> secondary_cache =
|
2022-04-11 20:28:33 +00:00
|
|
|
NewCompressedSecondaryCache(secondary_cache_opts);
|
2022-02-24 00:06:27 +00:00
|
|
|
|
|
|
|
LRUCacheOptions opts(1024, 0, false, 0.5, nullptr, kDefaultToAdaptiveMutex,
|
|
|
|
kDontChargeCacheMetadata);
|
|
|
|
opts.secondary_cache = secondary_cache;
|
|
|
|
std::shared_ptr<Cache> cache = NewLRUCache(opts);
|
|
|
|
|
|
|
|
Random rnd(301);
|
|
|
|
std::string str1 = rnd.RandomString(1020);
|
|
|
|
auto item1 =
|
|
|
|
std::unique_ptr<TestItem>(new TestItem(str1.data(), str1.length()));
|
|
|
|
ASSERT_NOK(cache->Insert("k1", item1.get(), nullptr, str1.length()));
|
2022-04-11 20:28:33 +00:00
|
|
|
ASSERT_OK(cache->Insert("k1", item1.get(),
|
|
|
|
&CompressedSecondaryCacheTest::helper_,
|
2022-02-24 00:06:27 +00:00
|
|
|
str1.length()));
|
|
|
|
item1.release(); // Appease clang-analyze "potential memory leak"
|
|
|
|
|
|
|
|
Cache::Handle* handle;
|
|
|
|
handle = cache->Lookup("k2", nullptr, test_item_creator,
|
|
|
|
Cache::Priority::LOW, true);
|
|
|
|
ASSERT_EQ(handle, nullptr);
|
2022-04-11 20:28:33 +00:00
|
|
|
handle = cache->Lookup("k2", &CompressedSecondaryCacheTest::helper_,
|
2022-02-24 00:06:27 +00:00
|
|
|
test_item_creator, Cache::Priority::LOW, false);
|
|
|
|
ASSERT_EQ(handle, nullptr);
|
|
|
|
|
|
|
|
cache.reset();
|
|
|
|
secondary_cache.reset();
|
|
|
|
}
|
|
|
|
|
|
|
|
void IntegrationSaveFailTest(bool sec_cache_is_compressed) {
|
2022-04-11 20:28:33 +00:00
|
|
|
CompressedSecondaryCacheOptions secondary_cache_opts;
|
2022-02-24 00:06:27 +00:00
|
|
|
|
|
|
|
if (sec_cache_is_compressed) {
|
|
|
|
if (!LZ4_Supported()) {
|
|
|
|
ROCKSDB_GTEST_SKIP("This test requires LZ4 support.");
|
|
|
|
secondary_cache_opts.compression_type = CompressionType::kNoCompression;
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
secondary_cache_opts.compression_type = CompressionType::kNoCompression;
|
|
|
|
}
|
|
|
|
|
|
|
|
secondary_cache_opts.capacity = 2048;
|
|
|
|
secondary_cache_opts.num_shard_bits = 0;
|
|
|
|
secondary_cache_opts.metadata_charge_policy = kDontChargeCacheMetadata;
|
|
|
|
|
|
|
|
std::shared_ptr<SecondaryCache> secondary_cache =
|
2022-04-11 20:28:33 +00:00
|
|
|
NewCompressedSecondaryCache(secondary_cache_opts);
|
2022-02-24 00:06:27 +00:00
|
|
|
|
|
|
|
LRUCacheOptions opts(1024, 0, false, 0.5, nullptr, kDefaultToAdaptiveMutex,
|
|
|
|
kDontChargeCacheMetadata);
|
|
|
|
opts.secondary_cache = secondary_cache;
|
|
|
|
std::shared_ptr<Cache> cache = NewLRUCache(opts);
|
|
|
|
|
|
|
|
Random rnd(301);
|
|
|
|
std::string str1 = rnd.RandomString(1020);
|
|
|
|
TestItem* item1 = new TestItem(str1.data(), str1.length());
|
2022-04-11 20:28:33 +00:00
|
|
|
ASSERT_OK(cache->Insert("k1", item1,
|
|
|
|
&CompressedSecondaryCacheTest::helper_fail_,
|
2022-02-24 00:06:27 +00:00
|
|
|
str1.length()));
|
|
|
|
std::string str2 = rnd.RandomString(1020);
|
|
|
|
TestItem* item2 = new TestItem(str2.data(), str2.length());
|
|
|
|
// k1 should be demoted to the secondary cache.
|
2022-04-11 20:28:33 +00:00
|
|
|
ASSERT_OK(cache->Insert("k2", item2,
|
|
|
|
&CompressedSecondaryCacheTest::helper_fail_,
|
2022-02-24 00:06:27 +00:00
|
|
|
str2.length()));
|
|
|
|
|
|
|
|
Cache::Handle* handle;
|
2022-04-11 20:28:33 +00:00
|
|
|
handle = cache->Lookup("k2", &CompressedSecondaryCacheTest::helper_fail_,
|
2022-02-24 00:06:27 +00:00
|
|
|
test_item_creator, Cache::Priority::LOW, true);
|
|
|
|
ASSERT_NE(handle, nullptr);
|
|
|
|
cache->Release(handle);
|
|
|
|
// This lookup should fail, since k1 demotion would have failed
|
2022-04-11 20:28:33 +00:00
|
|
|
handle = cache->Lookup("k1", &CompressedSecondaryCacheTest::helper_fail_,
|
2022-02-24 00:06:27 +00:00
|
|
|
test_item_creator, Cache::Priority::LOW, true);
|
|
|
|
ASSERT_EQ(handle, nullptr);
|
|
|
|
// Since k1 didn't get promoted, k2 should still be in cache
|
2022-04-11 20:28:33 +00:00
|
|
|
handle = cache->Lookup("k2", &CompressedSecondaryCacheTest::helper_fail_,
|
2022-02-24 00:06:27 +00:00
|
|
|
test_item_creator, Cache::Priority::LOW, true);
|
|
|
|
ASSERT_NE(handle, nullptr);
|
|
|
|
cache->Release(handle);
|
|
|
|
|
|
|
|
cache.reset();
|
|
|
|
secondary_cache.reset();
|
|
|
|
}
|
|
|
|
|
|
|
|
void IntegrationCreateFailTest(bool sec_cache_is_compressed) {
|
2022-04-11 20:28:33 +00:00
|
|
|
CompressedSecondaryCacheOptions secondary_cache_opts;
|
2022-02-24 00:06:27 +00:00
|
|
|
|
|
|
|
if (sec_cache_is_compressed) {
|
|
|
|
if (!LZ4_Supported()) {
|
|
|
|
ROCKSDB_GTEST_SKIP("This test requires LZ4 support.");
|
|
|
|
secondary_cache_opts.compression_type = CompressionType::kNoCompression;
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
secondary_cache_opts.compression_type = CompressionType::kNoCompression;
|
|
|
|
}
|
|
|
|
|
|
|
|
secondary_cache_opts.capacity = 2048;
|
|
|
|
secondary_cache_opts.num_shard_bits = 0;
|
|
|
|
secondary_cache_opts.metadata_charge_policy = kDontChargeCacheMetadata;
|
|
|
|
|
|
|
|
std::shared_ptr<SecondaryCache> secondary_cache =
|
2022-04-11 20:28:33 +00:00
|
|
|
NewCompressedSecondaryCache(secondary_cache_opts);
|
2022-02-24 00:06:27 +00:00
|
|
|
|
|
|
|
LRUCacheOptions opts(1024, 0, false, 0.5, nullptr, kDefaultToAdaptiveMutex,
|
|
|
|
kDontChargeCacheMetadata);
|
|
|
|
opts.secondary_cache = secondary_cache;
|
|
|
|
std::shared_ptr<Cache> cache = NewLRUCache(opts);
|
|
|
|
|
|
|
|
Random rnd(301);
|
|
|
|
std::string str1 = rnd.RandomString(1020);
|
|
|
|
TestItem* item1 = new TestItem(str1.data(), str1.length());
|
2022-04-11 20:28:33 +00:00
|
|
|
ASSERT_OK(cache->Insert("k1", item1, &CompressedSecondaryCacheTest::helper_,
|
2022-02-24 00:06:27 +00:00
|
|
|
str1.length()));
|
|
|
|
|
|
|
|
std::string str2 = rnd.RandomString(1020);
|
|
|
|
TestItem* item2 = new TestItem(str2.data(), str2.length());
|
|
|
|
// k1 should be demoted to the secondary cache.
|
2022-04-11 20:28:33 +00:00
|
|
|
ASSERT_OK(cache->Insert("k2", item2, &CompressedSecondaryCacheTest::helper_,
|
2022-02-24 00:06:27 +00:00
|
|
|
str2.length()));
|
|
|
|
|
|
|
|
Cache::Handle* handle;
|
|
|
|
SetFailCreate(true);
|
2022-04-11 20:28:33 +00:00
|
|
|
handle = cache->Lookup("k2", &CompressedSecondaryCacheTest::helper_,
|
2022-02-24 00:06:27 +00:00
|
|
|
test_item_creator, Cache::Priority::LOW, true);
|
|
|
|
ASSERT_NE(handle, nullptr);
|
|
|
|
cache->Release(handle);
|
|
|
|
// This lookup should fail, since k1 creation would have failed
|
2022-04-11 20:28:33 +00:00
|
|
|
handle = cache->Lookup("k1", &CompressedSecondaryCacheTest::helper_,
|
2022-02-24 00:06:27 +00:00
|
|
|
test_item_creator, Cache::Priority::LOW, true);
|
|
|
|
ASSERT_EQ(handle, nullptr);
|
|
|
|
// Since k1 didn't get promoted, k2 should still be in cache
|
2022-04-11 20:28:33 +00:00
|
|
|
handle = cache->Lookup("k2", &CompressedSecondaryCacheTest::helper_,
|
2022-02-24 00:06:27 +00:00
|
|
|
test_item_creator, Cache::Priority::LOW, true);
|
|
|
|
ASSERT_NE(handle, nullptr);
|
|
|
|
cache->Release(handle);
|
|
|
|
|
|
|
|
cache.reset();
|
|
|
|
secondary_cache.reset();
|
|
|
|
}
|
|
|
|
|
|
|
|
void IntegrationFullCapacityTest(bool sec_cache_is_compressed) {
|
2022-04-11 20:28:33 +00:00
|
|
|
CompressedSecondaryCacheOptions secondary_cache_opts;
|
2022-02-24 00:06:27 +00:00
|
|
|
|
|
|
|
if (sec_cache_is_compressed) {
|
|
|
|
if (!LZ4_Supported()) {
|
|
|
|
ROCKSDB_GTEST_SKIP("This test requires LZ4 support.");
|
|
|
|
secondary_cache_opts.compression_type = CompressionType::kNoCompression;
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
secondary_cache_opts.compression_type = CompressionType::kNoCompression;
|
|
|
|
}
|
|
|
|
|
|
|
|
secondary_cache_opts.capacity = 2048;
|
|
|
|
secondary_cache_opts.num_shard_bits = 0;
|
|
|
|
secondary_cache_opts.metadata_charge_policy = kDontChargeCacheMetadata;
|
|
|
|
|
|
|
|
std::shared_ptr<SecondaryCache> secondary_cache =
|
2022-04-11 20:28:33 +00:00
|
|
|
NewCompressedSecondaryCache(secondary_cache_opts);
|
2022-02-24 00:06:27 +00:00
|
|
|
|
|
|
|
LRUCacheOptions opts(1024, 0, /*_strict_capacity_limit=*/true, 0.5, nullptr,
|
|
|
|
kDefaultToAdaptiveMutex, kDontChargeCacheMetadata);
|
|
|
|
opts.secondary_cache = secondary_cache;
|
|
|
|
std::shared_ptr<Cache> cache = NewLRUCache(opts);
|
|
|
|
|
|
|
|
Random rnd(301);
|
|
|
|
std::string str1 = rnd.RandomString(1020);
|
|
|
|
TestItem* item1 = new TestItem(str1.data(), str1.length());
|
2022-04-11 20:28:33 +00:00
|
|
|
ASSERT_OK(cache->Insert("k1", item1, &CompressedSecondaryCacheTest::helper_,
|
2022-02-24 00:06:27 +00:00
|
|
|
str1.length()));
|
|
|
|
std::string str2 = rnd.RandomString(1020);
|
|
|
|
TestItem* item2 = new TestItem(str2.data(), str2.length());
|
|
|
|
// k1 should be demoted to the secondary cache.
|
2022-04-11 20:28:33 +00:00
|
|
|
ASSERT_OK(cache->Insert("k2", item2, &CompressedSecondaryCacheTest::helper_,
|
2022-02-24 00:06:27 +00:00
|
|
|
str2.length()));
|
|
|
|
|
|
|
|
Cache::Handle* handle2;
|
2022-04-11 20:28:33 +00:00
|
|
|
handle2 = cache->Lookup("k2", &CompressedSecondaryCacheTest::helper_,
|
2022-02-24 00:06:27 +00:00
|
|
|
test_item_creator, Cache::Priority::LOW, true);
|
|
|
|
ASSERT_NE(handle2, nullptr);
|
2022-04-11 20:28:33 +00:00
|
|
|
cache->Release(handle2);
|
|
|
|
// k1 promotion should fail due to the block cache being at capacity,
|
|
|
|
// but the lookup should still succeed
|
|
|
|
Cache::Handle* handle1;
|
|
|
|
handle1 = cache->Lookup("k1", &CompressedSecondaryCacheTest::helper_,
|
|
|
|
test_item_creator, Cache::Priority::LOW, true);
|
|
|
|
ASSERT_NE(handle1, nullptr);
|
|
|
|
cache->Release(handle1);
|
|
|
|
|
2022-02-24 00:06:27 +00:00
|
|
|
// Since k1 didn't get inserted, k2 should still be in cache
|
2022-04-11 20:28:33 +00:00
|
|
|
handle2 = cache->Lookup("k2", &CompressedSecondaryCacheTest::helper_,
|
|
|
|
test_item_creator, Cache::Priority::LOW, true);
|
|
|
|
ASSERT_NE(handle2, nullptr);
|
2022-02-24 00:06:27 +00:00
|
|
|
cache->Release(handle2);
|
|
|
|
|
|
|
|
cache.reset();
|
|
|
|
secondary_cache.reset();
|
|
|
|
}
|
|
|
|
|
|
|
|
private:
|
|
|
|
bool fail_create_;
|
|
|
|
};
|
|
|
|
|
2022-04-11 20:28:33 +00:00
|
|
|
Cache::CacheItemHelper CompressedSecondaryCacheTest::helper_(
|
|
|
|
CompressedSecondaryCacheTest::SizeCallback,
|
|
|
|
CompressedSecondaryCacheTest::SaveToCallback,
|
|
|
|
CompressedSecondaryCacheTest::DeletionCallback);
|
2022-02-24 00:06:27 +00:00
|
|
|
|
2022-04-11 20:28:33 +00:00
|
|
|
Cache::CacheItemHelper CompressedSecondaryCacheTest::helper_fail_(
|
|
|
|
CompressedSecondaryCacheTest::SizeCallback,
|
|
|
|
CompressedSecondaryCacheTest::SaveToCallbackFail,
|
|
|
|
CompressedSecondaryCacheTest::DeletionCallback);
|
2022-02-24 00:06:27 +00:00
|
|
|
|
2022-04-11 20:28:33 +00:00
|
|
|
TEST_F(CompressedSecondaryCacheTest, BasicTestWithNoCompression) {
|
2022-02-24 00:06:27 +00:00
|
|
|
BasicTest(false, false);
|
|
|
|
}
|
|
|
|
|
2022-04-11 20:28:33 +00:00
|
|
|
TEST_F(CompressedSecondaryCacheTest,
|
|
|
|
BasicTestWithMemoryAllocatorAndNoCompression) {
|
2022-02-24 00:06:27 +00:00
|
|
|
BasicTest(false, true);
|
|
|
|
}
|
|
|
|
|
2022-04-11 20:28:33 +00:00
|
|
|
TEST_F(CompressedSecondaryCacheTest, BasicTestWithCompression) {
|
2022-02-24 00:06:27 +00:00
|
|
|
BasicTest(true, false);
|
|
|
|
}
|
|
|
|
|
2022-04-11 20:28:33 +00:00
|
|
|
TEST_F(CompressedSecondaryCacheTest,
|
|
|
|
BasicTestWithMemoryAllocatorAndCompression) {
|
2022-02-24 00:06:27 +00:00
|
|
|
BasicTest(true, true);
|
|
|
|
}
|
|
|
|
|
2022-06-10 19:23:10 +00:00
|
|
|
#ifndef ROCKSDB_LITE
|
|
|
|
|
|
|
|
TEST_F(CompressedSecondaryCacheTest, BasicTestFromStringWithNoCompression) {
|
|
|
|
std::string sec_cache_uri =
|
|
|
|
"compressed_secondary_cache://"
|
|
|
|
"capacity=2048;num_shard_bits=0;compression_type=kNoCompression";
|
|
|
|
std::shared_ptr<SecondaryCache> sec_cache;
|
|
|
|
Status s = SecondaryCache::CreateFromString(ConfigOptions(), sec_cache_uri,
|
|
|
|
&sec_cache);
|
|
|
|
EXPECT_OK(s);
|
|
|
|
BasicTestHelper(sec_cache);
|
|
|
|
}
|
|
|
|
|
|
|
|
TEST_F(CompressedSecondaryCacheTest, BasicTestFromStringWithCompression) {
|
|
|
|
std::string sec_cache_uri;
|
|
|
|
if (LZ4_Supported()) {
|
|
|
|
sec_cache_uri =
|
|
|
|
"compressed_secondary_cache://"
|
|
|
|
"capacity=2048;num_shard_bits=0;compression_type=kLZ4Compression;"
|
|
|
|
"compress_format_version=2";
|
|
|
|
} else {
|
|
|
|
ROCKSDB_GTEST_SKIP("This test requires LZ4 support.");
|
|
|
|
sec_cache_uri =
|
|
|
|
"compressed_secondary_cache://"
|
|
|
|
"capacity=2048;num_shard_bits=0;compression_type=kNoCompression";
|
|
|
|
}
|
|
|
|
|
|
|
|
std::shared_ptr<SecondaryCache> sec_cache;
|
|
|
|
Status s = SecondaryCache::CreateFromString(ConfigOptions(), sec_cache_uri,
|
|
|
|
&sec_cache);
|
|
|
|
EXPECT_OK(s);
|
|
|
|
BasicTestHelper(sec_cache);
|
|
|
|
}
|
|
|
|
|
|
|
|
#endif // ROCKSDB_LITE
|
|
|
|
|
2022-04-11 20:28:33 +00:00
|
|
|
TEST_F(CompressedSecondaryCacheTest, FailsTestWithNoCompression) {
|
|
|
|
FailsTest(false);
|
|
|
|
}
|
2022-02-24 00:06:27 +00:00
|
|
|
|
2022-04-11 20:28:33 +00:00
|
|
|
TEST_F(CompressedSecondaryCacheTest, FailsTestWithCompression) {
|
|
|
|
FailsTest(true);
|
|
|
|
}
|
2022-02-24 00:06:27 +00:00
|
|
|
|
2022-04-11 20:28:33 +00:00
|
|
|
TEST_F(CompressedSecondaryCacheTest, BasicIntegrationTestWithNoCompression) {
|
2022-02-24 00:06:27 +00:00
|
|
|
BasicIntegrationTest(false);
|
|
|
|
}
|
|
|
|
|
2022-04-11 20:28:33 +00:00
|
|
|
TEST_F(CompressedSecondaryCacheTest, BasicIntegrationTestWithCompression) {
|
2022-02-24 00:06:27 +00:00
|
|
|
BasicIntegrationTest(true);
|
|
|
|
}
|
|
|
|
|
2022-04-11 20:28:33 +00:00
|
|
|
TEST_F(CompressedSecondaryCacheTest,
|
|
|
|
BasicIntegrationFailTestWithNoCompression) {
|
2022-02-24 00:06:27 +00:00
|
|
|
BasicIntegrationFailTest(false);
|
|
|
|
}
|
|
|
|
|
2022-04-11 20:28:33 +00:00
|
|
|
TEST_F(CompressedSecondaryCacheTest, BasicIntegrationFailTestWithCompression) {
|
2022-02-24 00:06:27 +00:00
|
|
|
BasicIntegrationFailTest(true);
|
|
|
|
}
|
|
|
|
|
2022-04-11 20:28:33 +00:00
|
|
|
TEST_F(CompressedSecondaryCacheTest, IntegrationSaveFailTestWithNoCompression) {
|
2022-02-24 00:06:27 +00:00
|
|
|
IntegrationSaveFailTest(false);
|
|
|
|
}
|
|
|
|
|
2022-04-11 20:28:33 +00:00
|
|
|
TEST_F(CompressedSecondaryCacheTest, IntegrationSaveFailTestWithCompression) {
|
2022-02-24 00:06:27 +00:00
|
|
|
IntegrationSaveFailTest(true);
|
|
|
|
}
|
|
|
|
|
2022-04-11 20:28:33 +00:00
|
|
|
TEST_F(CompressedSecondaryCacheTest,
|
|
|
|
IntegrationCreateFailTestWithNoCompression) {
|
2022-02-24 00:06:27 +00:00
|
|
|
IntegrationCreateFailTest(false);
|
|
|
|
}
|
|
|
|
|
2022-04-11 20:28:33 +00:00
|
|
|
TEST_F(CompressedSecondaryCacheTest, IntegrationCreateFailTestWithCompression) {
|
2022-02-24 00:06:27 +00:00
|
|
|
IntegrationCreateFailTest(true);
|
|
|
|
}
|
|
|
|
|
2022-04-11 20:28:33 +00:00
|
|
|
TEST_F(CompressedSecondaryCacheTest,
|
|
|
|
IntegrationFullCapacityTestWithNoCompression) {
|
2022-02-24 00:06:27 +00:00
|
|
|
IntegrationFullCapacityTest(false);
|
|
|
|
}
|
|
|
|
|
2022-04-11 20:28:33 +00:00
|
|
|
TEST_F(CompressedSecondaryCacheTest,
|
|
|
|
IntegrationFullCapacityTestWithCompression) {
|
2022-02-24 00:06:27 +00:00
|
|
|
IntegrationFullCapacityTest(true);
|
|
|
|
}
|
|
|
|
|
|
|
|
} // namespace ROCKSDB_NAMESPACE
|
|
|
|
|
|
|
|
int main(int argc, char** argv) {
|
|
|
|
::testing::InitGoogleTest(&argc, argv);
|
|
|
|
return RUN_ALL_TESTS();
|
|
|
|
}
|