mirror of
https://github.com/facebook/rocksdb.git
synced 2024-12-02 10:15:54 +00:00
8c4a3bfa5b
Summary: * [java] Add a java api for rocksdb::Options, currently only supports create_if_missing. * [java] Add a test for RocksDBException in RocksDBSample. Test Plan: make jtest Reviewers: haobo, sdong Reviewed By: haobo CC: leveldb, dhruba Differential Revision: https://reviews.facebook.net/D17385
114 lines
3.5 KiB
C++
114 lines
3.5 KiB
C++
// 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 is designed for caching those frequently used IDs and provide
|
|
// efficient portal (i.e, a set of static functions) to access java code
|
|
// from c++.
|
|
|
|
#ifndef JAVA_ROCKSJNI_PORTAL_H_
|
|
#define JAVA_ROCKSJNI_PORTAL_H_
|
|
|
|
#include <jni.h>
|
|
#include "rocksdb/db.h"
|
|
|
|
namespace rocksdb {
|
|
|
|
// The portal class for org.rocksdb.RocksDB
|
|
class RocksDBJni {
|
|
public:
|
|
// Get the java class id of org.rocksdb.RocksDB.
|
|
static jclass getJClass(JNIEnv* env) {
|
|
static jclass jclazz = env->FindClass("org/rocksdb/RocksDB");
|
|
assert(jclazz != nullptr);
|
|
return jclazz;
|
|
}
|
|
|
|
// Get the field id of the member variable of org.rocksdb.RocksDB
|
|
// that stores the pointer to rocksdb::DB.
|
|
static jfieldID getHandleFieldID(JNIEnv* env) {
|
|
static jfieldID fid = env->GetFieldID(
|
|
getJClass(env), "nativeHandle_", "J");
|
|
assert(fid != nullptr);
|
|
return fid;
|
|
}
|
|
|
|
// Get the pointer to rocksdb::DB of the specified org.rocksdb.RocksDB.
|
|
static rocksdb::DB* getHandle(JNIEnv* env, jobject jdb) {
|
|
return reinterpret_cast<rocksdb::DB*>(
|
|
env->GetLongField(jdb, getHandleFieldID(env)));
|
|
}
|
|
|
|
// Pass the rocksdb::DB pointer to the java side.
|
|
static void setHandle(JNIEnv* env, jobject jdb, rocksdb::DB* db) {
|
|
env->SetLongField(
|
|
jdb, getHandleFieldID(env),
|
|
reinterpret_cast<jlong>(db));
|
|
}
|
|
};
|
|
|
|
// The portal class for org.rocksdb.RocksDBException
|
|
class RocksDBExceptionJni {
|
|
public:
|
|
// Get the jclass of org.rocksdb.RocksDBException
|
|
static jclass getJClass(JNIEnv* env) {
|
|
static jclass jclazz = env->FindClass("org/rocksdb/RocksDBException");
|
|
assert(jclazz != nullptr);
|
|
return jclazz;
|
|
}
|
|
|
|
// Create and throw a java exception by converting the input
|
|
// Status to an RocksDBException.
|
|
//
|
|
// In case s.ok() is true, then this function will not throw any
|
|
// exception.
|
|
static void ThrowNew(JNIEnv* env, Status s) {
|
|
if (s.ok()) {
|
|
return;
|
|
}
|
|
jstring msg = env->NewStringUTF(s.ToString().c_str());
|
|
// get the constructor id of org.rocksdb.RocksDBException
|
|
static jmethodID mid = env->GetMethodID(
|
|
getJClass(env), "<init>", "(Ljava/lang/String;)V");
|
|
assert(mid != nullptr);
|
|
|
|
env->Throw((jthrowable)env->NewObject(getJClass(env), mid, msg));
|
|
}
|
|
};
|
|
|
|
class OptionsJni {
|
|
public:
|
|
// Get the java class id of org.rocksdb.Options.
|
|
static jclass getJClass(JNIEnv* env) {
|
|
static jclass jclazz = env->FindClass("org/rocksdb/Options");
|
|
assert(jclazz != nullptr);
|
|
return jclazz;
|
|
}
|
|
|
|
// Get the field id of the member variable of org.rocksdb.Options
|
|
// that stores the pointer to rocksdb::Options
|
|
static jfieldID getHandleFieldID(JNIEnv* env) {
|
|
static jfieldID fid = env->GetFieldID(
|
|
getJClass(env), "nativeHandle_", "J");
|
|
assert(fid != nullptr);
|
|
return fid;
|
|
}
|
|
|
|
// Get the pointer to rocksdb::Options
|
|
static rocksdb::Options* getHandle(JNIEnv* env, jobject jobj) {
|
|
return reinterpret_cast<rocksdb::Options*>(
|
|
env->GetLongField(jobj, getHandleFieldID(env)));
|
|
}
|
|
|
|
// Pass the rocksdb::Options pointer to the java side.
|
|
static void setHandle(JNIEnv* env, jobject jobj, rocksdb::Options* op) {
|
|
env->SetLongField(
|
|
jobj, getHandleFieldID(env),
|
|
reinterpret_cast<jlong>(op));
|
|
}
|
|
};
|
|
|
|
} // namespace rocksdb
|
|
#endif // JAVA_ROCKSJNI_PORTAL_H_
|