@ -15,6 +15,293 @@
namespace rocksdb {
// Instead of reconstructing a Transaction object, and calling rollback on it,
// we can be more efficient with RollbackRecoveredTransaction by skipping
// unnecessary steps (eg. updating CommitMap, reconstructing keyset)
Status WriteUnpreparedTxnDB : : RollbackRecoveredTransaction (
const DBImpl : : RecoveredTransaction * rtxn ) {
// TODO(lth): Reduce duplicate code with WritePrepared rollback logic.
assert ( rtxn - > unprepared_ ) ;
auto cf_map_shared_ptr = WritePreparedTxnDB : : GetCFHandleMap ( ) ;
auto cf_comp_map_shared_ptr = WritePreparedTxnDB : : GetCFComparatorMap ( ) ;
const bool kRollbackMergeOperands = true ;
WriteOptions w_options ;
// If we crash during recovery, we can just recalculate and rewrite the
// rollback batch.
w_options . disableWAL = true ;
// Iterate starting with largest sequence number.
for ( auto it = rtxn - > batches_ . rbegin ( ) ; it ! = rtxn - > batches_ . rend ( ) ; it + + ) {
auto last_visible_txn = it - > first - 1 ;
const auto & batch = it - > second . batch_ ;
WriteBatch rollback_batch ;
struct RollbackWriteBatchBuilder : public WriteBatch : : Handler {
DBImpl * db_ ;
ReadOptions roptions ;
WritePreparedTxnReadCallback callback ;
WriteBatch * rollback_batch_ ;
std : : map < uint32_t , const Comparator * > & comparators_ ;
std : : map < uint32_t , ColumnFamilyHandle * > & handles_ ;
using CFKeys = std : : set < Slice , SetComparator > ;
std : : map < uint32_t , CFKeys > keys_ ;
bool rollback_merge_operands_ ;
RollbackWriteBatchBuilder (
DBImpl * db , WritePreparedTxnDB * wpt_db , SequenceNumber snap_seq ,
WriteBatch * dst_batch ,
std : : map < uint32_t , const Comparator * > & comparators ,
std : : map < uint32_t , ColumnFamilyHandle * > & handles ,
bool rollback_merge_operands )
: db_ ( db ) ,
callback ( wpt_db , snap_seq ,
0 ) , // 0 disables min_uncommitted optimization
rollback_batch_ ( dst_batch ) ,
comparators_ ( comparators ) ,
handles_ ( handles ) ,
rollback_merge_operands_ ( rollback_merge_operands ) { }
Status Rollback ( uint32_t cf , const Slice & key ) {
Status s ;
CFKeys & cf_keys = keys_ [ cf ] ;
if ( cf_keys . size ( ) = = 0 ) { // just inserted
auto cmp = comparators_ [ cf ] ;
keys_ [ cf ] = CFKeys ( SetComparator ( cmp ) ) ;
}
auto res = cf_keys . insert ( key ) ;
if ( res . second = =
false ) { // second is false if a element already existed.
return s ;
}
PinnableSlice pinnable_val ;
bool not_used ;
auto cf_handle = handles_ [ cf ] ;
s = db_ - > GetImpl ( roptions , cf_handle , key , & pinnable_val , & not_used ,
& callback ) ;
assert ( s . ok ( ) | | s . IsNotFound ( ) ) ;
if ( s . ok ( ) ) {
s = rollback_batch_ - > Put ( cf_handle , key , pinnable_val ) ;
assert ( s . ok ( ) ) ;
} else if ( s . IsNotFound ( ) ) {
// There has been no readable value before txn. By adding a delete we
// make sure that there will be none afterwards either.
s = rollback_batch_ - > Delete ( cf_handle , key ) ;
assert ( s . ok ( ) ) ;
} else {
// Unexpected status. Return it to the user.
}
return s ;
}
Status PutCF ( uint32_t cf , const Slice & key ,
const Slice & /*val*/ ) override {
return Rollback ( cf , key ) ;
}
Status DeleteCF ( uint32_t cf , const Slice & key ) override {
return Rollback ( cf , key ) ;
}
Status SingleDeleteCF ( uint32_t cf , const Slice & key ) override {
return Rollback ( cf , key ) ;
}
Status MergeCF ( uint32_t cf , const Slice & key ,
const Slice & /*val*/ ) override {
if ( rollback_merge_operands_ ) {
return Rollback ( cf , key ) ;
} else {
return Status : : OK ( ) ;
}
}
// Recovered batches do not contain 2PC markers.
Status MarkNoop ( bool ) override { return Status : : InvalidArgument ( ) ; }
Status MarkBeginPrepare ( bool ) override {
return Status : : InvalidArgument ( ) ;
}
Status MarkEndPrepare ( const Slice & ) override {
return Status : : InvalidArgument ( ) ;
}
Status MarkCommit ( const Slice & ) override {
return Status : : InvalidArgument ( ) ;
}
Status MarkRollback ( const Slice & ) override {
return Status : : InvalidArgument ( ) ;
}
} rollback_handler ( db_impl_ , this , last_visible_txn , & rollback_batch ,
* cf_comp_map_shared_ptr . get ( ) , * cf_map_shared_ptr . get ( ) ,
! kRollbackMergeOperands ) ;
auto s = batch - > Iterate ( & rollback_handler ) ;
if ( ! s . ok ( ) ) {
return s ;
}
// The Rollback marker will be used as a batch separator
WriteBatchInternal : : MarkRollback ( & rollback_batch , rtxn - > name_ ) ;
const uint64_t kNoLogRef = 0 ;
const bool kDisableMemtable = true ;
const size_t kOneBatch = 1 ;
uint64_t seq_used = kMaxSequenceNumber ;
s = db_impl_ - > WriteImpl ( w_options , & rollback_batch , nullptr , nullptr ,
kNoLogRef , ! kDisableMemtable , & seq_used , kOneBatch ) ;
if ( ! s . ok ( ) ) {
return s ;
}
// If two_write_queues, we must manually release the sequence number to
// readers.
if ( db_impl_ - > immutable_db_options ( ) . two_write_queues ) {
db_impl_ - > SetLastPublishedSequence ( seq_used ) ;
}
}
return Status : : OK ( ) ;
}
Status WriteUnpreparedTxnDB : : Initialize (
const std : : vector < size_t > & compaction_enabled_cf_indices ,
const std : : vector < ColumnFamilyHandle * > & handles ) {
// TODO(lth): Reduce code duplication in this function.
auto dbimpl = reinterpret_cast < DBImpl * > ( GetRootDB ( ) ) ;
assert ( dbimpl ! = nullptr ) ;
db_impl_ - > SetSnapshotChecker ( new WritePreparedSnapshotChecker ( this ) ) ;
// A callback to commit a single sub-batch
class CommitSubBatchPreReleaseCallback : public PreReleaseCallback {
public :
explicit CommitSubBatchPreReleaseCallback ( WritePreparedTxnDB * db )
: db_ ( db ) { }
virtual Status Callback ( SequenceNumber commit_seq ,
bool is_mem_disabled ) override {
# ifdef NDEBUG
( void ) is_mem_disabled ;
# endif
assert ( ! is_mem_disabled ) ;
db_ - > AddCommitted ( commit_seq , commit_seq ) ;
return Status : : OK ( ) ;
}
private :
WritePreparedTxnDB * db_ ;
} ;
db_impl_ - > SetRecoverableStatePreReleaseCallback (
new CommitSubBatchPreReleaseCallback ( this ) ) ;
// PessimisticTransactionDB::Initialize
for ( auto cf_ptr : handles ) {
AddColumnFamily ( cf_ptr ) ;
}
// Verify cf options
for ( auto handle : handles ) {
ColumnFamilyDescriptor cfd ;
Status s = handle - > GetDescriptor ( & cfd ) ;
if ( ! s . ok ( ) ) {
return s ;
}
s = VerifyCFOptions ( cfd . options ) ;
if ( ! s . ok ( ) ) {
return s ;
}
}
// Re-enable compaction for the column families that initially had
// compaction enabled.
std : : vector < ColumnFamilyHandle * > compaction_enabled_cf_handles ;
compaction_enabled_cf_handles . reserve ( compaction_enabled_cf_indices . size ( ) ) ;
for ( auto index : compaction_enabled_cf_indices ) {
compaction_enabled_cf_handles . push_back ( handles [ index ] ) ;
}
Status s = EnableAutoCompaction ( compaction_enabled_cf_handles ) ;
if ( ! s . ok ( ) ) {
return s ;
}
// create 'real' transactions from recovered shell transactions
auto rtxns = dbimpl - > recovered_transactions ( ) ;
for ( auto rtxn : rtxns ) {
auto recovered_trx = rtxn . second ;
assert ( recovered_trx ) ;
assert ( recovered_trx - > batches_ . size ( ) > = 1 ) ;
assert ( recovered_trx - > name_ . length ( ) ) ;
// We can only rollback transactions after AdvanceMaxEvictedSeq is called,
// but AddPrepared must occur before AdvanceMaxEvictedSeq, which is why
// two iterations is required.
if ( recovered_trx - > unprepared_ ) {
continue ;
}
WriteOptions w_options ;
w_options . sync = true ;
TransactionOptions t_options ;
auto first_log_number = recovered_trx - > batches_ . begin ( ) - > second . log_number_ ;
auto last_seq = recovered_trx - > batches_ . rbegin ( ) - > first ;
auto last_prepare_batch_cnt =
recovered_trx - > batches_ . begin ( ) - > second . batch_cnt_ ;
Transaction * real_trx = BeginTransaction ( w_options , t_options , nullptr ) ;
assert ( real_trx ) ;
auto wupt =
static_cast_with_check < WriteUnpreparedTxn , Transaction > ( real_trx ) ;
real_trx - > SetLogNumber ( first_log_number ) ;
real_trx - > SetId ( last_seq ) ;
s = real_trx - > SetName ( recovered_trx - > name_ ) ;
if ( ! s . ok ( ) ) {
break ;
}
wupt - > prepare_batch_cnt_ = last_prepare_batch_cnt ;
for ( auto batch : recovered_trx - > batches_ ) {
const auto & seq = batch . first ;
const auto & batch_info = batch . second ;
auto cnt = batch_info . batch_cnt_ ? batch_info . batch_cnt_ : 1 ;
assert ( batch_info . log_number_ ) ;
for ( size_t i = 0 ; i < cnt ; i + + ) {
AddPrepared ( seq + i ) ;
}
assert ( wupt - > unprep_seqs_ . count ( seq ) = = 0 ) ;
wupt - > unprep_seqs_ [ seq ] = cnt ;
}
wupt - > write_batch_ . Clear ( ) ;
WriteBatchInternal : : InsertNoop ( wupt - > write_batch_ . GetWriteBatch ( ) ) ;
real_trx - > SetState ( Transaction : : PREPARED ) ;
if ( ! s . ok ( ) ) {
break ;
}
}
SequenceNumber prev_max = max_evicted_seq_ ;
SequenceNumber last_seq = db_impl_ - > GetLatestSequenceNumber ( ) ;
AdvanceMaxEvictedSeq ( prev_max , last_seq ) ;
// Rollback unprepared transactions.
for ( auto rtxn : rtxns ) {
auto recovered_trx = rtxn . second ;
if ( recovered_trx - > unprepared_ ) {
s = RollbackRecoveredTransaction ( recovered_trx ) ;
if ( ! s . ok ( ) ) {
return s ;
}
continue ;
}
}
if ( s . ok ( ) ) {
dbimpl - > DeleteAllRecoveredTransactions ( ) ;
}
return s ;
}
Transaction * WriteUnpreparedTxnDB : : BeginTransaction (
const WriteOptions & write_options , const TransactionOptions & txn_options ,
Transaction * old_txn ) {
Manuel Ung
7 years ago
committed by