Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
33 changes: 33 additions & 0 deletions src/core/bptree_set_test.cc
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,7 @@
//
#include "core/bptree_set.h"

#include <absl/container/btree_set.h>
#include <mimalloc.h>

#include <random>
Expand Down Expand Up @@ -320,6 +321,38 @@ TEST_F(BPTreeSetTest, DeleteRange) {
}
}

TEST_F(BPTreeSetTest, MemoryUsage) {
zskiplist* zsl = zslCreate();
std::vector<sds> sds_vec;
for (size_t i = 0; i < 10'000; ++i) {
sds_vec.push_back(sdsnew("f"));
}
size_t sz_before = zmalloc_used_memory_tl;
LOG(INFO) << "zskiplist before: " << sz_before << " bytes";

for (size_t i = 0; i < sds_vec.size(); ++i) {
zslInsert(zsl, i, sds_vec[i]);
}
LOG(INFO) << "zskiplist took: " << zmalloc_used_memory_tl - sz_before << " bytes";
zslFree(zsl);

sds_vec.clear();
for (size_t i = 0; i < 10'000; ++i) {
sds_vec.push_back(sdsnew("f"));
}

MiMemoryResource mi_alloc(mi_heap_get_backing());
using AllocType = PMR_NS::polymorphic_allocator<std::pair<double, sds>>;
AllocType alloc(&mi_alloc);
absl::btree_set<pair<double, sds>, std::greater<pair<double, sds>>, AllocType> btree(alloc);

LOG(INFO) << "btree before: " << mi_alloc.used() << " bytes";
for (size_t i = 0; i < sds_vec.size(); ++i) {
btree.emplace(i, sds_vec[i]);
}
LOG(INFO) << "btree after: " << mi_alloc.used() << " bytes";
}

struct ZsetPolicy {
struct KeyT {
double d;
Expand Down
4 changes: 0 additions & 4 deletions src/core/compact_object.h
Original file line number Diff line number Diff line change
Expand Up @@ -41,10 +41,6 @@ class RobjWrapper {
void SetString(std::string_view s, MemoryResource* mr);
void Init(unsigned type, unsigned encoding, void* inner);

// Equivalent to zsetAdd
int AddZsetMember(std::string_view member, double score, int in_flags, int* out_flags,
double* newscore);

unsigned type() const {
return type_;
}
Expand Down
Loading