67 lines
2.1 KiB
C++
67 lines
2.1 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::Env methods from Java side.
|
||
|
|
|
||
|
|
#include "include/org_rocksdb_RocksEnv.h"
|
||
|
|
#include "rocksdb/env.h"
|
||
|
|
|
||
|
|
/*
|
||
|
|
* Class: org_rocksdb_RocksEnv
|
||
|
|
* Method: getDefaultEnvInternal
|
||
|
|
* Signature: ()J
|
||
|
|
*/
|
||
|
|
jlong Java_org_rocksdb_RocksEnv_getDefaultEnvInternal(
|
||
|
|
JNIEnv* env, jclass jclass) {
|
||
|
|
return reinterpret_cast<jlong>(rocksdb::Env::Default());
|
||
|
|
}
|
||
|
|
|
||
|
|
/*
|
||
|
|
* Class: org_rocksdb_RocksEnv
|
||
|
|
* Method: setBackgroundThreads
|
||
|
|
* Signature: (JII)V
|
||
|
|
*/
|
||
|
|
void Java_org_rocksdb_RocksEnv_setBackgroundThreads(
|
||
|
|
JNIEnv* env, jobject jobj, jlong jhandle,
|
||
|
|
jint num, jint priority) {
|
||
|
|
auto* rocks_env = reinterpret_cast<rocksdb::Env*>(jhandle);
|
||
|
|
switch (priority) {
|
||
|
|
case org_rocksdb_RocksEnv_FLUSH_POOL:
|
||
|
|
rocks_env->SetBackgroundThreads(num, rocksdb::Env::Priority::LOW);
|
||
|
|
break;
|
||
|
|
case org_rocksdb_RocksEnv_COMPACTION_POOL:
|
||
|
|
rocks_env->SetBackgroundThreads(num, rocksdb::Env::Priority::HIGH);
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
/*
|
||
|
|
* Class: org_rocksdb_RocksEnv
|
||
|
|
* Method: getThreadPoolQueueLen
|
||
|
|
* Signature: (JI)I
|
||
|
|
*/
|
||
|
|
jint Java_org_rocksdb_RocksEnv_getThreadPoolQueueLen(
|
||
|
|
JNIEnv* env, jobject jobj, jlong jhandle, jint pool_id) {
|
||
|
|
auto* rocks_env = reinterpret_cast<rocksdb::Env*>(jhandle);
|
||
|
|
switch (pool_id) {
|
||
|
|
case org_rocksdb_RocksEnv_FLUSH_POOL:
|
||
|
|
return rocks_env->GetThreadPoolQueueLen(rocksdb::Env::Priority::LOW);
|
||
|
|
case org_rocksdb_RocksEnv_COMPACTION_POOL:
|
||
|
|
return rocks_env->GetThreadPoolQueueLen(rocksdb::Env::Priority::HIGH);
|
||
|
|
}
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
/*
|
||
|
|
* Class: org_rocksdb_RocksEnv
|
||
|
|
* Method: disposeInternal
|
||
|
|
* Signature: (J)V
|
||
|
|
*/
|
||
|
|
void Java_org_rocksdb_RocksEnv_disposeInternal(
|
||
|
|
JNIEnv* env, jobject jobj, jlong jhandle) {
|
||
|
|
delete reinterpret_cast<rocksdb::Env*>(jhandle);
|
||
|
|
}
|