d1b74b0c82
Summary: Update Compaction/Flush to support WritePreparedTxnDB: Add SnapshotChecker which is a proxy to query WritePreparedTxnDB::IsInSnapshot. Pass SnapshotChecker to DBImpl on WritePreparedTxnDB open. CompactionIterator use it to check if a key has been committed and if it is visible to a snapshot. In CompactionIterator: * check if key has been committed. If not, output uncommitted keys AS-IS. * use SnapshotChecker to check if key is visible to a snapshot when in need. * do not output key with seq = 0 if the key is not committed. Closes https://github.com/facebook/rocksdb/pull/2926 Differential Revision: D5902907 Pulled By: yiwu-arbug fbshipit-source-id: 945e037fdf0aa652dc5ba0ad879461040baa0320
38 lines
1.1 KiB
C++
38 lines
1.1 KiB
C++
// Copyright (c) 2011-present, Facebook, Inc. All rights reserved.
|
|
// This source code is licensed under both the GPLv2 (found in the
|
|
// COPYING file in the root directory) and Apache 2.0 License
|
|
// (found in the LICENSE.Apache file in the root directory).
|
|
|
|
#include "db/snapshot_checker.h"
|
|
|
|
#ifdef ROCKSDB_LITE
|
|
#include <assert.h>
|
|
#endif // ROCKSDB_LITE
|
|
|
|
#include "utilities/transactions/pessimistic_transaction_db.h"
|
|
|
|
namespace rocksdb {
|
|
|
|
#ifdef ROCKSDB_LITE
|
|
SnapshotChecker::SnapshotChecker(WritePreparedTxnDB* txn_db) {}
|
|
|
|
bool SnapshotChecker::IsInSnapshot(SequenceNumber sequence,
|
|
SequenceNumber snapshot_sequence) const {
|
|
// Should never be called in LITE mode.
|
|
assert(false);
|
|
return true;
|
|
}
|
|
|
|
#else
|
|
|
|
SnapshotChecker::SnapshotChecker(WritePreparedTxnDB* txn_db)
|
|
: txn_db_(txn_db){};
|
|
|
|
bool SnapshotChecker::IsInSnapshot(SequenceNumber sequence,
|
|
SequenceNumber snapshot_sequence) const {
|
|
return txn_db_->IsInSnapshot(sequence, snapshot_sequence);
|
|
}
|
|
#endif // ROCKSDB_LITE
|
|
|
|
} // namespace rocksdb
|