mirrors/dragonfly
1
0
Fork 0
mirror of https://github.com/dragonflydb/dragonfly.git synced 2025-05-11 02:15:45 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

feat(core): Add Ttl semantics to string_map (#813)

Browse source 
Now string_map respects the optional ttl argument that accepts ttl time in seconds.
Upon traversing and accessing expirerd elements they will be transparently removed and deleted
from the map.

Signed-off-by: Roman Gershman <roman@dragonflydb.io>
This commit is contained in:
Roman Gershman 2023-02-17 10:49:21 +02:00 • committed by GitHub
parent 18ce8c3a5b
commit f92403fb09
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
5 changed files with 108 additions and 49 deletions
Download patch file Download diff file Expand all files Collapse all files

31
src/core/dense_set.cc
View file

@ -86,7 +86,7 @@ size_t DenseSet::PushFront(DenseSet::ChainVectorIterator it, void* data, bool ha
}
if (has_ttl)
it->SetTtl();
it->SetTtl(true);
return ObjectAllocSize(data);
}
@ -97,7 +97,7 @@ void DenseSet::PushFront(DenseSet::ChainVectorIterator it, DenseSet::DensePtr pt
if (it->IsEmpty()) {
it->SetObject(ptr.GetObject());
if (ptr.HasTtl())
it->SetTtl();
it->SetTtl(true);
if (ptr.IsLink()) {
FreeLink(ptr.AsLink());
}
@ -112,7 +112,7 @@ void DenseSet::PushFront(DenseSet::ChainVectorIterator it, DenseSet::DensePtr pt
// allocate a new link if needed and copy the pointer to the new link
it->SetLink(NewLink(ptr.Raw(), *it));
if (ptr.HasTtl())
it->SetTtl();
it->SetTtl(true);
DCHECK(!it->AsLink()->next.IsEmpty());
}
}
@ -313,7 +313,7 @@ void DenseSet::Grow() {
}
}
void* DenseSet::AddOrFind(void* ptr, bool has_ttl) {
auto DenseSet::AddOrFindDense(void* ptr, bool has_ttl) -> DensePtr* {
uint64_t hc = Hash(ptr, 0);
if (entries_.empty()) {
@ -332,7 +332,7 @@ void* DenseSet::AddOrFind(void* ptr, bool has_ttl) {
uint32_t bucket_id = BucketId(hc);
DensePtr* dptr = Find(ptr, bucket_id, 0).second;
if (dptr != nullptr) {
return dptr->GetObject();
return dptr;
}
DCHECK_LT(bucket_id, entries_.size());
@ -373,7 +373,7 @@ void* DenseSet::AddOrFind(void* ptr, bool has_ttl) {
DensePtr to_insert(ptr);
if (has_ttl)
to_insert.SetTtl();
to_insert.SetTtl(true);
while (!entries_[bucket_id].IsEmpty() && entries_[bucket_id].IsDisplaced()) {
DensePtr unlinked = PopPtrFront(entries_.begin() + bucket_id);
@ -497,6 +497,25 @@ void* DenseSet::PopInternal() {
return ret;
}
void* DenseSet::AddOrReplaceObj(void* obj, bool has_ttl) {
DensePtr* ptr = AddOrFindDense(obj, has_ttl);
if (!ptr)
return nullptr;
if (ptr->IsLink()) {
ptr = ptr->AsLink();
}
void* res = ptr->Raw();
obj_malloc_used_ -= ObjectAllocSize(res);
obj_malloc_used_ += ObjectAllocSize(obj);
ptr->SetObject(obj);
ptr->SetTtl(has_ttl);
return res;
}
/**
* stable scanning api. has the same guarantees as redis scan command.
* we avoid doing bit-reverse by using a different function to derive a bucket id