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

feat(server): extend populate command with expire, fix bug that expire the same key for multiple time (#4757)

Browse source 
* feat(server): extend populate command with expire, fix bug that expire the same key for multiple time
This commit is contained in:
lichuang 2025-03-12 22:21:59 +08:00 • committed by GitHub
parent ac33cd871b
commit 93bd52ceb8
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 20 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

40
src/server/debugcmd.cc
View file

@ -195,28 +195,28 @@ void DoPopulateBatch(string_view type, string_view prefix, size_t val_size, bool
stub_tx->InitByArgs(cntx->ns, local_cntx.conn_state.db_index, args_span); stub_tx->InitByArgs(cntx->ns, local_cntx.conn_state.db_index, args_span);
sf->service().InvokeCmd(cid, args_span, &crb, &local_cntx); sf->service().InvokeCmd(cid, args_span, &crb, &local_cntx);
}
if (expire_ttl_range.has_value()) { if (expire_ttl_range.has_value()) {
uint32_t start = expire_ttl_range->first; uint32_t start = expire_ttl_range->first;
uint32_t end = expire_ttl_range->second; uint32_t end = expire_ttl_range->second;
uint32_t expire_ttl = rand() % (end - start) + start; uint32_t expire_ttl = rand() % (end - start) + start;
VLOG(1) << "set key " << key << " expire ttl as " << expire_ttl; VLOG(1) << "set key " << key << " expire ttl as " << expire_ttl;
auto cid = sf->service().mutable_registry()->Find("EXPIRE"); auto cid = sf->service().mutable_registry()->Find("EXPIRE");
absl::InlinedVector<string, 5> args; absl::InlinedVector<string, 5> args;
args.push_back(std::move(key)); args.push_back(std::move(key));
args.push_back(to_string(expire_ttl)); args.push_back(to_string(expire_ttl));
args_view.clear(); args_view.clear();
for (auto& arg : args) { for (auto& arg : args) {
args_view.push_back(arg); args_view.push_back(arg);
}
auto args_span = absl::MakeSpan(args_view);
stub_tx->MultiSwitchCmd(cid);
local_cntx.cid = cid;
crb.SetReplyMode(ReplyMode::NONE);
stub_tx->InitByArgs(cntx->ns, local_cntx.conn_state.db_index, args_span);
sf->service().InvokeCmd(cid, args_span, &crb, &local_cntx);
} }
auto args_span = absl::MakeSpan(args_view);
stub_tx->MultiSwitchCmd(cid);
local_cntx.cid = cid;
crb.SetReplyMode(ReplyMode::NONE);
stub_tx->InitByArgs(cntx->ns, local_cntx.conn_state.db_index, args_span);
sf->service().InvokeCmd(cid, args_span, &crb, &local_cntx);
} }
} }