diff --git a/utilities/transactions/transaction_base.cc b/utilities/transactions/transaction_base.cc index 72d12c607..01bab827a 100644 --- a/utilities/transactions/transaction_base.cc +++ b/utilities/transactions/transaction_base.cc @@ -22,7 +22,8 @@ TransactionBaseImpl::TransactionBaseImpl(DB* db, write_options_(write_options), cmp_(GetColumnFamilyUserComparator(db->DefaultColumnFamily())), start_time_(db_->GetEnv()->NowMicros()), - write_batch_(cmp_, 0, true) {} + write_batch_(cmp_, 0, true), + indexing_enabled_(true) {} TransactionBaseImpl::~TransactionBaseImpl() { // Release snapshot if snapshot is set @@ -38,10 +39,15 @@ void TransactionBaseImpl::Clear() { num_merges_ = 0; } -void TransactionBaseImpl::Reinitialize(const WriteOptions& write_options) { +void TransactionBaseImpl::Reinitialize(DB* db, + const WriteOptions& write_options) { Clear(); + ClearSnapshot(); + db_ = db; write_options_ = write_options; start_time_ = db_->GetEnv()->NowMicros(); + indexing_enabled_ = true; + cmp_ = GetColumnFamilyUserComparator(db_->DefaultColumnFamily()); } void TransactionBaseImpl::SetSnapshot() { diff --git a/utilities/transactions/transaction_base.h b/utilities/transactions/transaction_base.h index 86903ea1f..db33b6f65 100644 --- a/utilities/transactions/transaction_base.h +++ b/utilities/transactions/transaction_base.h @@ -32,7 +32,7 @@ class TransactionBaseImpl : public Transaction { // Remove pending operations queued in this transaction. virtual void Clear(); - void Reinitialize(const WriteOptions& write_options); + void Reinitialize(DB* db, const WriteOptions& write_options); // Called before executing Put, Merge, Delete, and GetForUpdate. If TryLock // returns non-OK, the Put/Merge/Delete/GetForUpdate will be failed. @@ -235,7 +235,7 @@ class TransactionBaseImpl : public Transaction { // Sets a snapshot if SetSnapshotOnNextOperation() has been called. void SetSnapshotIfNeeded(); - DB* const db_; + DB* db_; WriteOptions write_options_; @@ -294,7 +294,7 @@ class TransactionBaseImpl : public Transaction { // WriteBatchWithIndex. // If false, future Put/Merge/Deletes will be inserted directly into the // underlying WriteBatch and not indexed in the WriteBatchWithIndex. - bool indexing_enabled_ = true; + bool indexing_enabled_; // SetSnapshotOnNextOperation() has been called and the snapshot has not yet // been reset. diff --git a/utilities/transactions/transaction_db_impl.cc b/utilities/transactions/transaction_db_impl.cc index b02d7bd25..ef03f3454 100644 --- a/utilities/transactions/transaction_db_impl.cc +++ b/utilities/transactions/transaction_db_impl.cc @@ -312,7 +312,7 @@ void TransactionDBImpl::ReinitializeTransaction( assert(dynamic_cast(txn) != nullptr); auto txn_impl = reinterpret_cast(txn); - txn_impl->Reinitialize(write_options, txn_options); + txn_impl->Reinitialize(this, write_options, txn_options); } } // namespace rocksdb diff --git a/utilities/transactions/transaction_impl.cc b/utilities/transactions/transaction_impl.cc index 33393751d..8f80433a8 100644 --- a/utilities/transactions/transaction_impl.cc +++ b/utilities/transactions/transaction_impl.cc @@ -70,6 +70,7 @@ void TransactionImpl::Initialize(const TransactionOptions& txn_options) { if (txn_options.set_snapshot) { SetSnapshot(); } + if (expiration_time_ > 0) { txn_db_impl_->InsertExpirableTransaction(txn_id_, this); } @@ -87,9 +88,10 @@ void TransactionImpl::Clear() { TransactionBaseImpl::Clear(); } -void TransactionImpl::Reinitialize(const WriteOptions& write_options, +void TransactionImpl::Reinitialize(TransactionDB* txn_db, + const WriteOptions& write_options, const TransactionOptions& txn_options) { - TransactionBaseImpl::Reinitialize(write_options); + TransactionBaseImpl::Reinitialize(txn_db->GetBaseDB(), write_options); Initialize(txn_options); } diff --git a/utilities/transactions/transaction_impl.h b/utilities/transactions/transaction_impl.h index 8a8ed6531..cb02e2834 100644 --- a/utilities/transactions/transaction_impl.h +++ b/utilities/transactions/transaction_impl.h @@ -38,7 +38,7 @@ class TransactionImpl : public TransactionBaseImpl { virtual ~TransactionImpl(); - void Reinitialize(const WriteOptions& write_options, + void Reinitialize(TransactionDB* txn_db, const WriteOptions& write_options, const TransactionOptions& txn_options); Status Commit() override; diff --git a/utilities/transactions/transaction_test.cc b/utilities/transactions/transaction_test.cc index 809dc9506..6f40e5e6a 100644 --- a/utilities/transactions/transaction_test.cc +++ b/utilities/transactions/transaction_test.cc @@ -1222,7 +1222,7 @@ TEST_F(TransactionTest, ReinitializeTest) { // Reinitialize transaction to no long expire txn_options.expiration = -1; - db->BeginTransaction(write_options, txn_options, txn1); + txn1 = db->BeginTransaction(write_options, txn_options, txn1); s = txn1->Put("Z", "z"); ASSERT_OK(s); @@ -1231,13 +1231,13 @@ TEST_F(TransactionTest, ReinitializeTest) { s = txn1->Commit(); ASSERT_OK(s); - db->BeginTransaction(write_options, txn_options, txn1); + txn1 = db->BeginTransaction(write_options, txn_options, txn1); s = txn1->Put("Z", "zz"); ASSERT_OK(s); // Reinitilize txn1 and verify that Z gets unlocked - db->BeginTransaction(write_options, txn_options, txn1); + txn1 = db->BeginTransaction(write_options, txn_options, txn1); Transaction* txn2 = db->BeginTransaction(write_options, txn_options, nullptr); s = txn2->Put("Z", "zzz"); @@ -1262,12 +1262,12 @@ TEST_F(TransactionTest, ReinitializeTest) { ASSERT_OK(s); ASSERT_EQ(value, "zzzz"); - db->BeginTransaction(write_options, txn_options, txn1); + txn1 = db->BeginTransaction(write_options, txn_options, txn1); const Snapshot* snapshot = txn1->GetSnapshot(); - ASSERT_TRUE(snapshot); + ASSERT_FALSE(snapshot); txn_options.set_snapshot = true; - db->BeginTransaction(write_options, txn_options, txn1); + txn1 = db->BeginTransaction(write_options, txn_options, txn1); snapshot = txn1->GetSnapshot(); ASSERT_TRUE(snapshot); @@ -1280,8 +1280,9 @@ TEST_F(TransactionTest, ReinitializeTest) { ASSERT_OK(s); txn_options.set_snapshot = false; - db->BeginTransaction(write_options, txn_options, txn1); + txn1 = db->BeginTransaction(write_options, txn_options, txn1); snapshot = txn1->GetSnapshot(); + ASSERT_FALSE(snapshot); s = txn1->Put("X", "x"); ASSERT_OK(s);