8f52972cf9
Summary: An arbitrary string can be used as a delimiter in StringAppend merge operator flavor. In particular, it allows using an empty string, combining binary values for the same key byte-to-byte one next to another. Pull Request resolved: https://github.com/facebook/rocksdb/pull/8536 Reviewed By: mrambacher Differential Revision: D29962120 Pulled By: zhichao-cao fbshipit-source-id: 4ef5d846a47835cf428a11200409e30e2dbffc4f
96 lines
3.4 KiB
C++
96 lines
3.4 KiB
C++
// Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved.
|
|
// Copyright (c) 2014, Vlad Balan (vlad.gm@gmail.com). All rights reserved.
|
|
// This source code is licensed under both the GPLv2 (found in the
|
|
// COPYING file in the root directory) and Apache 2.0 License
|
|
// (found in the LICENSE.Apache file in the root directory).
|
|
//
|
|
// This file implements the "bridge" between Java and C++
|
|
// for ROCKSDB_NAMESPACE::MergeOperator.
|
|
|
|
#include <jni.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <memory>
|
|
#include <string>
|
|
|
|
#include "include/org_rocksdb_StringAppendOperator.h"
|
|
#include "include/org_rocksdb_UInt64AddOperator.h"
|
|
#include "rocksdb/db.h"
|
|
#include "rocksdb/memtablerep.h"
|
|
#include "rocksdb/merge_operator.h"
|
|
#include "rocksdb/options.h"
|
|
#include "rocksdb/slice_transform.h"
|
|
#include "rocksdb/statistics.h"
|
|
#include "rocksdb/table.h"
|
|
#include "rocksjni/portal.h"
|
|
#include "utilities/merge_operators.h"
|
|
|
|
/*
|
|
* Class: org_rocksdb_StringAppendOperator
|
|
* Method: newSharedStringAppendOperator
|
|
* Signature: (C)J
|
|
*/
|
|
jlong Java_org_rocksdb_StringAppendOperator_newSharedStringAppendOperator__C(
|
|
JNIEnv* /*env*/, jclass /*jclazz*/, jchar jdelim) {
|
|
auto* sptr_string_append_op =
|
|
new std::shared_ptr<ROCKSDB_NAMESPACE::MergeOperator>(
|
|
ROCKSDB_NAMESPACE::MergeOperators::CreateStringAppendOperator(
|
|
(char)jdelim));
|
|
return reinterpret_cast<jlong>(sptr_string_append_op);
|
|
}
|
|
|
|
jlong Java_org_rocksdb_StringAppendOperator_newSharedStringAppendOperator__Ljava_lang_String_2(
|
|
JNIEnv* env, jclass /*jclass*/, jstring jdelim) {
|
|
jboolean has_exception = JNI_FALSE;
|
|
auto delim =
|
|
ROCKSDB_NAMESPACE::JniUtil::copyStdString(env, jdelim, &has_exception);
|
|
if (has_exception == JNI_TRUE) {
|
|
return 0;
|
|
}
|
|
auto* sptr_string_append_op =
|
|
new std::shared_ptr<ROCKSDB_NAMESPACE::MergeOperator>(
|
|
ROCKSDB_NAMESPACE::MergeOperators::CreateStringAppendOperator(delim));
|
|
return reinterpret_cast<jlong>(sptr_string_append_op);
|
|
}
|
|
|
|
/*
|
|
* Class: org_rocksdb_StringAppendOperator
|
|
* Method: disposeInternal
|
|
* Signature: (J)V
|
|
*/
|
|
void Java_org_rocksdb_StringAppendOperator_disposeInternal(JNIEnv* /*env*/,
|
|
jobject /*jobj*/,
|
|
jlong jhandle) {
|
|
auto* sptr_string_append_op =
|
|
reinterpret_cast<std::shared_ptr<ROCKSDB_NAMESPACE::MergeOperator>*>(
|
|
jhandle);
|
|
delete sptr_string_append_op; // delete std::shared_ptr
|
|
}
|
|
|
|
/*
|
|
* Class: org_rocksdb_UInt64AddOperator
|
|
* Method: newSharedUInt64AddOperator
|
|
* Signature: ()J
|
|
*/
|
|
jlong Java_org_rocksdb_UInt64AddOperator_newSharedUInt64AddOperator(
|
|
JNIEnv* /*env*/, jclass /*jclazz*/) {
|
|
auto* sptr_uint64_add_op =
|
|
new std::shared_ptr<ROCKSDB_NAMESPACE::MergeOperator>(
|
|
ROCKSDB_NAMESPACE::MergeOperators::CreateUInt64AddOperator());
|
|
return reinterpret_cast<jlong>(sptr_uint64_add_op);
|
|
}
|
|
|
|
/*
|
|
* Class: org_rocksdb_UInt64AddOperator
|
|
* Method: disposeInternal
|
|
* Signature: (J)V
|
|
*/
|
|
void Java_org_rocksdb_UInt64AddOperator_disposeInternal(JNIEnv* /*env*/,
|
|
jobject /*jobj*/,
|
|
jlong jhandle) {
|
|
auto* sptr_uint64_add_op =
|
|
reinterpret_cast<std::shared_ptr<ROCKSDB_NAMESPACE::MergeOperator>*>(
|
|
jhandle);
|
|
delete sptr_uint64_add_op; // delete std::shared_ptr
|
|
}
|