Adam Retter
10 years ago
parent
00211f9c5b
commit
8fb4751d50
@ -0,0 +1,98 @@ |
|||||||
|
// Copyright (c) 2014, 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) { |
||||||
|
|
||||||
|
// Note: WriteBatchHandler methods may be accessed by multiple threads,
|
||||||
|
// so we ref the jvm not the env
|
||||||
|
const jint rs = env->GetJavaVM(&m_jvm); |
||||||
|
assert(rs == JNI_OK); |
||||||
|
|
||||||
|
// 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); |
||||||
|
} |
||||||
|
|
||||||
|
/**
|
||||||
|
* Attach/Get a JNIEnv for the current native thread |
||||||
|
*/ |
||||||
|
JNIEnv* WriteBatchHandlerJniCallback::getJniEnv() const { |
||||||
|
JNIEnv *env; |
||||||
|
jint rs = m_jvm->AttachCurrentThread(reinterpret_cast<void **>(&env), NULL); |
||||||
|
assert(rs == JNI_OK); |
||||||
|
return env; |
||||||
|
} |
||||||
|
|
||||||
|
void WriteBatchHandlerJniCallback::Put(const Slice& key, const Slice& value) { |
||||||
|
getJniEnv()->CallVoidMethod( |
||||||
|
m_jWriteBatchHandler, |
||||||
|
m_jPutMethodId, |
||||||
|
sliceToJArray(key), |
||||||
|
sliceToJArray(value)); |
||||||
|
} |
||||||
|
|
||||||
|
void WriteBatchHandlerJniCallback::Merge(const Slice& key, const Slice& value) { |
||||||
|
getJniEnv()->CallVoidMethod( |
||||||
|
m_jWriteBatchHandler, |
||||||
|
m_jMergeMethodId, |
||||||
|
sliceToJArray(key), |
||||||
|
sliceToJArray(value)); |
||||||
|
} |
||||||
|
|
||||||
|
void WriteBatchHandlerJniCallback::Delete(const Slice& key) { |
||||||
|
getJniEnv()->CallVoidMethod( |
||||||
|
m_jWriteBatchHandler, |
||||||
|
m_jDeleteMethodId, |
||||||
|
sliceToJArray(key)); |
||||||
|
} |
||||||
|
|
||||||
|
void WriteBatchHandlerJniCallback::LogData(const Slice& blob) { |
||||||
|
getJniEnv()->CallVoidMethod( |
||||||
|
m_jWriteBatchHandler, |
||||||
|
m_jLogDataMethodId, |
||||||
|
sliceToJArray(blob)); |
||||||
|
} |
||||||
|
|
||||||
|
bool WriteBatchHandlerJniCallback::Continue() { |
||||||
|
jboolean jContinue = getJniEnv()->CallBooleanMethod( |
||||||
|
m_jWriteBatchHandler, |
||||||
|
m_jContinueMethodId); |
||||||
|
|
||||||
|
return static_cast<bool>(jContinue == JNI_TRUE); |
||||||
|
} |
||||||
|
|
||||||
|
jbyteArray WriteBatchHandlerJniCallback::sliceToJArray(const Slice& s) { |
||||||
|
jbyteArray ja = getJniEnv()->NewByteArray(s.size()); |
||||||
|
getJniEnv()->SetByteArrayRegion( |
||||||
|
ja, 0, s.size(), |
||||||
|
reinterpret_cast<const jbyte*>(s.data())); |
||||||
|
return ja; |
||||||
|
} |
||||||
|
|
||||||
|
WriteBatchHandlerJniCallback::~WriteBatchHandlerJniCallback() { |
||||||
|
JNIEnv* m_env = getJniEnv(); |
||||||
|
|
||||||
|
m_env->DeleteGlobalRef(m_jWriteBatchHandler); |
||||||
|
|
||||||
|
// Note: do not need to explicitly detach, as this function is effectively
|
||||||
|
// called from the Java class's disposeInternal method, and so already
|
||||||
|
// has an attached thread, getJniEnv above is just a no-op Attach to get
|
||||||
|
// the env jvm->DetachCurrentThread();
|
||||||
|
} |
||||||
|
} // namespace rocksdb
|
||||||
@ -0,0 +1,47 @@ |
|||||||
|
// Copyright (c) 2014, 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::WriteBatch::Handler.
|
||||||
|
|
||||||
|
#ifndef JAVA_ROCKSJNI_WRITEBATCHHANDLERJNICALLBACK_H_ |
||||||
|
#define JAVA_ROCKSJNI_WRITEBATCHHANDLERJNICALLBACK_H_ |
||||||
|
|
||||||
|
#include <jni.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++) |
||||||
|
* we then callback to the appropriate Java method |
||||||
|
* this enables Write Batch Handlers to be implemented in Java. |
||||||
|
*/ |
||||||
|
class WriteBatchHandlerJniCallback : public WriteBatch::Handler { |
||||||
|
public: |
||||||
|
WriteBatchHandlerJniCallback( |
||||||
|
JNIEnv* env, jobject jWriteBackHandler); |
||||||
|
~WriteBatchHandlerJniCallback(); |
||||||
|
void Put(const Slice& key, const Slice& value); |
||||||
|
void Merge(const Slice& key, const Slice& value); |
||||||
|
void Delete(const Slice& key); |
||||||
|
void LogData(const Slice& blob); |
||||||
|
bool Continue(); |
||||||
|
|
||||||
|
private: |
||||||
|
JavaVM* m_jvm; |
||||||
|
jobject m_jWriteBatchHandler; |
||||||
|
JNIEnv* getJniEnv() const; |
||||||
|
jbyteArray sliceToJArray(const Slice& s); |
||||||
|
jmethodID m_jPutMethodId; |
||||||
|
jmethodID m_jMergeMethodId; |
||||||
|
jmethodID m_jDeleteMethodId; |
||||||
|
jmethodID m_jLogDataMethodId; |
||||||
|
jmethodID m_jContinueMethodId; |
||||||
|
}; |
||||||
|
} // namespace rocksdb
|
||||||
|
|
||||||
|
#endif // JAVA_ROCKSJNI_WRITEBATCHHANDLERJNICALLBACK_H_
|
||||||
Loading…
Reference in new issue