398d72fa61
Summary: Closes https://github.com/facebook/rocksdb/pull/3332 Differential Revision: D6667680 Pulled By: ajkr fbshipit-source-id: b2bb6889257850a4eb6f6cbd7106f62df7b82730
113 lines
3.4 KiB
C++
113 lines
3.4 KiB
C++
// Copyright (c) 2011-present, Facebook, Inc. 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 RateLimiter.
|
|
|
|
#include "rocksjni/portal.h"
|
|
#include "include/org_rocksdb_RateLimiter.h"
|
|
#include "rocksdb/rate_limiter.h"
|
|
|
|
/*
|
|
* Class: org_rocksdb_RateLimiter
|
|
* Method: newRateLimiterHandle
|
|
* Signature: (JJIBZ)J
|
|
*/
|
|
jlong Java_org_rocksdb_RateLimiter_newRateLimiterHandle(
|
|
JNIEnv* env, jclass jclazz, jlong jrate_bytes_per_second,
|
|
jlong jrefill_period_micros, jint jfairness, jbyte jrate_limiter_mode,
|
|
jboolean jauto_tune) {
|
|
auto rate_limiter_mode = rocksdb::RateLimiterModeJni::toCppRateLimiterMode(
|
|
jrate_limiter_mode);
|
|
auto * sptr_rate_limiter =
|
|
new std::shared_ptr<rocksdb::RateLimiter>(rocksdb::NewGenericRateLimiter(
|
|
static_cast<int64_t>(jrate_bytes_per_second),
|
|
static_cast<int64_t>(jrefill_period_micros),
|
|
static_cast<int32_t>(jfairness),
|
|
rate_limiter_mode,
|
|
jauto_tune));
|
|
|
|
return reinterpret_cast<jlong>(sptr_rate_limiter);
|
|
}
|
|
|
|
/*
|
|
* Class: org_rocksdb_RateLimiter
|
|
* Method: disposeInternal
|
|
* Signature: (J)V
|
|
*/
|
|
void Java_org_rocksdb_RateLimiter_disposeInternal(
|
|
JNIEnv* env, jobject jobj, jlong jhandle) {
|
|
auto* handle =
|
|
reinterpret_cast<std::shared_ptr<rocksdb::RateLimiter> *>(jhandle);
|
|
delete handle; // delete std::shared_ptr
|
|
}
|
|
|
|
/*
|
|
* Class: org_rocksdb_RateLimiter
|
|
* Method: setBytesPerSecond
|
|
* Signature: (JJ)V
|
|
*/
|
|
void Java_org_rocksdb_RateLimiter_setBytesPerSecond(
|
|
JNIEnv* env, jobject jobj, jlong handle,
|
|
jlong jbytes_per_second) {
|
|
reinterpret_cast<std::shared_ptr<rocksdb::RateLimiter> *>(handle)->get()->
|
|
SetBytesPerSecond(jbytes_per_second);
|
|
}
|
|
|
|
/*
|
|
* Class: org_rocksdb_RateLimiter
|
|
* Method: getBytesPerSecond
|
|
* Signature: (J)J
|
|
*/
|
|
jlong Java_org_rocksdb_RateLimiter_getBytesPerSecond(
|
|
JNIEnv* env, jobject jobj, jlong handle) {
|
|
return reinterpret_cast<std::shared_ptr<rocksdb::RateLimiter> *>(handle)->get()->
|
|
GetBytesPerSecond();
|
|
}
|
|
|
|
/*
|
|
* Class: org_rocksdb_RateLimiter
|
|
* Method: request
|
|
* Signature: (JJ)V
|
|
*/
|
|
void Java_org_rocksdb_RateLimiter_request(
|
|
JNIEnv* env, jobject jobj, jlong handle,
|
|
jlong jbytes) {
|
|
reinterpret_cast<std::shared_ptr<rocksdb::RateLimiter> *>(handle)->get()->
|
|
Request(jbytes, rocksdb::Env::IO_TOTAL);
|
|
}
|
|
|
|
/*
|
|
* Class: org_rocksdb_RateLimiter
|
|
* Method: getSingleBurstBytes
|
|
* Signature: (J)J
|
|
*/
|
|
jlong Java_org_rocksdb_RateLimiter_getSingleBurstBytes(
|
|
JNIEnv* env, jobject jobj, jlong handle) {
|
|
return reinterpret_cast<std::shared_ptr<rocksdb::RateLimiter> *>(handle)->
|
|
get()->GetSingleBurstBytes();
|
|
}
|
|
|
|
/*
|
|
* Class: org_rocksdb_RateLimiter
|
|
* Method: getTotalBytesThrough
|
|
* Signature: (J)J
|
|
*/
|
|
jlong Java_org_rocksdb_RateLimiter_getTotalBytesThrough(
|
|
JNIEnv* env, jobject jobj, jlong handle) {
|
|
return reinterpret_cast<std::shared_ptr<rocksdb::RateLimiter> *>(handle)->
|
|
get()->GetTotalBytesThrough();
|
|
}
|
|
|
|
/*
|
|
* Class: org_rocksdb_RateLimiter
|
|
* Method: getTotalRequests
|
|
* Signature: (J)J
|
|
*/
|
|
jlong Java_org_rocksdb_RateLimiter_getTotalRequests(
|
|
JNIEnv* env, jobject jobj, jlong handle) {
|
|
return reinterpret_cast<std::shared_ptr<rocksdb::RateLimiter> *>(handle)->
|
|
get()->GetTotalRequests();
|
|
}
|