mirror of
https://github.com/facebook/rocksdb.git
synced 2024-11-25 22:44:05 +00:00
0bf9079d44
Summary: This should give us some performance benefit calling native C++ code. Closes https://github.com/facebook/rocksdb/issues/4786 See https://github.com/evolvedbinary/jni-benchmarks/ for more info. Pull Request resolved: https://github.com/facebook/rocksdb/pull/11882 Reviewed By: pdillinger Differential Revision: D53066207 Pulled By: ajkr fbshipit-source-id: daedef185215d0d8e791cd85bef598900bcb5bf2
117 lines
4 KiB
C++
117 lines
4 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 "bridge" between Java and C++ and enables
|
|
// calling C++ ROCKSDB_NAMESPACE::SstFileReader methods
|
|
// from Java side.
|
|
|
|
#include <jni.h>
|
|
|
|
#include <string>
|
|
|
|
#include "include/org_rocksdb_SstFileReader.h"
|
|
#include "rocksdb/comparator.h"
|
|
#include "rocksdb/env.h"
|
|
#include "rocksdb/options.h"
|
|
#include "rocksdb/sst_file_reader.h"
|
|
#include "rocksjni/cplusplus_to_java_convert.h"
|
|
#include "rocksjni/portal.h"
|
|
|
|
/*
|
|
* Class: org_rocksdb_SstFileReader
|
|
* Method: newSstFileReader
|
|
* Signature: (J)J
|
|
*/
|
|
jlong Java_org_rocksdb_SstFileReader_newSstFileReader(JNIEnv * /*env*/,
|
|
jclass /*jcls*/,
|
|
jlong joptions) {
|
|
auto *options =
|
|
reinterpret_cast<const ROCKSDB_NAMESPACE::Options *>(joptions);
|
|
ROCKSDB_NAMESPACE::SstFileReader *sst_file_reader =
|
|
new ROCKSDB_NAMESPACE::SstFileReader(*options);
|
|
return GET_CPLUSPLUS_POINTER(sst_file_reader);
|
|
}
|
|
|
|
/*
|
|
* Class: org_rocksdb_SstFileReader
|
|
* Method: open
|
|
* Signature: (JLjava/lang/String;)V
|
|
*/
|
|
void Java_org_rocksdb_SstFileReader_open(JNIEnv *env, jclass /*jcls*/,
|
|
jlong jhandle, jstring jfile_path) {
|
|
const char *file_path = env->GetStringUTFChars(jfile_path, nullptr);
|
|
if (file_path == nullptr) {
|
|
// exception thrown: OutOfMemoryError
|
|
return;
|
|
}
|
|
ROCKSDB_NAMESPACE::Status s =
|
|
reinterpret_cast<ROCKSDB_NAMESPACE::SstFileReader *>(jhandle)->Open(
|
|
file_path);
|
|
env->ReleaseStringUTFChars(jfile_path, file_path);
|
|
|
|
if (!s.ok()) {
|
|
ROCKSDB_NAMESPACE::RocksDBExceptionJni::ThrowNew(env, s);
|
|
}
|
|
}
|
|
|
|
/*
|
|
* Class: org_rocksdb_SstFileReader
|
|
* Method: newIterator
|
|
* Signature: (JJ)J
|
|
*/
|
|
jlong Java_org_rocksdb_SstFileReader_newIterator(JNIEnv * /*env*/,
|
|
jclass /*jcls*/, jlong jhandle,
|
|
jlong jread_options_handle) {
|
|
auto *sst_file_reader =
|
|
reinterpret_cast<ROCKSDB_NAMESPACE::SstFileReader *>(jhandle);
|
|
auto *read_options =
|
|
reinterpret_cast<ROCKSDB_NAMESPACE::ReadOptions *>(jread_options_handle);
|
|
return GET_CPLUSPLUS_POINTER(sst_file_reader->NewIterator(*read_options));
|
|
}
|
|
|
|
/*
|
|
* Class: org_rocksdb_SstFileReader
|
|
* Method: disposeInternal
|
|
* Signature: (J)V
|
|
*/
|
|
void Java_org_rocksdb_SstFileReader_disposeInternalJni(JNIEnv * /*env*/,
|
|
jclass /*jcls*/,
|
|
jlong jhandle) {
|
|
delete reinterpret_cast<ROCKSDB_NAMESPACE::SstFileReader *>(jhandle);
|
|
}
|
|
|
|
/*
|
|
* Class: org_rocksdb_SstFileReader
|
|
* Method: verifyChecksum
|
|
* Signature: (J)V
|
|
*/
|
|
void Java_org_rocksdb_SstFileReader_verifyChecksum(JNIEnv *env, jclass /*jcls*/,
|
|
jlong jhandle) {
|
|
auto *sst_file_reader =
|
|
reinterpret_cast<ROCKSDB_NAMESPACE::SstFileReader *>(jhandle);
|
|
auto s = sst_file_reader->VerifyChecksum();
|
|
if (!s.ok()) {
|
|
ROCKSDB_NAMESPACE::RocksDBExceptionJni::ThrowNew(env, s);
|
|
}
|
|
}
|
|
|
|
/*
|
|
* Class: org_rocksdb_SstFileReader
|
|
* Method: getTableProperties
|
|
* Signature: (J)J
|
|
*/
|
|
jobject Java_org_rocksdb_SstFileReader_getTableProperties(JNIEnv *env,
|
|
jclass /*jcls*/,
|
|
jlong jhandle) {
|
|
auto *sst_file_reader =
|
|
reinterpret_cast<ROCKSDB_NAMESPACE::SstFileReader *>(jhandle);
|
|
std::shared_ptr<const ROCKSDB_NAMESPACE::TableProperties> tp =
|
|
sst_file_reader->GetTableProperties();
|
|
jobject jtable_properties =
|
|
ROCKSDB_NAMESPACE::TablePropertiesJni::fromCppTableProperties(
|
|
env, *(tp.get()));
|
|
return jtable_properties;
|
|
}
|