ab3e566699
Summary: 1. Move disOwnNativeHandle() function from RocksDB to RocksObject to allow other RocksObject to use disOwnNativeHandle() when its ownership of native handle has been transferred. 2. RocksObject now has an abstract implementation of dispose(), which does the following two things. First, it checks whether both isOwningNativeHandle() and isInitialized() return true. If so, it will call the protected abstract function dispose0(), which all the subclasses of RocksObject should implement. Second, it sets nativeHandle_ = 0. This redesign ensure all subclasses of RocksObject have the same dispose behavior. 3. All subclasses of RocksObject now should implement dispose0() instead of dispose(), and dispose0() will be called only when isInitialized() returns true. Test Plan: make rocksdbjava make jtest Reviewers: dhruba, sdong, ankgup87, rsumbaly, swapnilghike, zzbennett, haobo Reviewed By: haobo Subscribers: leveldb Differential Revision: https://reviews.facebook.net/D18801
146 lines
3.8 KiB
C++
146 lines
3.8 KiB
C++
// Copyright (c) 2014, Facebook, Inc. All rights reserved.
|
|
// This source code is licensed under the BSD-style license found in the
|
|
// LICENSE file in the root directory of this source tree. An additional grant
|
|
// of patent rights can be found in the PATENTS file in the same directory.
|
|
//
|
|
// This file implements the "bridge" between Java and C++ and enables
|
|
// calling c++ rocksdb::Iterator methods from Java side.
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <jni.h>
|
|
|
|
#include "include/org_rocksdb_RocksIterator.h"
|
|
#include "rocksjni/portal.h"
|
|
#include "rocksdb/iterator.h"
|
|
|
|
/*
|
|
* Class: org_rocksdb_RocksIterator
|
|
* Method: isValid0
|
|
* Signature: (J)Z
|
|
*/
|
|
jboolean Java_org_rocksdb_RocksIterator_isValid0(
|
|
JNIEnv* env, jobject jobj, jlong handle) {
|
|
return reinterpret_cast<rocksdb::Iterator*>(handle)->Valid();
|
|
}
|
|
|
|
/*
|
|
* Class: org_rocksdb_RocksIterator
|
|
* Method: seekToFirst0
|
|
* Signature: (J)V
|
|
*/
|
|
void Java_org_rocksdb_RocksIterator_seekToFirst0(
|
|
JNIEnv* env, jobject jobj, jlong handle) {
|
|
reinterpret_cast<rocksdb::Iterator*>(handle)->SeekToFirst();
|
|
}
|
|
|
|
/*
|
|
* Class: org_rocksdb_RocksIterator
|
|
* Method: seekToFirst0
|
|
* Signature: (J)V
|
|
*/
|
|
void Java_org_rocksdb_RocksIterator_seekToLast0(
|
|
JNIEnv* env, jobject jobj, jlong handle) {
|
|
reinterpret_cast<rocksdb::Iterator*>(handle)->SeekToLast();
|
|
}
|
|
|
|
/*
|
|
* Class: org_rocksdb_RocksIterator
|
|
* Method: seekToLast0
|
|
* Signature: (J)V
|
|
*/
|
|
void Java_org_rocksdb_RocksIterator_next0(
|
|
JNIEnv* env, jobject jobj, jlong handle) {
|
|
reinterpret_cast<rocksdb::Iterator*>(handle)->Next();
|
|
}
|
|
|
|
/*
|
|
* Class: org_rocksdb_RocksIterator
|
|
* Method: next0
|
|
* Signature: (J)V
|
|
*/
|
|
void Java_org_rocksdb_RocksIterator_prev0(
|
|
JNIEnv* env, jobject jobj, jlong handle) {
|
|
reinterpret_cast<rocksdb::Iterator*>(handle)->Prev();
|
|
}
|
|
|
|
/*
|
|
* Class: org_rocksdb_RocksIterator
|
|
* Method: prev0
|
|
* Signature: (J)V
|
|
*/
|
|
jbyteArray Java_org_rocksdb_RocksIterator_key0(
|
|
JNIEnv* env, jobject jobj, jlong handle) {
|
|
auto it = reinterpret_cast<rocksdb::Iterator*>(handle);
|
|
rocksdb::Slice key_slice = it->key();
|
|
|
|
jbyteArray jkey = env->NewByteArray(key_slice.size());
|
|
env->SetByteArrayRegion(
|
|
jkey, 0, key_slice.size(),
|
|
reinterpret_cast<const jbyte*>(key_slice.data()));
|
|
return jkey;
|
|
}
|
|
|
|
/*
|
|
* Class: org_rocksdb_RocksIterator
|
|
* Method: key0
|
|
* Signature: (J)[B
|
|
*/
|
|
jbyteArray Java_org_rocksdb_RocksIterator_value0(
|
|
JNIEnv* env, jobject jobj, jlong handle) {
|
|
auto it = reinterpret_cast<rocksdb::Iterator*>(handle);
|
|
rocksdb::Slice value_slice = it->value();
|
|
|
|
jbyteArray jvalue = env->NewByteArray(value_slice.size());
|
|
env->SetByteArrayRegion(
|
|
jvalue, 0, value_slice.size(),
|
|
reinterpret_cast<const jbyte*>(value_slice.data()));
|
|
return jvalue;
|
|
}
|
|
|
|
/*
|
|
* Class: org_rocksdb_RocksIterator
|
|
* Method: value0
|
|
* Signature: (J)[B
|
|
*/
|
|
void Java_org_rocksdb_RocksIterator_seek0(
|
|
JNIEnv* env, jobject jobj, jlong handle,
|
|
jbyteArray jtarget, jint jtarget_len) {
|
|
auto it = reinterpret_cast<rocksdb::Iterator*>(handle);
|
|
jbyte* target = env->GetByteArrayElements(jtarget, 0);
|
|
rocksdb::Slice target_slice(
|
|
reinterpret_cast<char*>(target), jtarget_len);
|
|
|
|
it->Seek(target_slice);
|
|
|
|
env->ReleaseByteArrayElements(jtarget, target, JNI_ABORT);
|
|
}
|
|
|
|
/*
|
|
* Class: org_rocksdb_RocksIterator
|
|
* Method: seek0
|
|
* Signature: (J[BI)V
|
|
*/
|
|
void Java_org_rocksdb_RocksIterator_status0(
|
|
JNIEnv* env, jobject jobj, jlong handle) {
|
|
auto it = reinterpret_cast<rocksdb::Iterator*>(handle);
|
|
rocksdb::Status s = it->status();
|
|
|
|
if (s.ok()) {
|
|
return;
|
|
}
|
|
|
|
rocksdb::RocksDBExceptionJni::ThrowNew(env, s);
|
|
}
|
|
|
|
/*
|
|
* Class: org_rocksdb_RocksIterator
|
|
* Method: disposeInternal
|
|
* Signature: (J)V
|
|
*/
|
|
void Java_org_rocksdb_RocksIterator_disposeInternal(
|
|
JNIEnv* env, jobject jobj, jlong handle) {
|
|
auto it = reinterpret_cast<rocksdb::Iterator*>(handle);
|
|
delete it;
|
|
}
|