fork of https://github.com/oxigraph/rocksdb and https://github.com/facebook/rocksdb for nextgraph and oxigraph
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
167 lines
5.2 KiB
167 lines
5.2 KiB
// 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 callback "bridge" between Java and C++ for
|
|
// rocksdb::Comparator.
|
|
|
|
#include "include/org_rocksdb_AbstractLogger.h"
|
|
|
|
#include "rocksjni/loggerjnicallback.h"
|
|
#include "rocksjni/portal.h"
|
|
|
|
namespace rocksdb {
|
|
|
|
LoggerJniCallback::LoggerJniCallback(
|
|
JNIEnv* env, jobject jAbstractLogger) {
|
|
|
|
const jint rs = env->GetJavaVM(&m_jvm);
|
|
assert(rs == JNI_OK);
|
|
|
|
// Note: we want to access the Java Logger instance
|
|
// across multiple method calls, so we create a global ref
|
|
m_jAbstractLogger = env->NewGlobalRef(jAbstractLogger);
|
|
m_jLogMethodId = AbstractLoggerJni::getLogMethodId(env);
|
|
}
|
|
|
|
/**
|
|
* Get JNIEnv for current native thread
|
|
*/
|
|
JNIEnv* LoggerJniCallback::getJniEnv() const {
|
|
JNIEnv *env;
|
|
jint rs = m_jvm->AttachCurrentThread(reinterpret_cast<void **>(&env), NULL);
|
|
assert(rs == JNI_OK);
|
|
return env;
|
|
}
|
|
|
|
void LoggerJniCallback::Logv(const char* format, va_list ap) {
|
|
// We implement this method because it is virtual but we don't
|
|
// use it because we need to know about the log level.
|
|
}
|
|
|
|
void LoggerJniCallback::Logv(const InfoLogLevel log_level,
|
|
const char* format, va_list ap) {
|
|
if (GetInfoLogLevel() <= log_level) {
|
|
JNIEnv* env = getJniEnv();
|
|
// We try twice: the first time with a fixed-size stack allocated buffer,
|
|
// and the second time with a much larger dynamically allocated buffer.
|
|
char buffer[500];
|
|
for (int iter = 0; iter < 2; iter++) {
|
|
char* base;
|
|
int bufsize;
|
|
if (iter == 0) {
|
|
bufsize = sizeof(buffer);
|
|
base = buffer;
|
|
} else {
|
|
bufsize = 30000;
|
|
base = new char[bufsize];
|
|
}
|
|
char* p = base;
|
|
char* limit = base + bufsize;
|
|
// Print the message
|
|
if (p < limit) {
|
|
va_list backup_ap;
|
|
va_copy(backup_ap, ap);
|
|
p += vsnprintf(p, limit - p, format, backup_ap);
|
|
va_end(backup_ap);
|
|
}
|
|
// Truncate to available space if necessary
|
|
if (p >= limit) {
|
|
if (iter == 0) {
|
|
continue; // Try again with larger buffer
|
|
} else {
|
|
p = limit - 1;
|
|
}
|
|
}
|
|
assert(p < limit);
|
|
*p++ = '\0';
|
|
|
|
// determine InfoLogLevel java enum instance
|
|
jobject jlog_level;
|
|
switch(log_level) {
|
|
case rocksdb::InfoLogLevel::DEBUG_LEVEL:
|
|
jlog_level = InfoLogLevelJni::DEBUG_LEVEL(env);
|
|
break;
|
|
case rocksdb::InfoLogLevel::INFO_LEVEL:
|
|
jlog_level = InfoLogLevelJni::INFO_LEVEL(env);
|
|
break;
|
|
case rocksdb::InfoLogLevel::ERROR_LEVEL:
|
|
jlog_level = InfoLogLevelJni::ERROR_LEVEL(env);
|
|
case rocksdb::InfoLogLevel::FATAL_LEVEL:
|
|
jlog_level = InfoLogLevelJni::FATAL_LEVEL(env);
|
|
default:
|
|
jlog_level = InfoLogLevelJni::FATAL_LEVEL(env);
|
|
break;
|
|
}
|
|
// pass java string to callback handler
|
|
env->CallVoidMethod(
|
|
m_jAbstractLogger,
|
|
m_jLogMethodId,
|
|
jlog_level,
|
|
env->NewStringUTF(base));
|
|
|
|
if (base != buffer) {
|
|
delete[] base;
|
|
}
|
|
break;
|
|
}
|
|
m_jvm->DetachCurrentThread();
|
|
}
|
|
}
|
|
|
|
LoggerJniCallback::~LoggerJniCallback() {
|
|
JNIEnv* env = getJniEnv();
|
|
env->DeleteGlobalRef(m_jAbstractLogger);
|
|
m_jvm->DetachCurrentThread();
|
|
}
|
|
|
|
} // namespace rocksdb
|
|
|
|
/*
|
|
* Class: org_rocksdb_AbstractLogger
|
|
* Method: createNewLoggerOptions
|
|
* Signature: (J)V
|
|
*/
|
|
void Java_org_rocksdb_AbstractLogger_createNewLoggerOptions(
|
|
JNIEnv* env, jobject jobj, jlong joptions) {
|
|
rocksdb::LoggerJniCallback* c =
|
|
new rocksdb::LoggerJniCallback(env, jobj);
|
|
// set log level
|
|
c->SetInfoLogLevel(reinterpret_cast<rocksdb::Options*>
|
|
(joptions)->info_log_level);
|
|
std::shared_ptr<rocksdb::LoggerJniCallback> *pLoggerJniCallback =
|
|
new std::shared_ptr<rocksdb::LoggerJniCallback>;
|
|
*pLoggerJniCallback = std::shared_ptr<rocksdb::LoggerJniCallback>(c);
|
|
rocksdb::AbstractLoggerJni::setHandle(env, jobj, pLoggerJniCallback);
|
|
}
|
|
|
|
/*
|
|
* Class: org_rocksdb_AbstractLogger
|
|
* Method: createNewLoggerDbOptions
|
|
* Signature: (J)V
|
|
*/
|
|
void Java_org_rocksdb_AbstractLogger_createNewLoggerDbOptions(
|
|
JNIEnv* env, jobject jobj, jlong jdb_options) {
|
|
rocksdb::LoggerJniCallback* c =
|
|
new rocksdb::LoggerJniCallback(env, jobj);
|
|
// set log level
|
|
c->SetInfoLogLevel(reinterpret_cast<rocksdb::DBOptions*>
|
|
(jdb_options)->info_log_level);
|
|
std::shared_ptr<rocksdb::LoggerJniCallback> *pLoggerJniCallback =
|
|
new std::shared_ptr<rocksdb::LoggerJniCallback>;
|
|
*pLoggerJniCallback = std::shared_ptr<rocksdb::LoggerJniCallback>(c);
|
|
rocksdb::AbstractLoggerJni::setHandle(env, jobj, pLoggerJniCallback);
|
|
}
|
|
|
|
/*
|
|
* Class: org_rocksdb_AbstractLogger
|
|
* Method: disposeInternal
|
|
* Signature: (J)V
|
|
*/
|
|
void Java_org_rocksdb_AbstractLogger_disposeInternal(
|
|
JNIEnv* env, jobject jobj, jlong jhandle) {
|
|
std::shared_ptr<rocksdb::LoggerJniCallback> *handle =
|
|
reinterpret_cast<std::shared_ptr<rocksdb::LoggerJniCallback> *>(jhandle);
|
|
handle->reset();
|
|
}
|
|
|