abby/kv_bench
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Align transaction benchmark semantics

Browse Source
This commit is contained in:
abbycin 2026-03-10 09:19:46 +08:00
parent 82a845429b
commit 0177b0afc1
2 changed files with 24 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
rocksdb/main.cpp
View File

@ -560,6 +560,8 @@ static bool run_one_op(OpKind op, rocksdb::OptimisticTransactionDB *db, rocksdb:
} }
auto *txn = db->BeginTransaction(wopt); auto *txn = db->BeginTransaction(wopt);
txn->SetSnapshot();
ropt.snapshot = txn->GetSnapshot();
auto st = txn->Get(ropt, handle, key, &out); auto st = txn->Get(ropt, handle, key, &out);
auto cst = txn->Commit(); auto cst = txn->Commit();
delete txn; delete txn;
@ -597,6 +599,21 @@ static bool run_one_op(OpKind op, rocksdb::OptimisticTransactionDB *db, rocksdb:
} }
auto *txn = db->BeginTransaction(wopt); auto *txn = db->BeginTransaction(wopt);
if (spec.insert_only) {
auto pst = txn->Put(handle, key.value(), value);
auto cst = txn->Commit();
delete txn;
return pst.ok() && cst.ok();
}
rocksdb::ReadOptions update_ropt;
txn->SetSnapshot();
update_ropt.snapshot = txn->GetSnapshot();
auto gst = txn->GetForUpdate(update_ropt, handle, key.value(), static_cast<std::string *>(nullptr));
if (!gst.ok()) {
delete txn;
return false;
}
auto pst = txn->Put(handle, key.value(), value); auto pst = txn->Put(handle, key.value(), value);
auto cst = txn->Commit(); auto cst = txn->Commit();
delete txn; delete txn;
@ -641,6 +658,8 @@ static bool run_one_op(OpKind op, rocksdb::OptimisticTransactionDB *db, rocksdb:
} }
auto *txn = db->BeginTransaction(wopt); auto *txn = db->BeginTransaction(wopt);
txn->SetSnapshot();
ropt.snapshot = txn->GetSnapshot();
auto *iter = txn->GetIterator(ropt); auto *iter = txn->GetIterator(ropt);
iter->Seek(prefix.value()); iter->Seek(prefix.value());
size_t scanned = 0; size_t scanned = 0;

6
src/main.rs
View File

@ -1110,7 +1110,11 @@ fn run_one_op(
if let Some(key) = key_opt { if let Some(key) = key_opt {
if let Ok(tx) = bucket.begin() { if let Ok(tx) = bucket.begin() {
let write_ok = tx.upsert(key.as_slice(), value.as_slice()).is_ok(); let write_ok = if spec.insert_only {
tx.upsert(key.as_slice(), value.as_slice()).is_ok()
} else {
tx.update(key.as_slice(), value.as_slice()).is_ok()
};
if !write_ok { if !write_ok {
false false
} else { } else {