b878f93c70
Summary: Transaction::GetForUpdate is extended with a do_validate parameter with default value of true. If false it skips validating the snapshot (if there is any) before doing the read. After the read it also returns the latest value (expects the ReadOptions::snapshot to be nullptr). This allows RocksDB applications to use GetForUpdate similarly to how InnoDB does. Similarly ::Merge, ::Put, ::Delete, and ::SingleDelete are extended with assume_exclusive_tracked with default value of false. It true it indicates that call is assumed to be after a ::GetForUpdate(do_validate=false). The Java APIs are accordingly updated. Pull Request resolved: https://github.com/facebook/rocksdb/pull/4680 Differential Revision: D13068508 Pulled By: maysamyabandeh fbshipit-source-id: f0b59db28f7f6a078b60844d902057140765e67d
99 lines
2.9 KiB
C++
99 lines
2.9 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).
|
|
|
|
#pragma once
|
|
|
|
#ifndef ROCKSDB_LITE
|
|
|
|
#include <stack>
|
|
#include <string>
|
|
#include <unordered_map>
|
|
#include <vector>
|
|
|
|
#include "db/write_callback.h"
|
|
#include "rocksdb/db.h"
|
|
#include "rocksdb/slice.h"
|
|
#include "rocksdb/snapshot.h"
|
|
#include "rocksdb/status.h"
|
|
#include "rocksdb/types.h"
|
|
#include "rocksdb/utilities/transaction.h"
|
|
#include "rocksdb/utilities/optimistic_transaction_db.h"
|
|
#include "rocksdb/utilities/write_batch_with_index.h"
|
|
#include "utilities/transactions/transaction_base.h"
|
|
#include "utilities/transactions/transaction_util.h"
|
|
|
|
namespace rocksdb {
|
|
|
|
class OptimisticTransaction : public TransactionBaseImpl {
|
|
public:
|
|
OptimisticTransaction(OptimisticTransactionDB* db,
|
|
const WriteOptions& write_options,
|
|
const OptimisticTransactionOptions& txn_options);
|
|
|
|
virtual ~OptimisticTransaction();
|
|
|
|
void Reinitialize(OptimisticTransactionDB* txn_db,
|
|
const WriteOptions& write_options,
|
|
const OptimisticTransactionOptions& txn_options);
|
|
|
|
Status Prepare() override;
|
|
|
|
Status Commit() override;
|
|
|
|
Status Rollback() override;
|
|
|
|
Status SetName(const TransactionName& name) override;
|
|
|
|
protected:
|
|
Status TryLock(ColumnFamilyHandle* column_family, const Slice& key,
|
|
bool read_only, bool exclusive, const bool do_validate = true,
|
|
const bool assume_tracked = false) override;
|
|
|
|
private:
|
|
OptimisticTransactionDB* const txn_db_;
|
|
|
|
friend class OptimisticTransactionCallback;
|
|
|
|
void Initialize(const OptimisticTransactionOptions& txn_options);
|
|
|
|
// Returns OK if it is safe to commit this transaction. Returns Status::Busy
|
|
// if there are read or write conflicts that would prevent us from committing
|
|
// OR if we can not determine whether there would be any such conflicts.
|
|
//
|
|
// Should only be called on writer thread.
|
|
Status CheckTransactionForConflicts(DB* db);
|
|
|
|
void Clear() override;
|
|
|
|
void UnlockGetForUpdate(ColumnFamilyHandle* /* unused */,
|
|
const Slice& /* unused */) override {
|
|
// Nothing to unlock.
|
|
}
|
|
|
|
// No copying allowed
|
|
OptimisticTransaction(const OptimisticTransaction&);
|
|
void operator=(const OptimisticTransaction&);
|
|
};
|
|
|
|
// Used at commit time to trigger transaction validation
|
|
class OptimisticTransactionCallback : public WriteCallback {
|
|
public:
|
|
explicit OptimisticTransactionCallback(OptimisticTransaction* txn)
|
|
: txn_(txn) {}
|
|
|
|
Status Callback(DB* db) override {
|
|
return txn_->CheckTransactionForConflicts(db);
|
|
}
|
|
|
|
bool AllowWriteBatching() override { return false; }
|
|
|
|
private:
|
|
OptimisticTransaction* txn_;
|
|
};
|
|
|
|
} // namespace rocksdb
|
|
|
|
#endif // ROCKSDB_LITE
|