b9846370e9
Summary: This adds support for recovering WriteUnprepared transactions through the following changes: - The information in `RecoveredTransaction` is extended so that it can reference multiple batches. - `MarkBeginPrepare` is extended with a bool indicating whether it is an unprepared begin, and this is passed down to `InsertRecoveredTransaction` to indicate whether the current transaction is prepared or not. - `WriteUnpreparedTxnDB::Initialize` is overridden so that it will rollback unprepared transactions from the recovered transactions. This can be done without updating the prepare heap/commit map, because this is before the DB has finished initializing, and after writing the rollback batch, those data structures should not contain information about the rolled back transaction anyway. Commit/Rollback of live transactions is still unimplemented and will come later. Pull Request resolved: https://github.com/facebook/rocksdb/pull/4078 Differential Revision: D8703382 Pulled By: lth fbshipit-source-id: 7e0aada6c23bd39299f1f20d6c060492e0e6b60a
85 lines
3.3 KiB
C++
85 lines
3.3 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).
|
|
//
|
|
// This file implements the callback "bridge" between Java and C++ for
|
|
// rocksdb::WriteBatch::Handler.
|
|
|
|
#ifndef JAVA_ROCKSJNI_WRITEBATCHHANDLERJNICALLBACK_H_
|
|
#define JAVA_ROCKSJNI_WRITEBATCHHANDLERJNICALLBACK_H_
|
|
|
|
#include <functional>
|
|
#include <jni.h>
|
|
#include <memory>
|
|
#include "rocksjni/jnicallback.h"
|
|
#include "rocksdb/write_batch.h"
|
|
|
|
namespace rocksdb {
|
|
/**
|
|
* This class acts as a bridge between C++
|
|
* and Java. The methods in this class will be
|
|
* called back from the RocksDB storage engine (C++)
|
|
* which calls the appropriate Java method.
|
|
* This enables Write Batch Handlers to be implemented in Java.
|
|
*/
|
|
class WriteBatchHandlerJniCallback : public JniCallback, public WriteBatch::Handler {
|
|
public:
|
|
WriteBatchHandlerJniCallback(
|
|
JNIEnv* env, jobject jWriteBackHandler);
|
|
Status PutCF(uint32_t column_family_id, const Slice& key,
|
|
const Slice& value);
|
|
void Put(const Slice& key, const Slice& value);
|
|
Status MergeCF(uint32_t column_family_id, const Slice& key,
|
|
const Slice& value);
|
|
void Merge(const Slice& key, const Slice& value);
|
|
Status DeleteCF(uint32_t column_family_id, const Slice& key);
|
|
void Delete(const Slice& key);
|
|
Status SingleDeleteCF(uint32_t column_family_id, const Slice& key);
|
|
void SingleDelete(const Slice& key);
|
|
Status DeleteRangeCF(uint32_t column_family_id, const Slice& beginKey,
|
|
const Slice& endKey);
|
|
void DeleteRange(const Slice& beginKey, const Slice& endKey);
|
|
void LogData(const Slice& blob);
|
|
Status PutBlobIndexCF(uint32_t column_family_id, const Slice& key,
|
|
const Slice& value);
|
|
Status MarkBeginPrepare(bool);
|
|
Status MarkEndPrepare(const Slice& xid);
|
|
Status MarkNoop(bool empty_batch);
|
|
Status MarkRollback(const Slice& xid);
|
|
Status MarkCommit(const Slice& xid);
|
|
bool Continue();
|
|
|
|
private:
|
|
JNIEnv* m_env;
|
|
jmethodID m_jPutCfMethodId;
|
|
jmethodID m_jPutMethodId;
|
|
jmethodID m_jMergeCfMethodId;
|
|
jmethodID m_jMergeMethodId;
|
|
jmethodID m_jDeleteCfMethodId;
|
|
jmethodID m_jDeleteMethodId;
|
|
jmethodID m_jSingleDeleteCfMethodId;
|
|
jmethodID m_jSingleDeleteMethodId;
|
|
jmethodID m_jDeleteRangeCfMethodId;
|
|
jmethodID m_jDeleteRangeMethodId;
|
|
jmethodID m_jLogDataMethodId;
|
|
jmethodID m_jPutBlobIndexCfMethodId;
|
|
jmethodID m_jMarkBeginPrepareMethodId;
|
|
jmethodID m_jMarkEndPrepareMethodId;
|
|
jmethodID m_jMarkNoopMethodId;
|
|
jmethodID m_jMarkRollbackMethodId;
|
|
jmethodID m_jMarkCommitMethodId;
|
|
jmethodID m_jContinueMethodId;
|
|
/**
|
|
* @return A pointer to a rocksdb::Status or nullptr if an unexpected exception occurred
|
|
*/
|
|
std::unique_ptr<rocksdb::Status> kv_op(const Slice& key, const Slice& value, std::function<void(jbyteArray, jbyteArray)> kvFn);
|
|
/**
|
|
* @return A pointer to a rocksdb::Status or nullptr if an unexpected exception occurred
|
|
*/
|
|
std::unique_ptr<rocksdb::Status> k_op(const Slice& key, std::function<void(jbyteArray)> kFn);
|
|
};
|
|
} // namespace rocksdb
|
|
|
|
#endif // JAVA_ROCKSJNI_WRITEBATCHHANDLERJNICALLBACK_H_
|