Add Java API for SstFileWriter. Closes https://github.com/facebook/rocksdb/issues/1248main
Injun Song
8 years ago
committed by
Adam Retter
parent
26388247aa
commit
7260662b39
@ -0,0 +1,315 @@ |
|||||||
|
// 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++ and enables
|
||||||
|
// calling c++ rocksdb::BackupableDB and rocksdb::BackupableDBOptions methods
|
||||||
|
// from Java side.
|
||||||
|
|
||||||
|
#include <jni.h> |
||||||
|
|
||||||
|
#include "include/org_rocksdb_EnvOptions.h" |
||||||
|
#include "rocksdb/env.h" |
||||||
|
|
||||||
|
#define ENV_OPTIONS_SET_BOOL(_jhandle, _opt) \ |
||||||
|
reinterpret_cast<rocksdb::EnvOptions *>(_jhandle)->_opt = \
|
||||||
|
static_cast<bool>(_opt) |
||||||
|
|
||||||
|
#define ENV_OPTIONS_SET_SIZE_T(_jhandle, _opt) \ |
||||||
|
reinterpret_cast<rocksdb::EnvOptions *>(_jhandle)->_opt = \
|
||||||
|
static_cast<size_t>(_opt) |
||||||
|
|
||||||
|
#define ENV_OPTIONS_SET_UINT64_T(_jhandle, _opt) \ |
||||||
|
reinterpret_cast<rocksdb::EnvOptions *>(_jhandle)->_opt = \
|
||||||
|
static_cast<uint64_t>(_opt) |
||||||
|
|
||||||
|
#define ENV_OPTIONS_GET(_jhandle, _opt) \ |
||||||
|
reinterpret_cast<rocksdb::EnvOptions *>(_jhandle)->_opt |
||||||
|
|
||||||
|
/*
|
||||||
|
* Class: org_rocksdb_EnvOptions |
||||||
|
* Method: newEnvOptions |
||||||
|
* Signature: ()J |
||||||
|
*/ |
||||||
|
jlong Java_org_rocksdb_EnvOptions_newEnvOptions(JNIEnv *env, jclass jcls) { |
||||||
|
auto *env_opt = new rocksdb::EnvOptions(); |
||||||
|
return reinterpret_cast<jlong>(env_opt); |
||||||
|
} |
||||||
|
|
||||||
|
/*
|
||||||
|
* Class: org_rocksdb_EnvOptions |
||||||
|
* Method: disposeInternal |
||||||
|
* Signature: (J)V |
||||||
|
*/ |
||||||
|
void Java_org_rocksdb_EnvOptions_disposeInternal(JNIEnv *env, jobject jobj, |
||||||
|
jlong jhandle) { |
||||||
|
delete reinterpret_cast<rocksdb::EnvOptions *>(jhandle); |
||||||
|
} |
||||||
|
|
||||||
|
/*
|
||||||
|
* Class: org_rocksdb_EnvOptions |
||||||
|
* Method: setUseOsBuffer |
||||||
|
* Signature: (JZ)V |
||||||
|
*/ |
||||||
|
void Java_org_rocksdb_EnvOptions_setUseOsBuffer(JNIEnv *env, jobject jobj, |
||||||
|
jlong jhandle, |
||||||
|
jboolean use_os_buffer) { |
||||||
|
ENV_OPTIONS_SET_BOOL(jhandle, use_os_buffer); |
||||||
|
} |
||||||
|
|
||||||
|
/*
|
||||||
|
* Class: org_rocksdb_EnvOptions |
||||||
|
* Method: useOsBuffer |
||||||
|
* Signature: (J)Z |
||||||
|
*/ |
||||||
|
jboolean Java_org_rocksdb_EnvOptions_useOsBuffer(JNIEnv *env, jobject jobj, |
||||||
|
jlong jhandle) { |
||||||
|
return ENV_OPTIONS_GET(jhandle, use_os_buffer); |
||||||
|
} |
||||||
|
|
||||||
|
/*
|
||||||
|
* Class: org_rocksdb_EnvOptions |
||||||
|
* Method: setUseMmapReads |
||||||
|
* Signature: (JZ)V |
||||||
|
*/ |
||||||
|
void Java_org_rocksdb_EnvOptions_setUseMmapReads(JNIEnv *env, jobject jobj, |
||||||
|
jlong jhandle, |
||||||
|
jboolean use_mmap_reads) { |
||||||
|
ENV_OPTIONS_SET_BOOL(jhandle, use_mmap_reads); |
||||||
|
} |
||||||
|
|
||||||
|
/*
|
||||||
|
* Class: org_rocksdb_EnvOptions |
||||||
|
* Method: useMmapReads |
||||||
|
* Signature: (J)Z |
||||||
|
*/ |
||||||
|
jboolean Java_org_rocksdb_EnvOptions_useMmapReads(JNIEnv *env, jobject jobj, |
||||||
|
jlong jhandle) { |
||||||
|
return ENV_OPTIONS_GET(jhandle, use_mmap_reads); |
||||||
|
} |
||||||
|
|
||||||
|
/*
|
||||||
|
* Class: org_rocksdb_EnvOptions |
||||||
|
* Method: setUseMmapWrites |
||||||
|
* Signature: (JZ)V |
||||||
|
*/ |
||||||
|
void Java_org_rocksdb_EnvOptions_setUseMmapWrites(JNIEnv *env, jobject jobj, |
||||||
|
jlong jhandle, |
||||||
|
jboolean use_mmap_writes) { |
||||||
|
ENV_OPTIONS_SET_BOOL(jhandle, use_mmap_writes); |
||||||
|
} |
||||||
|
|
||||||
|
/*
|
||||||
|
* Class: org_rocksdb_EnvOptions |
||||||
|
* Method: useMmapWrites |
||||||
|
* Signature: (J)Z |
||||||
|
*/ |
||||||
|
jboolean Java_org_rocksdb_EnvOptions_useMmapWrites(JNIEnv *env, jobject jobj, |
||||||
|
jlong jhandle) { |
||||||
|
return ENV_OPTIONS_GET(jhandle, use_mmap_writes); |
||||||
|
} |
||||||
|
|
||||||
|
/*
|
||||||
|
* Class: org_rocksdb_EnvOptions |
||||||
|
* Method: setUseDirectReads |
||||||
|
* Signature: (JZ)V |
||||||
|
*/ |
||||||
|
void Java_org_rocksdb_EnvOptions_setUseDirectReads(JNIEnv *env, jobject jobj, |
||||||
|
jlong jhandle, |
||||||
|
jboolean use_direct_reads) { |
||||||
|
ENV_OPTIONS_SET_BOOL(jhandle, use_direct_reads); |
||||||
|
} |
||||||
|
|
||||||
|
/*
|
||||||
|
* Class: org_rocksdb_EnvOptions |
||||||
|
* Method: useDirectReads |
||||||
|
* Signature: (J)Z |
||||||
|
*/ |
||||||
|
jboolean Java_org_rocksdb_EnvOptions_useDirectReads(JNIEnv *env, jobject jobj, |
||||||
|
jlong jhandle) { |
||||||
|
return ENV_OPTIONS_GET(jhandle, use_direct_reads); |
||||||
|
} |
||||||
|
|
||||||
|
/*
|
||||||
|
* Class: org_rocksdb_EnvOptions |
||||||
|
* Method: setUseDirectWrites |
||||||
|
* Signature: (JZ)V |
||||||
|
*/ |
||||||
|
void Java_org_rocksdb_EnvOptions_setUseDirectWrites( |
||||||
|
JNIEnv *env, jobject jobj, jlong jhandle, jboolean use_direct_writes) { |
||||||
|
ENV_OPTIONS_SET_BOOL(jhandle, use_direct_writes); |
||||||
|
} |
||||||
|
|
||||||
|
/*
|
||||||
|
* Class: org_rocksdb_EnvOptions |
||||||
|
* Method: useDirectWrites |
||||||
|
* Signature: (J)Z |
||||||
|
*/ |
||||||
|
jboolean Java_org_rocksdb_EnvOptions_useDirectWrites(JNIEnv *env, jobject jobj, |
||||||
|
jlong jhandle) { |
||||||
|
return ENV_OPTIONS_GET(jhandle, use_direct_writes); |
||||||
|
} |
||||||
|
|
||||||
|
/*
|
||||||
|
* Class: org_rocksdb_EnvOptions |
||||||
|
* Method: setAllowFallocate |
||||||
|
* Signature: (JZ)V |
||||||
|
*/ |
||||||
|
void Java_org_rocksdb_EnvOptions_setAllowFallocate(JNIEnv *env, jobject jobj, |
||||||
|
jlong jhandle, |
||||||
|
jboolean allow_fallocate) { |
||||||
|
ENV_OPTIONS_SET_BOOL(jhandle, allow_fallocate); |
||||||
|
} |
||||||
|
|
||||||
|
/*
|
||||||
|
* Class: org_rocksdb_EnvOptions |
||||||
|
* Method: allowFallocate |
||||||
|
* Signature: (J)Z |
||||||
|
*/ |
||||||
|
jboolean Java_org_rocksdb_EnvOptions_allowFallocate(JNIEnv *env, jobject jobj, |
||||||
|
jlong jhandle) { |
||||||
|
return ENV_OPTIONS_GET(jhandle, allow_fallocate); |
||||||
|
} |
||||||
|
|
||||||
|
/*
|
||||||
|
* Class: org_rocksdb_EnvOptions |
||||||
|
* Method: setSetFdCloexec |
||||||
|
* Signature: (JZ)V |
||||||
|
*/ |
||||||
|
void Java_org_rocksdb_EnvOptions_setSetFdCloexec(JNIEnv *env, jobject jobj, |
||||||
|
jlong jhandle, |
||||||
|
jboolean set_fd_cloexec) { |
||||||
|
ENV_OPTIONS_SET_BOOL(jhandle, set_fd_cloexec); |
||||||
|
} |
||||||
|
|
||||||
|
/*
|
||||||
|
* Class: org_rocksdb_EnvOptions |
||||||
|
* Method: setFdCloexec |
||||||
|
* Signature: (J)Z |
||||||
|
*/ |
||||||
|
jboolean Java_org_rocksdb_EnvOptions_setFdCloexec(JNIEnv *env, jobject jobj, |
||||||
|
jlong jhandle) { |
||||||
|
return ENV_OPTIONS_GET(jhandle, set_fd_cloexec); |
||||||
|
} |
||||||
|
|
||||||
|
/*
|
||||||
|
* Class: org_rocksdb_EnvOptions |
||||||
|
* Method: setBytesPerSync |
||||||
|
* Signature: (JJ)V |
||||||
|
*/ |
||||||
|
void Java_org_rocksdb_EnvOptions_setBytesPerSync(JNIEnv *env, jobject jobj, |
||||||
|
jlong jhandle, |
||||||
|
jlong bytes_per_sync) { |
||||||
|
ENV_OPTIONS_SET_UINT64_T(jhandle, bytes_per_sync); |
||||||
|
} |
||||||
|
|
||||||
|
/*
|
||||||
|
* Class: org_rocksdb_EnvOptions |
||||||
|
* Method: bytesPerSync |
||||||
|
* Signature: (J)J |
||||||
|
*/ |
||||||
|
jlong Java_org_rocksdb_EnvOptions_bytesPerSync(JNIEnv *env, jobject jobj, |
||||||
|
jlong jhandle) { |
||||||
|
return ENV_OPTIONS_GET(jhandle, bytes_per_sync); |
||||||
|
} |
||||||
|
|
||||||
|
/*
|
||||||
|
* Class: org_rocksdb_EnvOptions |
||||||
|
* Method: setFallocateWithKeepSize |
||||||
|
* Signature: (JZ)V |
||||||
|
*/ |
||||||
|
void Java_org_rocksdb_EnvOptions_setFallocateWithKeepSize( |
||||||
|
JNIEnv *env, jobject jobj, jlong jhandle, |
||||||
|
jboolean fallocate_with_keep_size) { |
||||||
|
ENV_OPTIONS_SET_BOOL(jhandle, fallocate_with_keep_size); |
||||||
|
} |
||||||
|
|
||||||
|
/*
|
||||||
|
* Class: org_rocksdb_EnvOptions |
||||||
|
* Method: fallocateWithKeepSize |
||||||
|
* Signature: (J)Z |
||||||
|
*/ |
||||||
|
jboolean Java_org_rocksdb_EnvOptions_fallocateWithKeepSize(JNIEnv *env, |
||||||
|
jobject jobj, |
||||||
|
jlong jhandle) { |
||||||
|
return ENV_OPTIONS_GET(jhandle, fallocate_with_keep_size); |
||||||
|
} |
||||||
|
|
||||||
|
/*
|
||||||
|
* Class: org_rocksdb_EnvOptions |
||||||
|
* Method: setCompactionReadaheadSize |
||||||
|
* Signature: (JJ)V |
||||||
|
*/ |
||||||
|
void Java_org_rocksdb_EnvOptions_setCompactionReadaheadSize( |
||||||
|
JNIEnv *env, jobject jobj, jlong jhandle, jlong compaction_readahead_size) { |
||||||
|
ENV_OPTIONS_SET_SIZE_T(jhandle, compaction_readahead_size); |
||||||
|
} |
||||||
|
|
||||||
|
/*
|
||||||
|
* Class: org_rocksdb_EnvOptions |
||||||
|
* Method: compactionReadaheadSize |
||||||
|
* Signature: (J)J |
||||||
|
*/ |
||||||
|
jlong Java_org_rocksdb_EnvOptions_compactionReadaheadSize(JNIEnv *env, |
||||||
|
jobject jobj, |
||||||
|
jlong jhandle) { |
||||||
|
return ENV_OPTIONS_GET(jhandle, compaction_readahead_size); |
||||||
|
} |
||||||
|
|
||||||
|
/*
|
||||||
|
* Class: org_rocksdb_EnvOptions |
||||||
|
* Method: setRandomAccessMaxBufferSize |
||||||
|
* Signature: (JJ)V |
||||||
|
*/ |
||||||
|
void Java_org_rocksdb_EnvOptions_setRandomAccessMaxBufferSize( |
||||||
|
JNIEnv *env, jobject jobj, jlong jhandle, |
||||||
|
jlong random_access_max_buffer_size) { |
||||||
|
ENV_OPTIONS_SET_SIZE_T(jhandle, random_access_max_buffer_size); |
||||||
|
} |
||||||
|
|
||||||
|
/*
|
||||||
|
* Class: org_rocksdb_EnvOptions |
||||||
|
* Method: randomAccessMaxBufferSize |
||||||
|
* Signature: (J)J |
||||||
|
*/ |
||||||
|
jlong Java_org_rocksdb_EnvOptions_randomAccessMaxBufferSize(JNIEnv *env, |
||||||
|
jobject jobj, |
||||||
|
jlong jhandle) { |
||||||
|
return ENV_OPTIONS_GET(jhandle, random_access_max_buffer_size); |
||||||
|
} |
||||||
|
|
||||||
|
/*
|
||||||
|
* Class: org_rocksdb_EnvOptions |
||||||
|
* Method: setWritableFileMaxBufferSize |
||||||
|
* Signature: (JJ)V |
||||||
|
*/ |
||||||
|
void Java_org_rocksdb_EnvOptions_setWritableFileMaxBufferSize( |
||||||
|
JNIEnv *env, jobject jobj, jlong jhandle, |
||||||
|
jlong writable_file_max_buffer_size) { |
||||||
|
ENV_OPTIONS_SET_SIZE_T(jhandle, writable_file_max_buffer_size); |
||||||
|
} |
||||||
|
|
||||||
|
/*
|
||||||
|
* Class: org_rocksdb_EnvOptions |
||||||
|
* Method: writableFileMaxBufferSize |
||||||
|
* Signature: (J)J |
||||||
|
*/ |
||||||
|
jlong Java_org_rocksdb_EnvOptions_writableFileMaxBufferSize(JNIEnv *env, |
||||||
|
jobject jobj, |
||||||
|
jlong jhandle) { |
||||||
|
return ENV_OPTIONS_GET(jhandle, writable_file_max_buffer_size); |
||||||
|
} |
||||||
|
|
||||||
|
/*
|
||||||
|
* Class: org_rocksdb_EnvOptions |
||||||
|
* Method: setRateLimiter |
||||||
|
* Signature: (JJ)V |
||||||
|
*/ |
||||||
|
void Java_org_rocksdb_EnvOptions_setRateLimiter(JNIEnv *env, jobject jobj, |
||||||
|
jlong jhandle, |
||||||
|
jlong rl_handle) { |
||||||
|
auto *rate_limiter = reinterpret_cast<rocksdb::RateLimiter *>(rl_handle); |
||||||
|
auto *env_opt = reinterpret_cast<rocksdb::EnvOptions *>(jhandle); |
||||||
|
env_opt->rate_limiter = rate_limiter; |
||||||
|
} |
||||||
@ -0,0 +1,242 @@ |
|||||||
|
// 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++ and enables
|
||||||
|
// calling c++ rocksdb::BackupableDB and rocksdb::BackupableDBOptions methods
|
||||||
|
// from Java side.
|
||||||
|
|
||||||
|
#include <jni.h> |
||||||
|
|
||||||
|
#include "include/org_rocksdb_ExternalSstFileInfo.h" |
||||||
|
#include "rocksdb/sst_file_writer.h" |
||||||
|
|
||||||
|
/*
|
||||||
|
* Class: org_rocksdb_ExternalSstFileInfo |
||||||
|
* Method: newExternalSstFileInfo |
||||||
|
* Signature: ()J |
||||||
|
*/ |
||||||
|
jlong Java_org_rocksdb_ExternalSstFileInfo_newExternalSstFileInfo__( |
||||||
|
JNIEnv *env, jclass jcls) { |
||||||
|
return reinterpret_cast<jlong>(new rocksdb::ExternalSstFileInfo()); |
||||||
|
} |
||||||
|
|
||||||
|
/*
|
||||||
|
* Class: org_rocksdb_ExternalSstFileInfo |
||||||
|
* Method: newExternalSstFileInfo |
||||||
|
* Signature: (Ljava/lang/String;Ljava/lang/String;Ljava/lang/String;JJII)J |
||||||
|
*/ |
||||||
|
jlong Java_org_rocksdb_ExternalSstFileInfo_newExternalSstFileInfo__Ljava_lang_String_2Ljava_lang_String_2Ljava_lang_String_2JJII( |
||||||
|
JNIEnv *env, jclass jcls, jstring jfile_path, jstring jsmallest_key, |
||||||
|
jstring jlargest_key, jlong jsequence_number, jlong jfile_size, |
||||||
|
jint jnum_entries, jint jversion) { |
||||||
|
const char *file_path = env->GetStringUTFChars(jfile_path, NULL); |
||||||
|
const char *smallest_key = env->GetStringUTFChars(jsmallest_key, NULL); |
||||||
|
const char *largest_key = env->GetStringUTFChars(jlargest_key, NULL); |
||||||
|
auto *external_sst_file_info = new rocksdb::ExternalSstFileInfo( |
||||||
|
file_path, smallest_key, largest_key, |
||||||
|
static_cast<rocksdb::SequenceNumber>(jsequence_number), |
||||||
|
static_cast<uint64_t>(jfile_size), static_cast<int32_t>(jnum_entries), |
||||||
|
static_cast<int32_t>(jversion)); |
||||||
|
env->ReleaseStringUTFChars(jfile_path, file_path); |
||||||
|
env->ReleaseStringUTFChars(jsmallest_key, smallest_key); |
||||||
|
env->ReleaseStringUTFChars(jlargest_key, largest_key); |
||||||
|
return reinterpret_cast<jlong>(external_sst_file_info); |
||||||
|
} |
||||||
|
|
||||||
|
/*
|
||||||
|
* Class: org_rocksdb_ExternalSstFileInfo |
||||||
|
* Method: setFilePath |
||||||
|
* Signature: (JLjava/lang/String;)V |
||||||
|
*/ |
||||||
|
void Java_org_rocksdb_ExternalSstFileInfo_setFilePath(JNIEnv *env, jobject jobj, |
||||||
|
jlong jhandle, |
||||||
|
jstring jfile_path) { |
||||||
|
auto *external_sst_file_info = |
||||||
|
reinterpret_cast<rocksdb::ExternalSstFileInfo *>(jhandle); |
||||||
|
const char *file_path = env->GetStringUTFChars(jfile_path, NULL); |
||||||
|
external_sst_file_info->file_path = file_path; |
||||||
|
env->ReleaseStringUTFChars(jfile_path, file_path); |
||||||
|
} |
||||||
|
|
||||||
|
/*
|
||||||
|
* Class: org_rocksdb_ExternalSstFileInfo |
||||||
|
* Method: filePath |
||||||
|
* Signature: (J)Ljava/lang/String; |
||||||
|
*/ |
||||||
|
jstring Java_org_rocksdb_ExternalSstFileInfo_filePath(JNIEnv *env, jobject jobj, |
||||||
|
jlong jhandle) { |
||||||
|
auto *external_sst_file_info = |
||||||
|
reinterpret_cast<rocksdb::ExternalSstFileInfo *>(jhandle); |
||||||
|
return env->NewStringUTF(external_sst_file_info->file_path.data()); |
||||||
|
} |
||||||
|
|
||||||
|
/*
|
||||||
|
* Class: org_rocksdb_ExternalSstFileInfo |
||||||
|
* Method: setSmallestKey |
||||||
|
* Signature: (JLjava/lang/String;)V |
||||||
|
*/ |
||||||
|
void Java_org_rocksdb_ExternalSstFileInfo_setSmallestKey( |
||||||
|
JNIEnv *env, jobject jobj, jlong jhandle, jstring jsmallest_key) { |
||||||
|
auto *external_sst_file_info = |
||||||
|
reinterpret_cast<rocksdb::ExternalSstFileInfo *>(jhandle); |
||||||
|
const char *smallest_key = env->GetStringUTFChars(jsmallest_key, NULL); |
||||||
|
external_sst_file_info->smallest_key = smallest_key; |
||||||
|
env->ReleaseStringUTFChars(jsmallest_key, smallest_key); |
||||||
|
} |
||||||
|
|
||||||
|
/*
|
||||||
|
* Class: org_rocksdb_ExternalSstFileInfo |
||||||
|
* Method: smallestKey |
||||||
|
* Signature: (J)Ljava/lang/String; |
||||||
|
*/ |
||||||
|
jstring Java_org_rocksdb_ExternalSstFileInfo_smallestKey(JNIEnv *env, |
||||||
|
jobject jobj, |
||||||
|
jlong jhandle) { |
||||||
|
auto *external_sst_file_info = |
||||||
|
reinterpret_cast<rocksdb::ExternalSstFileInfo *>(jhandle); |
||||||
|
return env->NewStringUTF(external_sst_file_info->smallest_key.data()); |
||||||
|
} |
||||||
|
|
||||||
|
/*
|
||||||
|
* Class: org_rocksdb_ExternalSstFileInfo |
||||||
|
* Method: setLargestKey |
||||||
|
* Signature: (JLjava/lang/String;)V |
||||||
|
*/ |
||||||
|
void Java_org_rocksdb_ExternalSstFileInfo_setLargestKey(JNIEnv *env, |
||||||
|
jobject jobj, |
||||||
|
jlong jhandle, |
||||||
|
jstring jlargest_key) { |
||||||
|
auto *external_sst_file_info = |
||||||
|
reinterpret_cast<rocksdb::ExternalSstFileInfo *>(jhandle); |
||||||
|
const char *largest_key = env->GetStringUTFChars(jlargest_key, NULL); |
||||||
|
external_sst_file_info->largest_key = largest_key; |
||||||
|
env->ReleaseStringUTFChars(jlargest_key, largest_key); |
||||||
|
} |
||||||
|
|
||||||
|
/*
|
||||||
|
* Class: org_rocksdb_ExternalSstFileInfo |
||||||
|
* Method: largestKey |
||||||
|
* Signature: (J)Ljava/lang/String; |
||||||
|
*/ |
||||||
|
jstring Java_org_rocksdb_ExternalSstFileInfo_largestKey(JNIEnv *env, |
||||||
|
jobject jobj, |
||||||
|
jlong jhandle) { |
||||||
|
auto *external_sst_file_info = |
||||||
|
reinterpret_cast<rocksdb::ExternalSstFileInfo *>(jhandle); |
||||||
|
return env->NewStringUTF(external_sst_file_info->largest_key.data()); |
||||||
|
} |
||||||
|
|
||||||
|
/*
|
||||||
|
* Class: org_rocksdb_ExternalSstFileInfo |
||||||
|
* Method: setSequenceNumber |
||||||
|
* Signature: (JJ)V |
||||||
|
*/ |
||||||
|
void Java_org_rocksdb_ExternalSstFileInfo_setSequenceNumber( |
||||||
|
JNIEnv *env, jobject jobj, jlong jhandle, jlong jsequence_number) { |
||||||
|
auto *external_sst_file_info = |
||||||
|
reinterpret_cast<rocksdb::ExternalSstFileInfo *>(jhandle); |
||||||
|
external_sst_file_info->sequence_number = |
||||||
|
static_cast<rocksdb::SequenceNumber>(jsequence_number); |
||||||
|
} |
||||||
|
|
||||||
|
/*
|
||||||
|
* Class: org_rocksdb_ExternalSstFileInfo |
||||||
|
* Method: sequenceNumber |
||||||
|
* Signature: (J)J |
||||||
|
*/ |
||||||
|
jlong Java_org_rocksdb_ExternalSstFileInfo_sequenceNumber(JNIEnv *env, |
||||||
|
jobject jobj, |
||||||
|
jlong jhandle) { |
||||||
|
auto *external_sst_file_info = |
||||||
|
reinterpret_cast<rocksdb::ExternalSstFileInfo *>(jhandle); |
||||||
|
return static_cast<jlong>(external_sst_file_info->sequence_number); |
||||||
|
} |
||||||
|
|
||||||
|
/*
|
||||||
|
* Class: org_rocksdb_ExternalSstFileInfo |
||||||
|
* Method: setFileSize |
||||||
|
* Signature: (JJ)V |
||||||
|
*/ |
||||||
|
void Java_org_rocksdb_ExternalSstFileInfo_setFileSize(JNIEnv *env, jobject jobj, |
||||||
|
jlong jhandle, |
||||||
|
jlong jfile_size) { |
||||||
|
auto *external_sst_file_info = |
||||||
|
reinterpret_cast<rocksdb::ExternalSstFileInfo *>(jhandle); |
||||||
|
external_sst_file_info->file_size = static_cast<uint64_t>(jfile_size); |
||||||
|
} |
||||||
|
|
||||||
|
/*
|
||||||
|
* Class: org_rocksdb_ExternalSstFileInfo |
||||||
|
* Method: fileSize |
||||||
|
* Signature: (J)J |
||||||
|
*/ |
||||||
|
jlong Java_org_rocksdb_ExternalSstFileInfo_fileSize(JNIEnv *env, jobject jobj, |
||||||
|
jlong jhandle) { |
||||||
|
auto *external_sst_file_info = |
||||||
|
reinterpret_cast<rocksdb::ExternalSstFileInfo *>(jhandle); |
||||||
|
return static_cast<jlong>(external_sst_file_info->file_size); |
||||||
|
} |
||||||
|
|
||||||
|
/*
|
||||||
|
* Class: org_rocksdb_ExternalSstFileInfo |
||||||
|
* Method: setNumEntries |
||||||
|
* Signature: (JI)V |
||||||
|
*/ |
||||||
|
void Java_org_rocksdb_ExternalSstFileInfo_setNumEntries(JNIEnv *env, |
||||||
|
jobject jobj, |
||||||
|
jlong jhandle, |
||||||
|
jint jnum_entries) { |
||||||
|
auto *external_sst_file_info = |
||||||
|
reinterpret_cast<rocksdb::ExternalSstFileInfo *>(jhandle); |
||||||
|
external_sst_file_info->num_entries = static_cast<uint64_t>(jnum_entries); |
||||||
|
} |
||||||
|
|
||||||
|
/*
|
||||||
|
* Class: org_rocksdb_ExternalSstFileInfo |
||||||
|
* Method: numEntries |
||||||
|
* Signature: (J)I |
||||||
|
*/ |
||||||
|
jint Java_org_rocksdb_ExternalSstFileInfo_numEntries(JNIEnv *env, jobject jobj, |
||||||
|
jlong jhandle) { |
||||||
|
auto *external_sst_file_info = |
||||||
|
reinterpret_cast<rocksdb::ExternalSstFileInfo *>(jhandle); |
||||||
|
return static_cast<jint>(external_sst_file_info->num_entries); |
||||||
|
} |
||||||
|
|
||||||
|
/*
|
||||||
|
* Class: org_rocksdb_ExternalSstFileInfo |
||||||
|
* Method: setVersion |
||||||
|
* Signature: (JI)V |
||||||
|
*/ |
||||||
|
void Java_org_rocksdb_ExternalSstFileInfo_setVersion(JNIEnv *env, jobject jobj, |
||||||
|
jlong jhandle, |
||||||
|
jint jversion) { |
||||||
|
auto *external_sst_file_info = |
||||||
|
reinterpret_cast<rocksdb::ExternalSstFileInfo *>(jhandle); |
||||||
|
external_sst_file_info->version = static_cast<int32_t>(jversion); |
||||||
|
} |
||||||
|
|
||||||
|
/*
|
||||||
|
* Class: org_rocksdb_ExternalSstFileInfo |
||||||
|
* Method: version |
||||||
|
* Signature: (J)I |
||||||
|
*/ |
||||||
|
jint Java_org_rocksdb_ExternalSstFileInfo_version(JNIEnv *env, jobject jobj, |
||||||
|
jlong jhandle) { |
||||||
|
auto *external_sst_file_info = |
||||||
|
reinterpret_cast<rocksdb::ExternalSstFileInfo *>(jhandle); |
||||||
|
return static_cast<jint>(external_sst_file_info->version); |
||||||
|
} |
||||||
|
|
||||||
|
/*
|
||||||
|
* Class: org_rocksdb_ExternalSstFileInfo |
||||||
|
* Method: disposeInternal |
||||||
|
* Signature: (J)V |
||||||
|
*/ |
||||||
|
void Java_org_rocksdb_ExternalSstFileInfo_disposeInternal(JNIEnv *env, |
||||||
|
jobject jobj, |
||||||
|
jlong jhandle) { |
||||||
|
delete reinterpret_cast<rocksdb::ExternalSstFileInfo *>(jhandle); |
||||||
|
} |
||||||
@ -0,0 +1,94 @@ |
|||||||
|
// 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++ and enables
|
||||||
|
// calling c++ rocksdb::BackupableDB and rocksdb::BackupableDBOptions methods
|
||||||
|
// from Java side.
|
||||||
|
|
||||||
|
#include <jni.h> |
||||||
|
#include <string> |
||||||
|
|
||||||
|
#include "include/org_rocksdb_SstFileWriter.h" |
||||||
|
#include "rocksdb/comparator.h" |
||||||
|
#include "rocksdb/env.h" |
||||||
|
#include "rocksdb/options.h" |
||||||
|
#include "rocksdb/sst_file_writer.h" |
||||||
|
#include "rocksjni/portal.h" |
||||||
|
|
||||||
|
/*
|
||||||
|
* Class: org_rocksdb_SstFileWriter |
||||||
|
* Method: newSstFileWriter |
||||||
|
* Signature: (JJJ)J |
||||||
|
*/ |
||||||
|
jlong Java_org_rocksdb_SstFileWriter_newSstFileWriter(JNIEnv *env, jclass jcls, |
||||||
|
jlong jenvoptions, |
||||||
|
jlong joptions, |
||||||
|
jlong jcomparator) { |
||||||
|
auto *env_options = |
||||||
|
reinterpret_cast<const rocksdb::EnvOptions *>(jenvoptions); |
||||||
|
auto *options = reinterpret_cast<const rocksdb::Options *>(joptions); |
||||||
|
auto *comparator = reinterpret_cast<const rocksdb::Comparator *>(jcomparator); |
||||||
|
rocksdb::SstFileWriter *sst_file_writer = |
||||||
|
new rocksdb::SstFileWriter(*env_options, *options, comparator); |
||||||
|
return reinterpret_cast<jlong>(sst_file_writer); |
||||||
|
} |
||||||
|
|
||||||
|
/*
|
||||||
|
* Class: org_rocksdb_SstFileWriter |
||||||
|
* Method: open |
||||||
|
* Signature: (JLjava/lang/String;)V |
||||||
|
*/ |
||||||
|
void Java_org_rocksdb_SstFileWriter_open(JNIEnv *env, jobject jobj, |
||||||
|
jlong jhandle, jstring jfile_path) { |
||||||
|
const char *file_path = env->GetStringUTFChars(jfile_path, NULL); |
||||||
|
rocksdb::Status s = |
||||||
|
reinterpret_cast<rocksdb::SstFileWriter *>(jhandle)->Open(file_path); |
||||||
|
env->ReleaseStringUTFChars(jfile_path, file_path); |
||||||
|
|
||||||
|
if (!s.ok()) { |
||||||
|
rocksdb::RocksDBExceptionJni::ThrowNew(env, s); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
/*
|
||||||
|
* Class: org_rocksdb_SstFileWriter |
||||||
|
* Method: add |
||||||
|
* Signature: (JJJ)V |
||||||
|
*/ |
||||||
|
void Java_org_rocksdb_SstFileWriter_add(JNIEnv *env, jobject jobj, |
||||||
|
jlong jhandle, jlong jkey_handle, |
||||||
|
jlong jvalue_handle) { |
||||||
|
auto *key_slice = reinterpret_cast<rocksdb::Slice *>(jkey_handle); |
||||||
|
auto *value_slice = reinterpret_cast<rocksdb::Slice *>(jvalue_handle); |
||||||
|
rocksdb::Status s = reinterpret_cast<rocksdb::SstFileWriter *>(jhandle)->Add( |
||||||
|
*key_slice, *value_slice); |
||||||
|
if (!s.ok()) { |
||||||
|
rocksdb::RocksDBExceptionJni::ThrowNew(env, s); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
/*
|
||||||
|
* Class: org_rocksdb_SstFileWriter |
||||||
|
* Method: finish |
||||||
|
* Signature: (J)V |
||||||
|
*/ |
||||||
|
void Java_org_rocksdb_SstFileWriter_finish(JNIEnv *env, jobject jobj, |
||||||
|
jlong jhandle) { |
||||||
|
rocksdb::Status s = |
||||||
|
reinterpret_cast<rocksdb::SstFileWriter *>(jhandle)->Finish(); |
||||||
|
if (!s.ok()) { |
||||||
|
rocksdb::RocksDBExceptionJni::ThrowNew(env, s); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
/*
|
||||||
|
* Class: org_rocksdb_SstFileWriter |
||||||
|
* Method: disposeInternal |
||||||
|
* Signature: (J)V |
||||||
|
*/ |
||||||
|
void Java_org_rocksdb_SstFileWriter_disposeInternal(JNIEnv *env, jobject jobj, |
||||||
|
jlong jhandle) { |
||||||
|
delete reinterpret_cast<rocksdb::SstFileWriter *>(jhandle); |
||||||
|
} |
||||||
@ -0,0 +1,207 @@ |
|||||||
|
// 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.
|
||||||
|
|
||||||
|
package org.rocksdb; |
||||||
|
|
||||||
|
public class EnvOptions extends RocksObject { |
||||||
|
static { |
||||||
|
RocksDB.loadLibrary(); |
||||||
|
} |
||||||
|
|
||||||
|
public EnvOptions() { |
||||||
|
super(newEnvOptions()); |
||||||
|
} |
||||||
|
|
||||||
|
public EnvOptions setUseOsBuffer(final boolean useOsBuffer) { |
||||||
|
setUseOsBuffer(nativeHandle_, useOsBuffer); |
||||||
|
return this; |
||||||
|
} |
||||||
|
|
||||||
|
public boolean useOsBuffer() { |
||||||
|
assert(isOwningHandle()); |
||||||
|
return useOsBuffer(nativeHandle_); |
||||||
|
} |
||||||
|
|
||||||
|
public EnvOptions setUseMmapReads(final boolean useMmapReads) { |
||||||
|
setUseMmapReads(nativeHandle_, useMmapReads); |
||||||
|
return this; |
||||||
|
} |
||||||
|
|
||||||
|
public boolean useMmapReads() { |
||||||
|
assert(isOwningHandle()); |
||||||
|
return useMmapReads(nativeHandle_); |
||||||
|
} |
||||||
|
|
||||||
|
public EnvOptions setUseMmapWrites(final boolean useMmapWrites) { |
||||||
|
setUseMmapWrites(nativeHandle_, useMmapWrites); |
||||||
|
return this; |
||||||
|
} |
||||||
|
|
||||||
|
public boolean useMmapWrites() { |
||||||
|
assert(isOwningHandle()); |
||||||
|
return useMmapWrites(nativeHandle_); |
||||||
|
} |
||||||
|
|
||||||
|
public EnvOptions setUseDirectReads(final boolean useDirectReads) { |
||||||
|
setUseDirectReads(nativeHandle_, useDirectReads); |
||||||
|
return this; |
||||||
|
} |
||||||
|
|
||||||
|
public boolean useDirectReads() { |
||||||
|
assert(isOwningHandle()); |
||||||
|
return useDirectReads(nativeHandle_); |
||||||
|
} |
||||||
|
|
||||||
|
public EnvOptions setUseDirectWrites(final boolean useDirectWrites) { |
||||||
|
setUseDirectWrites(nativeHandle_, useDirectWrites); |
||||||
|
return this; |
||||||
|
} |
||||||
|
|
||||||
|
public boolean useDirectWrites() { |
||||||
|
assert(isOwningHandle()); |
||||||
|
return useDirectWrites(nativeHandle_); |
||||||
|
} |
||||||
|
|
||||||
|
public EnvOptions setAllowFallocate(final boolean allowFallocate) { |
||||||
|
setAllowFallocate(nativeHandle_, allowFallocate); |
||||||
|
return this; |
||||||
|
} |
||||||
|
|
||||||
|
public boolean allowFallocate() { |
||||||
|
assert(isOwningHandle()); |
||||||
|
return allowFallocate(nativeHandle_); |
||||||
|
} |
||||||
|
|
||||||
|
public EnvOptions setSetFdCloexec(final boolean setFdCloexec) { |
||||||
|
setSetFdCloexec(nativeHandle_, setFdCloexec); |
||||||
|
return this; |
||||||
|
} |
||||||
|
|
||||||
|
public boolean setFdCloexec() { |
||||||
|
assert(isOwningHandle()); |
||||||
|
return setFdCloexec(nativeHandle_); |
||||||
|
} |
||||||
|
|
||||||
|
public EnvOptions setBytesPerSync(final long bytesPerSync) { |
||||||
|
setBytesPerSync(nativeHandle_, bytesPerSync); |
||||||
|
return this; |
||||||
|
} |
||||||
|
|
||||||
|
public long bytesPerSync() { |
||||||
|
assert(isOwningHandle()); |
||||||
|
return bytesPerSync(nativeHandle_); |
||||||
|
} |
||||||
|
|
||||||
|
public EnvOptions setFallocateWithKeepSize(final boolean fallocateWithKeepSize) { |
||||||
|
setFallocateWithKeepSize(nativeHandle_, fallocateWithKeepSize); |
||||||
|
return this; |
||||||
|
} |
||||||
|
|
||||||
|
public boolean fallocateWithKeepSize() { |
||||||
|
assert(isOwningHandle()); |
||||||
|
return fallocateWithKeepSize(nativeHandle_); |
||||||
|
} |
||||||
|
|
||||||
|
public EnvOptions setCompactionReadaheadSize(final long compactionReadaheadSize) { |
||||||
|
setCompactionReadaheadSize(nativeHandle_, compactionReadaheadSize); |
||||||
|
return this; |
||||||
|
} |
||||||
|
|
||||||
|
public long compactionReadaheadSize() { |
||||||
|
assert(isOwningHandle()); |
||||||
|
return compactionReadaheadSize(nativeHandle_); |
||||||
|
} |
||||||
|
|
||||||
|
public EnvOptions setRandomAccessMaxBufferSize(final long randomAccessMaxBufferSize) { |
||||||
|
setRandomAccessMaxBufferSize(nativeHandle_, randomAccessMaxBufferSize); |
||||||
|
return this; |
||||||
|
} |
||||||
|
|
||||||
|
public long randomAccessMaxBufferSize() { |
||||||
|
assert(isOwningHandle()); |
||||||
|
return randomAccessMaxBufferSize(nativeHandle_); |
||||||
|
} |
||||||
|
|
||||||
|
public EnvOptions setWritableFileMaxBufferSize(final long writableFileMaxBufferSize) { |
||||||
|
setWritableFileMaxBufferSize(nativeHandle_, writableFileMaxBufferSize); |
||||||
|
return this; |
||||||
|
} |
||||||
|
|
||||||
|
public long writableFileMaxBufferSize() { |
||||||
|
assert(isOwningHandle()); |
||||||
|
return writableFileMaxBufferSize(nativeHandle_); |
||||||
|
} |
||||||
|
|
||||||
|
public EnvOptions setRateLimiterConfig(final RateLimiterConfig rateLimiterConfig) { |
||||||
|
this.rateLimiterConfig = rateLimiterConfig; |
||||||
|
setRateLimiter(nativeHandle_, rateLimiterConfig.newRateLimiterHandle()); |
||||||
|
return this; |
||||||
|
} |
||||||
|
|
||||||
|
public RateLimiterConfig rateLimiterConfig() { |
||||||
|
assert(isOwningHandle()); |
||||||
|
return rateLimiterConfig; |
||||||
|
} |
||||||
|
|
||||||
|
private native static long newEnvOptions(); |
||||||
|
|
||||||
|
@Override protected final native void disposeInternal(final long handle); |
||||||
|
|
||||||
|
private native void setUseOsBuffer(final long handle, final boolean useOsBuffer); |
||||||
|
|
||||||
|
private native boolean useOsBuffer(final long handle); |
||||||
|
|
||||||
|
private native void setUseMmapReads(final long handle, final boolean useMmapReads); |
||||||
|
|
||||||
|
private native boolean useMmapReads(final long handle); |
||||||
|
|
||||||
|
private native void setUseMmapWrites(final long handle, final boolean useMmapWrites); |
||||||
|
|
||||||
|
private native boolean useMmapWrites(final long handle); |
||||||
|
|
||||||
|
private native void setUseDirectReads(final long handle, final boolean useDirectReads); |
||||||
|
|
||||||
|
private native boolean useDirectReads(final long handle); |
||||||
|
|
||||||
|
private native void setUseDirectWrites(final long handle, final boolean useDirectWrites); |
||||||
|
|
||||||
|
private native boolean useDirectWrites(final long handle); |
||||||
|
|
||||||
|
private native void setAllowFallocate(final long handle, final boolean allowFallocate); |
||||||
|
|
||||||
|
private native boolean allowFallocate(final long handle); |
||||||
|
|
||||||
|
private native void setSetFdCloexec(final long handle, final boolean setFdCloexec); |
||||||
|
|
||||||
|
private native boolean setFdCloexec(final long handle); |
||||||
|
|
||||||
|
private native void setBytesPerSync(final long handle, final long bytesPerSync); |
||||||
|
|
||||||
|
private native long bytesPerSync(final long handle); |
||||||
|
|
||||||
|
private native void setFallocateWithKeepSize( |
||||||
|
final long handle, final boolean fallocateWithKeepSize); |
||||||
|
|
||||||
|
private native boolean fallocateWithKeepSize(final long handle); |
||||||
|
|
||||||
|
private native void setCompactionReadaheadSize( |
||||||
|
final long handle, final long compactionReadaheadSize); |
||||||
|
|
||||||
|
private native long compactionReadaheadSize(final long handle); |
||||||
|
|
||||||
|
private native void setRandomAccessMaxBufferSize( |
||||||
|
final long handle, final long randomAccessMaxBufferSize); |
||||||
|
|
||||||
|
private native long randomAccessMaxBufferSize(final long handle); |
||||||
|
|
||||||
|
private native void setWritableFileMaxBufferSize( |
||||||
|
final long handle, final long writableFileMaxBufferSize); |
||||||
|
|
||||||
|
private native long writableFileMaxBufferSize(final long handle); |
||||||
|
|
||||||
|
private native void setRateLimiter(final long handle, final long rateLimiterHandle); |
||||||
|
|
||||||
|
private RateLimiterConfig rateLimiterConfig; |
||||||
|
} |
||||||
@ -0,0 +1,135 @@ |
|||||||
|
// 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.
|
||||||
|
|
||||||
|
package org.rocksdb; |
||||||
|
|
||||||
|
import java.io.File; |
||||||
|
|
||||||
|
public class ExternalSstFileInfo extends RocksObject { |
||||||
|
public ExternalSstFileInfo() { |
||||||
|
super(newExternalSstFileInfo()); |
||||||
|
} |
||||||
|
|
||||||
|
public ExternalSstFileInfo(final String filePath, final String smallestKey, |
||||||
|
final String largestKey, final long sequenceNumber, final long fileSize, final int numEntries, |
||||||
|
final int version) { |
||||||
|
super(newExternalSstFileInfo(ensureNotNullFilePath(filePath), |
||||||
|
ensureNotNullSmallestKey(smallestKey), ensureNotNullLargestKey(largestKey), sequenceNumber, |
||||||
|
fileSize, numEntries, version)); |
||||||
|
} |
||||||
|
|
||||||
|
private static String ensureNotNullFilePath(final String filePath) { |
||||||
|
if (filePath == null) { |
||||||
|
throw new NullPointerException("filePath is null."); |
||||||
|
} |
||||||
|
return filePath; |
||||||
|
} |
||||||
|
|
||||||
|
private static String ensureNotNullSmallestKey(final String smallestKey) { |
||||||
|
if (smallestKey == null) { |
||||||
|
throw new NullPointerException("smallestKey is null."); |
||||||
|
} |
||||||
|
return smallestKey; |
||||||
|
} |
||||||
|
|
||||||
|
private static String ensureNotNullLargestKey(final String largestKey) { |
||||||
|
if (largestKey == null) { |
||||||
|
throw new NullPointerException("largestKey is null."); |
||||||
|
} |
||||||
|
return largestKey; |
||||||
|
} |
||||||
|
|
||||||
|
public void setFilePath(final String filePath) { |
||||||
|
setFilePath(nativeHandle_, filePath); |
||||||
|
} |
||||||
|
|
||||||
|
public String filePath() { |
||||||
|
return filePath(nativeHandle_); |
||||||
|
} |
||||||
|
|
||||||
|
public void setSmallestKey(final String smallestKey) { |
||||||
|
setSmallestKey(nativeHandle_, smallestKey); |
||||||
|
} |
||||||
|
|
||||||
|
public String smallestKey() { |
||||||
|
return smallestKey(nativeHandle_); |
||||||
|
} |
||||||
|
|
||||||
|
public void setLargestKey(final String largestKey) { |
||||||
|
setLargestKey(nativeHandle_, largestKey); |
||||||
|
} |
||||||
|
|
||||||
|
public String largestKey() { |
||||||
|
return largestKey(nativeHandle_); |
||||||
|
} |
||||||
|
|
||||||
|
public void setSequenceNumber(final long sequenceNumber) { |
||||||
|
setSequenceNumber(nativeHandle_, sequenceNumber); |
||||||
|
} |
||||||
|
|
||||||
|
public long sequenceNumber() { |
||||||
|
return sequenceNumber(nativeHandle_); |
||||||
|
} |
||||||
|
|
||||||
|
public void setFileSize(final long fileSize) { |
||||||
|
setFileSize(nativeHandle_, fileSize); |
||||||
|
} |
||||||
|
|
||||||
|
public long fileSize() { |
||||||
|
return fileSize(nativeHandle_); |
||||||
|
} |
||||||
|
|
||||||
|
public void setNumEntries(final int numEntries) { |
||||||
|
setNumEntries(nativeHandle_, numEntries); |
||||||
|
} |
||||||
|
|
||||||
|
public int numEntries() { |
||||||
|
return numEntries(nativeHandle_); |
||||||
|
} |
||||||
|
|
||||||
|
public void setVersion(final int version) { |
||||||
|
setVersion(nativeHandle_, version); |
||||||
|
} |
||||||
|
|
||||||
|
public int version() { |
||||||
|
return version(nativeHandle_); |
||||||
|
} |
||||||
|
|
||||||
|
private native static long newExternalSstFileInfo(); |
||||||
|
|
||||||
|
private native static long newExternalSstFileInfo(final String filePath, final String smallestKey, |
||||||
|
final String largestKey, final long sequenceNumber, final long fileSize, final int numEntries, |
||||||
|
final int version); |
||||||
|
|
||||||
|
private native void setFilePath(final long handle, final String filePath); |
||||||
|
|
||||||
|
private native String filePath(final long handle); |
||||||
|
|
||||||
|
private native void setSmallestKey(final long handle, final String smallestKey); |
||||||
|
|
||||||
|
private native String smallestKey(final long handle); |
||||||
|
|
||||||
|
private native void setLargestKey(final long handle, final String largestKey); |
||||||
|
|
||||||
|
private native String largestKey(final long handle); |
||||||
|
|
||||||
|
private native void setSequenceNumber(final long handle, final long sequenceNumber); |
||||||
|
|
||||||
|
private native long sequenceNumber(final long handle); |
||||||
|
|
||||||
|
private native void setFileSize(final long handle, final long fileSize); |
||||||
|
|
||||||
|
private native long fileSize(final long handle); |
||||||
|
|
||||||
|
private native void setNumEntries(final long handle, final int numEntries); |
||||||
|
|
||||||
|
private native int numEntries(final long handle); |
||||||
|
|
||||||
|
private native void setVersion(final long handle, final int version); |
||||||
|
|
||||||
|
private native int version(final long handle); |
||||||
|
|
||||||
|
@Override protected final native void disposeInternal(final long handle); |
||||||
|
} |
||||||
@ -0,0 +1,46 @@ |
|||||||
|
// 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.
|
||||||
|
|
||||||
|
package org.rocksdb; |
||||||
|
|
||||||
|
public class SstFileWriter extends RocksObject { |
||||||
|
static { |
||||||
|
RocksDB.loadLibrary(); |
||||||
|
} |
||||||
|
|
||||||
|
public SstFileWriter(final EnvOptions envOptions, final Options options, |
||||||
|
final AbstractComparator<? extends AbstractSlice<?>> comparator) { |
||||||
|
super(newSstFileWriter( |
||||||
|
envOptions.nativeHandle_, options.nativeHandle_, comparator.getNativeHandle())); |
||||||
|
} |
||||||
|
|
||||||
|
public void open(final String filePath) throws RocksDBException { |
||||||
|
open(nativeHandle_, filePath); |
||||||
|
} |
||||||
|
|
||||||
|
public void add(final Slice key, final Slice value) throws RocksDBException { |
||||||
|
add(nativeHandle_, key.getNativeHandle(), value.getNativeHandle()); |
||||||
|
} |
||||||
|
|
||||||
|
public void add(final DirectSlice key, final DirectSlice value) throws RocksDBException { |
||||||
|
add(nativeHandle_, key.getNativeHandle(), value.getNativeHandle()); |
||||||
|
} |
||||||
|
|
||||||
|
public void finish() throws RocksDBException { |
||||||
|
finish(nativeHandle_); |
||||||
|
} |
||||||
|
|
||||||
|
private native static long newSstFileWriter( |
||||||
|
final long envOptionsHandle, final long optionsHandle, final long userComparatorHandle); |
||||||
|
|
||||||
|
private native void open(final long handle, final String filePath) throws RocksDBException; |
||||||
|
|
||||||
|
private native void add(final long handle, final long keyHandle, final long valueHandle) |
||||||
|
throws RocksDBException; |
||||||
|
|
||||||
|
private native void finish(final long handle) throws RocksDBException; |
||||||
|
|
||||||
|
@Override protected final native void disposeInternal(final long handle); |
||||||
|
} |
||||||
@ -0,0 +1,142 @@ |
|||||||
|
// 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.
|
||||||
|
|
||||||
|
package org.rocksdb; |
||||||
|
|
||||||
|
import org.junit.ClassRule; |
||||||
|
import org.junit.Test; |
||||||
|
|
||||||
|
import java.util.Random; |
||||||
|
|
||||||
|
import static org.assertj.core.api.Assertions.assertThat; |
||||||
|
|
||||||
|
public class EnvOptionsTest { |
||||||
|
@ClassRule |
||||||
|
public static final RocksMemoryResource rocksMemoryResource = new RocksMemoryResource(); |
||||||
|
|
||||||
|
public static final Random rand = PlatformRandomHelper.getPlatformSpecificRandomFactory(); |
||||||
|
|
||||||
|
@Test |
||||||
|
public void useOsBuffer() { |
||||||
|
try (final EnvOptions envOptions = new EnvOptions()) { |
||||||
|
final boolean boolValue = rand.nextBoolean(); |
||||||
|
envOptions.setUseOsBuffer(boolValue); |
||||||
|
assertThat(envOptions.useOsBuffer()).isEqualTo(boolValue); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
@Test |
||||||
|
public void useMmapReads() { |
||||||
|
try (final EnvOptions envOptions = new EnvOptions()) { |
||||||
|
final boolean boolValue = rand.nextBoolean(); |
||||||
|
envOptions.setUseMmapReads(boolValue); |
||||||
|
assertThat(envOptions.useMmapReads()).isEqualTo(boolValue); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
@Test |
||||||
|
public void useMmapWrites() { |
||||||
|
try (final EnvOptions envOptions = new EnvOptions()) { |
||||||
|
final boolean boolValue = rand.nextBoolean(); |
||||||
|
envOptions.setUseMmapWrites(boolValue); |
||||||
|
assertThat(envOptions.useMmapWrites()).isEqualTo(boolValue); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
@Test |
||||||
|
public void useDirectReads() { |
||||||
|
try (final EnvOptions envOptions = new EnvOptions()) { |
||||||
|
final boolean boolValue = rand.nextBoolean(); |
||||||
|
envOptions.setUseDirectReads(boolValue); |
||||||
|
assertThat(envOptions.useDirectReads()).isEqualTo(boolValue); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
@Test |
||||||
|
public void useDirectWrites() { |
||||||
|
try (final EnvOptions envOptions = new EnvOptions()) { |
||||||
|
final boolean boolValue = rand.nextBoolean(); |
||||||
|
envOptions.setUseDirectWrites(boolValue); |
||||||
|
assertThat(envOptions.useDirectWrites()).isEqualTo(boolValue); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
@Test |
||||||
|
public void allowFallocate() { |
||||||
|
try (final EnvOptions envOptions = new EnvOptions()) { |
||||||
|
final boolean boolValue = rand.nextBoolean(); |
||||||
|
envOptions.setAllowFallocate(boolValue); |
||||||
|
assertThat(envOptions.allowFallocate()).isEqualTo(boolValue); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
@Test |
||||||
|
public void setFdCloexecs() { |
||||||
|
try (final EnvOptions envOptions = new EnvOptions()) { |
||||||
|
final boolean boolValue = rand.nextBoolean(); |
||||||
|
envOptions.setSetFdCloexec(boolValue); |
||||||
|
assertThat(envOptions.setFdCloexec()).isEqualTo(boolValue); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
@Test |
||||||
|
public void bytesPerSync() { |
||||||
|
try (final EnvOptions envOptions = new EnvOptions()) { |
||||||
|
final long longValue = rand.nextLong(); |
||||||
|
envOptions.setBytesPerSync(longValue); |
||||||
|
assertThat(envOptions.bytesPerSync()).isEqualTo(longValue); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
@Test |
||||||
|
public void fallocateWithKeepSize() { |
||||||
|
try (final EnvOptions envOptions = new EnvOptions()) { |
||||||
|
final boolean boolValue = rand.nextBoolean(); |
||||||
|
envOptions.setFallocateWithKeepSize(boolValue); |
||||||
|
assertThat(envOptions.fallocateWithKeepSize()).isEqualTo(boolValue); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
@Test |
||||||
|
public void compactionReadaheadSize() { |
||||||
|
try (final EnvOptions envOptions = new EnvOptions()) { |
||||||
|
final int intValue = rand.nextInt(); |
||||||
|
envOptions.setCompactionReadaheadSize(intValue); |
||||||
|
assertThat(envOptions.compactionReadaheadSize()).isEqualTo(intValue); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
@Test |
||||||
|
public void randomAccessMaxBufferSize() { |
||||||
|
try (final EnvOptions envOptions = new EnvOptions()) { |
||||||
|
final int intValue = rand.nextInt(); |
||||||
|
envOptions.setRandomAccessMaxBufferSize(intValue); |
||||||
|
assertThat(envOptions.randomAccessMaxBufferSize()).isEqualTo(intValue); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
@Test |
||||||
|
public void writableFileMaxBufferSize() { |
||||||
|
try (final EnvOptions envOptions = new EnvOptions()) { |
||||||
|
final int intValue = rand.nextInt(); |
||||||
|
envOptions.setWritableFileMaxBufferSize(intValue); |
||||||
|
assertThat(envOptions.writableFileMaxBufferSize()).isEqualTo(intValue); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
@Test |
||||||
|
public void rateLimiterConfig() { |
||||||
|
try (final EnvOptions envOptions = new EnvOptions()) { |
||||||
|
final RateLimiterConfig rateLimiterConfig1 = |
||||||
|
new GenericRateLimiterConfig(1000, 100 * 1000, 1); |
||||||
|
envOptions.setRateLimiterConfig(rateLimiterConfig1); |
||||||
|
assertThat(envOptions.rateLimiterConfig()).isEqualTo(rateLimiterConfig1); |
||||||
|
|
||||||
|
final RateLimiterConfig rateLimiterConfig2 = new GenericRateLimiterConfig(1000); |
||||||
|
envOptions.setRateLimiterConfig(rateLimiterConfig2); |
||||||
|
assertThat(envOptions.rateLimiterConfig()).isEqualTo(rateLimiterConfig2); |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
@ -0,0 +1,105 @@ |
|||||||
|
// 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.
|
||||||
|
|
||||||
|
package org.rocksdb; |
||||||
|
|
||||||
|
import org.junit.ClassRule; |
||||||
|
import org.junit.Test; |
||||||
|
|
||||||
|
import java.util.Random; |
||||||
|
|
||||||
|
import static org.assertj.core.api.Assertions.assertThat; |
||||||
|
|
||||||
|
public class ExternalSstFileInfoTest { |
||||||
|
@ClassRule |
||||||
|
public static final RocksMemoryResource rocksMemoryResource = new RocksMemoryResource(); |
||||||
|
|
||||||
|
public static final Random rand = PlatformRandomHelper.getPlatformSpecificRandomFactory(); |
||||||
|
|
||||||
|
@Test |
||||||
|
public void createExternalSstFileInfoWithoutParameters() { |
||||||
|
try (final ExternalSstFileInfo info = new ExternalSstFileInfo()) { |
||||||
|
assertThat(info).isNotNull(); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
@Test |
||||||
|
public void createExternalSstFileInfoWithParameters() { |
||||||
|
final String filePath = "path/to/sst"; |
||||||
|
final String smallestKey = "min"; |
||||||
|
final String largestKey = "max"; |
||||||
|
final long sequenceNumber = rand.nextLong(); |
||||||
|
final long fileSize = rand.nextLong(); |
||||||
|
final int numEntries = rand.nextInt(); |
||||||
|
final int version = rand.nextInt(); |
||||||
|
try (final ExternalSstFileInfo info = new ExternalSstFileInfo( |
||||||
|
filePath, smallestKey, largestKey, sequenceNumber, fileSize, numEntries, version)) { |
||||||
|
assertThat(info).isNotNull(); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
@Test |
||||||
|
public void filePath() { |
||||||
|
try (final ExternalSstFileInfo info = new ExternalSstFileInfo()) { |
||||||
|
final String stringVale = "path/to/sst"; |
||||||
|
info.setFilePath(stringVale); |
||||||
|
assertThat(info.filePath()).isEqualTo(stringVale); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
@Test |
||||||
|
public void smallestKey() { |
||||||
|
try (final ExternalSstFileInfo info = new ExternalSstFileInfo()) { |
||||||
|
final String stringValue = "min"; |
||||||
|
info.setSmallestKey(stringValue); |
||||||
|
assertThat(info.smallestKey()).isEqualTo(stringValue); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
@Test |
||||||
|
public void largestKey() { |
||||||
|
try (final ExternalSstFileInfo info = new ExternalSstFileInfo()) { |
||||||
|
final String stringValue = "max"; |
||||||
|
info.setLargestKey(stringValue); |
||||||
|
assertThat(info.largestKey()).isEqualTo(stringValue); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
@Test |
||||||
|
public void sequenceNumber() { |
||||||
|
try (final ExternalSstFileInfo info = new ExternalSstFileInfo()) { |
||||||
|
final long longValue = rand.nextLong(); |
||||||
|
info.setSequenceNumber(longValue); |
||||||
|
assertThat(info.sequenceNumber()).isEqualTo(longValue); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
@Test |
||||||
|
public void fileSize() { |
||||||
|
try (final ExternalSstFileInfo info = new ExternalSstFileInfo()) { |
||||||
|
final long longValue = Math.max(1, rand.nextLong()); |
||||||
|
info.setFileSize(longValue); |
||||||
|
assertThat(info.fileSize()).isEqualTo(longValue); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
@Test |
||||||
|
public void numEntries() { |
||||||
|
try (final ExternalSstFileInfo info = new ExternalSstFileInfo()) { |
||||||
|
final int intValue = Math.max(1, rand.nextInt()); |
||||||
|
info.setNumEntries(intValue); |
||||||
|
assertThat(info.numEntries()).isEqualTo(intValue); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
@Test |
||||||
|
public void version() { |
||||||
|
try (final ExternalSstFileInfo info = new ExternalSstFileInfo()) { |
||||||
|
final int intValue = rand.nextInt(); |
||||||
|
info.setVersion(intValue); |
||||||
|
assertThat(info.version()).isEqualTo(intValue); |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
@ -0,0 +1,84 @@ |
|||||||
|
// 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.
|
||||||
|
|
||||||
|
package org.rocksdb; |
||||||
|
|
||||||
|
import org.junit.ClassRule; |
||||||
|
import org.junit.Rule; |
||||||
|
import org.junit.Test; |
||||||
|
import org.junit.rules.TemporaryFolder; |
||||||
|
import org.rocksdb.util.BytewiseComparator; |
||||||
|
|
||||||
|
import java.io.File; |
||||||
|
import java.io.IOException; |
||||||
|
import java.util.Map; |
||||||
|
import java.util.TreeMap; |
||||||
|
|
||||||
|
import static org.assertj.core.api.Assertions.assertThat; |
||||||
|
|
||||||
|
public class SstFileWriterTest { |
||||||
|
private static final String SST_FILE_NAME = "test.sst"; |
||||||
|
private static final String DB_DIRECTORY_NAME = "test_db"; |
||||||
|
|
||||||
|
@ClassRule |
||||||
|
public static final RocksMemoryResource rocksMemoryResource = new RocksMemoryResource(); |
||||||
|
|
||||||
|
@Rule public TemporaryFolder parentFolder = new TemporaryFolder(); |
||||||
|
|
||||||
|
private File newSstFile(final TreeMap<String, String> keyValues) |
||||||
|
throws IOException, RocksDBException { |
||||||
|
final EnvOptions envOptions = new EnvOptions(); |
||||||
|
final ComparatorOptions comparatorOptions = new ComparatorOptions(); |
||||||
|
final BytewiseComparator comparator = new BytewiseComparator(comparatorOptions); |
||||||
|
final Options options = new Options().setComparator(comparator); |
||||||
|
final SstFileWriter sstFileWriter = new SstFileWriter(envOptions, options, comparator); |
||||||
|
final File sstFile = parentFolder.newFile(SST_FILE_NAME); |
||||||
|
try { |
||||||
|
sstFileWriter.open(sstFile.getAbsolutePath()); |
||||||
|
for (Map.Entry<String, String> keyValue : keyValues.entrySet()) { |
||||||
|
Slice keySlice = new Slice(keyValue.getKey()); |
||||||
|
Slice valueSlice = new Slice(keyValue.getValue()); |
||||||
|
sstFileWriter.add(keySlice, valueSlice); |
||||||
|
keySlice.close(); |
||||||
|
valueSlice.close(); |
||||||
|
} |
||||||
|
sstFileWriter.finish(); |
||||||
|
} finally { |
||||||
|
assertThat(sstFileWriter).isNotNull(); |
||||||
|
sstFileWriter.close(); |
||||||
|
options.close(); |
||||||
|
envOptions.close(); |
||||||
|
comparatorOptions.close(); |
||||||
|
comparator.close(); |
||||||
|
} |
||||||
|
return sstFile; |
||||||
|
} |
||||||
|
|
||||||
|
@Test |
||||||
|
public void generateSstFile() throws RocksDBException, IOException { |
||||||
|
final TreeMap<String, String> keyValues = new TreeMap<>(); |
||||||
|
keyValues.put("key1", "value1"); |
||||||
|
keyValues.put("key2", "value2"); |
||||||
|
newSstFile(keyValues); |
||||||
|
} |
||||||
|
|
||||||
|
@Test |
||||||
|
public void ingestSstFile() throws RocksDBException, IOException { |
||||||
|
final TreeMap<String, String> keyValues = new TreeMap<>(); |
||||||
|
keyValues.put("key1", "value1"); |
||||||
|
keyValues.put("key2", "value2"); |
||||||
|
final File sstFile = newSstFile(keyValues); |
||||||
|
final File dbFolder = parentFolder.newFolder(DB_DIRECTORY_NAME); |
||||||
|
final Options options = new Options().setCreateIfMissing(true); |
||||||
|
final RocksDB db = RocksDB.open(options, dbFolder.getAbsolutePath()); |
||||||
|
db.addFileWithFilePath(sstFile.getAbsolutePath()); |
||||||
|
|
||||||
|
assertThat(db.get("key1".getBytes())).isEqualTo("value1".getBytes()); |
||||||
|
assertThat(db.get("key2".getBytes())).isEqualTo("value2".getBytes()); |
||||||
|
|
||||||
|
options.close(); |
||||||
|
db.close(); |
||||||
|
} |
||||||
|
} |
||||||
Loading…
Reference in new issue