// Copyright (c) 2014, Facebook, Inc. All rights reserved. // This source code is licensed under the BSD-style license found in the // LICENSE file in the root directory of this source tree. An additional grant // of patent rights can be found in the PATENTS file in the same directory. // // This file implements the "bridge" between Java and C++ and enables // calling c++ rocksdb::DB methods from Java side. #include #include #include #include #include #include "include/org_rocksdb_RocksDB.h" #include "rocksjni/portal.h" #include "rocksdb/db.h" #include "rocksdb/cache.h" ////////////////////////////////////////////////////////////////////////////// // rocksdb::DB::Open /* * Class: org_rocksdb_RocksDB * Method: open * Signature: (JLjava/lang/String;)V */ void Java_org_rocksdb_RocksDB_open__JLjava_lang_String_2( JNIEnv* env, jobject jdb, jlong jopt_handle, jstring jdb_path) { auto opt = reinterpret_cast(jopt_handle); rocksdb::DB* db = nullptr; const char* db_path = env->GetStringUTFChars(jdb_path, 0); rocksdb::Status s = rocksdb::DB::Open(*opt, db_path, &db); env->ReleaseStringUTFChars(jdb_path, db_path); if (s.ok()) { rocksdb::RocksDBJni::setHandle(env, jdb, db); return; } rocksdb::RocksDBExceptionJni::ThrowNew(env, s); } /* * Class: org_rocksdb_RocksDB * Method: openROnly * Signature: (JLjava/lang/String;)V */ void Java_org_rocksdb_RocksDB_openROnly__JLjava_lang_String_2( JNIEnv* env, jobject jdb, jlong jopt_handle, jstring jdb_path) { auto opt = reinterpret_cast(jopt_handle); rocksdb::DB* db = nullptr; const char* db_path = env->GetStringUTFChars(jdb_path, 0); rocksdb::Status s = rocksdb::DB::OpenForReadOnly(*opt, db_path, &db); env->ReleaseStringUTFChars(jdb_path, db_path); if (s.ok()) { rocksdb::RocksDBJni::setHandle(env, jdb, db); return; } rocksdb::RocksDBExceptionJni::ThrowNew(env, s); } /* * Class: org_rocksdb_RocksDB * Method: openROnly * Signature: (JLjava/lang/String;Ljava/util/List;I)Ljava/util/List; */ jobject Java_org_rocksdb_RocksDB_openROnly__JLjava_lang_String_2Ljava_util_List_2I( JNIEnv* env, jobject jdb, jlong jopt_handle, jstring jdb_path, jobject jcfname_list, jint jcfname_count) { auto opt = reinterpret_cast(jopt_handle); rocksdb::DB* db = nullptr; const char* db_path = env->GetStringUTFChars(jdb_path, 0); std::vector cfnames_to_free; std::vector jcfnames_for_free; std::vector column_families; std::vector handles; // get iterator for cfnames jobject iteratorObj = env->CallObjectMethod( jcfname_list, rocksdb::ListJni::getIteratorMethod(env)); // iterate over cfnames and convert cfnames to // ColumnFamilyDescriptor instances while (env->CallBooleanMethod( iteratorObj, rocksdb::ListJni::getHasNextMethod(env)) == JNI_TRUE) { jstring jstr = (jstring) env->CallObjectMethod(iteratorObj, rocksdb::ListJni::getNextMethod(env)); const char* cfname = env->GetStringUTFChars(jstr, 0); // free allocated cfnames after call to open cfnames_to_free.push_back(cfname); jcfnames_for_free.push_back(jstr); column_families.push_back(rocksdb::ColumnFamilyDescriptor(cfname, rocksdb::ColumnFamilyOptions())); } rocksdb::Status s = rocksdb::DB::OpenForReadOnly(*opt, db_path, column_families, &handles, &db); env->ReleaseStringUTFChars(jdb_path, db_path); // free jbyte allocations for (std::vector::size_type i = 0; i != cfnames_to_free.size(); i++) { // free cfnames env->ReleaseStringUTFChars(jcfnames_for_free[i], cfnames_to_free[i]); } jobject jcfhandle_list = nullptr; // check if open operation was successful if (s.ok()) { rocksdb::RocksDBJni::setHandle(env, jdb, db); jclass jclazz = env->FindClass("java/util/ArrayList"); jmethodID mid = rocksdb::ListJni::getArrayListConstructorMethodId( env, jclazz); jobject jcfhandle_list = env->NewObject(jclazz, mid, handles.size()); // insert in java list for (std::vector::size_type i = 0; i != handles.size(); i++) { // jlong must be converted to Long due to collections restrictions jclass jclazz = env->FindClass("java/lang/Long"); jmethodID mid = env->GetMethodID(jclazz, "", "(J)V"); jobject obj = env->NewObject(jclazz, mid, reinterpret_cast(handles[i])); env->CallBooleanMethod(jcfhandle_list, rocksdb::ListJni::getListAddMethodId(env), obj); } return jcfhandle_list; } rocksdb::RocksDBExceptionJni::ThrowNew(env, s); return jcfhandle_list; } /* * Class: org_rocksdb_RocksDB * Method: open * Signature: (JLjava/lang/String;Ljava/util/List;I)Ljava/util/List; */ jobject Java_org_rocksdb_RocksDB_open__JLjava_lang_String_2Ljava_util_List_2I( JNIEnv* env, jobject jdb, jlong jopt_handle, jstring jdb_path, jobject jcfname_list, jint jcfname_count) { auto opt = reinterpret_cast(jopt_handle); rocksdb::DB* db = nullptr; const char* db_path = env->GetStringUTFChars(jdb_path, 0); std::vector cfnames_to_free; std::vector jcfnames_for_free; std::vector column_families; std::vector handles; // get iterator for cfnames jobject iteratorObj = env->CallObjectMethod( jcfname_list, rocksdb::ListJni::getIteratorMethod(env)); // iterate over cfnames and convert cfnames to // ColumnFamilyDescriptor instances while (env->CallBooleanMethod( iteratorObj, rocksdb::ListJni::getHasNextMethod(env)) == JNI_TRUE) { jstring jstr = (jstring) env->CallObjectMethod(iteratorObj, rocksdb::ListJni::getNextMethod(env)); const char* cfname = env->GetStringUTFChars(jstr, 0); // free allocated cfnames after call to open cfnames_to_free.push_back(cfname); jcfnames_for_free.push_back(jstr); column_families.push_back(rocksdb::ColumnFamilyDescriptor(cfname, rocksdb::ColumnFamilyOptions())); } rocksdb::Status s = rocksdb::DB::Open(*opt, db_path, column_families, &handles, &db); env->ReleaseStringUTFChars(jdb_path, db_path); // free jbyte allocations for (std::vector::size_type i = 0; i != cfnames_to_free.size(); i++) { // free cfnames env->ReleaseStringUTFChars(jcfnames_for_free[i], cfnames_to_free[i]); } jobject jcfhandle_list = nullptr; // check if open operation was successful if (s.ok()) { rocksdb::RocksDBJni::setHandle(env, jdb, db); jclass jclazz = env->FindClass("java/util/ArrayList"); jmethodID mid = rocksdb::ListJni::getArrayListConstructorMethodId( env, jclazz); jobject jcfhandle_list = env->NewObject(jclazz, mid, handles.size()); // insert in java list for (std::vector::size_type i = 0; i != handles.size(); i++) { // jlong must be converted to Long due to collections restrictions jclass jclazz = env->FindClass("java/lang/Long"); jmethodID mid = env->GetMethodID(jclazz, "", "(J)V"); jobject obj = env->NewObject(jclazz, mid, reinterpret_cast(handles[i])); env->CallBooleanMethod(jcfhandle_list, rocksdb::ListJni::getListAddMethodId(env), obj); } return jcfhandle_list; } rocksdb::RocksDBExceptionJni::ThrowNew(env, s); return jcfhandle_list; } ////////////////////////////////////////////////////////////////////////////// // rocksdb::DB::ListColumnFamilies /* * Class: org_rocksdb_RocksDB * Method: listColumnFamilies * Signature: (JLjava/lang/String;)Ljava/util/List; */ jobject Java_org_rocksdb_RocksDB_listColumnFamilies( JNIEnv* env, jclass jclazz, jlong jopt_handle, jstring jdb_path) { std::vector column_family_names; auto opt = reinterpret_cast(jopt_handle); const char* db_path = env->GetStringUTFChars(jdb_path, 0); jobject jvalue_list = nullptr; rocksdb::Status s = rocksdb::DB::ListColumnFamilies(*opt, db_path, &column_family_names); env->ReleaseStringUTFChars(jdb_path, db_path); if (s.ok()) { // Don't reuse class pointer jclass jclazz = env->FindClass("java/util/ArrayList"); jmethodID mid = rocksdb::ListJni::getArrayListConstructorMethodId(env, jclazz); jvalue_list = env->NewObject(jclazz, mid, column_family_names.size()); for (std::vector::size_type i = 0; i < column_family_names.size(); i++) { jbyteArray jvalue = env->NewByteArray(column_family_names[i].size()); env->SetByteArrayRegion(jvalue, 0, column_family_names[i].size(), reinterpret_cast(column_family_names[i].c_str())); env->CallBooleanMethod(jvalue_list, rocksdb::ListJni::getListAddMethodId(env), jvalue); } } return jvalue_list; } ////////////////////////////////////////////////////////////////////////////// // rocksdb::DB::Put void rocksdb_put_helper( JNIEnv* env, rocksdb::DB* db, const rocksdb::WriteOptions& write_options, rocksdb::ColumnFamilyHandle* cf_handle, jbyteArray jkey, jint jkey_len, jbyteArray jvalue, jint jvalue_len) { jbyte* key = env->GetByteArrayElements(jkey, 0); jbyte* value = env->GetByteArrayElements(jvalue, 0); rocksdb::Slice key_slice(reinterpret_cast(key), jkey_len); rocksdb::Slice value_slice(reinterpret_cast(value), jvalue_len); 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); } // 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); env->ReleaseByteArrayElements(jvalue, value, JNI_ABORT); if (s.ok()) { return; } rocksdb::RocksDBExceptionJni::ThrowNew(env, s); } /* * Class: org_rocksdb_RocksDB * Method: put * Signature: (J[BI[BI)V */ void Java_org_rocksdb_RocksDB_put__J_3BI_3BI( JNIEnv* env, jobject jdb, jlong jdb_handle, jbyteArray jkey, jint jkey_len, jbyteArray jvalue, jint jvalue_len) { auto db = reinterpret_cast(jdb_handle); static const rocksdb::WriteOptions default_write_options = rocksdb::WriteOptions(); rocksdb_put_helper(env, db, default_write_options, nullptr, jkey, jkey_len, jvalue, jvalue_len); } /* * Class: org_rocksdb_RocksDB * Method: put * Signature: (J[BI[BIJ)V */ void Java_org_rocksdb_RocksDB_put__J_3BI_3BIJ( JNIEnv* env, jobject jdb, jlong jdb_handle, jbyteArray jkey, jint jkey_len, jbyteArray jvalue, jint jvalue_len, jlong jcf_handle) { auto db = reinterpret_cast(jdb_handle); static const rocksdb::WriteOptions default_write_options = rocksdb::WriteOptions(); auto cf_handle = reinterpret_cast(jcf_handle); if (cf_handle != nullptr) { rocksdb_put_helper(env, db, default_write_options, cf_handle, jkey, jkey_len, jvalue, jvalue_len); } else { rocksdb::RocksDBExceptionJni::ThrowNew(env, rocksdb::Status::InvalidArgument("Invalid ColumnFamilyHandle.")); } } /* * Class: org_rocksdb_RocksDB * Method: put * Signature: (JJ[BI[BI)V */ void Java_org_rocksdb_RocksDB_put__JJ_3BI_3BI( JNIEnv* env, jobject jdb, jlong jdb_handle, jlong jwrite_options_handle, jbyteArray jkey, jint jkey_len, jbyteArray jvalue, jint jvalue_len) { auto db = reinterpret_cast(jdb_handle); auto write_options = reinterpret_cast( jwrite_options_handle); rocksdb_put_helper(env, db, *write_options, nullptr, jkey, jkey_len, jvalue, jvalue_len); } /* * Class: org_rocksdb_RocksDB * Method: put * Signature: (JJ[BI[BIJ)V */ void Java_org_rocksdb_RocksDB_put__JJ_3BI_3BIJ( JNIEnv* env, jobject jdb, jlong jdb_handle, jlong jwrite_options_handle, jbyteArray jkey, jint jkey_len, jbyteArray jvalue, jint jvalue_len, jlong jcf_handle) { auto db = reinterpret_cast(jdb_handle); auto write_options = reinterpret_cast( jwrite_options_handle); auto cf_handle = reinterpret_cast(jcf_handle); if (cf_handle != nullptr) { rocksdb_put_helper(env, db, *write_options, cf_handle, jkey, jkey_len, jvalue, jvalue_len); } else { rocksdb::RocksDBExceptionJni::ThrowNew(env, rocksdb::Status::InvalidArgument("Invalid ColumnFamilyHandle.")); } } ////////////////////////////////////////////////////////////////////////////// // rocksdb::DB::Write /* * Class: org_rocksdb_RocksDB * Method: write * Signature: (JJ)V */ void Java_org_rocksdb_RocksDB_write( JNIEnv* env, jobject jdb, jlong jwrite_options_handle, jlong jbatch_handle) { rocksdb::DB* db = rocksdb::RocksDBJni::getHandle(env, jdb); auto write_options = reinterpret_cast( jwrite_options_handle); auto batch = reinterpret_cast(jbatch_handle); rocksdb::Status s = db->Write(*write_options, batch); if (!s.ok()) { rocksdb::RocksDBExceptionJni::ThrowNew(env, s); } } ////////////////////////////////////////////////////////////////////////////// // 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_len, jobject jvalue) { std::string value; bool value_found = false; jboolean isCopy; jbyte* key = env->GetByteArrayElements(jkey, &isCopy); rocksdb::Slice key_slice(reinterpret_cast(key), jkey_len); bool keyMaxExist = db->KeyMayExist(read_opt, cf_handle, key_slice, &value, &value_found); if (value_found && !value.empty()) { jclass clazz = env->GetObjectClass(jvalue); jmethodID mid = env->GetMethodID(clazz, "append", "(Ljava/lang/String;)Ljava/lang/StringBuffer;"); jstring new_value_str = env->NewStringUTF(value.c_str()); env->CallObjectMethod(jvalue, mid, new_value_str); } env->ReleaseByteArrayElements(jkey, key, JNI_ABORT); return static_cast(keyMaxExist); } /* * Class: org_rocksdb_RocksDB * Method: keyMayExist * Signature: ([BIJLjava/lang/StringBuffer;)Z */ jboolean Java_org_rocksdb_RocksDB_keyMayExist___3BIJLjava_lang_StringBuffer_2( JNIEnv* env, jobject jdb, jbyteArray jkey, jint jkey_len, jlong jcf_handle, jobject jvalue) { rocksdb::DB* db = rocksdb::RocksDBJni::getHandle(env, jdb); auto cf_handle = reinterpret_cast( jcf_handle); if (cf_handle != nullptr) { return key_may_exist_helper(env, db, rocksdb::ReadOptions(), cf_handle, jkey, jkey_len, jvalue); } else { rocksdb::RocksDBExceptionJni::ThrowNew(env, rocksdb::Status::InvalidArgument("Invalid ColumnFamilyHandle.")); } return true; } /* * Class: org_rocksdb_RocksDB * Method: keyMayExist * Signature: (J[BIJLjava/lang/StringBuffer;)Z */ jboolean Java_org_rocksdb_RocksDB_keyMayExist__J_3BIJLjava_lang_StringBuffer_2( JNIEnv* env, jobject jdb, jlong jread_options_handle, jbyteArray jkey, jint jkey_len, jlong jcf_handle, jobject jvalue) { rocksdb::DB* db = rocksdb::RocksDBJni::getHandle(env, jdb); auto& read_options = *reinterpret_cast( jread_options_handle); auto cf_handle = reinterpret_cast( jcf_handle); if (cf_handle != nullptr) { return key_may_exist_helper(env, db, read_options, cf_handle, jkey, jkey_len, jvalue); } else { rocksdb::RocksDBExceptionJni::ThrowNew(env, rocksdb::Status::InvalidArgument("Invalid ColumnFamilyHandle.")); } return true; } ////////////////////////////////////////////////////////////////////////////// // rocksdb::DB::Get jbyteArray rocksdb_get_helper( JNIEnv* env, rocksdb::DB* db, const rocksdb::ReadOptions& read_opt, rocksdb::ColumnFamilyHandle* column_family_handle, jbyteArray jkey, jint jkey_len) { jboolean isCopy; jbyte* key = env->GetByteArrayElements(jkey, &isCopy); rocksdb::Slice key_slice( reinterpret_cast(key), jkey_len); std::string value; rocksdb::Status s; if (column_family_handle != nullptr) { s = db->Get(read_opt, column_family_handle, key_slice, &value); } else { // backwards compatibility s = db->Get(read_opt, key_slice, &value); } // trigger java unref on key. // 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); if (s.IsNotFound()) { return nullptr; } if (s.ok()) { jbyteArray jvalue = env->NewByteArray(value.size()); env->SetByteArrayRegion( jvalue, 0, value.size(), reinterpret_cast(value.c_str())); return jvalue; } rocksdb::RocksDBExceptionJni::ThrowNew(env, s); return nullptr; } /* * Class: org_rocksdb_RocksDB * Method: get * Signature: (J[BI)[B */ jbyteArray Java_org_rocksdb_RocksDB_get__J_3BI( JNIEnv* env, jobject jdb, jlong jdb_handle, jbyteArray jkey, jint jkey_len) { return rocksdb_get_helper(env, reinterpret_cast(jdb_handle), rocksdb::ReadOptions(), nullptr, jkey, jkey_len); } /* * Class: org_rocksdb_RocksDB * Method: get * Signature: (J[BIJ)[B */ jbyteArray Java_org_rocksdb_RocksDB_get__J_3BIJ( JNIEnv* env, jobject jdb, jlong jdb_handle, jbyteArray jkey, jint jkey_len, jlong jcf_handle) { auto db_handle = reinterpret_cast(jdb_handle); auto cf_handle = reinterpret_cast(jcf_handle); if (cf_handle != nullptr) { return rocksdb_get_helper(env, db_handle, rocksdb::ReadOptions(), cf_handle, jkey, jkey_len); } else { rocksdb::RocksDBExceptionJni::ThrowNew(env, rocksdb::Status::InvalidArgument("Invalid ColumnFamilyHandle.")); // will never be evaluated return env->NewByteArray(0); } } /* * Class: org_rocksdb_RocksDB * Method: get * Signature: (JJ[BI)[B */ jbyteArray Java_org_rocksdb_RocksDB_get__JJ_3BI( JNIEnv* env, jobject jdb, jlong jdb_handle, jlong jropt_handle, jbyteArray jkey, jint jkey_len) { return rocksdb_get_helper(env, reinterpret_cast(jdb_handle), *reinterpret_cast(jropt_handle), nullptr, jkey, jkey_len); } /* * Class: org_rocksdb_RocksDB * Method: get * Signature: (JJ[BIJ)[B */ jbyteArray Java_org_rocksdb_RocksDB_get__JJ_3BIJ( JNIEnv* env, jobject jdb, jlong jdb_handle, jlong jropt_handle, jbyteArray jkey, jint jkey_len, jlong jcf_handle) { auto db_handle = reinterpret_cast(jdb_handle); auto& ro_opt = *reinterpret_cast(jropt_handle); auto cf_handle = reinterpret_cast(jcf_handle); if (cf_handle != nullptr) { return rocksdb_get_helper(env, db_handle, ro_opt, cf_handle, jkey, jkey_len); } else { rocksdb::RocksDBExceptionJni::ThrowNew(env, rocksdb::Status::InvalidArgument("Invalid ColumnFamilyHandle.")); // will never be evaluated return env->NewByteArray(0); } } jint rocksdb_get_helper( JNIEnv* env, rocksdb::DB* db, const rocksdb::ReadOptions& read_options, rocksdb::ColumnFamilyHandle* column_family_handle, jbyteArray jkey, jint jkey_len, jbyteArray jvalue, jint jvalue_len) { static const int kNotFound = -1; static const int kStatusError = -2; jbyte* key = env->GetByteArrayElements(jkey, 0); rocksdb::Slice key_slice( reinterpret_cast(key), jkey_len); // TODO(yhchiang): we might save one memory allocation here by adding // a DB::Get() function which takes preallocated jbyte* as input. std::string cvalue; rocksdb::Status s; if (column_family_handle != nullptr) { s = db->Get(read_options, column_family_handle, key_slice, &cvalue); } else { // backwards compatibility s = db->Get(read_options, key_slice, &cvalue); } // trigger java unref on key. // 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); if (s.IsNotFound()) { return kNotFound; } else if (!s.ok()) { // 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; } int cvalue_len = static_cast(cvalue.size()); int length = std::min(jvalue_len, cvalue_len); env->SetByteArrayRegion( jvalue, 0, length, reinterpret_cast(cvalue.c_str())); return cvalue_len; } // cf multi get jobject multi_get_helper(JNIEnv* env, jobject jdb, rocksdb::DB* db, const rocksdb::ReadOptions& rOpt, jobject jkey_list, jint jkeys_count, jobject jcfhandle_list) { std::vector keys; std::vector keys_to_free; std::vector cf_handles; if (jcfhandle_list != nullptr) { // get cf iterator jobject cfIteratorObj = env->CallObjectMethod( jcfhandle_list, rocksdb::ListJni::getIteratorMethod(env)); // iterate over keys and convert java byte array to slice while (env->CallBooleanMethod( cfIteratorObj, rocksdb::ListJni::getHasNextMethod(env)) == JNI_TRUE) { jobject jobj = (jbyteArray) env->CallObjectMethod( cfIteratorObj, rocksdb::ListJni::getNextMethod(env)); rocksdb::ColumnFamilyHandle* cfHandle = rocksdb::ColumnFamilyHandleJni::getHandle(env, jobj); cf_handles.push_back(cfHandle); } } // Process key list // get iterator jobject iteratorObj = env->CallObjectMethod( jkey_list, rocksdb::ListJni::getIteratorMethod(env)); // iterate over keys and convert java byte array to slice while (env->CallBooleanMethod( iteratorObj, rocksdb::ListJni::getHasNextMethod(env)) == JNI_TRUE) { jbyteArray jkey = (jbyteArray) env->CallObjectMethod( iteratorObj, rocksdb::ListJni::getNextMethod(env)); jint key_length = env->GetArrayLength(jkey); jbyte* key = new jbyte[key_length]; env->GetByteArrayRegion(jkey, 0, key_length, key); // store allocated jbyte to free it after multiGet call keys_to_free.push_back(key); rocksdb::Slice key_slice( reinterpret_cast(key), key_length); keys.push_back(key_slice); } std::vector values; std::vector s; if (cf_handles.size() == 0) { s = db->MultiGet(rOpt, keys, &values); } else { s = db->MultiGet(rOpt, cf_handles, keys, &values); } // Don't reuse class pointer jclass jclazz = env->FindClass("java/util/ArrayList"); jmethodID mid = rocksdb::ListJni::getArrayListConstructorMethodId( env, jclazz); jobject jvalue_list = env->NewObject(jclazz, mid, jkeys_count); // insert in java list for (std::vector::size_type i = 0; i != s.size(); i++) { if (s[i].ok()) { jbyteArray jvalue = env->NewByteArray(values[i].size()); env->SetByteArrayRegion( jvalue, 0, values[i].size(), reinterpret_cast(values[i].c_str())); env->CallBooleanMethod( jvalue_list, rocksdb::ListJni::getListAddMethodId(env), jvalue); } else { env->CallBooleanMethod( jvalue_list, rocksdb::ListJni::getListAddMethodId(env), nullptr); } } // free up allocated byte arrays for (std::vector::size_type i = 0; i != keys_to_free.size(); i++) { delete[] keys_to_free[i]; } keys_to_free.clear(); return jvalue_list; } /* * Class: org_rocksdb_RocksDB * Method: multiGet * Signature: (JLjava/util/List;I)Ljava/util/List; */ jobject Java_org_rocksdb_RocksDB_multiGet__JLjava_util_List_2I( JNIEnv* env, jobject jdb, jlong jdb_handle, jobject jkey_list, jint jkeys_count) { return multi_get_helper(env, jdb, reinterpret_cast(jdb_handle), rocksdb::ReadOptions(), jkey_list, jkeys_count, nullptr); } /* * Class: org_rocksdb_RocksDB * Method: multiGet * Signature: (JLjava/util/List;ILjava/util/List;)Ljava/util/List; */ jobject Java_org_rocksdb_RocksDB_multiGet__JLjava_util_List_2ILjava_util_List_2( JNIEnv* env, jobject jdb, jlong jdb_handle, jobject jkey_list, jint jkeys_count, jobject jcfhandle_list) { return multi_get_helper(env, jdb, reinterpret_cast(jdb_handle), rocksdb::ReadOptions(), jkey_list, jkeys_count, jcfhandle_list); } /* * Class: org_rocksdb_RocksDB * Method: multiGet * Signature: (JJLjava/util/List;I)Ljava/util/List; */ jobject Java_org_rocksdb_RocksDB_multiGet__JJLjava_util_List_2I( JNIEnv* env, jobject jdb, jlong jdb_handle, jlong jropt_handle, jobject jkey_list, jint jkeys_count) { return multi_get_helper(env, jdb, reinterpret_cast(jdb_handle), *reinterpret_cast(jropt_handle), jkey_list, jkeys_count, nullptr); } /* * Class: org_rocksdb_RocksDB * Method: multiGet * Signature: (JJLjava/util/List;ILjava/util/List;)Ljava/util/List; */ jobject Java_org_rocksdb_RocksDB_multiGet__JJLjava_util_List_2ILjava_util_List_2( JNIEnv* env, jobject jdb, jlong jdb_handle, jlong jropt_handle, jobject jkey_list, jint jkeys_count, jobject jcfhandle_list) { return multi_get_helper(env, jdb, reinterpret_cast(jdb_handle), *reinterpret_cast(jropt_handle), jkey_list, jkeys_count, jcfhandle_list); } /* * Class: org_rocksdb_RocksDB * Method: get * Signature: (J[BI[BI)I */ jint Java_org_rocksdb_RocksDB_get__J_3BI_3BI( JNIEnv* env, jobject jdb, jlong jdb_handle, jbyteArray jkey, jint jkey_len, jbyteArray jvalue, jint jvalue_len) { return rocksdb_get_helper(env, reinterpret_cast(jdb_handle), rocksdb::ReadOptions(), nullptr, jkey, jkey_len, jvalue, jvalue_len); } /* * Class: org_rocksdb_RocksDB * Method: get * Signature: (J[BI[BIJ)I */ jint Java_org_rocksdb_RocksDB_get__J_3BI_3BIJ( JNIEnv* env, jobject jdb, jlong jdb_handle, jbyteArray jkey, jint jkey_len, jbyteArray jvalue, jint jvalue_len, jlong jcf_handle) { auto db_handle = reinterpret_cast(jdb_handle); auto cf_handle = reinterpret_cast(jcf_handle); if (cf_handle != nullptr) { return rocksdb_get_helper(env, db_handle, rocksdb::ReadOptions(), cf_handle, jkey, jkey_len, jvalue, jvalue_len); } else { rocksdb::RocksDBExceptionJni::ThrowNew(env, rocksdb::Status::InvalidArgument("Invalid ColumnFamilyHandle.")); // will never be evaluated return 0; } } /* * Class: org_rocksdb_RocksDB * Method: get * Signature: (JJ[BI[BI)I */ jint Java_org_rocksdb_RocksDB_get__JJ_3BI_3BI( JNIEnv* env, jobject jdb, jlong jdb_handle, jlong jropt_handle, jbyteArray jkey, jint jkey_len, jbyteArray jvalue, jint jvalue_len) { return rocksdb_get_helper(env, reinterpret_cast(jdb_handle), *reinterpret_cast(jropt_handle), nullptr, jkey, jkey_len, jvalue, jvalue_len); } /* * Class: org_rocksdb_RocksDB * Method: get * Signature: (JJ[BI[BIJ)I */ jint Java_org_rocksdb_RocksDB_get__JJ_3BI_3BIJ( JNIEnv* env, jobject jdb, jlong jdb_handle, jlong jropt_handle, jbyteArray jkey, jint jkey_len, jbyteArray jvalue, jint jvalue_len, jlong jcf_handle) { auto db_handle = reinterpret_cast(jdb_handle); auto& ro_opt = *reinterpret_cast(jropt_handle); auto cf_handle = reinterpret_cast(jcf_handle); if (cf_handle != nullptr) { return rocksdb_get_helper(env, db_handle, ro_opt, cf_handle, jkey, jkey_len, jvalue, jvalue_len); } else { rocksdb::RocksDBExceptionJni::ThrowNew(env, rocksdb::Status::InvalidArgument("Invalid ColumnFamilyHandle.")); // will never be evaluated return 0; } } ////////////////////////////////////////////////////////////////////////////// // rocksdb::DB::Delete() void rocksdb_remove_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, 0); rocksdb::Slice key_slice(reinterpret_cast(key), jkey_len); rocksdb::Status s; if (cf_handle != nullptr) { s = db->Delete(write_options, cf_handle, key_slice); } else { // backwards compatibility s = db->Delete(write_options, key_slice); } // 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); if (!s.ok()) { rocksdb::RocksDBExceptionJni::ThrowNew(env, s); } return; } /* * Class: org_rocksdb_RocksDB * Method: remove * Signature: (J[BI)V */ void Java_org_rocksdb_RocksDB_remove__J_3BI( JNIEnv* env, jobject jdb, jlong jdb_handle, jbyteArray jkey, jint jkey_len) { auto db = reinterpret_cast(jdb_handle); static const rocksdb::WriteOptions default_write_options = rocksdb::WriteOptions(); rocksdb_remove_helper(env, db, default_write_options, nullptr, jkey, jkey_len); } /* * Class: org_rocksdb_RocksDB * Method: remove * Signature: (J[BIJ)V */ void Java_org_rocksdb_RocksDB_remove__J_3BIJ( JNIEnv* env, jobject jdb, jlong jdb_handle, jbyteArray jkey, jint jkey_len, jlong jcf_handle) { auto db = reinterpret_cast(jdb_handle); static const rocksdb::WriteOptions default_write_options = rocksdb::WriteOptions(); auto cf_handle = reinterpret_cast(jcf_handle); if (cf_handle != nullptr) { rocksdb_remove_helper(env, db, default_write_options, cf_handle, jkey, jkey_len); } else { rocksdb::RocksDBExceptionJni::ThrowNew(env, rocksdb::Status::InvalidArgument("Invalid ColumnFamilyHandle.")); } } /* * Class: org_rocksdb_RocksDB * Method: remove * Signature: (JJ[BIJ)V */ void Java_org_rocksdb_RocksDB_remove__JJ_3BI( JNIEnv* env, jobject jdb, jlong jdb_handle, jlong jwrite_options, jbyteArray jkey, jint jkey_len) { auto db = reinterpret_cast(jdb_handle); auto write_options = reinterpret_cast(jwrite_options); rocksdb_remove_helper(env, db, *write_options, nullptr, jkey, jkey_len); } /* * Class: org_rocksdb_RocksDB * Method: remove * Signature: (JJ[BIJ)V */ void Java_org_rocksdb_RocksDB_remove__JJ_3BIJ( JNIEnv* env, jobject jdb, jlong jdb_handle, jlong jwrite_options, jbyteArray jkey, jint jkey_len, jlong jcf_handle) { auto db = reinterpret_cast(jdb_handle); auto write_options = reinterpret_cast(jwrite_options); auto cf_handle = reinterpret_cast(jcf_handle); if (cf_handle != nullptr) { rocksdb_remove_helper(env, db, *write_options, cf_handle, jkey, jkey_len); } else { rocksdb::RocksDBExceptionJni::ThrowNew(env, rocksdb::Status::InvalidArgument("Invalid ColumnFamilyHandle.")); } } ////////////////////////////////////////////////////////////////////////////// // rocksdb::DB::Merge void rocksdb_merge_helper( JNIEnv* env, rocksdb::DB* db, const rocksdb::WriteOptions& write_options, jbyteArray jkey, jint jkey_len, jbyteArray jvalue, jint jvalue_len) { jbyte* key = env->GetByteArrayElements(jkey, 0); jbyte* value = env->GetByteArrayElements(jvalue, 0); rocksdb::Slice key_slice(reinterpret_cast(key), jkey_len); rocksdb::Slice value_slice(reinterpret_cast(value), jvalue_len); rocksdb::Status s = db->Merge(write_options, key_slice, value_slice); // 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); env->ReleaseByteArrayElements(jvalue, value, JNI_ABORT); if (s.ok()) { return; } rocksdb::RocksDBExceptionJni::ThrowNew(env, s); } /* * Class: org_rocksdb_RocksDB * Method: merge * Signature: (J[BI[BI)V */ void Java_org_rocksdb_RocksDB_merge__J_3BI_3BI( JNIEnv* env, jobject jdb, jlong jdb_handle, jbyteArray jkey, jint jkey_len, jbyteArray jvalue, jint jvalue_len) { auto db = reinterpret_cast(jdb_handle); static const rocksdb::WriteOptions default_write_options = rocksdb::WriteOptions(); rocksdb_merge_helper(env, db, default_write_options, jkey, jkey_len, jvalue, jvalue_len); } /* * Class: org_rocksdb_RocksDB * Method: merge * Signature: (JJ[BI[BI)V */ void Java_org_rocksdb_RocksDB_merge__JJ_3BI_3BI( JNIEnv* env, jobject jdb, jlong jdb_handle, jlong jwrite_options_handle, jbyteArray jkey, jint jkey_len, jbyteArray jvalue, jint jvalue_len) { auto db = reinterpret_cast(jdb_handle); auto write_options = reinterpret_cast( jwrite_options_handle); rocksdb_merge_helper(env, db, *write_options, jkey, jkey_len, jvalue, jvalue_len); } ////////////////////////////////////////////////////////////////////////////// // rocksdb::DB::~DB() /* * Class: org_rocksdb_RocksDB * Method: disposeInternal * Signature: (J)V */ void Java_org_rocksdb_RocksDB_disposeInternal( JNIEnv* env, jobject java_db, jlong jhandle) { delete reinterpret_cast(jhandle); } /* * Class: org_rocksdb_RocksDB * Method: iterator0 * Signature: (J)J */ jlong Java_org_rocksdb_RocksDB_iterator0__J( JNIEnv* env, jobject jdb, jlong db_handle) { auto db = reinterpret_cast(db_handle); rocksdb::Iterator* iterator = db->NewIterator(rocksdb::ReadOptions()); return reinterpret_cast(iterator); } /* * Class: org_rocksdb_RocksDB * Method: iterator0 * Signature: (JJ)J */ jlong Java_org_rocksdb_RocksDB_iterator0__JJ( JNIEnv* env, jobject jdb, jlong db_handle, jlong jcf_handle) { auto db = reinterpret_cast(db_handle); auto cf_handle = reinterpret_cast(jcf_handle); rocksdb::Iterator* iterator = db->NewIterator(rocksdb::ReadOptions(), cf_handle); return reinterpret_cast(iterator); } /* * Class: org_rocksdb_RocksDB * Method: iterators * Signature: (JLjava/util/List;)[J */ jlongArray Java_org_rocksdb_RocksDB_iterators( JNIEnv* env, jobject jdb, jlong db_handle, jobject jcfhandle_list) { auto db = reinterpret_cast(db_handle); std::vector cf_handles; std::vector iterators; if (jcfhandle_list != nullptr) { // get cf iterator jobject cfIteratorObj = env->CallObjectMethod( jcfhandle_list, rocksdb::ListJni::getIteratorMethod(env)); // iterate over keys and convert java byte array to slice while (env->CallBooleanMethod( cfIteratorObj, rocksdb::ListJni::getHasNextMethod(env)) == JNI_TRUE) { jobject jobj = (jbyteArray) env->CallObjectMethod( cfIteratorObj, rocksdb::ListJni::getNextMethod(env)); rocksdb::ColumnFamilyHandle* cfHandle = rocksdb::ColumnFamilyHandleJni::getHandle(env, jobj); cf_handles.push_back(cfHandle); } } rocksdb::Status s = db->NewIterators(rocksdb::ReadOptions(), cf_handles, &iterators); if (s.ok()) { jlongArray jLongArray = env->NewLongArray(iterators.size()); for (std::vector::size_type i = 0; i < iterators.size(); i++) { env->SetLongArrayRegion(jLongArray, i, 1, reinterpret_cast(&iterators[i])); } return jLongArray; } rocksdb::RocksDBExceptionJni::ThrowNew(env, s); return env->NewLongArray(0); } /* * Class: org_rocksdb_RocksDB * Method: createColumnFamily * Signature: (JLjava/lang/String;)J; */ jlong Java_org_rocksdb_RocksDB_createColumnFamily( JNIEnv* env, jobject jdb, jlong jdb_handle, jstring jcfname) { rocksdb::ColumnFamilyHandle* handle; const char* cfname = env->GetStringUTFChars(jcfname, 0); auto db_handle = reinterpret_cast(jdb_handle); rocksdb::Status s = db_handle->CreateColumnFamily( rocksdb::ColumnFamilyOptions(), cfname, &handle); env->ReleaseStringUTFChars(jcfname, cfname); if (s.ok()) { return reinterpret_cast(handle); } rocksdb::RocksDBExceptionJni::ThrowNew(env, s); return 0; } /* * Class: org_rocksdb_RocksDB * Method: dropColumnFamily * Signature: (JJ)V; */ void Java_org_rocksdb_RocksDB_dropColumnFamily( JNIEnv* env, jobject jdb, jlong jdb_handle, jlong jcf_handle) { auto cf_handle = reinterpret_cast(jcf_handle); auto db_handle = reinterpret_cast(jdb_handle); rocksdb::Status s = db_handle->DropColumnFamily(cf_handle); if (!s.ok()) { rocksdb::RocksDBExceptionJni::ThrowNew(env, s); } } /* * Method: getSnapshot * Signature: (J)J */ jlong Java_org_rocksdb_RocksDB_getSnapshot( JNIEnv* env, jobject jdb, jlong db_handle) { auto db = reinterpret_cast(db_handle); const rocksdb::Snapshot* snapshot = db->GetSnapshot(); return reinterpret_cast(snapshot); } /* * Method: releaseSnapshot * Signature: (JJ)V */ void Java_org_rocksdb_RocksDB_releaseSnapshot( JNIEnv* env, jobject jdb, jlong db_handle, jlong snapshot_handle) { auto db = reinterpret_cast(db_handle); auto snapshot = reinterpret_cast(snapshot_handle); db->ReleaseSnapshot(snapshot); } /* * Class: org_rocksdb_RocksDB * Method: getProperty0 * Signature: (JLjava/lang/String;I)Ljava/lang/String; */ jstring Java_org_rocksdb_RocksDB_getProperty0__JLjava_lang_String_2I( JNIEnv* env, jobject jdb, jlong db_handle, jstring jproperty, jint jproperty_len) { auto db = reinterpret_cast(db_handle); const char* property = env->GetStringUTFChars(jproperty, 0); rocksdb::Slice property_slice(property, jproperty_len); std::string property_value; bool retCode = db->GetProperty(property_slice, &property_value); env->ReleaseStringUTFChars(jproperty, property); if (!retCode) { rocksdb::RocksDBExceptionJni::ThrowNew(env, rocksdb::Status::NotFound()); } return env->NewStringUTF(property_value.data()); } /* * Class: org_rocksdb_RocksDB * Method: getProperty0 * Signature: (JJLjava/lang/String;I)Ljava/lang/String; */ jstring Java_org_rocksdb_RocksDB_getProperty0__JJLjava_lang_String_2I( JNIEnv* env, jobject jdb, jlong db_handle, jlong jcf_handle, jstring jproperty, jint jproperty_len) { auto db = reinterpret_cast(db_handle); auto cf_handle = reinterpret_cast(jcf_handle); const char* property = env->GetStringUTFChars(jproperty, 0); rocksdb::Slice property_slice(property, jproperty_len); std::string property_value; bool retCode = db->GetProperty(cf_handle, property_slice, &property_value); env->ReleaseStringUTFChars(jproperty, property); if (!retCode) { rocksdb::RocksDBExceptionJni::ThrowNew(env, rocksdb::Status::NotFound()); } return env->NewStringUTF(property_value.data()); }