#include "rocksdb/write_buffer_manager.h"
#include <jni.h>
#include <cassert>
#include "include/org_rocksdb_WriteBufferManager.h"
#include "rocksdb/cache.h"
#include "rocksjni/cplusplus_to_java_convert.h"
jlong Java_org_rocksdb_WriteBufferManager_newWriteBufferManager(
JNIEnv* , jclass , jlong jbuffer_size, jlong jcache_handle,
jboolean allow_stall) {
auto* cache_ptr =
reinterpret_cast<std::shared_ptr<ROCKSDB_NAMESPACE::Cache>*>(
jcache_handle);
auto* write_buffer_manager =
new std::shared_ptr<ROCKSDB_NAMESPACE::WriteBufferManager>(
std::make_shared<ROCKSDB_NAMESPACE::WriteBufferManager>(
jbuffer_size, *cache_ptr, allow_stall));
return GET_CPLUSPLUS_POINTER(write_buffer_manager);
}
void Java_org_rocksdb_WriteBufferManager_disposeInternalJni(JNIEnv* ,
jclass ,
jlong jhandle) {
auto* write_buffer_manager =
reinterpret_cast<std::shared_ptr<ROCKSDB_NAMESPACE::WriteBufferManager>*>(
jhandle);
assert(write_buffer_manager != nullptr);
delete write_buffer_manager;
}