summaryrefslogtreecommitdiffstats
path: root/storage/rocksdb/rocksdb/java/rocksjni/trace_writer_jnicallback.h
blob: eb2a8b0f8061a3cb7df6e4100940d5baefd3edb9 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
//  Copyright (c) 2011-present, Facebook, Inc.  All rights reserved.
//  This source code is licensed under both the GPLv2 (found in the
//  COPYING file in the root directory) and Apache 2.0 License
//  (found in the LICENSE.Apache file in the root directory).
//
// This file implements the callback "bridge" between Java and C++ for
// ROCKSDB_NAMESPACE::TraceWriter.

#ifndef JAVA_ROCKSJNI_TRACE_WRITER_JNICALLBACK_H_
#define JAVA_ROCKSJNI_TRACE_WRITER_JNICALLBACK_H_

#include <jni.h>
#include <memory>

#include "rocksdb/trace_reader_writer.h"
#include "rocksjni/jnicallback.h"

namespace ROCKSDB_NAMESPACE {

class TraceWriterJniCallback : public JniCallback, public TraceWriter {
 public:
    TraceWriterJniCallback(
        JNIEnv* env, jobject jtrace_writer);
    virtual Status Write(const Slice& data);
    virtual Status Close();
    virtual uint64_t GetFileSize();

 private:
    jmethodID m_jwrite_proxy_methodid;
    jmethodID m_jclose_writer_proxy_methodid;
    jmethodID m_jget_file_size_methodid;
};

}  // namespace ROCKSDB_NAMESPACE

#endif  // JAVA_ROCKSJNI_TRACE_WRITER_JNICALLBACK_H_