105 lines
3.1 KiB
C++
105 lines
3.1 KiB
C++
// Copyright (c) 2011-present, Facebook, Inc. All rights reserved.
|
|
// This source code is licensed under the BSD-style license found in the
|
|
// LICENSE file in the root directory of this source tree. An additional grant
|
|
// of patent rights can be found in the PATENTS file in the same directory.
|
|
//
|
|
// This file implements the callback "bridge" between Java and C++ for
|
|
// rocksdb::Comparator.
|
|
|
|
#include "rocksjni/writebatchhandlerjnicallback.h"
|
|
#include "rocksjni/portal.h"
|
|
|
|
namespace rocksdb {
|
|
WriteBatchHandlerJniCallback::WriteBatchHandlerJniCallback(
|
|
JNIEnv* env, jobject jWriteBatchHandler)
|
|
: m_env(env) {
|
|
|
|
// Note: we want to access the Java WriteBatchHandler instance
|
|
// across multiple method calls, so we create a global ref
|
|
m_jWriteBatchHandler = env->NewGlobalRef(jWriteBatchHandler);
|
|
|
|
m_jPutMethodId = WriteBatchHandlerJni::getPutMethodId(env);
|
|
m_jMergeMethodId = WriteBatchHandlerJni::getMergeMethodId(env);
|
|
m_jDeleteMethodId = WriteBatchHandlerJni::getDeleteMethodId(env);
|
|
m_jLogDataMethodId = WriteBatchHandlerJni::getLogDataMethodId(env);
|
|
m_jContinueMethodId = WriteBatchHandlerJni::getContinueMethodId(env);
|
|
}
|
|
|
|
void WriteBatchHandlerJniCallback::Put(const Slice& key, const Slice& value) {
|
|
const jbyteArray j_key = sliceToJArray(key);
|
|
const jbyteArray j_value = sliceToJArray(value);
|
|
|
|
m_env->CallVoidMethod(
|
|
m_jWriteBatchHandler,
|
|
m_jPutMethodId,
|
|
j_key,
|
|
j_value);
|
|
|
|
m_env->DeleteLocalRef(j_value);
|
|
m_env->DeleteLocalRef(j_key);
|
|
}
|
|
|
|
void WriteBatchHandlerJniCallback::Merge(const Slice& key, const Slice& value) {
|
|
const jbyteArray j_key = sliceToJArray(key);
|
|
const jbyteArray j_value = sliceToJArray(value);
|
|
|
|
m_env->CallVoidMethod(
|
|
m_jWriteBatchHandler,
|
|
m_jMergeMethodId,
|
|
j_key,
|
|
j_value);
|
|
|
|
m_env->DeleteLocalRef(j_value);
|
|
m_env->DeleteLocalRef(j_key);
|
|
}
|
|
|
|
void WriteBatchHandlerJniCallback::Delete(const Slice& key) {
|
|
const jbyteArray j_key = sliceToJArray(key);
|
|
|
|
m_env->CallVoidMethod(
|
|
m_jWriteBatchHandler,
|
|
m_jDeleteMethodId,
|
|
j_key);
|
|
|
|
m_env->DeleteLocalRef(j_key);
|
|
}
|
|
|
|
void WriteBatchHandlerJniCallback::LogData(const Slice& blob) {
|
|
const jbyteArray j_blob = sliceToJArray(blob);
|
|
|
|
m_env->CallVoidMethod(
|
|
m_jWriteBatchHandler,
|
|
m_jLogDataMethodId,
|
|
j_blob);
|
|
|
|
m_env->DeleteLocalRef(j_blob);
|
|
}
|
|
|
|
bool WriteBatchHandlerJniCallback::Continue() {
|
|
jboolean jContinue = m_env->CallBooleanMethod(
|
|
m_jWriteBatchHandler,
|
|
m_jContinueMethodId);
|
|
|
|
return static_cast<bool>(jContinue == JNI_TRUE);
|
|
}
|
|
|
|
/*
|
|
* Creates a Java Byte Array from the data in a Slice
|
|
*
|
|
* When calling this function
|
|
* you must remember to call env->DeleteLocalRef
|
|
* on the result after you have finished with it
|
|
*/
|
|
jbyteArray WriteBatchHandlerJniCallback::sliceToJArray(const Slice& s) {
|
|
jbyteArray ja = m_env->NewByteArray(static_cast<jsize>(s.size()));
|
|
m_env->SetByteArrayRegion(
|
|
ja, 0, static_cast<jsize>(s.size()),
|
|
reinterpret_cast<const jbyte*>(s.data()));
|
|
return ja;
|
|
}
|
|
|
|
WriteBatchHandlerJniCallback::~WriteBatchHandlerJniCallback() {
|
|
m_env->DeleteGlobalRef(m_jWriteBatchHandler);
|
|
}
|
|
} // namespace rocksdb
|