2016-02-10 00:12:00 +01:00
|
|
|
// Copyright (c) 2011-present, Facebook, Inc. All rights reserved.
|
2017-07-16 01:03:42 +02:00
|
|
|
// 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).
|
Add a jni library for rocksdb which supports Open, Get, Put, and Close.
Summary:
This diff contains a simple jni library for rocksdb which supports open, get,
put and closeusing default options (including Options, ReadOptions, and
WriteOptions.) In the usual case, Java developers can use the c++ rocksdb
library in the way similar to the following:
RocksDB db = RocksDB.open(path_to_db);
...
db.put("hello".getBytes(), "world".getBytes();
byte[] value = db.get("hello".getBytes());
...
db.close();
Specifically, this diff has the following major classes:
* RocksDB: a Java wrapper class which forwards the operations
from the java side to c++ rocksdb library.
* RocksDBException: ncapsulates the error of an operation.
This exception type is used to describe an internal error from
the c++ rocksdb library.
This diff also include a simple java sample code calling c++ rocksdb library.
To build the rocksdb jni library, simply run make jni, and make jtest will try to
build and run the sample code.
Note that if the rocksdb is not built with the default glibc that Java uses,
java will try to load the wrong glibc during the run time. As a result,
the sample code might not work properly during the run time.
Test Plan:
* make jni
* make jtest
Reviewers: haobo, dhruba, sdong, igor, ljin
Reviewed By: dhruba
CC: leveldb, xjin
Differential Revision: https://reviews.facebook.net/D17109
2014-03-28 22:19:21 +01:00
|
|
|
//
|
|
|
|
// This file implements the "bridge" between Java and C++ and enables
|
|
|
|
// calling c++ rocksdb::DB methods from Java side.
|
|
|
|
|
2015-01-30 21:05:45 +01:00
|
|
|
#include <jni.h>
|
Add a jni library for rocksdb which supports Open, Get, Put, and Close.
Summary:
This diff contains a simple jni library for rocksdb which supports open, get,
put and closeusing default options (including Options, ReadOptions, and
WriteOptions.) In the usual case, Java developers can use the c++ rocksdb
library in the way similar to the following:
RocksDB db = RocksDB.open(path_to_db);
...
db.put("hello".getBytes(), "world".getBytes();
byte[] value = db.get("hello".getBytes());
...
db.close();
Specifically, this diff has the following major classes:
* RocksDB: a Java wrapper class which forwards the operations
from the java side to c++ rocksdb library.
* RocksDBException: ncapsulates the error of an operation.
This exception type is used to describe an internal error from
the c++ rocksdb library.
This diff also include a simple java sample code calling c++ rocksdb library.
To build the rocksdb jni library, simply run make jni, and make jtest will try to
build and run the sample code.
Note that if the rocksdb is not built with the default glibc that Java uses,
java will try to load the wrong glibc during the run time. As a result,
the sample code might not work properly during the run time.
Test Plan:
* make jni
* make jtest
Reviewers: haobo, dhruba, sdong, igor, ljin
Reviewed By: dhruba
CC: leveldb, xjin
Differential Revision: https://reviews.facebook.net/D17109
2014-03-28 22:19:21 +01:00
|
|
|
#include <stdio.h>
|
|
|
|
#include <stdlib.h>
|
2016-12-16 20:17:26 +01:00
|
|
|
#include <algorithm>
|
|
|
|
#include <functional>
|
2015-01-25 22:05:29 +01:00
|
|
|
#include <memory>
|
Add a jni library for rocksdb which supports Open, Get, Put, and Close.
Summary:
This diff contains a simple jni library for rocksdb which supports open, get,
put and closeusing default options (including Options, ReadOptions, and
WriteOptions.) In the usual case, Java developers can use the c++ rocksdb
library in the way similar to the following:
RocksDB db = RocksDB.open(path_to_db);
...
db.put("hello".getBytes(), "world".getBytes();
byte[] value = db.get("hello".getBytes());
...
db.close();
Specifically, this diff has the following major classes:
* RocksDB: a Java wrapper class which forwards the operations
from the java side to c++ rocksdb library.
* RocksDBException: ncapsulates the error of an operation.
This exception type is used to describe an internal error from
the c++ rocksdb library.
This diff also include a simple java sample code calling c++ rocksdb library.
To build the rocksdb jni library, simply run make jni, and make jtest will try to
build and run the sample code.
Note that if the rocksdb is not built with the default glibc that Java uses,
java will try to load the wrong glibc during the run time. As a result,
the sample code might not work properly during the run time.
Test Plan:
* make jni
* make jtest
Reviewers: haobo, dhruba, sdong, igor, ljin
Reviewed By: dhruba
CC: leveldb, xjin
Differential Revision: https://reviews.facebook.net/D17109
2014-03-28 22:19:21 +01:00
|
|
|
#include <string>
|
2016-02-01 21:00:40 +01:00
|
|
|
#include <tuple>
|
2014-04-25 22:57:20 +02:00
|
|
|
#include <vector>
|
Add a jni library for rocksdb which supports Open, Get, Put, and Close.
Summary:
This diff contains a simple jni library for rocksdb which supports open, get,
put and closeusing default options (including Options, ReadOptions, and
WriteOptions.) In the usual case, Java developers can use the c++ rocksdb
library in the way similar to the following:
RocksDB db = RocksDB.open(path_to_db);
...
db.put("hello".getBytes(), "world".getBytes();
byte[] value = db.get("hello".getBytes());
...
db.close();
Specifically, this diff has the following major classes:
* RocksDB: a Java wrapper class which forwards the operations
from the java side to c++ rocksdb library.
* RocksDBException: ncapsulates the error of an operation.
This exception type is used to describe an internal error from
the c++ rocksdb library.
This diff also include a simple java sample code calling c++ rocksdb library.
To build the rocksdb jni library, simply run make jni, and make jtest will try to
build and run the sample code.
Note that if the rocksdb is not built with the default glibc that Java uses,
java will try to load the wrong glibc during the run time. As a result,
the sample code might not work properly during the run time.
Test Plan:
* make jni
* make jtest
Reviewers: haobo, dhruba, sdong, igor, ljin
Reviewed By: dhruba
CC: leveldb, xjin
Differential Revision: https://reviews.facebook.net/D17109
2014-03-28 22:19:21 +01:00
|
|
|
|
|
|
|
#include "include/org_rocksdb_RocksDB.h"
|
2014-04-14 22:42:36 +02:00
|
|
|
#include "rocksdb/cache.h"
|
2019-09-18 18:43:10 +02:00
|
|
|
#include "rocksdb/convenience.h"
|
2016-10-21 02:05:32 +02:00
|
|
|
#include "rocksdb/db.h"
|
|
|
|
#include "rocksdb/options.h"
|
2015-01-25 22:05:29 +01:00
|
|
|
#include "rocksdb/types.h"
|
2015-01-30 21:05:45 +01:00
|
|
|
#include "rocksjni/portal.h"
|
Add a jni library for rocksdb which supports Open, Get, Put, and Close.
Summary:
This diff contains a simple jni library for rocksdb which supports open, get,
put and closeusing default options (including Options, ReadOptions, and
WriteOptions.) In the usual case, Java developers can use the c++ rocksdb
library in the way similar to the following:
RocksDB db = RocksDB.open(path_to_db);
...
db.put("hello".getBytes(), "world".getBytes();
byte[] value = db.get("hello".getBytes());
...
db.close();
Specifically, this diff has the following major classes:
* RocksDB: a Java wrapper class which forwards the operations
from the java side to c++ rocksdb library.
* RocksDBException: ncapsulates the error of an operation.
This exception type is used to describe an internal error from
the c++ rocksdb library.
This diff also include a simple java sample code calling c++ rocksdb library.
To build the rocksdb jni library, simply run make jni, and make jtest will try to
build and run the sample code.
Note that if the rocksdb is not built with the default glibc that Java uses,
java will try to load the wrong glibc during the run time. As a result,
the sample code might not work properly during the run time.
Test Plan:
* make jni
* make jtest
Reviewers: haobo, dhruba, sdong, igor, ljin
Reviewed By: dhruba
CC: leveldb, xjin
Differential Revision: https://reviews.facebook.net/D17109
2014-03-28 22:19:21 +01:00
|
|
|
|
2016-01-28 15:44:31 +01:00
|
|
|
#ifdef min
|
|
|
|
#undef min
|
|
|
|
#endif
|
|
|
|
|
2018-04-13 02:55:14 +02:00
|
|
|
jlong rocksdb_open_helper(
|
|
|
|
JNIEnv* env, jlong jopt_handle, jstring jdb_path,
|
|
|
|
std::function<rocksdb::Status(const rocksdb::Options&, const std::string&,
|
|
|
|
rocksdb::DB**)>
|
|
|
|
open_fn) {
|
2017-02-28 01:26:12 +01:00
|
|
|
const char* db_path = env->GetStringUTFChars(jdb_path, nullptr);
|
2018-04-13 02:55:14 +02:00
|
|
|
if (db_path == nullptr) {
|
2017-02-28 01:26:12 +01:00
|
|
|
// exception thrown: OutOfMemoryError
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2016-01-20 18:05:41 +01:00
|
|
|
auto* opt = reinterpret_cast<rocksdb::Options*>(jopt_handle);
|
2014-04-14 22:42:36 +02:00
|
|
|
rocksdb::DB* db = nullptr;
|
2016-01-20 18:05:41 +01:00
|
|
|
rocksdb::Status s = open_fn(*opt, db_path, &db);
|
2017-02-28 01:26:12 +01:00
|
|
|
|
2014-04-02 01:59:05 +02:00
|
|
|
env->ReleaseStringUTFChars(jdb_path, db_path);
|
|
|
|
|
|
|
|
if (s.ok()) {
|
2016-01-20 18:05:41 +01:00
|
|
|
return reinterpret_cast<jlong>(db);
|
|
|
|
} else {
|
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(env, s);
|
|
|
|
return 0;
|
2014-04-02 01:59:05 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-10-13 10:34:52 +02:00
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
2016-01-20 18:05:41 +01:00
|
|
|
* Method: open
|
|
|
|
* Signature: (JLjava/lang/String;)J
|
2014-10-13 10:34:52 +02:00
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
jlong Java_org_rocksdb_RocksDB_open__JLjava_lang_String_2(
|
|
|
|
JNIEnv* env, jclass, jlong jopt_handle, jstring jdb_path) {
|
2018-04-13 02:55:14 +02:00
|
|
|
return rocksdb_open_helper(
|
|
|
|
env, jopt_handle, jdb_path,
|
|
|
|
(rocksdb::Status(*)(const rocksdb::Options&, const std::string&,
|
|
|
|
rocksdb::DB**)) &
|
|
|
|
rocksdb::DB::Open);
|
2014-10-13 10:34:52 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
|
|
|
* Method: openROnly
|
2016-01-20 18:05:41 +01:00
|
|
|
* Signature: (JLjava/lang/String;)J
|
2014-10-13 10:34:52 +02:00
|
|
|
*/
|
2016-01-20 18:05:41 +01:00
|
|
|
jlong Java_org_rocksdb_RocksDB_openROnly__JLjava_lang_String_2(
|
2019-02-22 23:36:38 +01:00
|
|
|
JNIEnv* env, jclass, jlong jopt_handle, jstring jdb_path) {
|
2018-04-13 02:55:14 +02:00
|
|
|
return rocksdb_open_helper(env, jopt_handle, jdb_path,
|
|
|
|
[](const rocksdb::Options& options,
|
|
|
|
const std::string& db_path, rocksdb::DB** db) {
|
|
|
|
return rocksdb::DB::OpenForReadOnly(options,
|
|
|
|
db_path, db);
|
|
|
|
});
|
2016-01-20 18:05:41 +01:00
|
|
|
}
|
2014-10-13 10:34:52 +02:00
|
|
|
|
2018-04-13 02:55:14 +02:00
|
|
|
jlongArray rocksdb_open_helper(
|
|
|
|
JNIEnv* env, jlong jopt_handle, jstring jdb_path,
|
|
|
|
jobjectArray jcolumn_names, jlongArray jcolumn_options,
|
2016-01-20 18:05:41 +01:00
|
|
|
std::function<rocksdb::Status(
|
2018-04-13 02:55:14 +02:00
|
|
|
const rocksdb::DBOptions&, const std::string&,
|
|
|
|
const std::vector<rocksdb::ColumnFamilyDescriptor>&,
|
|
|
|
std::vector<rocksdb::ColumnFamilyHandle*>*, rocksdb::DB**)>
|
|
|
|
open_fn) {
|
2017-02-28 01:26:12 +01:00
|
|
|
const char* db_path = env->GetStringUTFChars(jdb_path, nullptr);
|
2018-04-13 02:55:14 +02:00
|
|
|
if (db_path == nullptr) {
|
2017-02-28 01:26:12 +01:00
|
|
|
// exception thrown: OutOfMemoryError
|
|
|
|
return nullptr;
|
|
|
|
}
|
2014-10-13 10:34:52 +02:00
|
|
|
|
2017-02-28 01:26:12 +01:00
|
|
|
const jsize len_cols = env->GetArrayLength(jcolumn_names);
|
|
|
|
jlong* jco = env->GetLongArrayElements(jcolumn_options, nullptr);
|
2018-04-13 02:55:14 +02:00
|
|
|
if (jco == nullptr) {
|
2017-02-28 01:26:12 +01:00
|
|
|
// exception thrown: OutOfMemoryError
|
|
|
|
env->ReleaseStringUTFChars(jdb_path, db_path);
|
|
|
|
return nullptr;
|
|
|
|
}
|
2014-10-31 23:39:14 +01:00
|
|
|
|
2017-02-28 01:26:12 +01:00
|
|
|
std::vector<rocksdb::ColumnFamilyDescriptor> column_families;
|
|
|
|
jboolean has_exception = JNI_FALSE;
|
|
|
|
rocksdb::JniUtil::byteStrings<std::string>(
|
2018-04-13 02:55:14 +02:00
|
|
|
env, jcolumn_names,
|
|
|
|
[](const char* str_data, const size_t str_len) {
|
|
|
|
return std::string(str_data, str_len);
|
|
|
|
},
|
|
|
|
[&jco, &column_families](size_t idx, std::string cf_name) {
|
|
|
|
rocksdb::ColumnFamilyOptions* cf_options =
|
|
|
|
reinterpret_cast<rocksdb::ColumnFamilyOptions*>(jco[idx]);
|
|
|
|
column_families.push_back(
|
|
|
|
rocksdb::ColumnFamilyDescriptor(cf_name, *cf_options));
|
|
|
|
},
|
|
|
|
&has_exception);
|
2014-10-13 10:34:52 +02:00
|
|
|
|
2017-02-28 01:26:12 +01:00
|
|
|
env->ReleaseLongArrayElements(jcolumn_options, jco, JNI_ABORT);
|
2014-10-13 10:34:52 +02:00
|
|
|
|
2018-04-13 02:55:14 +02:00
|
|
|
if (has_exception == JNI_TRUE) {
|
2017-06-05 20:23:31 +02:00
|
|
|
// exception occurred
|
2017-02-28 01:26:12 +01:00
|
|
|
env->ReleaseStringUTFChars(jdb_path, db_path);
|
|
|
|
return nullptr;
|
2014-10-13 10:34:52 +02:00
|
|
|
}
|
2016-01-20 18:05:41 +01:00
|
|
|
|
2017-02-28 01:26:12 +01:00
|
|
|
auto* opt = reinterpret_cast<rocksdb::DBOptions*>(jopt_handle);
|
2019-02-22 23:36:38 +01:00
|
|
|
std::vector<rocksdb::ColumnFamilyHandle*> cf_handles;
|
2016-01-20 18:05:41 +01:00
|
|
|
rocksdb::DB* db = nullptr;
|
2019-02-22 23:36:38 +01:00
|
|
|
rocksdb::Status s = open_fn(*opt, db_path, column_families, &cf_handles, &db);
|
2014-10-13 10:34:52 +02:00
|
|
|
|
2017-02-28 01:26:12 +01:00
|
|
|
// we have now finished with db_path
|
|
|
|
env->ReleaseStringUTFChars(jdb_path, db_path);
|
|
|
|
|
2014-10-13 10:34:52 +02:00
|
|
|
// check if open operation was successful
|
2019-02-22 23:36:38 +01:00
|
|
|
if (!s.ok()) {
|
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(env, s);
|
|
|
|
return nullptr;
|
|
|
|
}
|
2014-10-13 10:34:52 +02:00
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
const jsize resultsLen = 1 + len_cols; // db handle + column family handles
|
|
|
|
std::unique_ptr<jlong[]> results =
|
|
|
|
std::unique_ptr<jlong[]>(new jlong[resultsLen]);
|
|
|
|
results[0] = reinterpret_cast<jlong>(db);
|
|
|
|
for (int i = 1; i <= len_cols; i++) {
|
|
|
|
results[i] = reinterpret_cast<jlong>(cf_handles[i - 1]);
|
|
|
|
}
|
2017-02-28 01:26:12 +01:00
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
jlongArray jresults = env->NewLongArray(resultsLen);
|
|
|
|
if (jresults == nullptr) {
|
|
|
|
// exception thrown: OutOfMemoryError
|
|
|
|
return nullptr;
|
|
|
|
}
|
2017-02-28 01:26:12 +01:00
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
env->SetLongArrayRegion(jresults, 0, resultsLen, results.get());
|
|
|
|
if (env->ExceptionCheck()) {
|
|
|
|
// exception thrown: ArrayIndexOutOfBoundsException
|
|
|
|
env->DeleteLocalRef(jresults);
|
2017-02-28 01:26:12 +01:00
|
|
|
return nullptr;
|
2014-10-13 10:34:52 +02:00
|
|
|
}
|
2019-02-22 23:36:38 +01:00
|
|
|
|
|
|
|
return jresults;
|
2014-10-13 10:34:52 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
2016-01-20 18:05:41 +01:00
|
|
|
* Method: openROnly
|
|
|
|
* Signature: (JLjava/lang/String;[[B[J)[J
|
2014-10-13 10:34:52 +02:00
|
|
|
*/
|
2016-01-20 18:05:41 +01:00
|
|
|
jlongArray Java_org_rocksdb_RocksDB_openROnly__JLjava_lang_String_2_3_3B_3J(
|
2019-02-22 23:36:38 +01:00
|
|
|
JNIEnv* env, jclass, jlong jopt_handle, jstring jdb_path,
|
2016-01-20 18:05:41 +01:00
|
|
|
jobjectArray jcolumn_names, jlongArray jcolumn_options) {
|
2018-04-13 02:55:14 +02:00
|
|
|
return rocksdb_open_helper(
|
|
|
|
env, jopt_handle, jdb_path, jcolumn_names, jcolumn_options,
|
|
|
|
[](const rocksdb::DBOptions& options, const std::string& db_path,
|
|
|
|
const std::vector<rocksdb::ColumnFamilyDescriptor>& column_families,
|
|
|
|
std::vector<rocksdb::ColumnFamilyHandle*>* handles, rocksdb::DB** db) {
|
|
|
|
return rocksdb::DB::OpenForReadOnly(options, db_path, column_families,
|
|
|
|
handles, db);
|
|
|
|
});
|
2016-01-20 18:05:41 +01:00
|
|
|
}
|
2014-10-13 10:34:52 +02:00
|
|
|
|
2016-01-20 18:05:41 +01:00
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
|
|
|
* Method: open
|
|
|
|
* Signature: (JLjava/lang/String;[[B[J)[J
|
|
|
|
*/
|
|
|
|
jlongArray Java_org_rocksdb_RocksDB_open__JLjava_lang_String_2_3_3B_3J(
|
2019-02-22 23:36:38 +01:00
|
|
|
JNIEnv* env, jclass, jlong jopt_handle, jstring jdb_path,
|
2016-01-20 18:05:41 +01:00
|
|
|
jobjectArray jcolumn_names, jlongArray jcolumn_options) {
|
2018-04-13 02:55:14 +02:00
|
|
|
return rocksdb_open_helper(
|
|
|
|
env, jopt_handle, jdb_path, jcolumn_names, jcolumn_options,
|
|
|
|
(rocksdb::Status(*)(const rocksdb::DBOptions&, const std::string&,
|
|
|
|
const std::vector<rocksdb::ColumnFamilyDescriptor>&,
|
|
|
|
std::vector<rocksdb::ColumnFamilyHandle*>*,
|
|
|
|
rocksdb::DB**)) &
|
|
|
|
rocksdb::DB::Open);
|
2014-10-13 10:34:52 +02:00
|
|
|
}
|
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
|
|
|
* Method: disposeInternal
|
|
|
|
* Signature: (J)V
|
|
|
|
*/
|
|
|
|
void Java_org_rocksdb_RocksDB_disposeInternal(
|
|
|
|
JNIEnv*, jobject, jlong jhandle) {
|
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jhandle);
|
|
|
|
assert(db != nullptr);
|
|
|
|
delete db;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
|
|
|
* Method: closeDatabase
|
|
|
|
* Signature: (J)V
|
|
|
|
*/
|
|
|
|
void Java_org_rocksdb_RocksDB_closeDatabase(
|
|
|
|
JNIEnv* env, jclass, jlong jhandle) {
|
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jhandle);
|
|
|
|
assert(db != nullptr);
|
|
|
|
rocksdb::Status s = db->Close();
|
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(env, s);
|
|
|
|
}
|
2014-10-13 10:34:52 +02:00
|
|
|
|
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
|
|
|
* Method: listColumnFamilies
|
2016-02-03 18:47:38 +01:00
|
|
|
* Signature: (JLjava/lang/String;)[[B
|
2014-10-13 10:34:52 +02:00
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
jobjectArray Java_org_rocksdb_RocksDB_listColumnFamilies(
|
|
|
|
JNIEnv* env, jclass, jlong jopt_handle, jstring jdb_path) {
|
2014-10-13 10:34:52 +02:00
|
|
|
std::vector<std::string> column_family_names;
|
2017-02-28 01:26:12 +01:00
|
|
|
const char* db_path = env->GetStringUTFChars(jdb_path, nullptr);
|
2018-04-13 02:55:14 +02:00
|
|
|
if (db_path == nullptr) {
|
2017-02-28 01:26:12 +01:00
|
|
|
// exception thrown: OutOfMemoryError
|
|
|
|
return nullptr;
|
|
|
|
}
|
|
|
|
|
2016-02-03 18:47:38 +01:00
|
|
|
auto* opt = reinterpret_cast<rocksdb::Options*>(jopt_handle);
|
2018-04-13 02:55:14 +02:00
|
|
|
rocksdb::Status s =
|
|
|
|
rocksdb::DB::ListColumnFamilies(*opt, db_path, &column_family_names);
|
2017-02-28 01:26:12 +01:00
|
|
|
|
2014-10-13 10:34:52 +02:00
|
|
|
env->ReleaseStringUTFChars(jdb_path, db_path);
|
|
|
|
|
2017-02-28 01:26:12 +01:00
|
|
|
jobjectArray jcolumn_family_names =
|
|
|
|
rocksdb::JniUtil::stringsBytes(env, column_family_names);
|
|
|
|
|
|
|
|
return jcolumn_family_names;
|
2014-10-13 10:34:52 +02:00
|
|
|
}
|
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
|
|
|
* Method: createColumnFamily
|
|
|
|
* Signature: (J[BIJ)J
|
|
|
|
*/
|
|
|
|
jlong Java_org_rocksdb_RocksDB_createColumnFamily(
|
|
|
|
JNIEnv* env, jobject, jlong jhandle, jbyteArray jcf_name,
|
|
|
|
jint jcf_name_len, jlong jcf_options_handle) {
|
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jhandle);
|
|
|
|
jboolean has_exception = JNI_FALSE;
|
|
|
|
const std::string cf_name =
|
|
|
|
rocksdb::JniUtil::byteString<std::string>(env, jcf_name, jcf_name_len,
|
|
|
|
[](const char* str, const size_t len) {
|
|
|
|
return std::string(str, len);
|
|
|
|
}, &has_exception);
|
|
|
|
if (has_exception == JNI_TRUE) {
|
|
|
|
// exception occurred
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
auto* cf_options =
|
|
|
|
reinterpret_cast<rocksdb::ColumnFamilyOptions*>(jcf_options_handle);
|
|
|
|
rocksdb::ColumnFamilyHandle *cf_handle;
|
|
|
|
rocksdb::Status s = db->CreateColumnFamily(*cf_options, cf_name, &cf_handle);
|
|
|
|
if (!s.ok()) {
|
|
|
|
// error occurred
|
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(env, s);
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
return reinterpret_cast<jlong>(cf_handle);
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
|
|
|
* Method: createColumnFamilies
|
|
|
|
* Signature: (JJ[[B)[J
|
|
|
|
*/
|
|
|
|
jlongArray Java_org_rocksdb_RocksDB_createColumnFamilies__JJ_3_3B(
|
|
|
|
JNIEnv* env, jobject, jlong jhandle, jlong jcf_options_handle,
|
|
|
|
jobjectArray jcf_names) {
|
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jhandle);
|
|
|
|
auto* cf_options =
|
|
|
|
reinterpret_cast<rocksdb::ColumnFamilyOptions*>(jcf_options_handle);
|
|
|
|
jboolean has_exception = JNI_FALSE;
|
|
|
|
std::vector<std::string> cf_names;
|
|
|
|
rocksdb::JniUtil::byteStrings<std::string>(env, jcf_names,
|
|
|
|
[](const char* str, const size_t len) {
|
|
|
|
return std::string(str, len);
|
|
|
|
},
|
|
|
|
[&cf_names](const size_t, std::string str) {
|
|
|
|
cf_names.push_back(str);
|
|
|
|
},
|
|
|
|
&has_exception);
|
|
|
|
if (has_exception == JNI_TRUE) {
|
|
|
|
// exception occurred
|
|
|
|
return nullptr;
|
|
|
|
}
|
|
|
|
|
|
|
|
std::vector<rocksdb::ColumnFamilyHandle*> cf_handles;
|
|
|
|
rocksdb::Status s = db->CreateColumnFamilies(*cf_options, cf_names, &cf_handles);
|
|
|
|
if (!s.ok()) {
|
|
|
|
// error occurred
|
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(env, s);
|
|
|
|
return nullptr;
|
|
|
|
}
|
|
|
|
|
|
|
|
jlongArray jcf_handles = rocksdb::JniUtil::toJPointers<rocksdb::ColumnFamilyHandle>(
|
|
|
|
env, cf_handles, &has_exception);
|
|
|
|
if (has_exception == JNI_TRUE) {
|
|
|
|
// exception occurred
|
|
|
|
return nullptr;
|
|
|
|
}
|
|
|
|
return jcf_handles;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
|
|
|
* Method: createColumnFamilies
|
|
|
|
* Signature: (J[J[[B)[J
|
|
|
|
*/
|
|
|
|
jlongArray Java_org_rocksdb_RocksDB_createColumnFamilies__J_3J_3_3B(
|
|
|
|
JNIEnv* env, jobject, jlong jhandle, jlongArray jcf_options_handles,
|
|
|
|
jobjectArray jcf_names) {
|
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jhandle);
|
|
|
|
const jsize jlen = env->GetArrayLength(jcf_options_handles);
|
|
|
|
std::vector<rocksdb::ColumnFamilyDescriptor> cf_descriptors;
|
|
|
|
cf_descriptors.reserve(jlen);
|
|
|
|
|
|
|
|
jboolean jcf_options_handles_is_copy = JNI_FALSE;
|
|
|
|
jlong *jcf_options_handles_elems = env->GetLongArrayElements(jcf_options_handles, &jcf_options_handles_is_copy);
|
|
|
|
if(jcf_options_handles_elems == nullptr) {
|
|
|
|
// exception thrown: OutOfMemoryError
|
|
|
|
return nullptr;
|
|
|
|
}
|
|
|
|
|
|
|
|
// extract the column family descriptors
|
|
|
|
jboolean has_exception = JNI_FALSE;
|
|
|
|
for (jsize i = 0; i < jlen; i++) {
|
|
|
|
auto* cf_options = reinterpret_cast<rocksdb::ColumnFamilyOptions*>(
|
|
|
|
jcf_options_handles_elems[i]);
|
|
|
|
jbyteArray jcf_name = static_cast<jbyteArray>(
|
|
|
|
env->GetObjectArrayElement(jcf_names, i));
|
|
|
|
if (env->ExceptionCheck()) {
|
|
|
|
// exception thrown: ArrayIndexOutOfBoundsException
|
|
|
|
env->ReleaseLongArrayElements(jcf_options_handles, jcf_options_handles_elems, JNI_ABORT);
|
|
|
|
return nullptr;
|
|
|
|
}
|
|
|
|
const std::string cf_name =
|
|
|
|
rocksdb::JniUtil::byteString<std::string>(env, jcf_name,
|
|
|
|
[](const char* str, const size_t len) {
|
|
|
|
return std::string(str, len);
|
|
|
|
},
|
|
|
|
&has_exception);
|
|
|
|
if (has_exception == JNI_TRUE) {
|
|
|
|
// exception occurred
|
|
|
|
env->DeleteLocalRef(jcf_name);
|
|
|
|
env->ReleaseLongArrayElements(jcf_options_handles, jcf_options_handles_elems, JNI_ABORT);
|
|
|
|
return nullptr;
|
|
|
|
}
|
|
|
|
|
|
|
|
cf_descriptors.push_back(rocksdb::ColumnFamilyDescriptor(cf_name, *cf_options));
|
|
|
|
|
|
|
|
env->DeleteLocalRef(jcf_name);
|
|
|
|
}
|
|
|
|
|
|
|
|
std::vector<rocksdb::ColumnFamilyHandle*> cf_handles;
|
|
|
|
rocksdb::Status s = db->CreateColumnFamilies(cf_descriptors, &cf_handles);
|
|
|
|
|
|
|
|
env->ReleaseLongArrayElements(jcf_options_handles, jcf_options_handles_elems, JNI_ABORT);
|
|
|
|
|
|
|
|
if (!s.ok()) {
|
|
|
|
// error occurred
|
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(env, s);
|
|
|
|
return nullptr;
|
|
|
|
}
|
|
|
|
|
|
|
|
jlongArray jcf_handles = rocksdb::JniUtil::toJPointers<rocksdb::ColumnFamilyHandle>(
|
|
|
|
env, cf_handles, &has_exception);
|
|
|
|
if (has_exception == JNI_TRUE) {
|
|
|
|
// exception occurred
|
|
|
|
return nullptr;
|
|
|
|
}
|
|
|
|
return jcf_handles;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
|
|
|
* Method: dropColumnFamily
|
|
|
|
* Signature: (JJ)V;
|
|
|
|
*/
|
|
|
|
void Java_org_rocksdb_RocksDB_dropColumnFamily(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle,
|
|
|
|
jlong jcf_handle) {
|
|
|
|
auto* db_handle = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
|
|
|
auto* cf_handle = reinterpret_cast<rocksdb::ColumnFamilyHandle*>(jcf_handle);
|
|
|
|
rocksdb::Status s = db_handle->DropColumnFamily(cf_handle);
|
|
|
|
if (!s.ok()) {
|
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(env, s);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
|
|
|
* Method: dropColumnFamilies
|
|
|
|
* Signature: (J[J)V
|
|
|
|
*/
|
|
|
|
void Java_org_rocksdb_RocksDB_dropColumnFamilies(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle,
|
|
|
|
jlongArray jcolumn_family_handles) {
|
|
|
|
auto* db_handle = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
|
|
|
|
|
|
|
std::vector<rocksdb::ColumnFamilyHandle*> cf_handles;
|
|
|
|
if (jcolumn_family_handles != nullptr) {
|
|
|
|
const jsize len_cols = env->GetArrayLength(jcolumn_family_handles);
|
|
|
|
|
|
|
|
jlong* jcfh = env->GetLongArrayElements(jcolumn_family_handles, nullptr);
|
|
|
|
if (jcfh == nullptr) {
|
|
|
|
// exception thrown: OutOfMemoryError
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
for (jsize i = 0; i < len_cols; i++) {
|
|
|
|
auto* cf_handle = reinterpret_cast<rocksdb::ColumnFamilyHandle*>(jcfh[i]);
|
|
|
|
cf_handles.push_back(cf_handle);
|
|
|
|
}
|
|
|
|
env->ReleaseLongArrayElements(jcolumn_family_handles, jcfh, JNI_ABORT);
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status s = db_handle->DropColumnFamilies(cf_handles);
|
|
|
|
if (!s.ok()) {
|
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(env, s);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-04-02 22:14:55 +02:00
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
// rocksdb::DB::Put
|
|
|
|
|
2017-02-28 01:26:12 +01:00
|
|
|
/**
|
|
|
|
* @return true if the put succeeded, false if a Java Exception was thrown
|
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
bool rocksdb_put_helper(
|
|
|
|
JNIEnv* env, rocksdb::DB* db,
|
|
|
|
const rocksdb::WriteOptions& write_options,
|
|
|
|
rocksdb::ColumnFamilyHandle* cf_handle, jbyteArray jkey,
|
|
|
|
jint jkey_off, jint jkey_len, jbyteArray jval,
|
|
|
|
jint jval_off, jint jval_len) {
|
2016-09-12 20:51:08 +02:00
|
|
|
jbyte* key = new jbyte[jkey_len];
|
|
|
|
env->GetByteArrayRegion(jkey, jkey_off, jkey_len, key);
|
2018-04-13 02:55:14 +02:00
|
|
|
if (env->ExceptionCheck()) {
|
2017-02-28 01:26:12 +01:00
|
|
|
// exception thrown: ArrayIndexOutOfBoundsException
|
2018-04-13 02:55:14 +02:00
|
|
|
delete[] key;
|
2017-02-28 01:26:12 +01:00
|
|
|
return false;
|
|
|
|
}
|
2016-09-12 20:51:08 +02:00
|
|
|
|
2016-09-14 22:12:55 +02:00
|
|
|
jbyte* value = new jbyte[jval_len];
|
|
|
|
env->GetByteArrayRegion(jval, jval_off, jval_len, value);
|
2018-04-13 02:55:14 +02:00
|
|
|
if (env->ExceptionCheck()) {
|
2017-02-28 01:26:12 +01:00
|
|
|
// exception thrown: ArrayIndexOutOfBoundsException
|
2018-04-13 02:55:14 +02:00
|
|
|
delete[] value;
|
|
|
|
delete[] key;
|
2017-02-28 01:26:12 +01:00
|
|
|
return false;
|
|
|
|
}
|
Add a jni library for rocksdb which supports Open, Get, Put, and Close.
Summary:
This diff contains a simple jni library for rocksdb which supports open, get,
put and closeusing default options (including Options, ReadOptions, and
WriteOptions.) In the usual case, Java developers can use the c++ rocksdb
library in the way similar to the following:
RocksDB db = RocksDB.open(path_to_db);
...
db.put("hello".getBytes(), "world".getBytes();
byte[] value = db.get("hello".getBytes());
...
db.close();
Specifically, this diff has the following major classes:
* RocksDB: a Java wrapper class which forwards the operations
from the java side to c++ rocksdb library.
* RocksDBException: ncapsulates the error of an operation.
This exception type is used to describe an internal error from
the c++ rocksdb library.
This diff also include a simple java sample code calling c++ rocksdb library.
To build the rocksdb jni library, simply run make jni, and make jtest will try to
build and run the sample code.
Note that if the rocksdb is not built with the default glibc that Java uses,
java will try to load the wrong glibc during the run time. As a result,
the sample code might not work properly during the run time.
Test Plan:
* make jni
* make jtest
Reviewers: haobo, dhruba, sdong, igor, ljin
Reviewed By: dhruba
CC: leveldb, xjin
Differential Revision: https://reviews.facebook.net/D17109
2014-03-28 22:19:21 +01:00
|
|
|
|
2014-04-02 22:14:55 +02:00
|
|
|
rocksdb::Slice key_slice(reinterpret_cast<char*>(key), jkey_len);
|
2016-09-14 22:12:55 +02:00
|
|
|
rocksdb::Slice value_slice(reinterpret_cast<char*>(value), jval_len);
|
Add a jni library for rocksdb which supports Open, Get, Put, and Close.
Summary:
This diff contains a simple jni library for rocksdb which supports open, get,
put and closeusing default options (including Options, ReadOptions, and
WriteOptions.) In the usual case, Java developers can use the c++ rocksdb
library in the way similar to the following:
RocksDB db = RocksDB.open(path_to_db);
...
db.put("hello".getBytes(), "world".getBytes();
byte[] value = db.get("hello".getBytes());
...
db.close();
Specifically, this diff has the following major classes:
* RocksDB: a Java wrapper class which forwards the operations
from the java side to c++ rocksdb library.
* RocksDBException: ncapsulates the error of an operation.
This exception type is used to describe an internal error from
the c++ rocksdb library.
This diff also include a simple java sample code calling c++ rocksdb library.
To build the rocksdb jni library, simply run make jni, and make jtest will try to
build and run the sample code.
Note that if the rocksdb is not built with the default glibc that Java uses,
java will try to load the wrong glibc during the run time. As a result,
the sample code might not work properly during the run time.
Test Plan:
* make jni
* make jtest
Reviewers: haobo, dhruba, sdong, igor, ljin
Reviewed By: dhruba
CC: leveldb, xjin
Differential Revision: https://reviews.facebook.net/D17109
2014-03-28 22:19:21 +01:00
|
|
|
|
2014-10-13 10:34:52 +02:00
|
|
|
rocksdb::Status s;
|
|
|
|
if (cf_handle != nullptr) {
|
|
|
|
s = db->Put(write_options, cf_handle, key_slice, value_slice);
|
|
|
|
} else {
|
|
|
|
// backwards compatibility
|
|
|
|
s = db->Put(write_options, key_slice, value_slice);
|
|
|
|
}
|
Add a jni library for rocksdb which supports Open, Get, Put, and Close.
Summary:
This diff contains a simple jni library for rocksdb which supports open, get,
put and closeusing default options (including Options, ReadOptions, and
WriteOptions.) In the usual case, Java developers can use the c++ rocksdb
library in the way similar to the following:
RocksDB db = RocksDB.open(path_to_db);
...
db.put("hello".getBytes(), "world".getBytes();
byte[] value = db.get("hello".getBytes());
...
db.close();
Specifically, this diff has the following major classes:
* RocksDB: a Java wrapper class which forwards the operations
from the java side to c++ rocksdb library.
* RocksDBException: ncapsulates the error of an operation.
This exception type is used to describe an internal error from
the c++ rocksdb library.
This diff also include a simple java sample code calling c++ rocksdb library.
To build the rocksdb jni library, simply run make jni, and make jtest will try to
build and run the sample code.
Note that if the rocksdb is not built with the default glibc that Java uses,
java will try to load the wrong glibc during the run time. As a result,
the sample code might not work properly during the run time.
Test Plan:
* make jni
* make jtest
Reviewers: haobo, dhruba, sdong, igor, ljin
Reviewed By: dhruba
CC: leveldb, xjin
Differential Revision: https://reviews.facebook.net/D17109
2014-03-28 22:19:21 +01:00
|
|
|
|
2016-09-12 20:51:08 +02:00
|
|
|
// cleanup
|
2018-04-13 02:55:14 +02:00
|
|
|
delete[] value;
|
|
|
|
delete[] key;
|
Add a jni library for rocksdb which supports Open, Get, Put, and Close.
Summary:
This diff contains a simple jni library for rocksdb which supports open, get,
put and closeusing default options (including Options, ReadOptions, and
WriteOptions.) In the usual case, Java developers can use the c++ rocksdb
library in the way similar to the following:
RocksDB db = RocksDB.open(path_to_db);
...
db.put("hello".getBytes(), "world".getBytes();
byte[] value = db.get("hello".getBytes());
...
db.close();
Specifically, this diff has the following major classes:
* RocksDB: a Java wrapper class which forwards the operations
from the java side to c++ rocksdb library.
* RocksDBException: ncapsulates the error of an operation.
This exception type is used to describe an internal error from
the c++ rocksdb library.
This diff also include a simple java sample code calling c++ rocksdb library.
To build the rocksdb jni library, simply run make jni, and make jtest will try to
build and run the sample code.
Note that if the rocksdb is not built with the default glibc that Java uses,
java will try to load the wrong glibc during the run time. As a result,
the sample code might not work properly during the run time.
Test Plan:
* make jni
* make jtest
Reviewers: haobo, dhruba, sdong, igor, ljin
Reviewed By: dhruba
CC: leveldb, xjin
Differential Revision: https://reviews.facebook.net/D17109
2014-03-28 22:19:21 +01:00
|
|
|
|
|
|
|
if (s.ok()) {
|
2017-02-28 01:26:12 +01:00
|
|
|
return true;
|
|
|
|
} else {
|
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(env, s);
|
|
|
|
return false;
|
Add a jni library for rocksdb which supports Open, Get, Put, and Close.
Summary:
This diff contains a simple jni library for rocksdb which supports open, get,
put and closeusing default options (including Options, ReadOptions, and
WriteOptions.) In the usual case, Java developers can use the c++ rocksdb
library in the way similar to the following:
RocksDB db = RocksDB.open(path_to_db);
...
db.put("hello".getBytes(), "world".getBytes();
byte[] value = db.get("hello".getBytes());
...
db.close();
Specifically, this diff has the following major classes:
* RocksDB: a Java wrapper class which forwards the operations
from the java side to c++ rocksdb library.
* RocksDBException: ncapsulates the error of an operation.
This exception type is used to describe an internal error from
the c++ rocksdb library.
This diff also include a simple java sample code calling c++ rocksdb library.
To build the rocksdb jni library, simply run make jni, and make jtest will try to
build and run the sample code.
Note that if the rocksdb is not built with the default glibc that Java uses,
java will try to load the wrong glibc during the run time. As a result,
the sample code might not work properly during the run time.
Test Plan:
* make jni
* make jtest
Reviewers: haobo, dhruba, sdong, igor, ljin
Reviewed By: dhruba
CC: leveldb, xjin
Differential Revision: https://reviews.facebook.net/D17109
2014-03-28 22:19:21 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-04-02 22:14:55 +02:00
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
|
|
|
* Method: put
|
2016-09-12 20:51:08 +02:00
|
|
|
* Signature: (J[BII[BII)V
|
2014-04-02 22:14:55 +02:00
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
void Java_org_rocksdb_RocksDB_put__J_3BII_3BII(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle,
|
|
|
|
jbyteArray jkey, jint jkey_off, jint jkey_len,
|
|
|
|
jbyteArray jval, jint jval_off, jint jval_len) {
|
2017-02-28 01:26:12 +01:00
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
2014-04-02 22:14:55 +02:00
|
|
|
static const rocksdb::WriteOptions default_write_options =
|
|
|
|
rocksdb::WriteOptions();
|
2016-09-14 22:12:55 +02:00
|
|
|
rocksdb_put_helper(env, db, default_write_options, nullptr, jkey, jkey_off,
|
2019-02-22 23:36:38 +01:00
|
|
|
jkey_len, jval, jval_off, jval_len);
|
2014-04-02 22:14:55 +02:00
|
|
|
}
|
2016-09-12 20:51:08 +02:00
|
|
|
|
2014-10-13 10:34:52 +02:00
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
|
|
|
* Method: put
|
2016-09-12 20:51:08 +02:00
|
|
|
* Signature: (J[BII[BIIJ)V
|
2014-10-13 10:34:52 +02:00
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
void Java_org_rocksdb_RocksDB_put__J_3BII_3BIIJ(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle,
|
|
|
|
jbyteArray jkey, jint jkey_off, jint jkey_len,
|
|
|
|
jbyteArray jval, jint jval_off, jint jval_len,
|
|
|
|
jlong jcf_handle) {
|
2017-02-28 01:26:12 +01:00
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
2014-10-13 10:34:52 +02:00
|
|
|
static const rocksdb::WriteOptions default_write_options =
|
|
|
|
rocksdb::WriteOptions();
|
2017-02-28 01:26:12 +01:00
|
|
|
auto* cf_handle = reinterpret_cast<rocksdb::ColumnFamilyHandle*>(jcf_handle);
|
2014-10-13 10:34:52 +02:00
|
|
|
if (cf_handle != nullptr) {
|
2016-09-14 22:12:55 +02:00
|
|
|
rocksdb_put_helper(env, db, default_write_options, cf_handle, jkey,
|
2019-02-22 23:36:38 +01:00
|
|
|
jkey_off, jkey_len, jval, jval_off, jval_len);
|
2014-10-13 10:34:52 +02:00
|
|
|
} else {
|
2018-04-13 02:55:14 +02:00
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(
|
|
|
|
env, rocksdb::Status::InvalidArgument("Invalid ColumnFamilyHandle."));
|
2014-10-13 10:34:52 +02:00
|
|
|
}
|
|
|
|
}
|
2014-04-02 22:14:55 +02:00
|
|
|
|
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
|
|
|
* Method: put
|
2016-09-12 20:51:08 +02:00
|
|
|
* Signature: (JJ[BII[BII)V
|
2014-04-02 22:14:55 +02:00
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
void Java_org_rocksdb_RocksDB_put__JJ_3BII_3BII(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle,
|
|
|
|
jlong jwrite_options_handle,
|
|
|
|
jbyteArray jkey, jint jkey_off, jint jkey_len,
|
|
|
|
jbyteArray jval, jint jval_off, jint jval_len) {
|
2017-02-28 01:26:12 +01:00
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
2018-04-13 02:55:14 +02:00
|
|
|
auto* write_options =
|
|
|
|
reinterpret_cast<rocksdb::WriteOptions*>(jwrite_options_handle);
|
2016-09-14 22:12:55 +02:00
|
|
|
rocksdb_put_helper(env, db, *write_options, nullptr, jkey, jkey_off, jkey_len,
|
2019-02-22 23:36:38 +01:00
|
|
|
jval, jval_off, jval_len);
|
2014-04-02 22:14:55 +02:00
|
|
|
}
|
|
|
|
|
2014-10-13 10:34:52 +02:00
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
|
|
|
* Method: put
|
2016-09-12 20:51:08 +02:00
|
|
|
* Signature: (JJ[BII[BIIJ)V
|
2014-10-13 10:34:52 +02:00
|
|
|
*/
|
2016-09-12 20:51:08 +02:00
|
|
|
void Java_org_rocksdb_RocksDB_put__JJ_3BII_3BIIJ(
|
2019-02-22 23:36:38 +01:00
|
|
|
JNIEnv* env, jobject, jlong jdb_handle, jlong jwrite_options_handle,
|
|
|
|
jbyteArray jkey, jint jkey_off, jint jkey_len,
|
|
|
|
jbyteArray jval, jint jval_off, jint jval_len,
|
|
|
|
jlong jcf_handle) {
|
2017-02-28 01:26:12 +01:00
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
2018-04-13 02:55:14 +02:00
|
|
|
auto* write_options =
|
|
|
|
reinterpret_cast<rocksdb::WriteOptions*>(jwrite_options_handle);
|
2017-02-28 01:26:12 +01:00
|
|
|
auto* cf_handle = reinterpret_cast<rocksdb::ColumnFamilyHandle*>(jcf_handle);
|
2014-10-13 10:34:52 +02:00
|
|
|
if (cf_handle != nullptr) {
|
2016-09-14 22:12:55 +02:00
|
|
|
rocksdb_put_helper(env, db, *write_options, cf_handle, jkey, jkey_off,
|
2019-02-22 23:36:38 +01:00
|
|
|
jkey_len, jval, jval_off, jval_len);
|
2014-10-13 10:34:52 +02:00
|
|
|
} else {
|
2018-04-13 02:55:14 +02:00
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(
|
|
|
|
env, rocksdb::Status::InvalidArgument("Invalid ColumnFamilyHandle."));
|
2014-10-13 10:34:52 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-04-09 09:48:20 +02:00
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
2019-02-22 23:36:38 +01:00
|
|
|
// rocksdb::DB::Delete()
|
2015-01-03 21:19:03 +01:00
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
/**
|
|
|
|
* @return true if the delete succeeded, false if a Java Exception was thrown
|
2015-01-03 21:19:03 +01:00
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
bool rocksdb_delete_helper(
|
|
|
|
JNIEnv* env, rocksdb::DB* db, const rocksdb::WriteOptions& write_options,
|
|
|
|
rocksdb::ColumnFamilyHandle* cf_handle,
|
|
|
|
jbyteArray jkey, jint jkey_off, jint jkey_len) {
|
2016-09-12 20:51:08 +02:00
|
|
|
jbyte* key = new jbyte[jkey_len];
|
|
|
|
env->GetByteArrayRegion(jkey, jkey_off, jkey_len, key);
|
2018-04-13 02:55:14 +02:00
|
|
|
if (env->ExceptionCheck()) {
|
2017-02-28 01:26:12 +01:00
|
|
|
// exception thrown: ArrayIndexOutOfBoundsException
|
2018-04-13 02:55:14 +02:00
|
|
|
delete[] key;
|
2017-02-28 01:26:12 +01:00
|
|
|
return false;
|
|
|
|
}
|
2014-10-13 10:34:52 +02:00
|
|
|
rocksdb::Slice key_slice(reinterpret_cast<char*>(key), jkey_len);
|
2017-05-10 19:19:52 +02:00
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
rocksdb::Status s;
|
2014-11-01 01:02:12 +01:00
|
|
|
if (cf_handle != nullptr) {
|
2019-02-22 23:36:38 +01:00
|
|
|
s = db->Delete(write_options, cf_handle, key_slice);
|
2014-11-01 01:02:12 +01:00
|
|
|
} else {
|
2019-02-22 23:36:38 +01:00
|
|
|
// backwards compatibility
|
|
|
|
s = db->Delete(write_options, key_slice);
|
2014-11-01 01:02:12 +01:00
|
|
|
}
|
|
|
|
|
2016-09-12 20:51:08 +02:00
|
|
|
// cleanup
|
2018-04-13 02:55:14 +02:00
|
|
|
delete[] key;
|
2016-09-12 20:51:08 +02:00
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
if (s.ok()) {
|
|
|
|
return true;
|
2014-10-13 10:34:52 +02:00
|
|
|
}
|
2017-02-28 01:26:12 +01:00
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(env, s);
|
|
|
|
return false;
|
2014-11-01 01:02:12 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
2019-02-22 23:36:38 +01:00
|
|
|
* Method: delete
|
|
|
|
* Signature: (J[BII)V
|
2014-11-01 01:02:12 +01:00
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
void Java_org_rocksdb_RocksDB_delete__J_3BII(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle,
|
|
|
|
jbyteArray jkey, jint jkey_off, jint jkey_len) {
|
2016-02-01 21:00:40 +01:00
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
2019-02-22 23:36:38 +01:00
|
|
|
static const rocksdb::WriteOptions default_write_options =
|
|
|
|
rocksdb::WriteOptions();
|
|
|
|
rocksdb_delete_helper(env, db, default_write_options, nullptr, jkey, jkey_off,
|
|
|
|
jkey_len);
|
2014-10-13 10:34:52 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
2019-02-22 23:36:38 +01:00
|
|
|
* Method: delete
|
|
|
|
* Signature: (J[BIIJ)V
|
2014-10-13 10:34:52 +02:00
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
void Java_org_rocksdb_RocksDB_delete__J_3BIIJ(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle,
|
|
|
|
jbyteArray jkey, jint jkey_off, jint jkey_len,
|
|
|
|
jlong jcf_handle) {
|
2016-02-01 21:00:40 +01:00
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
2019-02-22 23:36:38 +01:00
|
|
|
static const rocksdb::WriteOptions default_write_options =
|
|
|
|
rocksdb::WriteOptions();
|
|
|
|
auto* cf_handle = reinterpret_cast<rocksdb::ColumnFamilyHandle*>(jcf_handle);
|
2014-10-13 10:34:52 +02:00
|
|
|
if (cf_handle != nullptr) {
|
2019-02-22 23:36:38 +01:00
|
|
|
rocksdb_delete_helper(env, db, default_write_options, cf_handle, jkey,
|
|
|
|
jkey_off, jkey_len);
|
2014-10-13 10:34:52 +02:00
|
|
|
} else {
|
2018-04-13 02:55:14 +02:00
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(
|
|
|
|
env, rocksdb::Status::InvalidArgument("Invalid ColumnFamilyHandle."));
|
2014-10-13 10:34:52 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-11-01 01:02:12 +01:00
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
2019-02-22 23:36:38 +01:00
|
|
|
* Method: delete
|
|
|
|
* Signature: (JJ[BII)V
|
2014-11-01 01:02:12 +01:00
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
void Java_org_rocksdb_RocksDB_delete__JJ_3BII(
|
|
|
|
JNIEnv* env, jobject,
|
|
|
|
jlong jdb_handle,
|
|
|
|
jlong jwrite_options,
|
|
|
|
jbyteArray jkey, jint jkey_off, jint jkey_len) {
|
2016-02-01 21:00:40 +01:00
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
2019-02-22 23:36:38 +01:00
|
|
|
auto* write_options =
|
|
|
|
reinterpret_cast<rocksdb::WriteOptions*>(jwrite_options);
|
|
|
|
rocksdb_delete_helper(env, db, *write_options, nullptr, jkey, jkey_off,
|
|
|
|
jkey_len);
|
2014-11-01 01:02:12 +01:00
|
|
|
}
|
|
|
|
|
2014-10-13 10:34:52 +02:00
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
2019-02-22 23:36:38 +01:00
|
|
|
* Method: delete
|
|
|
|
* Signature: (JJ[BIIJ)V
|
2014-10-13 10:34:52 +02:00
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
void Java_org_rocksdb_RocksDB_delete__JJ_3BIIJ(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle, jlong jwrite_options,
|
|
|
|
jbyteArray jkey, jint jkey_off, jint jkey_len, jlong jcf_handle) {
|
2016-02-01 21:00:40 +01:00
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
2019-02-22 23:36:38 +01:00
|
|
|
auto* write_options =
|
|
|
|
reinterpret_cast<rocksdb::WriteOptions*>(jwrite_options);
|
2018-04-13 02:55:14 +02:00
|
|
|
auto* cf_handle = reinterpret_cast<rocksdb::ColumnFamilyHandle*>(jcf_handle);
|
2014-10-13 10:34:52 +02:00
|
|
|
if (cf_handle != nullptr) {
|
2019-02-22 23:36:38 +01:00
|
|
|
rocksdb_delete_helper(env, db, *write_options, cf_handle, jkey, jkey_off,
|
|
|
|
jkey_len);
|
2014-10-13 10:34:52 +02:00
|
|
|
} else {
|
2018-04-13 02:55:14 +02:00
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(
|
|
|
|
env, rocksdb::Status::InvalidArgument("Invalid ColumnFamilyHandle."));
|
2016-02-10 15:21:23 +01:00
|
|
|
}
|
2014-10-13 10:34:52 +02:00
|
|
|
}
|
|
|
|
|
2014-04-02 22:14:55 +02:00
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
2019-02-22 23:36:38 +01:00
|
|
|
// rocksdb::DB::SingleDelete()
|
|
|
|
/**
|
|
|
|
* @return true if the single delete succeeded, false if a Java Exception
|
|
|
|
* was thrown
|
|
|
|
*/
|
|
|
|
bool rocksdb_single_delete_helper(
|
|
|
|
JNIEnv* env, rocksdb::DB* db,
|
|
|
|
const rocksdb::WriteOptions& write_options,
|
|
|
|
rocksdb::ColumnFamilyHandle* cf_handle,
|
|
|
|
jbyteArray jkey, jint jkey_len) {
|
|
|
|
jbyte* key = env->GetByteArrayElements(jkey, nullptr);
|
|
|
|
if (key == nullptr) {
|
|
|
|
// exception thrown: OutOfMemoryError
|
|
|
|
return false;
|
2017-02-28 01:26:12 +01:00
|
|
|
}
|
2018-04-13 02:55:14 +02:00
|
|
|
rocksdb::Slice key_slice(reinterpret_cast<char*>(key), jkey_len);
|
Add a jni library for rocksdb which supports Open, Get, Put, and Close.
Summary:
This diff contains a simple jni library for rocksdb which supports open, get,
put and closeusing default options (including Options, ReadOptions, and
WriteOptions.) In the usual case, Java developers can use the c++ rocksdb
library in the way similar to the following:
RocksDB db = RocksDB.open(path_to_db);
...
db.put("hello".getBytes(), "world".getBytes();
byte[] value = db.get("hello".getBytes());
...
db.close();
Specifically, this diff has the following major classes:
* RocksDB: a Java wrapper class which forwards the operations
from the java side to c++ rocksdb library.
* RocksDBException: ncapsulates the error of an operation.
This exception type is used to describe an internal error from
the c++ rocksdb library.
This diff also include a simple java sample code calling c++ rocksdb library.
To build the rocksdb jni library, simply run make jni, and make jtest will try to
build and run the sample code.
Note that if the rocksdb is not built with the default glibc that Java uses,
java will try to load the wrong glibc during the run time. As a result,
the sample code might not work properly during the run time.
Test Plan:
* make jni
* make jtest
Reviewers: haobo, dhruba, sdong, igor, ljin
Reviewed By: dhruba
CC: leveldb, xjin
Differential Revision: https://reviews.facebook.net/D17109
2014-03-28 22:19:21 +01:00
|
|
|
|
2014-10-13 10:34:52 +02:00
|
|
|
rocksdb::Status s;
|
2019-02-22 23:36:38 +01:00
|
|
|
if (cf_handle != nullptr) {
|
|
|
|
s = db->SingleDelete(write_options, cf_handle, key_slice);
|
2014-10-13 10:34:52 +02:00
|
|
|
} else {
|
|
|
|
// backwards compatibility
|
2019-02-22 23:36:38 +01:00
|
|
|
s = db->SingleDelete(write_options, key_slice);
|
2014-10-13 10:34:52 +02:00
|
|
|
}
|
Add a jni library for rocksdb which supports Open, Get, Put, and Close.
Summary:
This diff contains a simple jni library for rocksdb which supports open, get,
put and closeusing default options (including Options, ReadOptions, and
WriteOptions.) In the usual case, Java developers can use the c++ rocksdb
library in the way similar to the following:
RocksDB db = RocksDB.open(path_to_db);
...
db.put("hello".getBytes(), "world".getBytes();
byte[] value = db.get("hello".getBytes());
...
db.close();
Specifically, this diff has the following major classes:
* RocksDB: a Java wrapper class which forwards the operations
from the java side to c++ rocksdb library.
* RocksDBException: ncapsulates the error of an operation.
This exception type is used to describe an internal error from
the c++ rocksdb library.
This diff also include a simple java sample code calling c++ rocksdb library.
To build the rocksdb jni library, simply run make jni, and make jtest will try to
build and run the sample code.
Note that if the rocksdb is not built with the default glibc that Java uses,
java will try to load the wrong glibc during the run time. As a result,
the sample code might not work properly during the run time.
Test Plan:
* make jni
* make jtest
Reviewers: haobo, dhruba, sdong, igor, ljin
Reviewed By: dhruba
CC: leveldb, xjin
Differential Revision: https://reviews.facebook.net/D17109
2014-03-28 22:19:21 +01:00
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
// trigger java unref on key and value.
|
|
|
|
// by passing JNI_ABORT, it will simply release the reference without
|
|
|
|
// copying the result back to the java byte array.
|
|
|
|
env->ReleaseByteArrayElements(jkey, key, JNI_ABORT);
|
Add a jni library for rocksdb which supports Open, Get, Put, and Close.
Summary:
This diff contains a simple jni library for rocksdb which supports open, get,
put and closeusing default options (including Options, ReadOptions, and
WriteOptions.) In the usual case, Java developers can use the c++ rocksdb
library in the way similar to the following:
RocksDB db = RocksDB.open(path_to_db);
...
db.put("hello".getBytes(), "world".getBytes();
byte[] value = db.get("hello".getBytes());
...
db.close();
Specifically, this diff has the following major classes:
* RocksDB: a Java wrapper class which forwards the operations
from the java side to c++ rocksdb library.
* RocksDBException: ncapsulates the error of an operation.
This exception type is used to describe an internal error from
the c++ rocksdb library.
This diff also include a simple java sample code calling c++ rocksdb library.
To build the rocksdb jni library, simply run make jni, and make jtest will try to
build and run the sample code.
Note that if the rocksdb is not built with the default glibc that Java uses,
java will try to load the wrong glibc during the run time. As a result,
the sample code might not work properly during the run time.
Test Plan:
* make jni
* make jtest
Reviewers: haobo, dhruba, sdong, igor, ljin
Reviewed By: dhruba
CC: leveldb, xjin
Differential Revision: https://reviews.facebook.net/D17109
2014-03-28 22:19:21 +01:00
|
|
|
|
|
|
|
if (s.ok()) {
|
2019-02-22 23:36:38 +01:00
|
|
|
return true;
|
Add a jni library for rocksdb which supports Open, Get, Put, and Close.
Summary:
This diff contains a simple jni library for rocksdb which supports open, get,
put and closeusing default options (including Options, ReadOptions, and
WriteOptions.) In the usual case, Java developers can use the c++ rocksdb
library in the way similar to the following:
RocksDB db = RocksDB.open(path_to_db);
...
db.put("hello".getBytes(), "world".getBytes();
byte[] value = db.get("hello".getBytes());
...
db.close();
Specifically, this diff has the following major classes:
* RocksDB: a Java wrapper class which forwards the operations
from the java side to c++ rocksdb library.
* RocksDBException: ncapsulates the error of an operation.
This exception type is used to describe an internal error from
the c++ rocksdb library.
This diff also include a simple java sample code calling c++ rocksdb library.
To build the rocksdb jni library, simply run make jni, and make jtest will try to
build and run the sample code.
Note that if the rocksdb is not built with the default glibc that Java uses,
java will try to load the wrong glibc during the run time. As a result,
the sample code might not work properly during the run time.
Test Plan:
* make jni
* make jtest
Reviewers: haobo, dhruba, sdong, igor, ljin
Reviewed By: dhruba
CC: leveldb, xjin
Differential Revision: https://reviews.facebook.net/D17109
2014-03-28 22:19:21 +01:00
|
|
|
}
|
|
|
|
|
2017-02-28 01:26:12 +01:00
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(env, s);
|
2019-02-22 23:36:38 +01:00
|
|
|
return false;
|
Add a jni library for rocksdb which supports Open, Get, Put, and Close.
Summary:
This diff contains a simple jni library for rocksdb which supports open, get,
put and closeusing default options (including Options, ReadOptions, and
WriteOptions.) In the usual case, Java developers can use the c++ rocksdb
library in the way similar to the following:
RocksDB db = RocksDB.open(path_to_db);
...
db.put("hello".getBytes(), "world".getBytes();
byte[] value = db.get("hello".getBytes());
...
db.close();
Specifically, this diff has the following major classes:
* RocksDB: a Java wrapper class which forwards the operations
from the java side to c++ rocksdb library.
* RocksDBException: ncapsulates the error of an operation.
This exception type is used to describe an internal error from
the c++ rocksdb library.
This diff also include a simple java sample code calling c++ rocksdb library.
To build the rocksdb jni library, simply run make jni, and make jtest will try to
build and run the sample code.
Note that if the rocksdb is not built with the default glibc that Java uses,
java will try to load the wrong glibc during the run time. As a result,
the sample code might not work properly during the run time.
Test Plan:
* make jni
* make jtest
Reviewers: haobo, dhruba, sdong, igor, ljin
Reviewed By: dhruba
CC: leveldb, xjin
Differential Revision: https://reviews.facebook.net/D17109
2014-03-28 22:19:21 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
2019-02-22 23:36:38 +01:00
|
|
|
* Method: singleDelete
|
|
|
|
* Signature: (J[BI)V
|
Add a jni library for rocksdb which supports Open, Get, Put, and Close.
Summary:
This diff contains a simple jni library for rocksdb which supports open, get,
put and closeusing default options (including Options, ReadOptions, and
WriteOptions.) In the usual case, Java developers can use the c++ rocksdb
library in the way similar to the following:
RocksDB db = RocksDB.open(path_to_db);
...
db.put("hello".getBytes(), "world".getBytes();
byte[] value = db.get("hello".getBytes());
...
db.close();
Specifically, this diff has the following major classes:
* RocksDB: a Java wrapper class which forwards the operations
from the java side to c++ rocksdb library.
* RocksDBException: ncapsulates the error of an operation.
This exception type is used to describe an internal error from
the c++ rocksdb library.
This diff also include a simple java sample code calling c++ rocksdb library.
To build the rocksdb jni library, simply run make jni, and make jtest will try to
build and run the sample code.
Note that if the rocksdb is not built with the default glibc that Java uses,
java will try to load the wrong glibc during the run time. As a result,
the sample code might not work properly during the run time.
Test Plan:
* make jni
* make jtest
Reviewers: haobo, dhruba, sdong, igor, ljin
Reviewed By: dhruba
CC: leveldb, xjin
Differential Revision: https://reviews.facebook.net/D17109
2014-03-28 22:19:21 +01:00
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
void Java_org_rocksdb_RocksDB_singleDelete__J_3BI(
|
|
|
|
JNIEnv* env, jobject,
|
|
|
|
jlong jdb_handle,
|
|
|
|
jbyteArray jkey,
|
|
|
|
jint jkey_len) {
|
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
|
|
|
static const rocksdb::WriteOptions default_write_options =
|
|
|
|
rocksdb::WriteOptions();
|
|
|
|
rocksdb_single_delete_helper(env, db, default_write_options, nullptr,
|
|
|
|
jkey, jkey_len);
|
2014-04-22 00:52:59 +02:00
|
|
|
}
|
|
|
|
|
2014-10-13 10:34:52 +02:00
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
2019-02-22 23:36:38 +01:00
|
|
|
* Method: singleDelete
|
|
|
|
* Signature: (J[BIJ)V
|
2014-10-13 10:34:52 +02:00
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
void Java_org_rocksdb_RocksDB_singleDelete__J_3BIJ(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle,
|
|
|
|
jbyteArray jkey, jint jkey_len, jlong jcf_handle) {
|
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
|
|
|
static const rocksdb::WriteOptions default_write_options =
|
|
|
|
rocksdb::WriteOptions();
|
|
|
|
auto* cf_handle = reinterpret_cast<rocksdb::ColumnFamilyHandle*>(jcf_handle);
|
2014-10-13 10:34:52 +02:00
|
|
|
if (cf_handle != nullptr) {
|
2019-02-22 23:36:38 +01:00
|
|
|
rocksdb_single_delete_helper(env, db, default_write_options, cf_handle,
|
|
|
|
jkey, jkey_len);
|
2014-10-13 10:34:52 +02:00
|
|
|
} else {
|
2018-04-13 02:55:14 +02:00
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(
|
|
|
|
env, rocksdb::Status::InvalidArgument("Invalid ColumnFamilyHandle."));
|
2014-10-13 10:34:52 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-04-22 00:52:59 +02:00
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
2019-02-22 23:36:38 +01:00
|
|
|
* Method: singleDelete
|
|
|
|
* Signature: (JJ[BIJ)V
|
2014-04-22 00:52:59 +02:00
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
void Java_org_rocksdb_RocksDB_singleDelete__JJ_3BI(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle,
|
|
|
|
jlong jwrite_options,
|
|
|
|
jbyteArray jkey,
|
|
|
|
jint jkey_len) {
|
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
|
|
|
auto* write_options =
|
|
|
|
reinterpret_cast<rocksdb::WriteOptions*>(jwrite_options);
|
|
|
|
rocksdb_single_delete_helper(env, db, *write_options, nullptr, jkey,
|
|
|
|
jkey_len);
|
2014-04-22 00:52:59 +02:00
|
|
|
}
|
|
|
|
|
2014-10-13 10:34:52 +02:00
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
2019-02-22 23:36:38 +01:00
|
|
|
* Method: singleDelete
|
|
|
|
* Signature: (JJ[BIJ)V
|
2014-10-13 10:34:52 +02:00
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
void Java_org_rocksdb_RocksDB_singleDelete__JJ_3BIJ(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle, jlong jwrite_options,
|
|
|
|
jbyteArray jkey, jint jkey_len, jlong jcf_handle) {
|
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
|
|
|
auto* write_options =
|
|
|
|
reinterpret_cast<rocksdb::WriteOptions*>(jwrite_options);
|
2017-02-28 01:26:12 +01:00
|
|
|
auto* cf_handle = reinterpret_cast<rocksdb::ColumnFamilyHandle*>(jcf_handle);
|
2014-10-13 10:34:52 +02:00
|
|
|
if (cf_handle != nullptr) {
|
2019-02-22 23:36:38 +01:00
|
|
|
rocksdb_single_delete_helper(env, db, *write_options, cf_handle, jkey,
|
|
|
|
jkey_len);
|
2014-10-13 10:34:52 +02:00
|
|
|
} else {
|
2018-04-13 02:55:14 +02:00
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(
|
|
|
|
env, rocksdb::Status::InvalidArgument("Invalid ColumnFamilyHandle."));
|
2014-10-13 10:34:52 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
// rocksdb::DB::DeleteRange()
|
|
|
|
/**
|
|
|
|
* @return true if the delete range succeeded, false if a Java Exception
|
|
|
|
* was thrown
|
|
|
|
*/
|
|
|
|
bool rocksdb_delete_range_helper(
|
|
|
|
JNIEnv* env, rocksdb::DB* db,
|
|
|
|
const rocksdb::WriteOptions& write_options,
|
|
|
|
rocksdb::ColumnFamilyHandle* cf_handle,
|
|
|
|
jbyteArray jbegin_key, jint jbegin_key_off, jint jbegin_key_len,
|
|
|
|
jbyteArray jend_key, jint jend_key_off, jint jend_key_len) {
|
|
|
|
jbyte* begin_key = new jbyte[jbegin_key_len];
|
|
|
|
env->GetByteArrayRegion(jbegin_key, jbegin_key_off, jbegin_key_len,
|
|
|
|
begin_key);
|
2018-04-13 02:55:14 +02:00
|
|
|
if (env->ExceptionCheck()) {
|
2019-02-22 23:36:38 +01:00
|
|
|
// exception thrown: ArrayIndexOutOfBoundsException
|
|
|
|
delete[] begin_key;
|
|
|
|
return false;
|
2017-02-28 01:26:12 +01:00
|
|
|
}
|
2019-02-22 23:36:38 +01:00
|
|
|
rocksdb::Slice begin_key_slice(reinterpret_cast<char*>(begin_key),
|
|
|
|
jbegin_key_len);
|
Add a jni library for rocksdb which supports Open, Get, Put, and Close.
Summary:
This diff contains a simple jni library for rocksdb which supports open, get,
put and closeusing default options (including Options, ReadOptions, and
WriteOptions.) In the usual case, Java developers can use the c++ rocksdb
library in the way similar to the following:
RocksDB db = RocksDB.open(path_to_db);
...
db.put("hello".getBytes(), "world".getBytes();
byte[] value = db.get("hello".getBytes());
...
db.close();
Specifically, this diff has the following major classes:
* RocksDB: a Java wrapper class which forwards the operations
from the java side to c++ rocksdb library.
* RocksDBException: ncapsulates the error of an operation.
This exception type is used to describe an internal error from
the c++ rocksdb library.
This diff also include a simple java sample code calling c++ rocksdb library.
To build the rocksdb jni library, simply run make jni, and make jtest will try to
build and run the sample code.
Note that if the rocksdb is not built with the default glibc that Java uses,
java will try to load the wrong glibc during the run time. As a result,
the sample code might not work properly during the run time.
Test Plan:
* make jni
* make jtest
Reviewers: haobo, dhruba, sdong, igor, ljin
Reviewed By: dhruba
CC: leveldb, xjin
Differential Revision: https://reviews.facebook.net/D17109
2014-03-28 22:19:21 +01:00
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
jbyte* end_key = new jbyte[jend_key_len];
|
|
|
|
env->GetByteArrayRegion(jend_key, jend_key_off, jend_key_len, end_key);
|
|
|
|
if (env->ExceptionCheck()) {
|
|
|
|
// exception thrown: ArrayIndexOutOfBoundsException
|
|
|
|
delete[] begin_key;
|
|
|
|
delete[] end_key;
|
|
|
|
return false;
|
2014-10-13 10:34:52 +02:00
|
|
|
}
|
2019-02-22 23:36:38 +01:00
|
|
|
rocksdb::Slice end_key_slice(reinterpret_cast<char*>(end_key), jend_key_len);
|
Add a jni library for rocksdb which supports Open, Get, Put, and Close.
Summary:
This diff contains a simple jni library for rocksdb which supports open, get,
put and closeusing default options (including Options, ReadOptions, and
WriteOptions.) In the usual case, Java developers can use the c++ rocksdb
library in the way similar to the following:
RocksDB db = RocksDB.open(path_to_db);
...
db.put("hello".getBytes(), "world".getBytes();
byte[] value = db.get("hello".getBytes());
...
db.close();
Specifically, this diff has the following major classes:
* RocksDB: a Java wrapper class which forwards the operations
from the java side to c++ rocksdb library.
* RocksDBException: ncapsulates the error of an operation.
This exception type is used to describe an internal error from
the c++ rocksdb library.
This diff also include a simple java sample code calling c++ rocksdb library.
To build the rocksdb jni library, simply run make jni, and make jtest will try to
build and run the sample code.
Note that if the rocksdb is not built with the default glibc that Java uses,
java will try to load the wrong glibc during the run time. As a result,
the sample code might not work properly during the run time.
Test Plan:
* make jni
* make jtest
Reviewers: haobo, dhruba, sdong, igor, ljin
Reviewed By: dhruba
CC: leveldb, xjin
Differential Revision: https://reviews.facebook.net/D17109
2014-03-28 22:19:21 +01:00
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
rocksdb::Status s =
|
|
|
|
db->DeleteRange(write_options, cf_handle, begin_key_slice, end_key_slice);
|
Add a jni library for rocksdb which supports Open, Get, Put, and Close.
Summary:
This diff contains a simple jni library for rocksdb which supports open, get,
put and closeusing default options (including Options, ReadOptions, and
WriteOptions.) In the usual case, Java developers can use the c++ rocksdb
library in the way similar to the following:
RocksDB db = RocksDB.open(path_to_db);
...
db.put("hello".getBytes(), "world".getBytes();
byte[] value = db.get("hello".getBytes());
...
db.close();
Specifically, this diff has the following major classes:
* RocksDB: a Java wrapper class which forwards the operations
from the java side to c++ rocksdb library.
* RocksDBException: ncapsulates the error of an operation.
This exception type is used to describe an internal error from
the c++ rocksdb library.
This diff also include a simple java sample code calling c++ rocksdb library.
To build the rocksdb jni library, simply run make jni, and make jtest will try to
build and run the sample code.
Note that if the rocksdb is not built with the default glibc that Java uses,
java will try to load the wrong glibc during the run time. As a result,
the sample code might not work properly during the run time.
Test Plan:
* make jni
* make jtest
Reviewers: haobo, dhruba, sdong, igor, ljin
Reviewed By: dhruba
CC: leveldb, xjin
Differential Revision: https://reviews.facebook.net/D17109
2014-03-28 22:19:21 +01:00
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
// cleanup
|
|
|
|
delete[] begin_key;
|
|
|
|
delete[] end_key;
|
2014-03-30 07:00:52 +02:00
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
if (s.ok()) {
|
|
|
|
return true;
|
Add a jni library for rocksdb which supports Open, Get, Put, and Close.
Summary:
This diff contains a simple jni library for rocksdb which supports open, get,
put and closeusing default options (including Options, ReadOptions, and
WriteOptions.) In the usual case, Java developers can use the c++ rocksdb
library in the way similar to the following:
RocksDB db = RocksDB.open(path_to_db);
...
db.put("hello".getBytes(), "world".getBytes();
byte[] value = db.get("hello".getBytes());
...
db.close();
Specifically, this diff has the following major classes:
* RocksDB: a Java wrapper class which forwards the operations
from the java side to c++ rocksdb library.
* RocksDBException: ncapsulates the error of an operation.
This exception type is used to describe an internal error from
the c++ rocksdb library.
This diff also include a simple java sample code calling c++ rocksdb library.
To build the rocksdb jni library, simply run make jni, and make jtest will try to
build and run the sample code.
Note that if the rocksdb is not built with the default glibc that Java uses,
java will try to load the wrong glibc during the run time. As a result,
the sample code might not work properly during the run time.
Test Plan:
* make jni
* make jtest
Reviewers: haobo, dhruba, sdong, igor, ljin
Reviewed By: dhruba
CC: leveldb, xjin
Differential Revision: https://reviews.facebook.net/D17109
2014-03-28 22:19:21 +01:00
|
|
|
}
|
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(env, s);
|
|
|
|
return false;
|
|
|
|
}
|
2017-02-28 01:26:12 +01:00
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
|
|
|
* Method: deleteRange
|
|
|
|
* Signature: (J[BII[BII)V
|
|
|
|
*/
|
|
|
|
void Java_org_rocksdb_RocksDB_deleteRange__J_3BII_3BII(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle,
|
|
|
|
jbyteArray jbegin_key, jint jbegin_key_off, jint jbegin_key_len,
|
|
|
|
jbyteArray jend_key, jint jend_key_off, jint jend_key_len) {
|
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
|
|
|
static const rocksdb::WriteOptions default_write_options =
|
|
|
|
rocksdb::WriteOptions();
|
|
|
|
rocksdb_delete_range_helper(env, db, default_write_options, nullptr,
|
|
|
|
jbegin_key, jbegin_key_off, jbegin_key_len,
|
|
|
|
jend_key, jend_key_off, jend_key_len);
|
Add a jni library for rocksdb which supports Open, Get, Put, and Close.
Summary:
This diff contains a simple jni library for rocksdb which supports open, get,
put and closeusing default options (including Options, ReadOptions, and
WriteOptions.) In the usual case, Java developers can use the c++ rocksdb
library in the way similar to the following:
RocksDB db = RocksDB.open(path_to_db);
...
db.put("hello".getBytes(), "world".getBytes();
byte[] value = db.get("hello".getBytes());
...
db.close();
Specifically, this diff has the following major classes:
* RocksDB: a Java wrapper class which forwards the operations
from the java side to c++ rocksdb library.
* RocksDBException: ncapsulates the error of an operation.
This exception type is used to describe an internal error from
the c++ rocksdb library.
This diff also include a simple java sample code calling c++ rocksdb library.
To build the rocksdb jni library, simply run make jni, and make jtest will try to
build and run the sample code.
Note that if the rocksdb is not built with the default glibc that Java uses,
java will try to load the wrong glibc during the run time. As a result,
the sample code might not work properly during the run time.
Test Plan:
* make jni
* make jtest
Reviewers: haobo, dhruba, sdong, igor, ljin
Reviewed By: dhruba
CC: leveldb, xjin
Differential Revision: https://reviews.facebook.net/D17109
2014-03-28 22:19:21 +01:00
|
|
|
}
|
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
|
|
|
* Method: deleteRange
|
|
|
|
* Signature: (J[BII[BIIJ)V
|
|
|
|
*/
|
|
|
|
void Java_org_rocksdb_RocksDB_deleteRange__J_3BII_3BIIJ(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle,
|
|
|
|
jbyteArray jbegin_key, jint jbegin_key_off, jint jbegin_key_len,
|
|
|
|
jbyteArray jend_key, jint jend_key_off, jint jend_key_len,
|
|
|
|
jlong jcf_handle) {
|
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
|
|
|
static const rocksdb::WriteOptions default_write_options =
|
|
|
|
rocksdb::WriteOptions();
|
|
|
|
auto* cf_handle = reinterpret_cast<rocksdb::ColumnFamilyHandle*>(jcf_handle);
|
|
|
|
if (cf_handle != nullptr) {
|
|
|
|
rocksdb_delete_range_helper(env, db, default_write_options, cf_handle,
|
|
|
|
jbegin_key, jbegin_key_off, jbegin_key_len,
|
|
|
|
jend_key, jend_key_off, jend_key_len);
|
|
|
|
} else {
|
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(
|
|
|
|
env, rocksdb::Status::InvalidArgument("Invalid ColumnFamilyHandle."));
|
2017-02-28 01:26:12 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
|
|
|
* Method: deleteRange
|
|
|
|
* Signature: (JJ[BII[BII)V
|
|
|
|
*/
|
|
|
|
void Java_org_rocksdb_RocksDB_deleteRange__JJ_3BII_3BII(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle, jlong jwrite_options,
|
|
|
|
jbyteArray jbegin_key, jint jbegin_key_off, jint jbegin_key_len,
|
|
|
|
jbyteArray jend_key, jint jend_key_off, jint jend_key_len) {
|
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
|
|
|
auto* write_options =
|
|
|
|
reinterpret_cast<rocksdb::WriteOptions*>(jwrite_options);
|
|
|
|
rocksdb_delete_range_helper(env, db, *write_options, nullptr, jbegin_key,
|
|
|
|
jbegin_key_off, jbegin_key_len, jend_key,
|
|
|
|
jend_key_off, jend_key_len);
|
|
|
|
}
|
2014-04-26 05:59:16 +02:00
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
|
|
|
* Method: deleteRange
|
|
|
|
* Signature: (JJ[BII[BIIJ)V
|
|
|
|
*/
|
|
|
|
void Java_org_rocksdb_RocksDB_deleteRange__JJ_3BII_3BIIJ(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle, jlong jwrite_options,
|
|
|
|
jbyteArray jbegin_key, jint jbegin_key_off, jint jbegin_key_len,
|
|
|
|
jbyteArray jend_key, jint jend_key_off, jint jend_key_len,
|
|
|
|
jlong jcf_handle) {
|
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
|
|
|
auto* write_options =
|
|
|
|
reinterpret_cast<rocksdb::WriteOptions*>(jwrite_options);
|
|
|
|
auto* cf_handle = reinterpret_cast<rocksdb::ColumnFamilyHandle*>(jcf_handle);
|
|
|
|
if (cf_handle != nullptr) {
|
|
|
|
rocksdb_delete_range_helper(env, db, *write_options, cf_handle,
|
|
|
|
jbegin_key, jbegin_key_off, jbegin_key_len,
|
|
|
|
jend_key, jend_key_off, jend_key_len);
|
|
|
|
} else {
|
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(
|
|
|
|
env, rocksdb::Status::InvalidArgument("Invalid ColumnFamilyHandle."));
|
2016-02-01 21:00:40 +01:00
|
|
|
}
|
2019-02-22 23:36:38 +01:00
|
|
|
}
|
2016-09-12 20:51:08 +02:00
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
// rocksdb::DB::Merge
|
2017-02-28 01:26:12 +01:00
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
/**
|
|
|
|
* @return true if the merge succeeded, false if a Java Exception was thrown
|
|
|
|
*/
|
|
|
|
bool rocksdb_merge_helper(
|
|
|
|
JNIEnv* env, rocksdb::DB* db, const rocksdb::WriteOptions& write_options,
|
|
|
|
rocksdb::ColumnFamilyHandle* cf_handle,
|
|
|
|
jbyteArray jkey, jint jkey_off, jint jkey_len,
|
|
|
|
jbyteArray jval, jint jval_off, jint jval_len) {
|
|
|
|
jbyte* key = new jbyte[jkey_len];
|
|
|
|
env->GetByteArrayRegion(jkey, jkey_off, jkey_len, key);
|
|
|
|
if (env->ExceptionCheck()) {
|
|
|
|
// exception thrown: ArrayIndexOutOfBoundsException
|
|
|
|
delete[] key;
|
|
|
|
return false;
|
2017-02-28 01:26:12 +01:00
|
|
|
}
|
2019-02-22 23:36:38 +01:00
|
|
|
rocksdb::Slice key_slice(reinterpret_cast<char*>(key), jkey_len);
|
2016-09-12 20:51:08 +02:00
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
jbyte* value = new jbyte[jval_len];
|
|
|
|
env->GetByteArrayRegion(jval, jval_off, jval_len, value);
|
|
|
|
if (env->ExceptionCheck()) {
|
|
|
|
// exception thrown: ArrayIndexOutOfBoundsException
|
|
|
|
delete[] value;
|
|
|
|
delete[] key;
|
|
|
|
return false;
|
2014-04-25 22:57:20 +02:00
|
|
|
}
|
2019-02-22 23:36:38 +01:00
|
|
|
rocksdb::Slice value_slice(reinterpret_cast<char*>(value), jval_len);
|
2014-04-26 05:59:16 +02:00
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
rocksdb::Status s;
|
|
|
|
if (cf_handle != nullptr) {
|
|
|
|
s = db->Merge(write_options, cf_handle, key_slice, value_slice);
|
2014-10-13 10:34:52 +02:00
|
|
|
} else {
|
2019-02-22 23:36:38 +01:00
|
|
|
s = db->Merge(write_options, key_slice, value_slice);
|
2017-02-28 01:26:12 +01:00
|
|
|
}
|
2014-04-26 05:59:16 +02:00
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
// cleanup
|
|
|
|
delete[] value;
|
|
|
|
delete[] key;
|
2017-02-28 01:26:12 +01:00
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
if (s.ok()) {
|
|
|
|
return true;
|
2014-04-25 22:57:20 +02:00
|
|
|
}
|
2016-02-01 21:00:40 +01:00
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(env, s);
|
|
|
|
return false;
|
2014-04-25 22:57:20 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
2019-02-22 23:36:38 +01:00
|
|
|
* Method: merge
|
|
|
|
* Signature: (J[BII[BII)V
|
2014-04-25 22:57:20 +02:00
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
void Java_org_rocksdb_RocksDB_merge__J_3BII_3BII(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle,
|
|
|
|
jbyteArray jkey, jint jkey_off, jint jkey_len,
|
|
|
|
jbyteArray jval, jint jval_off, jint jval_len) {
|
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
|
|
|
static const rocksdb::WriteOptions default_write_options =
|
|
|
|
rocksdb::WriteOptions();
|
|
|
|
rocksdb_merge_helper(env, db, default_write_options, nullptr, jkey, jkey_off,
|
|
|
|
jkey_len, jval, jval_off, jval_len);
|
2014-10-13 10:34:52 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
2019-02-22 23:36:38 +01:00
|
|
|
* Method: merge
|
|
|
|
* Signature: (J[BII[BIIJ)V
|
2014-10-13 10:34:52 +02:00
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
void Java_org_rocksdb_RocksDB_merge__J_3BII_3BIIJ(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle,
|
|
|
|
jbyteArray jkey, jint jkey_off, jint jkey_len,
|
|
|
|
jbyteArray jval, jint jval_off, jint jval_len,
|
|
|
|
jlong jcf_handle) {
|
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
|
|
|
static const rocksdb::WriteOptions default_write_options =
|
|
|
|
rocksdb::WriteOptions();
|
|
|
|
auto* cf_handle = reinterpret_cast<rocksdb::ColumnFamilyHandle*>(jcf_handle);
|
|
|
|
if (cf_handle != nullptr) {
|
|
|
|
rocksdb_merge_helper(env, db, default_write_options, cf_handle, jkey,
|
|
|
|
jkey_off, jkey_len, jval, jval_off, jval_len);
|
|
|
|
} else {
|
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(
|
|
|
|
env, rocksdb::Status::InvalidArgument("Invalid ColumnFamilyHandle."));
|
|
|
|
}
|
2014-04-25 22:57:20 +02:00
|
|
|
}
|
|
|
|
|
2014-04-22 00:52:59 +02:00
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
2019-02-22 23:36:38 +01:00
|
|
|
* Method: merge
|
|
|
|
* Signature: (JJ[BII[BII)V
|
2014-04-22 00:52:59 +02:00
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
void Java_org_rocksdb_RocksDB_merge__JJ_3BII_3BII(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle, jlong jwrite_options_handle,
|
|
|
|
jbyteArray jkey, jint jkey_off, jint jkey_len,
|
|
|
|
jbyteArray jval, jint jval_off, jint jval_len) {
|
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
|
|
|
auto* write_options =
|
|
|
|
reinterpret_cast<rocksdb::WriteOptions*>(jwrite_options_handle);
|
|
|
|
rocksdb_merge_helper(env, db, *write_options, nullptr, jkey, jkey_off,
|
|
|
|
jkey_len, jval, jval_off, jval_len);
|
2014-04-22 00:52:59 +02:00
|
|
|
}
|
|
|
|
|
2014-10-13 10:34:52 +02:00
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
2019-02-22 23:36:38 +01:00
|
|
|
* Method: merge
|
|
|
|
* Signature: (JJ[BII[BIIJ)V
|
2014-10-13 10:34:52 +02:00
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
void Java_org_rocksdb_RocksDB_merge__JJ_3BII_3BIIJ(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle, jlong jwrite_options_handle,
|
|
|
|
jbyteArray jkey, jint jkey_off, jint jkey_len,
|
|
|
|
jbyteArray jval, jint jval_off, jint jval_len, jlong jcf_handle) {
|
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
|
|
|
auto* write_options =
|
|
|
|
reinterpret_cast<rocksdb::WriteOptions*>(jwrite_options_handle);
|
2017-02-28 01:26:12 +01:00
|
|
|
auto* cf_handle = reinterpret_cast<rocksdb::ColumnFamilyHandle*>(jcf_handle);
|
2014-10-13 10:34:52 +02:00
|
|
|
if (cf_handle != nullptr) {
|
2019-02-22 23:36:38 +01:00
|
|
|
rocksdb_merge_helper(env, db, *write_options, cf_handle, jkey, jkey_off,
|
|
|
|
jkey_len, jval, jval_off, jval_len);
|
2014-10-13 10:34:52 +02:00
|
|
|
} else {
|
2018-04-13 02:55:14 +02:00
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(
|
|
|
|
env, rocksdb::Status::InvalidArgument("Invalid ColumnFamilyHandle."));
|
2014-10-13 10:34:52 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
jlong rocksdb_iterator_helper(rocksdb::DB* db,
|
|
|
|
rocksdb::ReadOptions read_options,
|
|
|
|
rocksdb::ColumnFamilyHandle* cf_handle) {
|
|
|
|
rocksdb::Iterator* iterator = nullptr;
|
|
|
|
if (cf_handle != nullptr) {
|
|
|
|
iterator = db->NewIterator(read_options, cf_handle);
|
|
|
|
} else {
|
|
|
|
iterator = db->NewIterator(read_options);
|
|
|
|
}
|
|
|
|
return reinterpret_cast<jlong>(iterator);
|
|
|
|
}
|
|
|
|
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
// rocksdb::DB::Write
|
2014-04-22 00:52:59 +02:00
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
2019-02-22 23:36:38 +01:00
|
|
|
* Method: write0
|
|
|
|
* Signature: (JJJ)V
|
2014-04-22 00:52:59 +02:00
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
void Java_org_rocksdb_RocksDB_write0(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle,
|
|
|
|
jlong jwrite_options_handle, jlong jwb_handle) {
|
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
|
|
|
auto* write_options =
|
|
|
|
reinterpret_cast<rocksdb::WriteOptions*>(jwrite_options_handle);
|
|
|
|
auto* wb = reinterpret_cast<rocksdb::WriteBatch*>(jwb_handle);
|
|
|
|
|
|
|
|
rocksdb::Status s = db->Write(*write_options, wb);
|
|
|
|
|
|
|
|
if (!s.ok()) {
|
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(env, s);
|
|
|
|
}
|
2014-04-22 00:52:59 +02:00
|
|
|
}
|
|
|
|
|
2014-10-13 10:34:52 +02:00
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
2019-02-22 23:36:38 +01:00
|
|
|
* Method: write1
|
|
|
|
* Signature: (JJJ)V
|
2014-10-13 10:34:52 +02:00
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
void Java_org_rocksdb_RocksDB_write1(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle,
|
|
|
|
jlong jwrite_options_handle, jlong jwbwi_handle) {
|
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
|
|
|
auto* write_options =
|
|
|
|
reinterpret_cast<rocksdb::WriteOptions*>(jwrite_options_handle);
|
|
|
|
auto* wbwi = reinterpret_cast<rocksdb::WriteBatchWithIndex*>(jwbwi_handle);
|
|
|
|
auto* wb = wbwi->GetWriteBatch();
|
|
|
|
|
|
|
|
rocksdb::Status s = db->Write(*write_options, wb);
|
|
|
|
|
|
|
|
if (!s.ok()) {
|
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(env, s);
|
2014-10-13 10:34:52 +02:00
|
|
|
}
|
|
|
|
}
|
2017-02-28 01:26:12 +01:00
|
|
|
|
2014-04-02 22:14:55 +02:00
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
2019-02-22 23:36:38 +01:00
|
|
|
// rocksdb::DB::Get
|
2017-02-28 01:26:12 +01:00
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
jbyteArray rocksdb_get_helper(
|
|
|
|
JNIEnv* env, rocksdb::DB* db,
|
|
|
|
const rocksdb::ReadOptions& read_opt,
|
|
|
|
rocksdb::ColumnFamilyHandle* column_family_handle,
|
|
|
|
jbyteArray jkey, jint jkey_off, jint jkey_len) {
|
2016-09-12 20:51:08 +02:00
|
|
|
jbyte* key = new jbyte[jkey_len];
|
|
|
|
env->GetByteArrayRegion(jkey, jkey_off, jkey_len, key);
|
2018-04-13 02:55:14 +02:00
|
|
|
if (env->ExceptionCheck()) {
|
2017-02-28 01:26:12 +01:00
|
|
|
// exception thrown: ArrayIndexOutOfBoundsException
|
2018-04-13 02:55:14 +02:00
|
|
|
delete[] key;
|
2019-02-22 23:36:38 +01:00
|
|
|
return nullptr;
|
2017-02-28 01:26:12 +01:00
|
|
|
}
|
2019-02-22 23:36:38 +01:00
|
|
|
|
2014-04-02 22:14:55 +02:00
|
|
|
rocksdb::Slice key_slice(reinterpret_cast<char*>(key), jkey_len);
|
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
std::string value;
|
2014-10-13 10:34:52 +02:00
|
|
|
rocksdb::Status s;
|
2019-02-22 23:36:38 +01:00
|
|
|
if (column_family_handle != nullptr) {
|
|
|
|
s = db->Get(read_opt, column_family_handle, key_slice, &value);
|
2014-10-13 10:34:52 +02:00
|
|
|
} else {
|
|
|
|
// backwards compatibility
|
2019-02-22 23:36:38 +01:00
|
|
|
s = db->Get(read_opt, key_slice, &value);
|
2014-10-13 10:34:52 +02:00
|
|
|
}
|
2016-09-12 20:51:08 +02:00
|
|
|
|
|
|
|
// cleanup
|
2018-04-13 02:55:14 +02:00
|
|
|
delete[] key;
|
2014-04-02 22:14:55 +02:00
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
if (s.IsNotFound()) {
|
|
|
|
return nullptr;
|
|
|
|
}
|
|
|
|
|
2017-02-28 01:26:12 +01:00
|
|
|
if (s.ok()) {
|
2019-02-22 23:36:38 +01:00
|
|
|
jbyteArray jret_value = rocksdb::JniUtil::copyBytes(env, value);
|
|
|
|
if (jret_value == nullptr) {
|
|
|
|
// exception occurred
|
|
|
|
return nullptr;
|
|
|
|
}
|
|
|
|
return jret_value;
|
2014-04-02 22:14:55 +02:00
|
|
|
}
|
2017-02-28 01:26:12 +01:00
|
|
|
|
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(env, s);
|
2019-02-22 23:36:38 +01:00
|
|
|
return nullptr;
|
2014-04-02 22:14:55 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
2019-02-22 23:36:38 +01:00
|
|
|
* Method: get
|
|
|
|
* Signature: (J[BII)[B
|
2014-04-02 22:14:55 +02:00
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
jbyteArray Java_org_rocksdb_RocksDB_get__J_3BII(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle,
|
|
|
|
jbyteArray jkey, jint jkey_off, jint jkey_len) {
|
|
|
|
return rocksdb_get_helper(env, reinterpret_cast<rocksdb::DB*>(jdb_handle),
|
|
|
|
rocksdb::ReadOptions(), nullptr, jkey, jkey_off, jkey_len);
|
2014-10-13 10:34:52 +02:00
|
|
|
}
|
2014-04-02 22:14:55 +02:00
|
|
|
|
2014-10-13 10:34:52 +02:00
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
2019-02-22 23:36:38 +01:00
|
|
|
* Method: get
|
|
|
|
* Signature: (J[BIIJ)[B
|
2014-10-13 10:34:52 +02:00
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
jbyteArray Java_org_rocksdb_RocksDB_get__J_3BIIJ(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle,
|
|
|
|
jbyteArray jkey, jint jkey_off, jint jkey_len, jlong jcf_handle) {
|
|
|
|
auto db_handle = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
|
|
|
auto cf_handle = reinterpret_cast<rocksdb::ColumnFamilyHandle*>(jcf_handle);
|
2014-10-13 10:34:52 +02:00
|
|
|
if (cf_handle != nullptr) {
|
2019-02-22 23:36:38 +01:00
|
|
|
return rocksdb_get_helper(env, db_handle, rocksdb::ReadOptions(), cf_handle,
|
|
|
|
jkey, jkey_off, jkey_len);
|
2014-10-13 10:34:52 +02:00
|
|
|
} else {
|
2018-04-13 02:55:14 +02:00
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(
|
|
|
|
env, rocksdb::Status::InvalidArgument("Invalid ColumnFamilyHandle."));
|
2019-02-22 23:36:38 +01:00
|
|
|
return nullptr;
|
2014-10-13 10:34:52 +02:00
|
|
|
}
|
2014-04-02 22:14:55 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
2019-02-22 23:36:38 +01:00
|
|
|
* Method: get
|
|
|
|
* Signature: (JJ[BII)[B
|
2014-04-02 22:14:55 +02:00
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
jbyteArray Java_org_rocksdb_RocksDB_get__JJ_3BII(
|
|
|
|
JNIEnv* env, jobject,
|
|
|
|
jlong jdb_handle, jlong jropt_handle,
|
|
|
|
jbyteArray jkey, jint jkey_off, jint jkey_len) {
|
|
|
|
return rocksdb_get_helper(
|
|
|
|
env, reinterpret_cast<rocksdb::DB*>(jdb_handle),
|
|
|
|
*reinterpret_cast<rocksdb::ReadOptions*>(jropt_handle), nullptr, jkey,
|
|
|
|
jkey_off, jkey_len);
|
2014-04-02 22:14:55 +02:00
|
|
|
}
|
|
|
|
|
2014-10-13 10:34:52 +02:00
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
2019-02-22 23:36:38 +01:00
|
|
|
* Method: get
|
|
|
|
* Signature: (JJ[BIIJ)[B
|
2014-10-13 10:34:52 +02:00
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
jbyteArray Java_org_rocksdb_RocksDB_get__JJ_3BIIJ(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle, jlong jropt_handle,
|
2018-04-13 02:55:14 +02:00
|
|
|
jbyteArray jkey, jint jkey_off, jint jkey_len, jlong jcf_handle) {
|
2019-02-22 23:36:38 +01:00
|
|
|
auto* db_handle = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
|
|
|
auto& ro_opt = *reinterpret_cast<rocksdb::ReadOptions*>(jropt_handle);
|
2017-02-28 01:26:12 +01:00
|
|
|
auto* cf_handle = reinterpret_cast<rocksdb::ColumnFamilyHandle*>(jcf_handle);
|
2014-10-13 10:34:52 +02:00
|
|
|
if (cf_handle != nullptr) {
|
2019-02-22 23:36:38 +01:00
|
|
|
return rocksdb_get_helper(
|
|
|
|
env, db_handle, ro_opt, cf_handle, jkey, jkey_off, jkey_len);
|
2014-10-13 10:34:52 +02:00
|
|
|
} else {
|
2018-04-13 02:55:14 +02:00
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(
|
|
|
|
env, rocksdb::Status::InvalidArgument("Invalid ColumnFamilyHandle."));
|
2019-02-22 23:36:38 +01:00
|
|
|
return nullptr;
|
2014-10-13 10:34:52 +02:00
|
|
|
}
|
|
|
|
}
|
2016-08-22 20:02:31 +02:00
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
jint rocksdb_get_helper(
|
|
|
|
JNIEnv* env, rocksdb::DB* db, const rocksdb::ReadOptions& read_options,
|
|
|
|
rocksdb::ColumnFamilyHandle* column_family_handle,
|
|
|
|
jbyteArray jkey, jint jkey_off, jint jkey_len,
|
|
|
|
jbyteArray jval, jint jval_off, jint jval_len,
|
|
|
|
bool* has_exception) {
|
|
|
|
static const int kNotFound = -1;
|
|
|
|
static const int kStatusError = -2;
|
|
|
|
|
|
|
|
jbyte* key = new jbyte[jkey_len];
|
|
|
|
env->GetByteArrayRegion(jkey, jkey_off, jkey_len, key);
|
|
|
|
if (env->ExceptionCheck()) {
|
2017-02-28 01:26:12 +01:00
|
|
|
// exception thrown: OutOfMemoryError
|
2019-02-22 23:36:38 +01:00
|
|
|
delete[] key;
|
|
|
|
*has_exception = true;
|
|
|
|
return kStatusError;
|
2017-02-28 01:26:12 +01:00
|
|
|
}
|
2016-08-22 20:02:31 +02:00
|
|
|
rocksdb::Slice key_slice(reinterpret_cast<char*>(key), jkey_len);
|
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
// TODO(yhchiang): we might save one memory allocation here by adding
|
|
|
|
// a DB::Get() function which takes preallocated jbyte* as input.
|
|
|
|
std::string cvalue;
|
2016-08-22 20:02:31 +02:00
|
|
|
rocksdb::Status s;
|
2019-02-22 23:36:38 +01:00
|
|
|
if (column_family_handle != nullptr) {
|
|
|
|
s = db->Get(read_options, column_family_handle, key_slice, &cvalue);
|
2016-08-22 20:02:31 +02:00
|
|
|
} else {
|
|
|
|
// backwards compatibility
|
2019-02-22 23:36:38 +01:00
|
|
|
s = db->Get(read_options, key_slice, &cvalue);
|
2016-08-22 20:02:31 +02:00
|
|
|
}
|
2017-02-28 01:26:12 +01:00
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
// cleanup
|
|
|
|
delete[] key;
|
2016-08-22 20:02:31 +02:00
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
if (s.IsNotFound()) {
|
|
|
|
*has_exception = false;
|
|
|
|
return kNotFound;
|
|
|
|
} else if (!s.ok()) {
|
|
|
|
*has_exception = true;
|
|
|
|
// Here since we are throwing a Java exception from c++ side.
|
|
|
|
// As a result, c++ does not know calling this function will in fact
|
|
|
|
// throwing an exception. As a result, the execution flow will
|
|
|
|
// not stop here, and codes after this throw will still be
|
|
|
|
// executed.
|
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(env, s);
|
|
|
|
|
|
|
|
// Return a dummy const value to avoid compilation error, although
|
|
|
|
// java side might not have a chance to get the return value :)
|
|
|
|
return kStatusError;
|
2016-08-22 20:02:31 +02:00
|
|
|
}
|
2017-02-28 01:26:12 +01:00
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
const jint cvalue_len = static_cast<jint>(cvalue.size());
|
|
|
|
const jint length = std::min(jval_len, cvalue_len);
|
|
|
|
|
|
|
|
env->SetByteArrayRegion(
|
|
|
|
jval, jval_off, length,
|
|
|
|
const_cast<jbyte*>(reinterpret_cast<const jbyte*>(cvalue.c_str())));
|
|
|
|
if (env->ExceptionCheck()) {
|
|
|
|
// exception thrown: OutOfMemoryError
|
|
|
|
*has_exception = true;
|
|
|
|
return kStatusError;
|
|
|
|
}
|
|
|
|
|
|
|
|
*has_exception = false;
|
|
|
|
return cvalue_len;
|
2016-08-22 20:02:31 +02:00
|
|
|
}
|
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
|
2016-08-22 20:02:31 +02:00
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
2019-02-22 23:36:38 +01:00
|
|
|
* Method: get
|
|
|
|
* Signature: (J[BII[BII)I
|
2016-08-22 20:02:31 +02:00
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
jint Java_org_rocksdb_RocksDB_get__J_3BII_3BII(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle,
|
|
|
|
jbyteArray jkey, jint jkey_off, jint jkey_len,
|
|
|
|
jbyteArray jval, jint jval_off, jint jval_len) {
|
|
|
|
bool has_exception = false;
|
|
|
|
return rocksdb_get_helper(env, reinterpret_cast<rocksdb::DB*>(jdb_handle),
|
|
|
|
rocksdb::ReadOptions(), nullptr, jkey, jkey_off,
|
|
|
|
jkey_len, jval, jval_off, jval_len, &has_exception);
|
2016-08-22 20:02:31 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
2019-02-22 23:36:38 +01:00
|
|
|
* Method: get
|
|
|
|
* Signature: (J[BII[BIIJ)I
|
|
|
|
*/
|
|
|
|
jint Java_org_rocksdb_RocksDB_get__J_3BII_3BIIJ(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle,
|
|
|
|
jbyteArray jkey, jint jkey_off, jint jkey_len,
|
|
|
|
jbyteArray jval, jint jval_off, jint jval_len,
|
|
|
|
jlong jcf_handle) {
|
|
|
|
auto* db_handle = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
2017-02-28 01:26:12 +01:00
|
|
|
auto* cf_handle = reinterpret_cast<rocksdb::ColumnFamilyHandle*>(jcf_handle);
|
2016-08-22 20:02:31 +02:00
|
|
|
if (cf_handle != nullptr) {
|
2019-02-22 23:36:38 +01:00
|
|
|
bool has_exception = false;
|
|
|
|
return rocksdb_get_helper(env, db_handle, rocksdb::ReadOptions(), cf_handle,
|
|
|
|
jkey, jkey_off, jkey_len, jval, jval_off,
|
|
|
|
jval_len, &has_exception);
|
2016-08-22 20:02:31 +02:00
|
|
|
} else {
|
2018-04-13 02:55:14 +02:00
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(
|
|
|
|
env, rocksdb::Status::InvalidArgument("Invalid ColumnFamilyHandle."));
|
2019-02-22 23:36:38 +01:00
|
|
|
// will never be evaluated
|
|
|
|
return 0;
|
2016-08-22 20:02:31 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
2019-02-22 23:36:38 +01:00
|
|
|
* Method: get
|
|
|
|
* Signature: (JJ[BII[BII)I
|
2016-08-22 20:02:31 +02:00
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
jint Java_org_rocksdb_RocksDB_get__JJ_3BII_3BII(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle, jlong jropt_handle,
|
|
|
|
jbyteArray jkey, jint jkey_off, jint jkey_len,
|
|
|
|
jbyteArray jval, jint jval_off, jint jval_len) {
|
|
|
|
bool has_exception = false;
|
|
|
|
return rocksdb_get_helper(
|
|
|
|
env, reinterpret_cast<rocksdb::DB*>(jdb_handle),
|
|
|
|
*reinterpret_cast<rocksdb::ReadOptions*>(jropt_handle), nullptr, jkey,
|
|
|
|
jkey_off, jkey_len, jval, jval_off, jval_len, &has_exception);
|
2016-08-22 20:02:31 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
2019-02-22 23:36:38 +01:00
|
|
|
* Method: get
|
|
|
|
* Signature: (JJ[BII[BIIJ)I
|
2016-08-22 20:02:31 +02:00
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
jint Java_org_rocksdb_RocksDB_get__JJ_3BII_3BIIJ(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle, jlong jropt_handle,
|
|
|
|
jbyteArray jkey, jint jkey_off, jint jkey_len,
|
|
|
|
jbyteArray jval, jint jval_off, jint jval_len,
|
|
|
|
jlong jcf_handle) {
|
|
|
|
auto* db_handle = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
|
|
|
auto& ro_opt = *reinterpret_cast<rocksdb::ReadOptions*>(jropt_handle);
|
2017-02-28 01:26:12 +01:00
|
|
|
auto* cf_handle = reinterpret_cast<rocksdb::ColumnFamilyHandle*>(jcf_handle);
|
2016-08-22 20:02:31 +02:00
|
|
|
if (cf_handle != nullptr) {
|
2019-02-22 23:36:38 +01:00
|
|
|
bool has_exception = false;
|
|
|
|
return rocksdb_get_helper(env, db_handle, ro_opt, cf_handle,
|
|
|
|
jkey, jkey_off, jkey_len,
|
|
|
|
jval, jval_off, jval_len,
|
|
|
|
&has_exception);
|
2016-08-22 20:02:31 +02:00
|
|
|
} else {
|
2018-04-13 02:55:14 +02:00
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(
|
|
|
|
env, rocksdb::Status::InvalidArgument("Invalid ColumnFamilyHandle."));
|
2019-02-22 23:36:38 +01:00
|
|
|
// will never be evaluated
|
|
|
|
return 0;
|
2016-08-22 20:02:31 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
inline void multi_get_helper_release_keys(
|
|
|
|
JNIEnv* env, std::vector<std::pair<jbyte*, jobject>>& keys_to_free) {
|
|
|
|
auto end = keys_to_free.end();
|
|
|
|
for (auto it = keys_to_free.begin(); it != end; ++it) {
|
|
|
|
delete[] it->first;
|
|
|
|
env->DeleteLocalRef(it->second);
|
|
|
|
}
|
|
|
|
keys_to_free.clear();
|
|
|
|
}
|
|
|
|
|
2017-03-07 07:13:53 +01:00
|
|
|
/**
|
2019-02-22 23:36:38 +01:00
|
|
|
* cf multi get
|
|
|
|
*
|
|
|
|
* @return byte[][] of values or nullptr if an exception occurs
|
2017-03-07 07:13:53 +01:00
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
jobjectArray multi_get_helper(
|
|
|
|
JNIEnv* env, jobject, rocksdb::DB* db, const rocksdb::ReadOptions& rOpt,
|
|
|
|
jobjectArray jkeys, jintArray jkey_offs, jintArray jkey_lens,
|
|
|
|
jlongArray jcolumn_family_handles) {
|
|
|
|
std::vector<rocksdb::ColumnFamilyHandle*> cf_handles;
|
|
|
|
if (jcolumn_family_handles != nullptr) {
|
|
|
|
const jsize len_cols = env->GetArrayLength(jcolumn_family_handles);
|
|
|
|
|
|
|
|
jlong* jcfh = env->GetLongArrayElements(jcolumn_family_handles, nullptr);
|
|
|
|
if (jcfh == nullptr) {
|
|
|
|
// exception thrown: OutOfMemoryError
|
|
|
|
return nullptr;
|
|
|
|
}
|
|
|
|
|
|
|
|
for (jsize i = 0; i < len_cols; i++) {
|
|
|
|
auto* cf_handle = reinterpret_cast<rocksdb::ColumnFamilyHandle*>(jcfh[i]);
|
|
|
|
cf_handles.push_back(cf_handle);
|
|
|
|
}
|
|
|
|
env->ReleaseLongArrayElements(jcolumn_family_handles, jcfh, JNI_ABORT);
|
2017-03-07 07:13:53 +01:00
|
|
|
}
|
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
const jsize len_keys = env->GetArrayLength(jkeys);
|
|
|
|
if (env->EnsureLocalCapacity(len_keys) != 0) {
|
|
|
|
// exception thrown: OutOfMemoryError
|
|
|
|
return nullptr;
|
2017-03-07 07:13:53 +01:00
|
|
|
}
|
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
jint* jkey_off = env->GetIntArrayElements(jkey_offs, nullptr);
|
|
|
|
if (jkey_off == nullptr) {
|
|
|
|
// exception thrown: OutOfMemoryError
|
|
|
|
return nullptr;
|
|
|
|
}
|
2017-03-07 07:13:53 +01:00
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
jint* jkey_len = env->GetIntArrayElements(jkey_lens, nullptr);
|
|
|
|
if (jkey_len == nullptr) {
|
|
|
|
// exception thrown: OutOfMemoryError
|
|
|
|
env->ReleaseIntArrayElements(jkey_offs, jkey_off, JNI_ABORT);
|
|
|
|
return nullptr;
|
|
|
|
}
|
2017-03-07 07:13:53 +01:00
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
std::vector<rocksdb::Slice> keys;
|
|
|
|
std::vector<std::pair<jbyte*, jobject>> keys_to_free;
|
|
|
|
for (jsize i = 0; i < len_keys; i++) {
|
|
|
|
jobject jkey = env->GetObjectArrayElement(jkeys, i);
|
|
|
|
if (env->ExceptionCheck()) {
|
|
|
|
// exception thrown: ArrayIndexOutOfBoundsException
|
|
|
|
env->ReleaseIntArrayElements(jkey_lens, jkey_len, JNI_ABORT);
|
|
|
|
env->ReleaseIntArrayElements(jkey_offs, jkey_off, JNI_ABORT);
|
|
|
|
multi_get_helper_release_keys(env, keys_to_free);
|
|
|
|
return nullptr;
|
|
|
|
}
|
|
|
|
|
|
|
|
jbyteArray jkey_ba = reinterpret_cast<jbyteArray>(jkey);
|
|
|
|
|
|
|
|
const jint len_key = jkey_len[i];
|
|
|
|
jbyte* key = new jbyte[len_key];
|
|
|
|
env->GetByteArrayRegion(jkey_ba, jkey_off[i], len_key, key);
|
|
|
|
if (env->ExceptionCheck()) {
|
|
|
|
// exception thrown: ArrayIndexOutOfBoundsException
|
|
|
|
delete[] key;
|
|
|
|
env->DeleteLocalRef(jkey);
|
|
|
|
env->ReleaseIntArrayElements(jkey_lens, jkey_len, JNI_ABORT);
|
|
|
|
env->ReleaseIntArrayElements(jkey_offs, jkey_off, JNI_ABORT);
|
|
|
|
multi_get_helper_release_keys(env, keys_to_free);
|
|
|
|
return nullptr;
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Slice key_slice(reinterpret_cast<char*>(key), len_key);
|
|
|
|
keys.push_back(key_slice);
|
|
|
|
|
|
|
|
keys_to_free.push_back(std::pair<jbyte*, jobject>(key, jkey));
|
2017-03-07 07:13:53 +01:00
|
|
|
}
|
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
// cleanup jkey_off and jken_len
|
|
|
|
env->ReleaseIntArrayElements(jkey_lens, jkey_len, JNI_ABORT);
|
|
|
|
env->ReleaseIntArrayElements(jkey_offs, jkey_off, JNI_ABORT);
|
|
|
|
|
|
|
|
std::vector<std::string> values;
|
|
|
|
std::vector<rocksdb::Status> s;
|
|
|
|
if (cf_handles.size() == 0) {
|
|
|
|
s = db->MultiGet(rOpt, keys, &values);
|
|
|
|
} else {
|
|
|
|
s = db->MultiGet(rOpt, cf_handles, keys, &values);
|
|
|
|
}
|
|
|
|
|
|
|
|
// free up allocated byte arrays
|
|
|
|
multi_get_helper_release_keys(env, keys_to_free);
|
|
|
|
|
|
|
|
// prepare the results
|
|
|
|
jobjectArray jresults =
|
|
|
|
rocksdb::ByteJni::new2dByteArray(env, static_cast<jsize>(s.size()));
|
|
|
|
if (jresults == nullptr) {
|
|
|
|
// exception occurred
|
|
|
|
return nullptr;
|
|
|
|
}
|
|
|
|
|
|
|
|
// TODO(AR) it is not clear to me why EnsureLocalCapacity is needed for the
|
|
|
|
// loop as we cleanup references with env->DeleteLocalRef(jentry_value);
|
|
|
|
if (env->EnsureLocalCapacity(static_cast<jint>(s.size())) != 0) {
|
|
|
|
// exception thrown: OutOfMemoryError
|
|
|
|
return nullptr;
|
|
|
|
}
|
|
|
|
// add to the jresults
|
|
|
|
for (std::vector<rocksdb::Status>::size_type i = 0; i != s.size(); i++) {
|
|
|
|
if (s[i].ok()) {
|
|
|
|
std::string* value = &values[i];
|
|
|
|
const jsize jvalue_len = static_cast<jsize>(value->size());
|
|
|
|
jbyteArray jentry_value = env->NewByteArray(jvalue_len);
|
|
|
|
if (jentry_value == nullptr) {
|
|
|
|
// exception thrown: OutOfMemoryError
|
|
|
|
return nullptr;
|
|
|
|
}
|
|
|
|
|
|
|
|
env->SetByteArrayRegion(
|
|
|
|
jentry_value, 0, static_cast<jsize>(jvalue_len),
|
|
|
|
const_cast<jbyte*>(reinterpret_cast<const jbyte*>(value->c_str())));
|
|
|
|
if (env->ExceptionCheck()) {
|
|
|
|
// exception thrown: ArrayIndexOutOfBoundsException
|
|
|
|
env->DeleteLocalRef(jentry_value);
|
|
|
|
return nullptr;
|
|
|
|
}
|
|
|
|
|
|
|
|
env->SetObjectArrayElement(jresults, static_cast<jsize>(i), jentry_value);
|
|
|
|
if (env->ExceptionCheck()) {
|
|
|
|
// exception thrown: ArrayIndexOutOfBoundsException
|
|
|
|
env->DeleteLocalRef(jentry_value);
|
|
|
|
return nullptr;
|
|
|
|
}
|
|
|
|
|
|
|
|
env->DeleteLocalRef(jentry_value);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return jresults;
|
2017-03-07 07:13:53 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
2019-02-22 23:36:38 +01:00
|
|
|
* Method: multiGet
|
|
|
|
* Signature: (J[[B[I[I)[[B
|
2017-03-07 07:13:53 +01:00
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
jobjectArray Java_org_rocksdb_RocksDB_multiGet__J_3_3B_3I_3I(
|
|
|
|
JNIEnv* env, jobject jdb, jlong jdb_handle,
|
|
|
|
jobjectArray jkeys, jintArray jkey_offs, jintArray jkey_lens) {
|
|
|
|
return multi_get_helper(env, jdb, reinterpret_cast<rocksdb::DB*>(jdb_handle),
|
|
|
|
rocksdb::ReadOptions(), jkeys, jkey_offs, jkey_lens,
|
|
|
|
nullptr);
|
2017-03-07 07:13:53 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
2019-02-22 23:36:38 +01:00
|
|
|
* Method: multiGet
|
|
|
|
* Signature: (J[[B[I[I[J)[[B
|
2017-03-07 07:13:53 +01:00
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
jobjectArray Java_org_rocksdb_RocksDB_multiGet__J_3_3B_3I_3I_3J(
|
|
|
|
JNIEnv* env, jobject jdb, jlong jdb_handle,
|
|
|
|
jobjectArray jkeys, jintArray jkey_offs, jintArray jkey_lens,
|
|
|
|
jlongArray jcolumn_family_handles) {
|
|
|
|
return multi_get_helper(env, jdb, reinterpret_cast<rocksdb::DB*>(jdb_handle),
|
|
|
|
rocksdb::ReadOptions(), jkeys, jkey_offs, jkey_lens,
|
|
|
|
jcolumn_family_handles);
|
2017-03-07 07:13:53 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
2019-02-22 23:36:38 +01:00
|
|
|
* Method: multiGet
|
|
|
|
* Signature: (JJ[[B[I[I)[[B
|
2017-03-07 07:13:53 +01:00
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
jobjectArray Java_org_rocksdb_RocksDB_multiGet__JJ_3_3B_3I_3I(
|
|
|
|
JNIEnv* env, jobject jdb, jlong jdb_handle, jlong jropt_handle,
|
|
|
|
jobjectArray jkeys, jintArray jkey_offs, jintArray jkey_lens) {
|
|
|
|
return multi_get_helper(
|
|
|
|
env, jdb, reinterpret_cast<rocksdb::DB*>(jdb_handle),
|
|
|
|
*reinterpret_cast<rocksdb::ReadOptions*>(jropt_handle), jkeys, jkey_offs,
|
|
|
|
jkey_lens, nullptr);
|
2017-03-07 07:13:53 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
2019-02-22 23:36:38 +01:00
|
|
|
* Class: org_rocksdb_RocksDB
|
|
|
|
* Method: multiGet
|
|
|
|
* Signature: (JJ[[B[I[I[J)[[B
|
2017-03-07 07:13:53 +01:00
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
jobjectArray Java_org_rocksdb_RocksDB_multiGet__JJ_3_3B_3I_3I_3J(
|
|
|
|
JNIEnv* env, jobject jdb, jlong jdb_handle, jlong jropt_handle,
|
|
|
|
jobjectArray jkeys, jintArray jkey_offs, jintArray jkey_lens,
|
|
|
|
jlongArray jcolumn_family_handles) {
|
|
|
|
return multi_get_helper(
|
|
|
|
env, jdb, reinterpret_cast<rocksdb::DB*>(jdb_handle),
|
|
|
|
*reinterpret_cast<rocksdb::ReadOptions*>(jropt_handle), jkeys, jkey_offs,
|
|
|
|
jkey_lens, jcolumn_family_handles);
|
2017-03-07 07:13:53 +01:00
|
|
|
}
|
|
|
|
|
2014-09-16 22:58:49 +02:00
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
2019-02-22 23:36:38 +01:00
|
|
|
// rocksdb::DB::KeyMayExist
|
|
|
|
jboolean key_may_exist_helper(JNIEnv* env, rocksdb::DB* db,
|
|
|
|
const rocksdb::ReadOptions& read_opt,
|
|
|
|
rocksdb::ColumnFamilyHandle* cf_handle,
|
|
|
|
jbyteArray jkey, jint jkey_off, jint jkey_len,
|
|
|
|
jobject jstring_builder, bool* has_exception) {
|
2016-09-12 20:51:08 +02:00
|
|
|
jbyte* key = new jbyte[jkey_len];
|
|
|
|
env->GetByteArrayRegion(jkey, jkey_off, jkey_len, key);
|
2018-04-13 02:55:14 +02:00
|
|
|
if (env->ExceptionCheck()) {
|
2017-02-28 01:26:12 +01:00
|
|
|
// exception thrown: ArrayIndexOutOfBoundsException
|
2018-04-13 02:55:14 +02:00
|
|
|
delete[] key;
|
2019-02-22 23:36:38 +01:00
|
|
|
*has_exception = true;
|
2017-02-28 01:26:12 +01:00
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
rocksdb::Slice key_slice(reinterpret_cast<char*>(key), jkey_len);
|
2014-09-16 22:58:49 +02:00
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
std::string value;
|
|
|
|
bool value_found = false;
|
|
|
|
bool keyMayExist;
|
2014-10-29 18:40:44 +01:00
|
|
|
if (cf_handle != nullptr) {
|
2019-02-22 23:36:38 +01:00
|
|
|
keyMayExist =
|
|
|
|
db->KeyMayExist(read_opt, cf_handle, key_slice, &value, &value_found);
|
2014-10-29 18:40:44 +01:00
|
|
|
} else {
|
2019-02-22 23:36:38 +01:00
|
|
|
keyMayExist = db->KeyMayExist(read_opt, key_slice, &value, &value_found);
|
2014-10-29 18:40:44 +01:00
|
|
|
}
|
2014-09-16 22:58:49 +02:00
|
|
|
|
2016-09-12 20:51:08 +02:00
|
|
|
// cleanup
|
2018-04-13 02:55:14 +02:00
|
|
|
delete[] key;
|
2014-09-16 22:58:49 +02:00
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
// extract the value
|
|
|
|
if (value_found && !value.empty()) {
|
|
|
|
jobject jresult_string_builder =
|
|
|
|
rocksdb::StringBuilderJni::append(env, jstring_builder, value.c_str());
|
|
|
|
if (jresult_string_builder == nullptr) {
|
|
|
|
*has_exception = true;
|
|
|
|
return false;
|
|
|
|
}
|
2014-09-16 22:58:49 +02:00
|
|
|
}
|
2017-02-28 01:26:12 +01:00
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
*has_exception = false;
|
|
|
|
return static_cast<jboolean>(keyMayExist);
|
2014-09-16 22:58:49 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
2019-02-22 23:36:38 +01:00
|
|
|
* Method: keyMayExist
|
|
|
|
* Signature: (J[BIILjava/lang/StringBuilder;)Z
|
2014-09-16 22:58:49 +02:00
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
jboolean Java_org_rocksdb_RocksDB_keyMayExist__J_3BIILjava_lang_StringBuilder_2(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle,
|
|
|
|
jbyteArray jkey, jint jkey_off, jint jkey_len, jobject jstring_builder) {
|
2017-02-28 01:26:12 +01:00
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
2019-02-22 23:36:38 +01:00
|
|
|
bool has_exception = false;
|
|
|
|
return key_may_exist_helper(env, db, rocksdb::ReadOptions(), nullptr, jkey,
|
|
|
|
jkey_off, jkey_len, jstring_builder, &has_exception);
|
2014-10-29 18:40:44 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
2019-02-22 23:36:38 +01:00
|
|
|
* Method: keyMayExist
|
|
|
|
* Signature: (J[BIIJLjava/lang/StringBuilder;)Z
|
2014-10-29 18:40:44 +01:00
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
jboolean
|
|
|
|
Java_org_rocksdb_RocksDB_keyMayExist__J_3BIIJLjava_lang_StringBuilder_2(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle,
|
|
|
|
jbyteArray jkey, jint jkey_off, jint jkey_len,
|
|
|
|
jlong jcf_handle, jobject jstring_builder) {
|
2017-02-28 01:26:12 +01:00
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
|
|
|
auto* cf_handle = reinterpret_cast<rocksdb::ColumnFamilyHandle*>(jcf_handle);
|
2014-10-29 18:40:44 +01:00
|
|
|
if (cf_handle != nullptr) {
|
2019-02-22 23:36:38 +01:00
|
|
|
bool has_exception = false;
|
|
|
|
return key_may_exist_helper(env, db, rocksdb::ReadOptions(), cf_handle,
|
|
|
|
jkey, jkey_off, jkey_len, jstring_builder,
|
|
|
|
&has_exception);
|
2014-10-29 18:40:44 +01:00
|
|
|
} else {
|
2018-04-13 02:55:14 +02:00
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(
|
|
|
|
env, rocksdb::Status::InvalidArgument("Invalid ColumnFamilyHandle."));
|
2019-02-22 23:36:38 +01:00
|
|
|
return true;
|
2014-10-29 18:40:44 +01:00
|
|
|
}
|
2014-09-16 22:58:49 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
2019-02-22 23:36:38 +01:00
|
|
|
* Method: keyMayExist
|
|
|
|
* Signature: (JJ[BIILjava/lang/StringBuilder;)Z
|
2014-09-16 22:58:49 +02:00
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
jboolean
|
|
|
|
Java_org_rocksdb_RocksDB_keyMayExist__JJ_3BIILjava_lang_StringBuilder_2(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle, jlong jread_options_handle,
|
|
|
|
jbyteArray jkey, jint jkey_off, jint jkey_len, jobject jstring_builder) {
|
2017-02-28 01:26:12 +01:00
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
2019-02-22 23:36:38 +01:00
|
|
|
auto& read_options =
|
|
|
|
*reinterpret_cast<rocksdb::ReadOptions*>(jread_options_handle);
|
|
|
|
bool has_exception = false;
|
|
|
|
return key_may_exist_helper(env, db, read_options, nullptr, jkey, jkey_off,
|
|
|
|
jkey_len, jstring_builder, &has_exception);
|
2014-10-29 18:40:44 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
2019-02-22 23:36:38 +01:00
|
|
|
* Method: keyMayExist
|
|
|
|
* Signature: (JJ[BIIJLjava/lang/StringBuilder;)Z
|
2014-10-29 18:40:44 +01:00
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
jboolean
|
|
|
|
Java_org_rocksdb_RocksDB_keyMayExist__JJ_3BIIJLjava_lang_StringBuilder_2(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle, jlong jread_options_handle,
|
|
|
|
jbyteArray jkey, jint jkey_off, jint jkey_len, jlong jcf_handle,
|
|
|
|
jobject jstring_builder) {
|
2017-02-28 01:26:12 +01:00
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
2019-02-22 23:36:38 +01:00
|
|
|
auto& read_options =
|
|
|
|
*reinterpret_cast<rocksdb::ReadOptions*>(jread_options_handle);
|
2017-02-28 01:26:12 +01:00
|
|
|
auto* cf_handle = reinterpret_cast<rocksdb::ColumnFamilyHandle*>(jcf_handle);
|
2014-10-29 18:40:44 +01:00
|
|
|
if (cf_handle != nullptr) {
|
2019-02-22 23:36:38 +01:00
|
|
|
bool has_exception = false;
|
|
|
|
return key_may_exist_helper(env, db, read_options, cf_handle, jkey,
|
|
|
|
jkey_off, jkey_len, jstring_builder, &has_exception);
|
2014-10-29 18:40:44 +01:00
|
|
|
} else {
|
2018-04-13 02:55:14 +02:00
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(
|
|
|
|
env, rocksdb::Status::InvalidArgument("Invalid ColumnFamilyHandle."));
|
2019-02-22 23:36:38 +01:00
|
|
|
return true;
|
2014-10-29 18:40:44 +01:00
|
|
|
}
|
2014-09-16 22:58:49 +02:00
|
|
|
}
|
|
|
|
|
2014-04-19 22:21:06 +02:00
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
2015-01-16 23:35:21 +01:00
|
|
|
* Method: iterator
|
2014-04-19 22:21:06 +02:00
|
|
|
* Signature: (J)J
|
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
jlong Java_org_rocksdb_RocksDB_iterator__J(
|
|
|
|
JNIEnv*, jobject, jlong db_handle) {
|
2017-02-28 01:26:12 +01:00
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(db_handle);
|
2018-04-13 02:55:14 +02:00
|
|
|
return rocksdb_iterator_helper(db, rocksdb::ReadOptions(), nullptr);
|
2014-04-19 22:05:21 +02:00
|
|
|
}
|
2014-09-24 20:43:35 +02:00
|
|
|
|
2014-10-13 10:34:52 +02:00
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
2015-01-16 23:35:21 +01:00
|
|
|
* Method: iterator
|
2014-10-13 10:34:52 +02:00
|
|
|
* Signature: (JJ)J
|
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
jlong Java_org_rocksdb_RocksDB_iterator__JJ(
|
|
|
|
JNIEnv*, jobject, jlong db_handle, jlong jread_options_handle) {
|
2017-02-28 01:26:12 +01:00
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(db_handle);
|
2018-04-13 02:55:14 +02:00
|
|
|
auto& read_options =
|
|
|
|
*reinterpret_cast<rocksdb::ReadOptions*>(jread_options_handle);
|
|
|
|
return rocksdb_iterator_helper(db, read_options, nullptr);
|
2015-01-16 23:35:21 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
|
|
|
* Method: iteratorCF
|
|
|
|
* Signature: (JJ)J
|
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
jlong Java_org_rocksdb_RocksDB_iteratorCF__JJ(
|
|
|
|
JNIEnv*, jobject, jlong db_handle, jlong jcf_handle) {
|
2017-02-28 01:26:12 +01:00
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(db_handle);
|
|
|
|
auto* cf_handle = reinterpret_cast<rocksdb::ColumnFamilyHandle*>(jcf_handle);
|
2018-04-13 02:55:14 +02:00
|
|
|
return rocksdb_iterator_helper(db, rocksdb::ReadOptions(), cf_handle);
|
2015-01-16 23:35:21 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
|
|
|
* Method: iteratorCF
|
|
|
|
* Signature: (JJJ)J
|
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
jlong Java_org_rocksdb_RocksDB_iteratorCF__JJJ(
|
|
|
|
JNIEnv*, jobject,
|
|
|
|
jlong db_handle, jlong jcf_handle, jlong jread_options_handle) {
|
2017-02-28 01:26:12 +01:00
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(db_handle);
|
|
|
|
auto* cf_handle = reinterpret_cast<rocksdb::ColumnFamilyHandle*>(jcf_handle);
|
2018-04-13 02:55:14 +02:00
|
|
|
auto& read_options =
|
|
|
|
*reinterpret_cast<rocksdb::ReadOptions*>(jread_options_handle);
|
|
|
|
return rocksdb_iterator_helper(db, read_options, cf_handle);
|
2014-10-13 10:34:52 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
|
|
|
* Method: iterators
|
2016-02-01 21:00:40 +01:00
|
|
|
* Signature: (J[JJ)[J
|
2014-10-13 10:34:52 +02:00
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
jlongArray Java_org_rocksdb_RocksDB_iterators(
|
|
|
|
JNIEnv* env, jobject, jlong db_handle,
|
|
|
|
jlongArray jcolumn_family_handles,
|
|
|
|
jlong jread_options_handle) {
|
2016-02-01 21:00:40 +01:00
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(db_handle);
|
2018-04-13 02:55:14 +02:00
|
|
|
auto& read_options =
|
|
|
|
*reinterpret_cast<rocksdb::ReadOptions*>(jread_options_handle);
|
2017-02-28 01:26:12 +01:00
|
|
|
|
2014-10-13 10:34:52 +02:00
|
|
|
std::vector<rocksdb::ColumnFamilyHandle*> cf_handles;
|
2016-02-01 21:00:40 +01:00
|
|
|
if (jcolumn_family_handles != nullptr) {
|
2017-02-28 01:26:12 +01:00
|
|
|
const jsize len_cols = env->GetArrayLength(jcolumn_family_handles);
|
|
|
|
jlong* jcfh = env->GetLongArrayElements(jcolumn_family_handles, nullptr);
|
2018-04-13 02:55:14 +02:00
|
|
|
if (jcfh == nullptr) {
|
2017-02-28 01:26:12 +01:00
|
|
|
// exception thrown: OutOfMemoryError
|
|
|
|
return nullptr;
|
|
|
|
}
|
|
|
|
|
|
|
|
for (jsize i = 0; i < len_cols; i++) {
|
2018-04-13 02:55:14 +02:00
|
|
|
auto* cf_handle = reinterpret_cast<rocksdb::ColumnFamilyHandle*>(jcfh[i]);
|
2016-02-01 21:00:40 +01:00
|
|
|
cf_handles.push_back(cf_handle);
|
2014-10-13 10:34:52 +02:00
|
|
|
}
|
2017-02-28 01:26:12 +01:00
|
|
|
|
2016-02-01 21:00:40 +01:00
|
|
|
env->ReleaseLongArrayElements(jcolumn_family_handles, jcfh, JNI_ABORT);
|
2014-10-13 10:34:52 +02:00
|
|
|
}
|
|
|
|
|
2016-02-01 21:00:40 +01:00
|
|
|
std::vector<rocksdb::Iterator*> iterators;
|
2018-04-13 02:55:14 +02:00
|
|
|
rocksdb::Status s = db->NewIterators(read_options, cf_handles, &iterators);
|
2014-10-13 10:34:52 +02:00
|
|
|
if (s.ok()) {
|
2014-11-11 22:47:22 +01:00
|
|
|
jlongArray jLongArray =
|
|
|
|
env->NewLongArray(static_cast<jsize>(iterators.size()));
|
2018-04-13 02:55:14 +02:00
|
|
|
if (jLongArray == nullptr) {
|
2017-02-28 01:26:12 +01:00
|
|
|
// exception thrown: OutOfMemoryError
|
|
|
|
return nullptr;
|
|
|
|
}
|
|
|
|
|
2018-04-13 02:55:14 +02:00
|
|
|
for (std::vector<rocksdb::Iterator*>::size_type i = 0; i < iterators.size();
|
|
|
|
i++) {
|
|
|
|
env->SetLongArrayRegion(
|
|
|
|
jLongArray, static_cast<jsize>(i), 1,
|
|
|
|
const_cast<jlong*>(reinterpret_cast<const jlong*>(&iterators[i])));
|
|
|
|
if (env->ExceptionCheck()) {
|
2017-02-28 01:26:12 +01:00
|
|
|
// exception thrown: ArrayIndexOutOfBoundsException
|
|
|
|
env->DeleteLocalRef(jLongArray);
|
|
|
|
return nullptr;
|
|
|
|
}
|
2014-10-13 10:34:52 +02:00
|
|
|
}
|
2017-02-28 01:26:12 +01:00
|
|
|
|
2014-10-13 10:34:52 +02:00
|
|
|
return jLongArray;
|
2016-02-01 21:00:40 +01:00
|
|
|
} else {
|
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(env, s);
|
2017-02-28 01:26:12 +01:00
|
|
|
return nullptr;
|
2014-10-13 10:34:52 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-10-13 20:48:44 +02:00
|
|
|
/*
|
|
|
|
* Method: getSnapshot
|
|
|
|
* Signature: (J)J
|
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
jlong Java_org_rocksdb_RocksDB_getSnapshot(
|
|
|
|
JNIEnv*, jobject, jlong db_handle) {
|
2017-02-28 01:26:12 +01:00
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(db_handle);
|
2014-10-13 20:48:44 +02:00
|
|
|
const rocksdb::Snapshot* snapshot = db->GetSnapshot();
|
|
|
|
return reinterpret_cast<jlong>(snapshot);
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Method: releaseSnapshot
|
|
|
|
* Signature: (JJ)V
|
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
void Java_org_rocksdb_RocksDB_releaseSnapshot(
|
|
|
|
JNIEnv*, jobject, jlong db_handle,
|
|
|
|
jlong snapshot_handle) {
|
2017-02-28 01:26:12 +01:00
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(db_handle);
|
|
|
|
auto* snapshot = reinterpret_cast<rocksdb::Snapshot*>(snapshot_handle);
|
2014-10-13 20:48:44 +02:00
|
|
|
db->ReleaseSnapshot(snapshot);
|
|
|
|
}
|
|
|
|
|
2014-09-24 20:43:35 +02:00
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
2019-02-22 23:36:38 +01:00
|
|
|
* Method: getProperty
|
|
|
|
* Signature: (JJLjava/lang/String;I)Ljava/lang/String;
|
2014-09-24 20:43:35 +02:00
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
jstring Java_org_rocksdb_RocksDB_getProperty(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle, jlong jcf_handle,
|
|
|
|
jstring jproperty, jint jproperty_len) {
|
2017-02-28 01:26:12 +01:00
|
|
|
const char* property = env->GetStringUTFChars(jproperty, nullptr);
|
2018-04-13 02:55:14 +02:00
|
|
|
if (property == nullptr) {
|
2017-02-28 01:26:12 +01:00
|
|
|
// exception thrown: OutOfMemoryError
|
|
|
|
return nullptr;
|
|
|
|
}
|
2019-02-22 23:36:38 +01:00
|
|
|
rocksdb::Slice property_name(property, jproperty_len);
|
|
|
|
|
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
|
|
|
rocksdb::ColumnFamilyHandle* cf_handle;
|
|
|
|
if (jcf_handle == 0) {
|
|
|
|
cf_handle = db->DefaultColumnFamily();
|
|
|
|
} else {
|
|
|
|
cf_handle =
|
|
|
|
reinterpret_cast<rocksdb::ColumnFamilyHandle*>(jcf_handle);
|
|
|
|
}
|
2014-10-09 23:16:41 +02:00
|
|
|
|
2014-09-24 20:43:35 +02:00
|
|
|
std::string property_value;
|
2019-02-22 23:36:38 +01:00
|
|
|
bool retCode = db->GetProperty(cf_handle, property_name, &property_value);
|
2014-09-24 20:43:35 +02:00
|
|
|
env->ReleaseStringUTFChars(jproperty, property);
|
2014-10-09 23:16:41 +02:00
|
|
|
|
2017-02-28 01:26:12 +01:00
|
|
|
if (retCode) {
|
|
|
|
return env->NewStringUTF(property_value.c_str());
|
2014-09-24 20:43:35 +02:00
|
|
|
}
|
2014-10-09 23:16:41 +02:00
|
|
|
|
2017-02-28 01:26:12 +01:00
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(env, rocksdb::Status::NotFound());
|
|
|
|
return nullptr;
|
2014-09-24 20:43:35 +02:00
|
|
|
}
|
2014-10-13 10:34:52 +02:00
|
|
|
|
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
2019-02-22 23:36:38 +01:00
|
|
|
* Method: getMapProperty
|
|
|
|
* Signature: (JJLjava/lang/String;I)Ljava/util/Map;
|
2014-10-13 10:34:52 +02:00
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
jobject Java_org_rocksdb_RocksDB_getMapProperty(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle, jlong jcf_handle,
|
2014-10-13 10:34:52 +02:00
|
|
|
jstring jproperty, jint jproperty_len) {
|
2019-02-22 23:36:38 +01:00
|
|
|
const char* property = env->GetStringUTFChars(jproperty, nullptr);
|
2018-04-13 02:55:14 +02:00
|
|
|
if (property == nullptr) {
|
2017-02-28 01:26:12 +01:00
|
|
|
// exception thrown: OutOfMemoryError
|
|
|
|
return nullptr;
|
|
|
|
}
|
2019-02-22 23:36:38 +01:00
|
|
|
rocksdb::Slice property_name(property, jproperty_len);
|
|
|
|
|
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
|
|
|
rocksdb::ColumnFamilyHandle* cf_handle;
|
|
|
|
if (jcf_handle == 0) {
|
|
|
|
cf_handle = db->DefaultColumnFamily();
|
|
|
|
} else {
|
|
|
|
cf_handle =
|
|
|
|
reinterpret_cast<rocksdb::ColumnFamilyHandle*>(jcf_handle);
|
2017-02-28 01:26:12 +01:00
|
|
|
}
|
2014-11-10 12:55:58 +01:00
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
std::map<std::string, std::string> property_value;
|
|
|
|
bool retCode = db->GetMapProperty(cf_handle, property_name, &property_value);
|
2014-11-10 12:55:58 +01:00
|
|
|
env->ReleaseStringUTFChars(jproperty, property);
|
|
|
|
|
2017-02-28 01:26:12 +01:00
|
|
|
if (retCode) {
|
2019-02-22 23:36:38 +01:00
|
|
|
return rocksdb::HashMapJni::fromCppMap(env, &property_value);
|
2014-11-10 12:55:58 +01:00
|
|
|
}
|
2017-02-28 01:26:12 +01:00
|
|
|
|
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(env, rocksdb::Status::NotFound());
|
2019-02-22 23:36:38 +01:00
|
|
|
return nullptr;
|
2014-11-10 12:55:58 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
|
|
|
* Method: getLongProperty
|
2019-02-22 23:36:38 +01:00
|
|
|
* Signature: (JJLjava/lang/String;I)J
|
2014-11-10 12:55:58 +01:00
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
jlong Java_org_rocksdb_RocksDB_getLongProperty(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle, jlong jcf_handle,
|
2014-11-10 12:55:58 +01:00
|
|
|
jstring jproperty, jint jproperty_len) {
|
2017-02-28 01:26:12 +01:00
|
|
|
const char* property = env->GetStringUTFChars(jproperty, nullptr);
|
2018-04-13 02:55:14 +02:00
|
|
|
if (property == nullptr) {
|
2017-02-28 01:26:12 +01:00
|
|
|
// exception thrown: OutOfMemoryError
|
|
|
|
return 0;
|
|
|
|
}
|
2019-02-22 23:36:38 +01:00
|
|
|
rocksdb::Slice property_name(property, jproperty_len);
|
|
|
|
|
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
|
|
|
rocksdb::ColumnFamilyHandle* cf_handle;
|
|
|
|
if (jcf_handle == 0) {
|
|
|
|
cf_handle = db->DefaultColumnFamily();
|
|
|
|
} else {
|
|
|
|
cf_handle =
|
|
|
|
reinterpret_cast<rocksdb::ColumnFamilyHandle*>(jcf_handle);
|
|
|
|
}
|
2014-11-10 12:55:58 +01:00
|
|
|
|
|
|
|
uint64_t property_value;
|
2019-02-22 23:36:38 +01:00
|
|
|
bool retCode = db->GetIntProperty(cf_handle, property_name, &property_value);
|
2014-11-10 12:55:58 +01:00
|
|
|
env->ReleaseStringUTFChars(jproperty, property);
|
|
|
|
|
2017-02-28 01:26:12 +01:00
|
|
|
if (retCode) {
|
|
|
|
return property_value;
|
2014-11-10 12:55:58 +01:00
|
|
|
}
|
2017-02-28 01:26:12 +01:00
|
|
|
|
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(env, rocksdb::Status::NotFound());
|
|
|
|
return 0;
|
2014-11-10 12:55:58 +01:00
|
|
|
}
|
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
|
|
|
* Method: resetStats
|
|
|
|
* Signature: (J)V
|
|
|
|
*/
|
|
|
|
void Java_org_rocksdb_RocksDB_resetStats(
|
|
|
|
JNIEnv *, jobject, jlong jdb_handle) {
|
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
|
|
|
db->ResetStats();
|
|
|
|
}
|
|
|
|
|
2018-09-20 02:44:12 +02:00
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
|
|
|
* Method: getAggregatedLongProperty
|
|
|
|
* Signature: (JLjava/lang/String;I)J
|
|
|
|
*/
|
|
|
|
jlong Java_org_rocksdb_RocksDB_getAggregatedLongProperty(
|
2019-02-22 23:36:38 +01:00
|
|
|
JNIEnv* env, jobject, jlong db_handle,
|
|
|
|
jstring jproperty, jint jproperty_len) {
|
2018-09-20 02:44:12 +02:00
|
|
|
const char* property = env->GetStringUTFChars(jproperty, nullptr);
|
|
|
|
if (property == nullptr) {
|
|
|
|
return 0;
|
|
|
|
}
|
2019-02-22 23:36:38 +01:00
|
|
|
rocksdb::Slice property_name(property, jproperty_len);
|
2018-09-20 02:44:12 +02:00
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(db_handle);
|
|
|
|
uint64_t property_value = 0;
|
2019-02-22 23:36:38 +01:00
|
|
|
bool retCode = db->GetAggregatedIntProperty(property_name, &property_value);
|
2018-09-20 02:44:12 +02:00
|
|
|
env->ReleaseStringUTFChars(jproperty, property);
|
|
|
|
|
|
|
|
if (retCode) {
|
|
|
|
return property_value;
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(env, rocksdb::Status::NotFound());
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
|
|
|
* Method: getApproximateSizes
|
|
|
|
* Signature: (JJ[JB)[J
|
|
|
|
*/
|
|
|
|
jlongArray Java_org_rocksdb_RocksDB_getApproximateSizes(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle, jlong jcf_handle,
|
|
|
|
jlongArray jrange_slice_handles, jbyte jinclude_flags) {
|
|
|
|
const jsize jlen = env->GetArrayLength(jrange_slice_handles);
|
|
|
|
const size_t range_count = jlen / 2;
|
2018-09-20 02:44:12 +02:00
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
jboolean jranges_is_copy = JNI_FALSE;
|
|
|
|
jlong* jranges = env->GetLongArrayElements(jrange_slice_handles,
|
|
|
|
&jranges_is_copy);
|
|
|
|
if (jranges == nullptr) {
|
|
|
|
// exception thrown: OutOfMemoryError
|
|
|
|
return nullptr;
|
|
|
|
}
|
2014-11-09 20:09:39 +01:00
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
auto ranges = std::unique_ptr<rocksdb::Range[]>(
|
|
|
|
new rocksdb::Range[range_count]);
|
|
|
|
for (jsize i = 0; i < jlen; ++i) {
|
|
|
|
auto* start = reinterpret_cast<rocksdb::Slice*>(jranges[i]);
|
|
|
|
auto* limit = reinterpret_cast<rocksdb::Slice*>(jranges[++i]);
|
|
|
|
ranges.get()[i] = rocksdb::Range(*start, *limit);
|
|
|
|
}
|
|
|
|
|
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
|
|
|
rocksdb::ColumnFamilyHandle* cf_handle;
|
|
|
|
if (jcf_handle == 0) {
|
|
|
|
cf_handle = db->DefaultColumnFamily();
|
2014-11-09 20:09:39 +01:00
|
|
|
} else {
|
2019-02-22 23:36:38 +01:00
|
|
|
cf_handle =
|
|
|
|
reinterpret_cast<rocksdb::ColumnFamilyHandle*>(jcf_handle);
|
2014-11-09 20:09:39 +01:00
|
|
|
}
|
2019-02-22 23:36:38 +01:00
|
|
|
|
|
|
|
auto sizes = std::unique_ptr<uint64_t[]>(new uint64_t[range_count]);
|
|
|
|
db->GetApproximateSizes(cf_handle, ranges.get(),
|
|
|
|
static_cast<int>(range_count), sizes.get(),
|
|
|
|
static_cast<uint8_t>(jinclude_flags));
|
|
|
|
|
|
|
|
// release LongArrayElements
|
|
|
|
env->ReleaseLongArrayElements(jrange_slice_handles, jranges, JNI_ABORT);
|
|
|
|
|
|
|
|
// prepare results
|
|
|
|
auto results = std::unique_ptr<jlong[]>(new jlong[range_count]);
|
|
|
|
for (size_t i = 0; i < range_count; ++i) {
|
|
|
|
results.get()[i] = static_cast<jlong>(sizes.get()[i]);
|
|
|
|
}
|
|
|
|
|
|
|
|
const jsize jrange_count = jlen / 2;
|
|
|
|
jlongArray jresults = env->NewLongArray(jrange_count);
|
|
|
|
if (jresults == nullptr) {
|
|
|
|
// exception thrown: OutOfMemoryError
|
|
|
|
return nullptr;
|
|
|
|
}
|
|
|
|
|
|
|
|
env->SetLongArrayRegion(jresults, 0, jrange_count, results.get());
|
|
|
|
if (env->ExceptionCheck()) {
|
|
|
|
// exception thrown: ArrayIndexOutOfBoundsException
|
|
|
|
env->DeleteLocalRef(jresults);
|
|
|
|
return nullptr;
|
2014-11-09 20:09:39 +01:00
|
|
|
}
|
2019-02-22 23:36:38 +01:00
|
|
|
|
|
|
|
return jresults;
|
2014-11-09 20:09:39 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
2019-02-22 23:36:38 +01:00
|
|
|
* Method: getApproximateMemTableStats
|
|
|
|
* Signature: (JJJJ)[J
|
2014-11-09 20:09:39 +01:00
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
jlongArray Java_org_rocksdb_RocksDB_getApproximateMemTableStats(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle, jlong jcf_handle,
|
|
|
|
jlong jstartHandle, jlong jlimitHandle) {
|
|
|
|
auto* start = reinterpret_cast<rocksdb::Slice*>(jstartHandle);
|
|
|
|
auto* limit = reinterpret_cast<rocksdb::Slice*>( jlimitHandle);
|
|
|
|
const rocksdb::Range range(*start, *limit);
|
|
|
|
|
2017-02-28 01:26:12 +01:00
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
2019-02-22 23:36:38 +01:00
|
|
|
rocksdb::ColumnFamilyHandle* cf_handle;
|
|
|
|
if (jcf_handle == 0) {
|
|
|
|
cf_handle = db->DefaultColumnFamily();
|
|
|
|
} else {
|
|
|
|
cf_handle =
|
|
|
|
reinterpret_cast<rocksdb::ColumnFamilyHandle*>(jcf_handle);
|
|
|
|
}
|
|
|
|
|
|
|
|
uint64_t count = 0;
|
|
|
|
uint64_t sizes = 0;
|
|
|
|
db->GetApproximateMemTableStats(cf_handle, range, &count, &sizes);
|
|
|
|
|
|
|
|
// prepare results
|
|
|
|
jlong results[2] = {
|
|
|
|
static_cast<jlong>(count),
|
|
|
|
static_cast<jlong>(sizes)};
|
|
|
|
|
|
|
|
const jsize jcount = static_cast<jsize>(count);
|
|
|
|
jlongArray jsizes = env->NewLongArray(jcount);
|
|
|
|
if (jsizes == nullptr) {
|
|
|
|
// exception thrown: OutOfMemoryError
|
|
|
|
return nullptr;
|
|
|
|
}
|
|
|
|
|
|
|
|
env->SetLongArrayRegion(jsizes, 0, jcount, results);
|
|
|
|
if (env->ExceptionCheck()) {
|
|
|
|
// exception thrown: ArrayIndexOutOfBoundsException
|
|
|
|
env->DeleteLocalRef(jsizes);
|
|
|
|
return nullptr;
|
|
|
|
}
|
|
|
|
|
|
|
|
return jsizes;
|
2014-11-09 20:09:39 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
2019-02-22 23:36:38 +01:00
|
|
|
* Method: compactRange
|
|
|
|
* Signature: (J[BI[BIJJ)V
|
2014-11-09 20:09:39 +01:00
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
void Java_org_rocksdb_RocksDB_compactRange(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle,
|
|
|
|
jbyteArray jbegin, jint jbegin_len,
|
|
|
|
jbyteArray jend, jint jend_len,
|
|
|
|
jlong jcompact_range_opts_handle,
|
|
|
|
jlong jcf_handle) {
|
|
|
|
jboolean has_exception = JNI_FALSE;
|
|
|
|
|
|
|
|
std::string str_begin;
|
|
|
|
if (jbegin_len > 0) {
|
|
|
|
str_begin = rocksdb::JniUtil::byteString<std::string>(env, jbegin, jbegin_len,
|
|
|
|
[](const char* str, const size_t len) {
|
|
|
|
return std::string(str, len);
|
|
|
|
},
|
|
|
|
&has_exception);
|
|
|
|
if (has_exception == JNI_TRUE) {
|
|
|
|
// exception occurred
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
std::string str_end;
|
|
|
|
if (jend_len > 0) {
|
|
|
|
str_end = rocksdb::JniUtil::byteString<std::string>(env, jend, jend_len,
|
|
|
|
[](const char* str, const size_t len) {
|
|
|
|
return std::string(str, len);
|
|
|
|
},
|
|
|
|
&has_exception);
|
|
|
|
if (has_exception == JNI_TRUE) {
|
|
|
|
// exception occurred
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::CompactRangeOptions *compact_range_opts = nullptr;
|
|
|
|
if (jcompact_range_opts_handle == 0) {
|
|
|
|
// NOTE: we DO own the pointer!
|
|
|
|
compact_range_opts = new rocksdb::CompactRangeOptions();
|
|
|
|
} else {
|
|
|
|
// NOTE: we do NOT own the pointer!
|
|
|
|
compact_range_opts =
|
|
|
|
reinterpret_cast<rocksdb::CompactRangeOptions*>(jcompact_range_opts_handle);
|
|
|
|
}
|
|
|
|
|
2017-02-28 01:26:12 +01:00
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
2014-11-17 23:29:52 +01:00
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
rocksdb::ColumnFamilyHandle* cf_handle;
|
|
|
|
if (jcf_handle == 0) {
|
|
|
|
cf_handle = db->DefaultColumnFamily();
|
|
|
|
} else {
|
|
|
|
cf_handle =
|
|
|
|
reinterpret_cast<rocksdb::ColumnFamilyHandle*>(jcf_handle);
|
|
|
|
}
|
2014-11-17 23:29:52 +01:00
|
|
|
|
|
|
|
rocksdb::Status s;
|
2019-02-22 23:36:38 +01:00
|
|
|
if (jbegin_len > 0 || jend_len > 0) {
|
|
|
|
const rocksdb::Slice begin(str_begin);
|
|
|
|
const rocksdb::Slice end(str_end);
|
|
|
|
s = db->CompactRange(*compact_range_opts, cf_handle, &begin, &end);
|
2014-11-17 23:29:52 +01:00
|
|
|
} else {
|
2019-02-22 23:36:38 +01:00
|
|
|
s = db->CompactRange(*compact_range_opts, cf_handle, nullptr, nullptr);
|
2014-11-17 23:29:52 +01:00
|
|
|
}
|
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
if (jcompact_range_opts_handle == 0) {
|
|
|
|
delete compact_range_opts;
|
2014-11-17 23:29:52 +01:00
|
|
|
}
|
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(env, s);
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
2019-02-22 23:36:38 +01:00
|
|
|
* Method: setOptions
|
|
|
|
* Signature: (JJ[Ljava/lang/String;[Ljava/lang/String;)V
|
2014-11-17 23:29:52 +01:00
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
void Java_org_rocksdb_RocksDB_setOptions(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle, jlong jcf_handle,
|
|
|
|
jobjectArray jkeys, jobjectArray jvalues) {
|
|
|
|
const jsize len = env->GetArrayLength(jkeys);
|
|
|
|
assert(len == env->GetArrayLength(jvalues));
|
|
|
|
|
|
|
|
std::unordered_map<std::string, std::string> options_map;
|
|
|
|
for (jsize i = 0; i < len; i++) {
|
|
|
|
jobject jobj_key = env->GetObjectArrayElement(jkeys, i);
|
|
|
|
if (env->ExceptionCheck()) {
|
|
|
|
// exception thrown: ArrayIndexOutOfBoundsException
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
jobject jobj_value = env->GetObjectArrayElement(jvalues, i);
|
|
|
|
if (env->ExceptionCheck()) {
|
|
|
|
// exception thrown: ArrayIndexOutOfBoundsException
|
|
|
|
env->DeleteLocalRef(jobj_key);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
jboolean has_exception = JNI_FALSE;
|
|
|
|
std::string s_key =
|
|
|
|
rocksdb::JniUtil::copyStdString(
|
|
|
|
env, reinterpret_cast<jstring>(jobj_key), &has_exception);
|
|
|
|
if (has_exception == JNI_TRUE) {
|
|
|
|
// exception occurred
|
|
|
|
env->DeleteLocalRef(jobj_value);
|
|
|
|
env->DeleteLocalRef(jobj_key);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
std::string s_value =
|
|
|
|
rocksdb::JniUtil::copyStdString(
|
|
|
|
env, reinterpret_cast<jstring>(jobj_value), &has_exception);
|
|
|
|
if (has_exception == JNI_TRUE) {
|
|
|
|
// exception occurred
|
|
|
|
env->DeleteLocalRef(jobj_value);
|
|
|
|
env->DeleteLocalRef(jobj_key);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
options_map[s_key] = s_value;
|
|
|
|
|
|
|
|
env->DeleteLocalRef(jobj_key);
|
|
|
|
env->DeleteLocalRef(jobj_value);
|
|
|
|
}
|
|
|
|
|
2017-02-28 01:26:12 +01:00
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
2019-02-22 23:36:38 +01:00
|
|
|
auto* cf_handle = reinterpret_cast<rocksdb::ColumnFamilyHandle*>(jcf_handle);
|
|
|
|
auto s = db->SetOptions(cf_handle, options_map);
|
|
|
|
if (!s.ok()) {
|
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(env, s);
|
|
|
|
}
|
2014-11-17 23:29:52 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
2019-02-22 23:36:38 +01:00
|
|
|
* Method: setDBOptions
|
|
|
|
* Signature: (J[Ljava/lang/String;[Ljava/lang/String;)V
|
2014-11-17 23:29:52 +01:00
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
void Java_org_rocksdb_RocksDB_setDBOptions(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle,
|
|
|
|
jobjectArray jkeys, jobjectArray jvalues) {
|
|
|
|
const jsize len = env->GetArrayLength(jkeys);
|
|
|
|
assert(len == env->GetArrayLength(jvalues));
|
|
|
|
|
|
|
|
std::unordered_map<std::string, std::string> options_map;
|
|
|
|
for (jsize i = 0; i < len; i++) {
|
|
|
|
jobject jobj_key = env->GetObjectArrayElement(jkeys, i);
|
|
|
|
if (env->ExceptionCheck()) {
|
|
|
|
// exception thrown: ArrayIndexOutOfBoundsException
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
jobject jobj_value = env->GetObjectArrayElement(jvalues, i);
|
|
|
|
if (env->ExceptionCheck()) {
|
|
|
|
// exception thrown: ArrayIndexOutOfBoundsException
|
|
|
|
env->DeleteLocalRef(jobj_key);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
jboolean has_exception = JNI_FALSE;
|
|
|
|
std::string s_key =
|
|
|
|
rocksdb::JniUtil::copyStdString(
|
|
|
|
env, reinterpret_cast<jstring>(jobj_key), &has_exception);
|
|
|
|
if (has_exception == JNI_TRUE) {
|
|
|
|
// exception occurred
|
|
|
|
env->DeleteLocalRef(jobj_value);
|
|
|
|
env->DeleteLocalRef(jobj_key);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
std::string s_value =
|
|
|
|
rocksdb::JniUtil::copyStdString(
|
|
|
|
env, reinterpret_cast<jstring>(jobj_value), &has_exception);
|
|
|
|
if (has_exception == JNI_TRUE) {
|
|
|
|
// exception occurred
|
|
|
|
env->DeleteLocalRef(jobj_value);
|
|
|
|
env->DeleteLocalRef(jobj_key);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
options_map[s_key] = s_value;
|
|
|
|
|
|
|
|
env->DeleteLocalRef(jobj_key);
|
|
|
|
env->DeleteLocalRef(jobj_value);
|
|
|
|
}
|
|
|
|
|
2017-02-28 01:26:12 +01:00
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
2019-02-22 23:36:38 +01:00
|
|
|
auto s = db->SetDBOptions(options_map);
|
|
|
|
if (!s.ok()) {
|
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(env, s);
|
|
|
|
}
|
2014-11-17 23:29:52 +01:00
|
|
|
}
|
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
|
|
|
* Method: compactFiles
|
|
|
|
* Signature: (JJJ[Ljava/lang/String;IIJ)[Ljava/lang/String;
|
2017-02-28 01:26:12 +01:00
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
jobjectArray Java_org_rocksdb_RocksDB_compactFiles(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle, jlong jcompaction_opts_handle,
|
|
|
|
jlong jcf_handle, jobjectArray jinput_file_names, jint joutput_level,
|
|
|
|
jint joutput_path_id, jlong jcompaction_job_info_handle) {
|
|
|
|
jboolean has_exception = JNI_FALSE;
|
|
|
|
const std::vector<std::string> input_file_names =
|
|
|
|
rocksdb::JniUtil::copyStrings(env, jinput_file_names, &has_exception);
|
|
|
|
if (has_exception == JNI_TRUE) {
|
|
|
|
// exception occurred
|
|
|
|
return nullptr;
|
2017-02-28 01:26:12 +01:00
|
|
|
}
|
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
auto* compaction_opts =
|
|
|
|
reinterpret_cast<rocksdb::CompactionOptions*>(jcompaction_opts_handle);
|
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
|
|
|
rocksdb::ColumnFamilyHandle* cf_handle;
|
|
|
|
if (jcf_handle == 0) {
|
|
|
|
cf_handle = db->DefaultColumnFamily();
|
|
|
|
} else {
|
|
|
|
cf_handle =
|
|
|
|
reinterpret_cast<rocksdb::ColumnFamilyHandle*>(jcf_handle);
|
2017-02-28 01:26:12 +01:00
|
|
|
}
|
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
rocksdb::CompactionJobInfo* compaction_job_info = nullptr;
|
|
|
|
if (jcompaction_job_info_handle != 0) {
|
|
|
|
compaction_job_info =
|
|
|
|
reinterpret_cast<rocksdb::CompactionJobInfo*>(jcompaction_job_info_handle);
|
|
|
|
}
|
2014-11-17 23:29:52 +01:00
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
std::vector<std::string> output_file_names;
|
|
|
|
auto s = db->CompactFiles(*compaction_opts, cf_handle, input_file_names,
|
|
|
|
static_cast<int>(joutput_level), static_cast<int>(joutput_path_id),
|
|
|
|
&output_file_names, compaction_job_info);
|
|
|
|
if (!s.ok()) {
|
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(env, s);
|
|
|
|
return nullptr;
|
2014-11-17 23:29:52 +01:00
|
|
|
}
|
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
return rocksdb::JniUtil::toJavaStrings(env, &output_file_names);
|
|
|
|
}
|
2014-11-17 23:29:52 +01:00
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
|
|
|
* Method: pauseBackgroundWork
|
|
|
|
* Signature: (J)V
|
|
|
|
*/
|
|
|
|
void Java_org_rocksdb_RocksDB_pauseBackgroundWork(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle) {
|
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
|
|
|
auto s = db->PauseBackgroundWork();
|
|
|
|
if (!s.ok()) {
|
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(env, s);
|
2014-11-17 23:29:52 +01:00
|
|
|
}
|
2019-02-22 23:36:38 +01:00
|
|
|
}
|
2017-02-28 01:26:12 +01:00
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
|
|
|
* Method: continueBackgroundWork
|
|
|
|
* Signature: (J)V
|
|
|
|
*/
|
|
|
|
void Java_org_rocksdb_RocksDB_continueBackgroundWork(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle) {
|
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
|
|
|
auto s = db->ContinueBackgroundWork();
|
|
|
|
if (!s.ok()) {
|
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(env, s);
|
|
|
|
}
|
2014-11-17 23:29:52 +01:00
|
|
|
}
|
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
|
|
|
* Method: enableAutoCompaction
|
|
|
|
* Signature: (J[J)V
|
|
|
|
*/
|
|
|
|
void Java_org_rocksdb_RocksDB_enableAutoCompaction(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle, jlongArray jcf_handles) {
|
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
|
|
|
jboolean has_exception = JNI_FALSE;
|
|
|
|
const std::vector<rocksdb::ColumnFamilyHandle*> cf_handles =
|
|
|
|
rocksdb::JniUtil::fromJPointers<rocksdb::ColumnFamilyHandle>(env, jcf_handles, &has_exception);
|
|
|
|
if (has_exception == JNI_TRUE) {
|
|
|
|
// exception occurred
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
db->EnableAutoCompaction(cf_handles);
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
|
|
|
* Method: numberLevels
|
|
|
|
* Signature: (JJ)I
|
|
|
|
*/
|
|
|
|
jint Java_org_rocksdb_RocksDB_numberLevels(
|
|
|
|
JNIEnv*, jobject, jlong jdb_handle, jlong jcf_handle) {
|
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
|
|
|
rocksdb::ColumnFamilyHandle* cf_handle;
|
|
|
|
if (jcf_handle == 0) {
|
|
|
|
cf_handle = db->DefaultColumnFamily();
|
|
|
|
} else {
|
|
|
|
cf_handle =
|
|
|
|
reinterpret_cast<rocksdb::ColumnFamilyHandle*>(jcf_handle);
|
|
|
|
}
|
|
|
|
return static_cast<jint>(db->NumberLevels(cf_handle));
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
|
|
|
* Method: maxMemCompactionLevel
|
|
|
|
* Signature: (JJ)I
|
2018-08-17 19:52:58 +02:00
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
jint Java_org_rocksdb_RocksDB_maxMemCompactionLevel(
|
|
|
|
JNIEnv*, jobject, jlong jdb_handle, jlong jcf_handle) {
|
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
|
|
|
rocksdb::ColumnFamilyHandle* cf_handle;
|
|
|
|
if (jcf_handle == 0) {
|
|
|
|
cf_handle = db->DefaultColumnFamily();
|
|
|
|
} else {
|
|
|
|
cf_handle =
|
|
|
|
reinterpret_cast<rocksdb::ColumnFamilyHandle*>(jcf_handle);
|
|
|
|
}
|
|
|
|
return static_cast<jint>(db->MaxMemCompactionLevel(cf_handle));
|
|
|
|
}
|
2018-08-17 19:52:58 +02:00
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
|
|
|
* Method: level0StopWriteTrigger
|
|
|
|
* Signature: (JJ)I
|
|
|
|
*/
|
|
|
|
jint Java_org_rocksdb_RocksDB_level0StopWriteTrigger(
|
|
|
|
JNIEnv*, jobject, jlong jdb_handle, jlong jcf_handle) {
|
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
|
|
|
rocksdb::ColumnFamilyHandle* cf_handle;
|
|
|
|
if (jcf_handle == 0) {
|
|
|
|
cf_handle = db->DefaultColumnFamily();
|
|
|
|
} else {
|
|
|
|
cf_handle =
|
|
|
|
reinterpret_cast<rocksdb::ColumnFamilyHandle*>(jcf_handle);
|
|
|
|
}
|
|
|
|
return static_cast<jint>(db->Level0StopWriteTrigger(cf_handle));
|
2018-08-17 19:52:58 +02:00
|
|
|
}
|
|
|
|
|
2014-11-17 23:29:52 +01:00
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
2019-02-22 23:36:38 +01:00
|
|
|
* Method: getName
|
|
|
|
* Signature: (J)Ljava/lang/String;
|
2014-11-17 23:29:52 +01:00
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
jstring Java_org_rocksdb_RocksDB_getName(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle) {
|
2017-02-28 01:26:12 +01:00
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
2019-02-22 23:36:38 +01:00
|
|
|
std::string name = db->GetName();
|
|
|
|
return rocksdb::JniUtil::toJavaString(env, &name, false);
|
2014-11-17 23:29:52 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
2019-02-22 23:36:38 +01:00
|
|
|
* Method: getEnv
|
|
|
|
* Signature: (J)J
|
2014-11-17 23:29:52 +01:00
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
jlong Java_org_rocksdb_RocksDB_getEnv(
|
|
|
|
JNIEnv*, jobject, jlong jdb_handle) {
|
2017-02-28 01:26:12 +01:00
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
2019-02-22 23:36:38 +01:00
|
|
|
return reinterpret_cast<jlong>(db->GetEnv());
|
2014-11-17 23:29:52 +01:00
|
|
|
}
|
2015-01-25 22:05:29 +01:00
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
|
|
|
* Method: flush
|
|
|
|
* Signature: (JJ[J)V
|
|
|
|
*/
|
|
|
|
void Java_org_rocksdb_RocksDB_flush(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle, jlong jflush_opts_handle,
|
|
|
|
jlongArray jcf_handles) {
|
2018-08-17 19:52:58 +02:00
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
2019-02-22 23:36:38 +01:00
|
|
|
auto* flush_opts =
|
|
|
|
reinterpret_cast<rocksdb::FlushOptions*>(jflush_opts_handle);
|
|
|
|
std::vector<rocksdb::ColumnFamilyHandle*> cf_handles;
|
|
|
|
if (jcf_handles == nullptr) {
|
|
|
|
cf_handles.push_back(db->DefaultColumnFamily());
|
|
|
|
} else {
|
|
|
|
jboolean has_exception = JNI_FALSE;
|
|
|
|
cf_handles =
|
|
|
|
rocksdb::JniUtil::fromJPointers<rocksdb::ColumnFamilyHandle>(
|
|
|
|
env, jcf_handles, &has_exception);
|
|
|
|
if (has_exception) {
|
|
|
|
// exception occurred
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
auto s = db->Flush(*flush_opts, cf_handles);
|
|
|
|
if (!s.ok()) {
|
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(env, s);
|
|
|
|
}
|
2018-08-17 19:52:58 +02:00
|
|
|
}
|
|
|
|
|
2016-05-07 00:04:13 +02:00
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
2019-02-22 23:36:38 +01:00
|
|
|
* Method: flushWal
|
|
|
|
* Signature: (JZ)V
|
2016-05-07 00:04:13 +02:00
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
void Java_org_rocksdb_RocksDB_flushWal(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle, jboolean jsync) {
|
2016-05-07 00:04:13 +02:00
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
2019-02-22 23:36:38 +01:00
|
|
|
auto s = db->FlushWAL(jsync == JNI_TRUE);
|
2017-02-28 01:26:12 +01:00
|
|
|
if (!s.ok()) {
|
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(env, s);
|
2016-05-07 00:04:13 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
2019-02-22 23:36:38 +01:00
|
|
|
* Method: syncWal
|
2016-05-07 00:04:13 +02:00
|
|
|
* Signature: (J)V
|
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
void Java_org_rocksdb_RocksDB_syncWal(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle) {
|
2016-05-07 00:04:13 +02:00
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
2019-02-22 23:36:38 +01:00
|
|
|
auto s = db->SyncWAL();
|
2017-02-28 01:26:12 +01:00
|
|
|
if (!s.ok()) {
|
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(env, s);
|
2016-05-07 00:04:13 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-01-25 22:46:11 +01:00
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
|
|
|
* Method: getLatestSequenceNumber
|
|
|
|
* Signature: (J)V
|
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
jlong Java_org_rocksdb_RocksDB_getLatestSequenceNumber(
|
|
|
|
JNIEnv*, jobject, jlong jdb_handle) {
|
2015-01-25 22:47:29 +01:00
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
2015-01-25 22:46:11 +01:00
|
|
|
return db->GetLatestSequenceNumber();
|
|
|
|
}
|
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
|
|
|
* Method: setPreserveDeletesSequenceNumber
|
|
|
|
* Signature: (JJ)Z
|
|
|
|
*/
|
|
|
|
jboolean JNICALL Java_org_rocksdb_RocksDB_setPreserveDeletesSequenceNumber(
|
|
|
|
JNIEnv*, jobject, jlong jdb_handle, jlong jseq_number) {
|
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
|
|
|
if (db->SetPreserveDeletesSequenceNumber(
|
|
|
|
static_cast<uint64_t>(jseq_number))) {
|
|
|
|
return JNI_TRUE;
|
|
|
|
} else {
|
|
|
|
return JNI_FALSE;
|
|
|
|
}
|
|
|
|
}
|
2015-01-25 22:46:11 +01:00
|
|
|
|
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
2019-02-22 23:36:38 +01:00
|
|
|
* Method: disableFileDeletions
|
2015-01-25 22:46:11 +01:00
|
|
|
* Signature: (J)V
|
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
void Java_org_rocksdb_RocksDB_disableFileDeletions(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle) {
|
2015-01-25 22:47:29 +01:00
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
2015-01-25 22:46:11 +01:00
|
|
|
rocksdb::Status s = db->DisableFileDeletions();
|
|
|
|
if (!s.ok()) {
|
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(env, s);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
|
|
|
* Method: enableFileDeletions
|
|
|
|
* Signature: (JZ)V
|
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
void Java_org_rocksdb_RocksDB_enableFileDeletions(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle, jboolean jforce) {
|
2015-01-25 22:47:29 +01:00
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
2015-01-25 22:46:11 +01:00
|
|
|
rocksdb::Status s = db->EnableFileDeletions(jforce);
|
|
|
|
if (!s.ok()) {
|
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(env, s);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
|
|
|
* Method: getLiveFiles
|
|
|
|
* Signature: (JZ)[Ljava/lang/String;
|
|
|
|
*/
|
|
|
|
jobjectArray Java_org_rocksdb_RocksDB_getLiveFiles(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle, jboolean jflush_memtable) {
|
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
|
|
|
std::vector<std::string> live_files;
|
|
|
|
uint64_t manifest_file_size = 0;
|
|
|
|
auto s = db->GetLiveFiles(
|
|
|
|
live_files, &manifest_file_size, jflush_memtable == JNI_TRUE);
|
|
|
|
if (!s.ok()) {
|
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(env, s);
|
|
|
|
return nullptr;
|
|
|
|
}
|
|
|
|
|
|
|
|
// append the manifest_file_size to the vector
|
|
|
|
// for passing back to java
|
|
|
|
live_files.push_back(std::to_string(manifest_file_size));
|
|
|
|
|
|
|
|
return rocksdb::JniUtil::toJavaStrings(env, &live_files);
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
|
|
|
* Method: getSortedWalFiles
|
|
|
|
* Signature: (J)[Lorg/rocksdb/LogFile;
|
|
|
|
*/
|
|
|
|
jobjectArray Java_org_rocksdb_RocksDB_getSortedWalFiles(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle) {
|
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
|
|
|
std::vector<std::unique_ptr<rocksdb::LogFile>> sorted_wal_files;
|
|
|
|
auto s = db->GetSortedWalFiles(sorted_wal_files);
|
|
|
|
if (!s.ok()) {
|
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(env, s);
|
|
|
|
return nullptr;
|
|
|
|
}
|
|
|
|
|
|
|
|
// convert to Java type
|
|
|
|
const jsize jlen = static_cast<jsize>(sorted_wal_files.size());
|
|
|
|
jobjectArray jsorted_wal_files = env->NewObjectArray(
|
|
|
|
jlen, rocksdb::LogFileJni::getJClass(env), nullptr);
|
|
|
|
if(jsorted_wal_files == nullptr) {
|
|
|
|
// exception thrown: OutOfMemoryError
|
|
|
|
return nullptr;
|
|
|
|
}
|
|
|
|
|
|
|
|
jsize i = 0;
|
|
|
|
for (auto it = sorted_wal_files.begin(); it != sorted_wal_files.end(); ++it) {
|
|
|
|
jobject jlog_file = rocksdb::LogFileJni::fromCppLogFile(env, it->get());
|
|
|
|
if (jlog_file == nullptr) {
|
|
|
|
// exception occurred
|
|
|
|
env->DeleteLocalRef(jsorted_wal_files);
|
|
|
|
return nullptr;
|
|
|
|
}
|
|
|
|
|
|
|
|
env->SetObjectArrayElement(jsorted_wal_files, i++, jlog_file);
|
|
|
|
if (env->ExceptionCheck()) {
|
|
|
|
// exception occurred
|
|
|
|
env->DeleteLocalRef(jlog_file);
|
|
|
|
env->DeleteLocalRef(jsorted_wal_files);
|
|
|
|
return nullptr;
|
|
|
|
}
|
|
|
|
|
|
|
|
env->DeleteLocalRef(jlog_file);
|
|
|
|
}
|
|
|
|
|
|
|
|
return jsorted_wal_files;
|
|
|
|
}
|
2015-01-25 22:05:29 +01:00
|
|
|
|
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
|
|
|
* Method: getUpdatesSince
|
|
|
|
* Signature: (JJ)J
|
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
jlong Java_org_rocksdb_RocksDB_getUpdatesSince(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle, jlong jsequence_number) {
|
2015-01-25 22:47:29 +01:00
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
2015-01-25 22:05:29 +01:00
|
|
|
rocksdb::SequenceNumber sequence_number =
|
|
|
|
static_cast<rocksdb::SequenceNumber>(jsequence_number);
|
|
|
|
std::unique_ptr<rocksdb::TransactionLogIterator> iter;
|
|
|
|
rocksdb::Status s = db->GetUpdatesSince(sequence_number, &iter);
|
|
|
|
if (s.ok()) {
|
|
|
|
return reinterpret_cast<jlong>(iter.release());
|
|
|
|
}
|
2017-02-28 01:26:12 +01:00
|
|
|
|
2015-01-25 22:05:29 +01:00
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(env, s);
|
|
|
|
return 0;
|
|
|
|
}
|
2016-08-06 21:03:47 +02:00
|
|
|
|
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
2019-02-22 23:36:38 +01:00
|
|
|
* Method: deleteFile
|
|
|
|
* Signature: (JLjava/lang/String;)V
|
2016-08-06 21:03:47 +02:00
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
void Java_org_rocksdb_RocksDB_deleteFile(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle, jstring jname) {
|
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
|
|
|
jboolean has_exception = JNI_FALSE;
|
|
|
|
std::string name =
|
|
|
|
rocksdb::JniUtil::copyStdString(env, jname, &has_exception);
|
|
|
|
if (has_exception == JNI_TRUE) {
|
|
|
|
// exception occurred
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
db->DeleteFile(name);
|
|
|
|
}
|
2017-02-28 01:26:12 +01:00
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
|
|
|
* Method: getLiveFilesMetaData
|
|
|
|
* Signature: (J)[Lorg/rocksdb/LiveFileMetaData;
|
|
|
|
*/
|
|
|
|
jobjectArray Java_org_rocksdb_RocksDB_getLiveFilesMetaData(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle) {
|
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
|
|
|
std::vector<rocksdb::LiveFileMetaData> live_files_meta_data;
|
|
|
|
db->GetLiveFilesMetaData(&live_files_meta_data);
|
|
|
|
|
|
|
|
// convert to Java type
|
|
|
|
const jsize jlen = static_cast<jsize>(live_files_meta_data.size());
|
|
|
|
jobjectArray jlive_files_meta_data = env->NewObjectArray(
|
|
|
|
jlen, rocksdb::LiveFileMetaDataJni::getJClass(env), nullptr);
|
|
|
|
if(jlive_files_meta_data == nullptr) {
|
|
|
|
// exception thrown: OutOfMemoryError
|
|
|
|
return nullptr;
|
|
|
|
}
|
2017-02-28 01:26:12 +01:00
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
jsize i = 0;
|
|
|
|
for (auto it = live_files_meta_data.begin(); it != live_files_meta_data.end(); ++it) {
|
|
|
|
jobject jlive_file_meta_data =
|
|
|
|
rocksdb::LiveFileMetaDataJni::fromCppLiveFileMetaData(env, &(*it));
|
|
|
|
if (jlive_file_meta_data == nullptr) {
|
|
|
|
// exception occurred
|
|
|
|
env->DeleteLocalRef(jlive_files_meta_data);
|
|
|
|
return nullptr;
|
2017-02-28 01:26:12 +01:00
|
|
|
}
|
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
env->SetObjectArrayElement(jlive_files_meta_data, i++, jlive_file_meta_data);
|
|
|
|
if (env->ExceptionCheck()) {
|
|
|
|
// exception occurred
|
|
|
|
env->DeleteLocalRef(jlive_file_meta_data);
|
|
|
|
env->DeleteLocalRef(jlive_files_meta_data);
|
|
|
|
return nullptr;
|
2017-02-28 01:26:12 +01:00
|
|
|
}
|
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
env->DeleteLocalRef(jlive_file_meta_data);
|
2016-08-06 21:03:47 +02:00
|
|
|
}
|
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
return jlive_files_meta_data;
|
2016-08-06 21:03:47 +02:00
|
|
|
}
|
2016-09-29 23:04:41 +02:00
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
|
|
|
* Method: getColumnFamilyMetaData
|
|
|
|
* Signature: (JJ)Lorg/rocksdb/ColumnFamilyMetaData;
|
|
|
|
*/
|
|
|
|
jobject Java_org_rocksdb_RocksDB_getColumnFamilyMetaData(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle, jlong jcf_handle) {
|
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
|
|
|
rocksdb::ColumnFamilyHandle* cf_handle;
|
|
|
|
if (jcf_handle == 0) {
|
|
|
|
cf_handle = db->DefaultColumnFamily();
|
|
|
|
} else {
|
|
|
|
cf_handle =
|
|
|
|
reinterpret_cast<rocksdb::ColumnFamilyHandle*>(jcf_handle);
|
|
|
|
}
|
|
|
|
rocksdb::ColumnFamilyMetaData cf_metadata;
|
|
|
|
db->GetColumnFamilyMetaData(cf_handle, &cf_metadata);
|
|
|
|
return rocksdb::ColumnFamilyMetaDataJni::fromCppColumnFamilyMetaData(
|
|
|
|
env, &cf_metadata);
|
|
|
|
}
|
2016-09-29 23:04:41 +02:00
|
|
|
|
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
2017-05-22 19:22:49 +02:00
|
|
|
* Method: ingestExternalFile
|
|
|
|
* Signature: (JJ[Ljava/lang/String;IJ)V
|
2016-09-29 23:04:41 +02:00
|
|
|
*/
|
2017-05-22 19:22:49 +02:00
|
|
|
void Java_org_rocksdb_RocksDB_ingestExternalFile(
|
2019-02-22 23:36:38 +01:00
|
|
|
JNIEnv* env, jobject, jlong jdb_handle, jlong jcf_handle,
|
2016-09-29 23:04:41 +02:00
|
|
|
jobjectArray jfile_path_list, jint jfile_path_list_len,
|
2017-05-22 19:22:49 +02:00
|
|
|
jlong jingest_external_file_options_handle) {
|
2017-02-28 01:26:12 +01:00
|
|
|
jboolean has_exception = JNI_FALSE;
|
2018-04-13 02:55:14 +02:00
|
|
|
std::vector<std::string> file_path_list = rocksdb::JniUtil::copyStrings(
|
|
|
|
env, jfile_path_list, jfile_path_list_len, &has_exception);
|
|
|
|
if (has_exception == JNI_TRUE) {
|
2017-06-05 20:23:31 +02:00
|
|
|
// exception occurred
|
2017-02-28 01:26:12 +01:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2016-09-29 23:04:41 +02:00
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
|
|
|
auto* column_family =
|
|
|
|
reinterpret_cast<rocksdb::ColumnFamilyHandle*>(jcf_handle);
|
2018-04-13 02:55:14 +02:00
|
|
|
auto* ifo = reinterpret_cast<rocksdb::IngestExternalFileOptions*>(
|
|
|
|
jingest_external_file_options_handle);
|
2016-09-29 23:04:41 +02:00
|
|
|
rocksdb::Status s =
|
2017-05-22 19:22:49 +02:00
|
|
|
db->IngestExternalFile(column_family, file_path_list, *ifo);
|
2016-09-29 23:04:41 +02:00
|
|
|
if (!s.ok()) {
|
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(env, s);
|
|
|
|
}
|
|
|
|
}
|
2017-09-27 01:27:28 +02:00
|
|
|
|
2019-02-22 23:36:38 +01:00
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
|
|
|
* Method: verifyChecksum
|
|
|
|
* Signature: (J)V
|
|
|
|
*/
|
|
|
|
void Java_org_rocksdb_RocksDB_verifyChecksum(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle) {
|
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
|
|
|
auto s = db->VerifyChecksum();
|
|
|
|
if (!s.ok()) {
|
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(env, s);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
|
|
|
* Method: getDefaultColumnFamily
|
|
|
|
* Signature: (J)J
|
|
|
|
*/
|
|
|
|
jlong Java_org_rocksdb_RocksDB_getDefaultColumnFamily(
|
|
|
|
JNIEnv*, jobject, jlong jdb_handle) {
|
|
|
|
auto* db_handle = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
|
|
|
auto* cf_handle = db_handle->DefaultColumnFamily();
|
|
|
|
return reinterpret_cast<jlong>(cf_handle);
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
|
|
|
* Method: getPropertiesOfAllTables
|
|
|
|
* Signature: (JJ)Ljava/util/Map;
|
|
|
|
*/
|
|
|
|
jobject Java_org_rocksdb_RocksDB_getPropertiesOfAllTables(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle, jlong jcf_handle) {
|
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
|
|
|
rocksdb::ColumnFamilyHandle* cf_handle;
|
|
|
|
if (jcf_handle == 0) {
|
|
|
|
cf_handle = db->DefaultColumnFamily();
|
|
|
|
} else {
|
|
|
|
cf_handle =
|
|
|
|
reinterpret_cast<rocksdb::ColumnFamilyHandle*>(jcf_handle);
|
|
|
|
}
|
|
|
|
rocksdb::TablePropertiesCollection table_properties_collection;
|
|
|
|
auto s = db->GetPropertiesOfAllTables(cf_handle,
|
|
|
|
&table_properties_collection);
|
|
|
|
if (!s.ok()) {
|
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(env, s);
|
|
|
|
}
|
|
|
|
|
|
|
|
// convert to Java type
|
|
|
|
jobject jhash_map = rocksdb::HashMapJni::construct(
|
|
|
|
env, static_cast<uint32_t>(table_properties_collection.size()));
|
|
|
|
if (jhash_map == nullptr) {
|
|
|
|
// exception occurred
|
|
|
|
return nullptr;
|
|
|
|
}
|
|
|
|
|
|
|
|
const rocksdb::HashMapJni::FnMapKV<const std::string, const std::shared_ptr<const rocksdb::TableProperties>, jobject, jobject> fn_map_kv =
|
|
|
|
[env](const std::pair<const std::string, const std::shared_ptr<const rocksdb::TableProperties>>& kv) {
|
|
|
|
jstring jkey = rocksdb::JniUtil::toJavaString(env, &(kv.first), false);
|
|
|
|
if (env->ExceptionCheck()) {
|
|
|
|
// an error occurred
|
|
|
|
return std::unique_ptr<std::pair<jobject, jobject>>(nullptr);
|
|
|
|
}
|
|
|
|
|
|
|
|
jobject jtable_properties = rocksdb::TablePropertiesJni::fromCppTableProperties(env, *(kv.second.get()));
|
|
|
|
if (jtable_properties == nullptr) {
|
|
|
|
// an error occurred
|
|
|
|
env->DeleteLocalRef(jkey);
|
|
|
|
return std::unique_ptr<std::pair<jobject, jobject>>(nullptr);
|
|
|
|
}
|
|
|
|
|
|
|
|
return std::unique_ptr<std::pair<jobject, jobject>>(new std::pair<jobject, jobject>(static_cast<jobject>(jkey), static_cast<jobject>(jtable_properties)));
|
|
|
|
};
|
|
|
|
|
|
|
|
if (!rocksdb::HashMapJni::putAll(env, jhash_map, table_properties_collection.begin(), table_properties_collection.end(), fn_map_kv)) {
|
|
|
|
// exception occurred
|
|
|
|
return nullptr;
|
|
|
|
}
|
|
|
|
|
|
|
|
return jhash_map;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
|
|
|
* Method: getPropertiesOfTablesInRange
|
|
|
|
* Signature: (JJ[J)Ljava/util/Map;
|
|
|
|
*/
|
|
|
|
jobject Java_org_rocksdb_RocksDB_getPropertiesOfTablesInRange(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle, jlong jcf_handle,
|
|
|
|
jlongArray jrange_slice_handles) {
|
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
|
|
|
rocksdb::ColumnFamilyHandle* cf_handle;
|
|
|
|
if (jcf_handle == 0) {
|
|
|
|
cf_handle = db->DefaultColumnFamily();
|
|
|
|
} else {
|
|
|
|
cf_handle =
|
|
|
|
reinterpret_cast<rocksdb::ColumnFamilyHandle*>(jcf_handle);
|
|
|
|
}
|
|
|
|
const jsize jlen = env->GetArrayLength(jrange_slice_handles);
|
|
|
|
jboolean jrange_slice_handles_is_copy = JNI_FALSE;
|
|
|
|
jlong *jrange_slice_handle = env->GetLongArrayElements(
|
|
|
|
jrange_slice_handles, &jrange_slice_handles_is_copy);
|
|
|
|
if (jrange_slice_handle == nullptr) {
|
|
|
|
// exception occurred
|
|
|
|
return nullptr;
|
|
|
|
}
|
|
|
|
|
|
|
|
const size_t ranges_len = static_cast<size_t>(jlen / 2);
|
|
|
|
auto ranges = std::unique_ptr<rocksdb::Range[]>(new rocksdb::Range[ranges_len]);
|
|
|
|
for (jsize i = 0, j = 0; i < jlen; ++i) {
|
|
|
|
auto* start = reinterpret_cast<rocksdb::Slice*>(
|
|
|
|
jrange_slice_handle[i]);
|
|
|
|
auto* limit = reinterpret_cast<rocksdb::Slice*>(
|
|
|
|
jrange_slice_handle[++i]);
|
|
|
|
ranges[j++] = rocksdb::Range(*start, *limit);
|
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::TablePropertiesCollection table_properties_collection;
|
|
|
|
auto s = db->GetPropertiesOfTablesInRange(
|
|
|
|
cf_handle, ranges.get(), ranges_len, &table_properties_collection);
|
|
|
|
if (!s.ok()) {
|
|
|
|
// error occurred
|
|
|
|
env->ReleaseLongArrayElements(jrange_slice_handles, jrange_slice_handle, JNI_ABORT);
|
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(env, s);
|
|
|
|
return nullptr;
|
|
|
|
}
|
|
|
|
|
|
|
|
// cleanup
|
|
|
|
env->ReleaseLongArrayElements(jrange_slice_handles, jrange_slice_handle, JNI_ABORT);
|
|
|
|
|
|
|
|
return jrange_slice_handles;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
|
|
|
* Method: suggestCompactRange
|
|
|
|
* Signature: (JJ)[J
|
|
|
|
*/
|
|
|
|
jlongArray Java_org_rocksdb_RocksDB_suggestCompactRange(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle, jlong jcf_handle) {
|
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
|
|
|
rocksdb::ColumnFamilyHandle* cf_handle;
|
|
|
|
if (jcf_handle == 0) {
|
|
|
|
cf_handle = db->DefaultColumnFamily();
|
|
|
|
} else {
|
|
|
|
cf_handle =
|
|
|
|
reinterpret_cast<rocksdb::ColumnFamilyHandle*>(jcf_handle);
|
|
|
|
}
|
|
|
|
auto* begin = new rocksdb::Slice();
|
|
|
|
auto* end = new rocksdb::Slice();
|
|
|
|
auto s = db->SuggestCompactRange(cf_handle, begin, end);
|
|
|
|
if (!s.ok()) {
|
|
|
|
// error occurred
|
|
|
|
delete begin;
|
|
|
|
delete end;
|
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(env, s);
|
|
|
|
return nullptr;
|
|
|
|
}
|
|
|
|
|
|
|
|
jlongArray jslice_handles = env->NewLongArray(2);
|
|
|
|
if (jslice_handles == nullptr) {
|
|
|
|
// exception thrown: OutOfMemoryError
|
|
|
|
delete begin;
|
|
|
|
delete end;
|
|
|
|
return nullptr;
|
|
|
|
}
|
|
|
|
|
|
|
|
jlong slice_handles[2];
|
|
|
|
slice_handles[0] = reinterpret_cast<jlong>(begin);
|
|
|
|
slice_handles[1] = reinterpret_cast<jlong>(end);
|
|
|
|
env->SetLongArrayRegion(jslice_handles, 0, 2, slice_handles);
|
|
|
|
if (env->ExceptionCheck()) {
|
|
|
|
// exception thrown: ArrayIndexOutOfBoundsException
|
|
|
|
delete begin;
|
|
|
|
delete end;
|
|
|
|
env->DeleteLocalRef(jslice_handles);
|
|
|
|
return nullptr;
|
|
|
|
}
|
|
|
|
|
|
|
|
return jslice_handles;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
|
|
|
* Method: promoteL0
|
|
|
|
* Signature: (JJI)V
|
|
|
|
*/
|
|
|
|
void Java_org_rocksdb_RocksDB_promoteL0(
|
|
|
|
JNIEnv*, jobject, jlong jdb_handle, jlong jcf_handle, jint jtarget_level) {
|
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
|
|
|
rocksdb::ColumnFamilyHandle* cf_handle;
|
|
|
|
if (jcf_handle == 0) {
|
|
|
|
cf_handle = db->DefaultColumnFamily();
|
|
|
|
} else {
|
|
|
|
cf_handle =
|
|
|
|
reinterpret_cast<rocksdb::ColumnFamilyHandle*>(jcf_handle);
|
|
|
|
}
|
|
|
|
db->PromoteL0(cf_handle, static_cast<int>(jtarget_level));
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
|
|
|
* Method: startTrace
|
|
|
|
* Signature: (JJJ)V
|
|
|
|
*/
|
|
|
|
void Java_org_rocksdb_RocksDB_startTrace(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle, jlong jmax_trace_file_size,
|
|
|
|
jlong jtrace_writer_jnicallback_handle) {
|
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
|
|
|
rocksdb::TraceOptions trace_options;
|
|
|
|
trace_options.max_trace_file_size =
|
|
|
|
static_cast<uint64_t>(jmax_trace_file_size);
|
|
|
|
// transfer ownership of trace writer from Java to C++
|
|
|
|
auto trace_writer = std::unique_ptr<rocksdb::TraceWriterJniCallback>(
|
|
|
|
reinterpret_cast<rocksdb::TraceWriterJniCallback*>(
|
|
|
|
jtrace_writer_jnicallback_handle));
|
|
|
|
auto s = db->StartTrace(trace_options, std::move(trace_writer));
|
|
|
|
if (!s.ok()) {
|
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(env, s);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
|
|
|
* Method: endTrace
|
|
|
|
* Signature: (J)V
|
|
|
|
*/
|
|
|
|
JNIEXPORT void JNICALL Java_org_rocksdb_RocksDB_endTrace(
|
|
|
|
JNIEnv* env, jobject, jlong jdb_handle) {
|
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
|
|
|
auto s = db->EndTrace();
|
|
|
|
if (!s.ok()) {
|
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(env, s);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2017-09-27 01:27:28 +02:00
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
|
|
|
* Method: destroyDB
|
|
|
|
* Signature: (Ljava/lang/String;J)V
|
|
|
|
*/
|
2019-02-22 23:36:38 +01:00
|
|
|
void Java_org_rocksdb_RocksDB_destroyDB(
|
|
|
|
JNIEnv* env, jclass, jstring jdb_path, jlong joptions_handle) {
|
2017-09-27 01:27:28 +02:00
|
|
|
const char* db_path = env->GetStringUTFChars(jdb_path, nullptr);
|
2018-04-13 02:55:14 +02:00
|
|
|
if (db_path == nullptr) {
|
2017-09-27 01:27:28 +02:00
|
|
|
// exception thrown: OutOfMemoryError
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
auto* options = reinterpret_cast<rocksdb::Options*>(joptions_handle);
|
|
|
|
if (options == nullptr) {
|
2018-04-13 02:55:14 +02:00
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(
|
|
|
|
env, rocksdb::Status::InvalidArgument("Invalid Options."));
|
2017-09-27 01:27:28 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
rocksdb::Status s = rocksdb::DestroyDB(db_path, *options);
|
|
|
|
env->ReleaseStringUTFChars(jdb_path, db_path);
|
|
|
|
|
|
|
|
if (!s.ok()) {
|
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(env, s);
|
|
|
|
}
|
|
|
|
}
|
2019-09-18 18:43:10 +02:00
|
|
|
|
|
|
|
bool get_slice_helper(JNIEnv* env, jobjectArray ranges, jsize index,
|
|
|
|
std::unique_ptr<rocksdb::Slice>& slice,
|
|
|
|
std::vector<std::unique_ptr<jbyte[]>>& ranges_to_free) {
|
|
|
|
jobject jArray = env->GetObjectArrayElement(ranges, index);
|
|
|
|
if (env->ExceptionCheck()) {
|
|
|
|
// exception thrown: ArrayIndexOutOfBoundsException
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (jArray == nullptr) {
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
jbyteArray jba = reinterpret_cast<jbyteArray>(jArray);
|
|
|
|
jsize len_ba = env->GetArrayLength(jba);
|
|
|
|
ranges_to_free.push_back(std::unique_ptr<jbyte[]>(new jbyte[len_ba]));
|
|
|
|
env->GetByteArrayRegion(jba, 0, len_ba, ranges_to_free.back().get());
|
|
|
|
if (env->ExceptionCheck()) {
|
|
|
|
// exception thrown: ArrayIndexOutOfBoundsException
|
|
|
|
env->DeleteLocalRef(jArray);
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
env->DeleteLocalRef(jArray);
|
|
|
|
slice.reset(new rocksdb::Slice(
|
|
|
|
reinterpret_cast<char*>(ranges_to_free.back().get()), len_ba));
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
/*
|
|
|
|
* Class: org_rocksdb_RocksDB
|
|
|
|
* Method: deleteFilesInRanges
|
|
|
|
* Signature: (JJLjava/util/List;Z)V
|
|
|
|
*/
|
|
|
|
JNIEXPORT void JNICALL Java_org_rocksdb_RocksDB_deleteFilesInRanges(
|
|
|
|
JNIEnv* env, jobject /*jdb*/, jlong jdb_handle, jlong jcf_handle,
|
|
|
|
jobjectArray ranges, jboolean include_end) {
|
|
|
|
jsize length = env->GetArrayLength(ranges);
|
|
|
|
|
|
|
|
std::vector<rocksdb::RangePtr> rangesVector;
|
|
|
|
std::vector<std::unique_ptr<rocksdb::Slice>> slices;
|
|
|
|
std::vector<std::unique_ptr<jbyte[]>> ranges_to_free;
|
|
|
|
for (jsize i = 0; (i + 1) < length; i += 2) {
|
|
|
|
slices.push_back(std::unique_ptr<rocksdb::Slice>());
|
|
|
|
if (!get_slice_helper(env, ranges, i, slices.back(), ranges_to_free)) {
|
|
|
|
// exception thrown
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
slices.push_back(std::unique_ptr<rocksdb::Slice>());
|
|
|
|
if (!get_slice_helper(env, ranges, i + 1, slices.back(), ranges_to_free)) {
|
|
|
|
// exception thrown
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
rangesVector.push_back(rocksdb::RangePtr(slices[slices.size() - 2].get(),
|
|
|
|
slices[slices.size() - 1].get()));
|
|
|
|
}
|
|
|
|
|
|
|
|
auto* db = reinterpret_cast<rocksdb::DB*>(jdb_handle);
|
|
|
|
auto* column_family =
|
|
|
|
reinterpret_cast<rocksdb::ColumnFamilyHandle*>(jcf_handle);
|
|
|
|
|
|
|
|
rocksdb::Status s = rocksdb::DeleteFilesInRanges(
|
|
|
|
db, column_family == nullptr ? db->DefaultColumnFamily() : column_family,
|
|
|
|
rangesVector.data(), rangesVector.size(), include_end);
|
|
|
|
|
|
|
|
if (!s.ok()) {
|
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(env, s);
|
|
|
|
}
|
|
|
|
}
|