1 // Copyright (c) 2011-present, Facebook, Inc.  All rights reserved.
2 //  This source code is licensed under both the GPLv2 (found in the
3 //  COPYING file in the root directory) and Apache 2.0 License
4 //  (found in the LICENSE.Apache file in the root directory).
5 //
6 // This file implements the "bridge" between Java and C++ and enables
7 // calling c++ ROCKSDB_NAMESPACE::ThreadStatus methods from Java side.
8 
9 #include <jni.h>
10 
11 #include "portal.h"
12 #include "include/org_rocksdb_ThreadStatus.h"
13 #include "rocksdb/thread_status.h"
14 
15 /*
16  * Class:     org_rocksdb_ThreadStatus
17  * Method:    getThreadTypeName
18  * Signature: (B)Ljava/lang/String;
19  */
Java_org_rocksdb_ThreadStatus_getThreadTypeName(JNIEnv * env,jclass,jbyte jthread_type_value)20 jstring Java_org_rocksdb_ThreadStatus_getThreadTypeName(
21     JNIEnv* env, jclass, jbyte jthread_type_value) {
22   auto name = ROCKSDB_NAMESPACE::ThreadStatus::GetThreadTypeName(
23       ROCKSDB_NAMESPACE::ThreadTypeJni::toCppThreadType(jthread_type_value));
24   return ROCKSDB_NAMESPACE::JniUtil::toJavaString(env, &name, true);
25 }
26 
27 /*
28  * Class:     org_rocksdb_ThreadStatus
29  * Method:    getOperationName
30  * Signature: (B)Ljava/lang/String;
31  */
Java_org_rocksdb_ThreadStatus_getOperationName(JNIEnv * env,jclass,jbyte joperation_type_value)32 jstring Java_org_rocksdb_ThreadStatus_getOperationName(
33     JNIEnv* env, jclass, jbyte joperation_type_value) {
34   auto name = ROCKSDB_NAMESPACE::ThreadStatus::GetOperationName(
35       ROCKSDB_NAMESPACE::OperationTypeJni::toCppOperationType(
36           joperation_type_value));
37   return ROCKSDB_NAMESPACE::JniUtil::toJavaString(env, &name, true);
38 }
39 
40 /*
41  * Class:     org_rocksdb_ThreadStatus
42  * Method:    microsToStringNative
43  * Signature: (J)Ljava/lang/String;
44  */
Java_org_rocksdb_ThreadStatus_microsToStringNative(JNIEnv * env,jclass,jlong jmicros)45 jstring Java_org_rocksdb_ThreadStatus_microsToStringNative(
46     JNIEnv* env, jclass, jlong jmicros) {
47   auto str = ROCKSDB_NAMESPACE::ThreadStatus::MicrosToString(
48       static_cast<uint64_t>(jmicros));
49   return ROCKSDB_NAMESPACE::JniUtil::toJavaString(env, &str, true);
50 }
51 
52 /*
53  * Class:     org_rocksdb_ThreadStatus
54  * Method:    getOperationStageName
55  * Signature: (B)Ljava/lang/String;
56  */
Java_org_rocksdb_ThreadStatus_getOperationStageName(JNIEnv * env,jclass,jbyte joperation_stage_value)57 jstring Java_org_rocksdb_ThreadStatus_getOperationStageName(
58     JNIEnv* env, jclass, jbyte joperation_stage_value) {
59   auto name = ROCKSDB_NAMESPACE::ThreadStatus::GetOperationStageName(
60       ROCKSDB_NAMESPACE::OperationStageJni::toCppOperationStage(
61           joperation_stage_value));
62   return ROCKSDB_NAMESPACE::JniUtil::toJavaString(env, &name, true);
63 }
64 
65 /*
66  * Class:     org_rocksdb_ThreadStatus
67  * Method:    getOperationPropertyName
68  * Signature: (BI)Ljava/lang/String;
69  */
Java_org_rocksdb_ThreadStatus_getOperationPropertyName(JNIEnv * env,jclass,jbyte joperation_type_value,jint jindex)70 jstring Java_org_rocksdb_ThreadStatus_getOperationPropertyName(
71     JNIEnv* env, jclass, jbyte joperation_type_value, jint jindex) {
72   auto name = ROCKSDB_NAMESPACE::ThreadStatus::GetOperationPropertyName(
73       ROCKSDB_NAMESPACE::OperationTypeJni::toCppOperationType(
74           joperation_type_value),
75       static_cast<int>(jindex));
76   return ROCKSDB_NAMESPACE::JniUtil::toJavaString(env, &name, true);
77 }
78 
79 /*
80  * Class:     org_rocksdb_ThreadStatus
81  * Method:    interpretOperationProperties
82  * Signature: (B[J)Ljava/util/Map;
83  */
Java_org_rocksdb_ThreadStatus_interpretOperationProperties(JNIEnv * env,jclass,jbyte joperation_type_value,jlongArray joperation_properties)84 jobject Java_org_rocksdb_ThreadStatus_interpretOperationProperties(
85     JNIEnv* env, jclass, jbyte joperation_type_value,
86     jlongArray joperation_properties) {
87 
88   //convert joperation_properties
89   const jsize len = env->GetArrayLength(joperation_properties);
90   const std::unique_ptr<uint64_t[]> op_properties(new uint64_t[len]);
91   jlong* jop = env->GetLongArrayElements(joperation_properties, nullptr);
92   if (jop == nullptr) {
93     // exception thrown: OutOfMemoryError
94     return nullptr;
95   }
96   for (jsize i = 0; i < len; i++) {
97     op_properties[i] = static_cast<uint64_t>(jop[i]);
98   }
99   env->ReleaseLongArrayElements(joperation_properties, jop, JNI_ABORT);
100 
101   // call the function
102   auto result = ROCKSDB_NAMESPACE::ThreadStatus::InterpretOperationProperties(
103       ROCKSDB_NAMESPACE::OperationTypeJni::toCppOperationType(
104           joperation_type_value),
105       op_properties.get());
106   jobject jresult = ROCKSDB_NAMESPACE::HashMapJni::fromCppMap(env, &result);
107   if (env->ExceptionCheck()) {
108     // exception occurred
109     return nullptr;
110   }
111 
112   return jresult;
113 }
114 
115 /*
116  * Class:     org_rocksdb_ThreadStatus
117  * Method:    getStateName
118  * Signature: (B)Ljava/lang/String;
119  */
Java_org_rocksdb_ThreadStatus_getStateName(JNIEnv * env,jclass,jbyte jstate_type_value)120 jstring Java_org_rocksdb_ThreadStatus_getStateName(
121   JNIEnv* env, jclass, jbyte jstate_type_value) {
122   auto name = ROCKSDB_NAMESPACE::ThreadStatus::GetStateName(
123       ROCKSDB_NAMESPACE::StateTypeJni::toCppStateType(jstate_type_value));
124   return ROCKSDB_NAMESPACE::JniUtil::toJavaString(env, &name, true);
125 }
126