// Copyright (c) 2011-present, 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++ for // rocksdb::Slice. #include <stdio.h> #include <stdlib.h> #include <jni.h> #include <string> #include "include/org_rocksdb_AbstractSlice.h" #include "include/org_rocksdb_Slice.h" #include "include/org_rocksdb_DirectSlice.h" #include "rocksdb/slice.h" #include "rocksjni/portal.h" // <editor-fold desc="org.rocksdb.AbstractSlice> /* * Class: org_rocksdb_AbstractSlice * Method: createNewSliceFromString * Signature: (Ljava/lang/String;)J */ jlong Java_org_rocksdb_AbstractSlice_createNewSliceFromString( JNIEnv * env, jclass jcls, jstring jstr) { const auto* str = env->GetStringUTFChars(jstr, NULL); const size_t len = strlen(str); char* buf = new char[len + 1]; memcpy(buf, str, len); buf[len] = 0; env->ReleaseStringUTFChars(jstr, str); const auto* slice = new rocksdb::Slice(buf); return reinterpret_cast<jlong>(slice); } /* * Class: org_rocksdb_AbstractSlice * Method: size0 * Signature: (J)I */ jint Java_org_rocksdb_AbstractSlice_size0( JNIEnv* env, jobject jobj, jlong handle) { const auto* slice = reinterpret_cast<rocksdb::Slice*>(handle); return static_cast<jint>(slice->size()); } /* * Class: org_rocksdb_AbstractSlice * Method: empty0 * Signature: (J)Z */ jboolean Java_org_rocksdb_AbstractSlice_empty0( JNIEnv* env, jobject jobj, jlong handle) { const auto* slice = reinterpret_cast<rocksdb::Slice*>(handle); return slice->empty(); } /* * Class: org_rocksdb_AbstractSlice * Method: toString0 * Signature: (JZ)Ljava/lang/String; */ jstring Java_org_rocksdb_AbstractSlice_toString0( JNIEnv* env, jobject jobj, jlong handle, jboolean hex) { const auto* slice = reinterpret_cast<rocksdb::Slice*>(handle); const std::string s = slice->ToString(hex); return env->NewStringUTF(s.c_str()); } /* * Class: org_rocksdb_AbstractSlice * Method: compare0 * Signature: (JJ)I; */ jint Java_org_rocksdb_AbstractSlice_compare0( JNIEnv* env, jobject jobj, jlong handle, jlong otherHandle) { const auto* slice = reinterpret_cast<rocksdb::Slice*>(handle); const auto* otherSlice = reinterpret_cast<rocksdb::Slice*>(otherHandle); return slice->compare(*otherSlice); } /* * Class: org_rocksdb_AbstractSlice * Method: startsWith0 * Signature: (JJ)Z; */ jboolean Java_org_rocksdb_AbstractSlice_startsWith0( JNIEnv* env, jobject jobj, jlong handle, jlong otherHandle) { const auto* slice = reinterpret_cast<rocksdb::Slice*>(handle); const auto* otherSlice = reinterpret_cast<rocksdb::Slice*>(otherHandle); return slice->starts_with(*otherSlice); } /* * Class: org_rocksdb_AbstractSlice * Method: disposeInternal * Signature: (J)V */ void Java_org_rocksdb_AbstractSlice_disposeInternal( JNIEnv* env, jobject jobj, jlong handle) { delete reinterpret_cast<rocksdb::Slice*>(handle); } // </editor-fold> // <editor-fold desc="org.rocksdb.Slice> /* * Class: org_rocksdb_Slice * Method: createNewSlice0 * Signature: ([BI)J */ jlong Java_org_rocksdb_Slice_createNewSlice0( JNIEnv * env, jclass jcls, jbyteArray data, jint offset) { const jsize dataSize = env->GetArrayLength(data); const int len = dataSize - offset; jbyte* ptrData = new jbyte[len]; env->GetByteArrayRegion(data, offset, len, ptrData); const auto* slice = new rocksdb::Slice((const char*)ptrData, len); return reinterpret_cast<jlong>(slice); } /* * Class: org_rocksdb_Slice * Method: createNewSlice1 * Signature: ([B)J */ jlong Java_org_rocksdb_Slice_createNewSlice1( JNIEnv * env, jclass jcls, jbyteArray data) { const int len = env->GetArrayLength(data) + 1; jboolean isCopy; jbyte* ptrData = env->GetByteArrayElements(data, &isCopy); // NOTE: buf will be deleted in the org.rocksdb.Slice#dispose method char* buf = new char[len]; memcpy(buf, ptrData, len - 1); buf[len-1]='\0'; const auto* slice = new rocksdb::Slice(buf, len - 1); env->ReleaseByteArrayElements(data, ptrData, JNI_ABORT); return reinterpret_cast<jlong>(slice); } /* * Class: org_rocksdb_Slice * Method: data0 * Signature: (J)[B */ jbyteArray Java_org_rocksdb_Slice_data0( JNIEnv* env, jobject jobj, jlong handle) { const auto* slice = reinterpret_cast<rocksdb::Slice*>(handle); const int len = static_cast<int>(slice->size()); const jbyteArray data = env->NewByteArray(len); env->SetByteArrayRegion(data, 0, len, reinterpret_cast<const jbyte*>(slice->data())); return data; } /* * Class: org_rocksdb_Slice * Method: disposeInternalBuf * Signature: (J)V */ void Java_org_rocksdb_Slice_disposeInternalBuf( JNIEnv * env, jobject jobj, jlong handle) { const auto* slice = reinterpret_cast<rocksdb::Slice*>(handle); delete [] slice->data_; } // </editor-fold> // <editor-fold desc="org.rocksdb.DirectSlice> /* * Class: org_rocksdb_DirectSlice * Method: createNewDirectSlice0 * Signature: (Ljava/nio/ByteBuffer;I)J */ jlong Java_org_rocksdb_DirectSlice_createNewDirectSlice0( JNIEnv* env, jclass jcls, jobject data, jint length) { const auto* ptrData = reinterpret_cast<char*>(env->GetDirectBufferAddress(data)); const auto* slice = new rocksdb::Slice(ptrData, length); return reinterpret_cast<jlong>(slice); } /* * Class: org_rocksdb_DirectSlice * Method: createNewDirectSlice1 * Signature: (Ljava/nio/ByteBuffer;)J */ jlong Java_org_rocksdb_DirectSlice_createNewDirectSlice1( JNIEnv* env, jclass jcls, jobject data) { const auto* ptrData = reinterpret_cast<char*>(env->GetDirectBufferAddress(data)); const auto* slice = new rocksdb::Slice(ptrData); return reinterpret_cast<jlong>(slice); } /* * Class: org_rocksdb_DirectSlice * Method: data0 * Signature: (J)Ljava/lang/Object; */ jobject Java_org_rocksdb_DirectSlice_data0( JNIEnv* env, jobject jobj, jlong handle) { const auto* slice = reinterpret_cast<rocksdb::Slice*>(handle); return env->NewDirectByteBuffer(const_cast<char*>(slice->data()), slice->size()); } /* * Class: org_rocksdb_DirectSlice * Method: get0 * Signature: (JI)B */ jbyte Java_org_rocksdb_DirectSlice_get0( JNIEnv* env, jobject jobj, jlong handle, jint offset) { const auto* slice = reinterpret_cast<rocksdb::Slice*>(handle); return (*slice)[offset]; } /* * Class: org_rocksdb_DirectSlice * Method: clear0 * Signature: (J)V */ void Java_org_rocksdb_DirectSlice_clear0( JNIEnv* env, jobject jobj, jlong handle) { auto* slice = reinterpret_cast<rocksdb::Slice*>(handle); delete [] slice->data_; slice->clear(); } /* * Class: org_rocksdb_DirectSlice * Method: removePrefix0 * Signature: (JI)V */ void Java_org_rocksdb_DirectSlice_removePrefix0( JNIEnv* env, jobject jobj, jlong handle, jint length) { auto* slice = reinterpret_cast<rocksdb::Slice*>(handle); slice->remove_prefix(length); } // </editor-fold>