mirror of https://github.com/facebook/rocksdb.git
131 lines
4.3 KiB
C++
131 lines
4.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 "bridge" between Java and C++ and enables
|
|
// calling C++ rocksdb::OptionsUtil methods from Java side.
|
|
|
|
#include <jni.h>
|
|
#include <string>
|
|
|
|
#include "include/org_rocksdb_OptionsUtil.h"
|
|
|
|
#include "rocksdb/db.h"
|
|
#include "rocksdb/env.h"
|
|
#include "rocksdb/utilities/options_util.h"
|
|
#include "rocksjni/portal.h"
|
|
|
|
void build_column_family_descriptor_list(
|
|
JNIEnv* env, jobject jcfds,
|
|
std::vector<rocksdb::ColumnFamilyDescriptor>& cf_descs) {
|
|
jmethodID add_mid = rocksdb::ListJni::getListAddMethodId(env);
|
|
if (add_mid == nullptr) {
|
|
// exception occurred accessing method
|
|
return;
|
|
}
|
|
|
|
// Column family descriptor
|
|
for (rocksdb::ColumnFamilyDescriptor& cfd : cf_descs) {
|
|
// Construct a ColumnFamilyDescriptor java object
|
|
jobject jcfd = rocksdb::ColumnFamilyDescriptorJni::construct(env, &cfd);
|
|
if (env->ExceptionCheck()) {
|
|
// exception occurred constructing object
|
|
if (jcfd != nullptr) {
|
|
env->DeleteLocalRef(jcfd);
|
|
}
|
|
return;
|
|
}
|
|
|
|
// Add the object to java list.
|
|
jboolean rs = env->CallBooleanMethod(jcfds, add_mid, jcfd);
|
|
if (env->ExceptionCheck() || rs == JNI_FALSE) {
|
|
// exception occurred calling method, or could not add
|
|
if (jcfd != nullptr) {
|
|
env->DeleteLocalRef(jcfd);
|
|
}
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
|
|
/*
|
|
* Class: org_rocksdb_OptionsUtil
|
|
* Method: loadLatestOptions
|
|
* Signature: (Ljava/lang/String;JLjava/util/List;Z)V
|
|
*/
|
|
void Java_org_rocksdb_OptionsUtil_loadLatestOptions(
|
|
JNIEnv* env, jclass /*jcls*/, jstring jdbpath, jlong jenv_handle,
|
|
jlong jdb_opts_handle, jobject jcfds, jboolean ignore_unknown_options) {
|
|
jboolean has_exception = JNI_FALSE;
|
|
auto db_path = rocksdb::JniUtil::copyStdString(env, jdbpath, &has_exception);
|
|
if (has_exception == JNI_TRUE) {
|
|
// exception occurred
|
|
return;
|
|
}
|
|
std::vector<rocksdb::ColumnFamilyDescriptor> cf_descs;
|
|
rocksdb::Status s = rocksdb::LoadLatestOptions(
|
|
db_path, reinterpret_cast<rocksdb::Env*>(jenv_handle),
|
|
reinterpret_cast<rocksdb::DBOptions*>(jdb_opts_handle), &cf_descs,
|
|
ignore_unknown_options);
|
|
if (!s.ok()) {
|
|
// error, raise an exception
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(env, s);
|
|
} else {
|
|
build_column_family_descriptor_list(env, jcfds, cf_descs);
|
|
}
|
|
}
|
|
|
|
/*
|
|
* Class: org_rocksdb_OptionsUtil
|
|
* Method: loadOptionsFromFile
|
|
* Signature: (Ljava/lang/String;JJLjava/util/List;Z)V
|
|
*/
|
|
void Java_org_rocksdb_OptionsUtil_loadOptionsFromFile(
|
|
JNIEnv* env, jclass /*jcls*/, jstring jopts_file_name, jlong jenv_handle,
|
|
jlong jdb_opts_handle, jobject jcfds, jboolean ignore_unknown_options) {
|
|
jboolean has_exception = JNI_FALSE;
|
|
auto opts_file_name = rocksdb::JniUtil::copyStdString(env, jopts_file_name, &has_exception);
|
|
if (has_exception == JNI_TRUE) {
|
|
// exception occurred
|
|
return;
|
|
}
|
|
std::vector<rocksdb::ColumnFamilyDescriptor> cf_descs;
|
|
rocksdb::Status s = rocksdb::LoadOptionsFromFile(
|
|
opts_file_name, reinterpret_cast<rocksdb::Env*>(jenv_handle),
|
|
reinterpret_cast<rocksdb::DBOptions*>(jdb_opts_handle), &cf_descs,
|
|
ignore_unknown_options);
|
|
if (!s.ok()) {
|
|
// error, raise an exception
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(env, s);
|
|
} else {
|
|
build_column_family_descriptor_list(env, jcfds, cf_descs);
|
|
}
|
|
}
|
|
|
|
/*
|
|
* Class: org_rocksdb_OptionsUtil
|
|
* Method: getLatestOptionsFileName
|
|
* Signature: (Ljava/lang/String;J)Ljava/lang/String;
|
|
*/
|
|
jstring Java_org_rocksdb_OptionsUtil_getLatestOptionsFileName(
|
|
JNIEnv* env, jclass /*jcls*/, jstring jdbpath, jlong jenv_handle) {
|
|
jboolean has_exception = JNI_FALSE;
|
|
auto db_path = rocksdb::JniUtil::copyStdString(env, jdbpath, &has_exception);
|
|
if (has_exception == JNI_TRUE) {
|
|
// exception occurred
|
|
return nullptr;
|
|
}
|
|
std::string options_file_name;
|
|
rocksdb::Status s = rocksdb::GetLatestOptionsFileName(
|
|
db_path, reinterpret_cast<rocksdb::Env*>(jenv_handle),
|
|
&options_file_name);
|
|
if (!s.ok()) {
|
|
// error, raise an exception
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(env, s);
|
|
return nullptr;
|
|
} else {
|
|
return env->NewStringUTF(options_file_name.c_str());
|
|
}
|
|
}
|