JNI implementations for new Java bindings

Author: mike-evolved

java/rocksjni/memory_allocator.cc

@@ -0,0 +1,62 @@
+// This file implements the JNI bindings for MemoryAllocator
+
+#include <jni.h>
+
+#include "include/org_rocksdb_MemoryAllocator.h"
+#include "rocksdb/memory_allocator.h"
+#include "rocksjni/cplusplus_to_java_convert.h"
+#include "rocksjni/portal.h"
+
+/*
+ * Class:     org_rocksdb_MemoryAllocator_DefaultMemoryAllocator
+ * Method:    newDefaultMemoryAllocatorInstance
+ * Signature: ()J
+ */
+jlong Java_org_rocksdb_MemoryAllocator_00024DefaultMemoryAllocator_newDefaultMemoryAllocatorInstance(
+    JNIEnv* /*env*/, jclass /*jcls*/) {
+  // Create default memory allocator (nullptr means use system allocator)
+  std::shared_ptr<ROCKSDB_NAMESPACE::MemoryAllocator>* allocator =
+      new std::shared_ptr<ROCKSDB_NAMESPACE::MemoryAllocator>(nullptr);
+  return GET_CPLUSPLUS_POINTER(allocator);
+}
+
+/*
+ * Class:     org_rocksdb_MemoryAllocator_JEMallocMemoryAllocator
+ * Method:    newJEMallocMemoryAllocatorInstance
+ * Signature: ()J
+ */
+jlong Java_org_rocksdb_MemoryAllocator_00024JEMallocMemoryAllocator_newJEMallocMemoryAllocatorInstance(
+    JNIEnv* env, jclass /*jcls*/) {
+#ifdef ROCKSDB_JEMALLOC
+  auto allocator = ROCKSDB_NAMESPACE::NewJemallocNodumpAllocator();
+  if (allocator.ok()) {
+    std::shared_ptr<ROCKSDB_NAMESPACE::MemoryAllocator>* allocator_ptr =
+        new std::shared_ptr<ROCKSDB_NAMESPACE::MemoryAllocator>(
+            std::move(allocator.value()));
+    return GET_CPLUSPLUS_POINTER(allocator_ptr);
+  } else {
+    ROCKSDB_NAMESPACE::RocksDBExceptionJni::ThrowNew(
+        env, "JEMalloc allocator creation failed: " + allocator.status().ToString());
+    return 0;
+  }
+#else
+  ROCKSDB_NAMESPACE::RocksDBExceptionJni::ThrowNew(
+      env, "JEMalloc is not available in this build");
+  return 0;
+#endif
+}
+
+/*
+ * Class:     org_rocksdb_MemoryAllocator
+ * Method:    disposeInternalJni
+ * Signature: (J)V
+ */
+void Java_org_rocksdb_MemoryAllocator_disposeInternalJni(JNIEnv* /*env*/,
+                                                          jobject /*jobj*/,
+                                                          jlong jhandle) {
+  auto* allocator =
+      reinterpret_cast<std::shared_ptr<ROCKSDB_NAMESPACE::MemoryAllocator>*>(
+          jhandle);
+  assert(allocator != nullptr);
+  delete allocator;
+}

java/rocksjni/ribbon_filter.cc

@@ -0,0 +1,23 @@
+// This file implements the JNI bindings for RibbonFilter
+
+#include <jni.h>
+
+#include "include/org_rocksdb_RibbonFilter.h"
+#include "rocksdb/filter_policy.h"
+#include "rocksjni/cplusplus_to_java_convert.h"
+
+/*
+ * Class:     org_rocksdb_RibbonFilter
+ * Method:    createNewRibbonFilter
+ * Signature: (DI)J
+ */
+jlong Java_org_rocksdb_RibbonFilter_createNewRibbonFilter(
+    JNIEnv* /*env*/, jclass /*jcls*/, jdouble jbits_per_key,
+    jint jbloom_before_level) {
+  auto* filter_policy =
+      new std::shared_ptr<const ROCKSDB_NAMESPACE::FilterPolicy>(
+          ROCKSDB_NAMESPACE::NewRibbonFilterPolicy(
+              static_cast<double>(jbits_per_key),
+              static_cast<int>(jbloom_before_level)));
+  return GET_CPLUSPLUS_POINTER(filter_policy);
+}

java/rocksjni/secondary_cache.cc

@@ -0,0 +1,50 @@
+// This file implements the JNI bindings for SecondaryCache
+
+#include <jni.h>
+
+#include "include/org_rocksdb_SecondaryCache.h"
+#include "rocksdb/secondary_cache.h"
+#include "rocksjni/cplusplus_to_java_convert.h"
+#include "rocksjni/portal.h"
+
+/*
+ * Class:     org_rocksdb_SecondaryCache_CompressedSecondaryCache
+ * Method:    newCompressedSecondaryCacheInstance
+ * Signature: (J)J
+ */
+jlong Java_org_rocksdb_SecondaryCache_00024CompressedSecondaryCache_newCompressedSecondaryCacheInstance__J(
+    JNIEnv* /*env*/, jclass /*jcls*/, jlong jcapacity) {
+  auto cache = ROCKSDB_NAMESPACE::NewCompressedSecondaryCache(
+      static_cast<size_t>(jcapacity));
+  auto* cache_ptr = new std::shared_ptr<ROCKSDB_NAMESPACE::SecondaryCache>(cache);
+  return GET_CPLUSPLUS_POINTER(cache_ptr);
+}
+
+/*
+ * Class:     org_rocksdb_SecondaryCache_CompressedSecondaryCache
+ * Method:    newCompressedSecondaryCacheInstance
+ * Signature: (JIZD)J
+ */
+jlong Java_org_rocksdb_SecondaryCache_00024CompressedSecondaryCache_newCompressedSecondaryCacheInstance__JIZD(
+    JNIEnv* /*env*/, jclass /*jcls*/, jlong jcapacity, jint jnum_shard_bits,
+    jboolean jstrict_capacity_limit, jdouble jhigh_pri_pool_ratio) {
+  auto cache = ROCKSDB_NAMESPACE::NewCompressedSecondaryCache(
+      static_cast<size_t>(jcapacity), static_cast<int>(jnum_shard_bits),
+      static_cast<bool>(jstrict_capacity_limit),
+      static_cast<double>(jhigh_pri_pool_ratio));
+  auto* cache_ptr = new std::shared_ptr<ROCKSDB_NAMESPACE::SecondaryCache>(cache);
+  return GET_CPLUSPLUS_POINTER(cache_ptr);
+}
+
+/*
+ * Class:     org_rocksdb_SecondaryCache
+ * Method:    disposeInternalJni
+ * Signature: (J)V
+ */
+void Java_org_rocksdb_SecondaryCache_disposeInternalJni(JNIEnv* /*env*/,
+                                                         jobject /*jobj*/,
+                                                         jlong jhandle) {
+  auto* cache = reinterpret_cast<std::shared_ptr<ROCKSDB_NAMESPACE::SecondaryCache>*>(jhandle);
+  assert(cache != nullptr);
+  delete cache;
+}