org.forstdb.AbstractTraceWriter Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of forstjni Show documentation

ForSt fat jar with modifications specific for Apache Flink that contains .so files for linux32 and linux64 (glibc and musl-libc), jnilib files for Mac OSX, and a .dll for Windows x64.

The newest version!

// 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).

package org.forstdb;

/**
 * Base class for TraceWriters.
 */
public abstract class AbstractTraceWriter
    extends RocksCallbackObject implements TraceWriter {

  @Override
  protected long initializeNative(final long... nativeParameterHandles) {
    return createNewTraceWriter();
  }

  /**
   * Called from JNI, proxy for {@link TraceWriter#write(Slice)}.
   *
   * @param sliceHandle the native handle of the slice (which we do not own)
   *
   * @return short (2 bytes) where the first byte is the
   *     {@link Status.Code#getValue()} and the second byte is the
   *     {@link Status.SubCode#getValue()}.
   */
  @SuppressWarnings("PMD.UnusedPrivateMethod")
  private short writeProxy(final long sliceHandle) {
    try {
      write(new Slice(sliceHandle));
      return statusToShort(Status.Code.Ok, Status.SubCode.None);
    } catch (final RocksDBException e) {
      return statusToShort(e.getStatus());
    }
  }

  /**
   * Called from JNI, proxy for {@link TraceWriter#closeWriter()}.
   *
   * @return short (2 bytes) where the first byte is the
   *     {@link Status.Code#getValue()} and the second byte is the
   *     {@link Status.SubCode#getValue()}.
   */
  @SuppressWarnings("PMD.UnusedPrivateMethod")
  private short closeWriterProxy() {
    try {
      closeWriter();
      return statusToShort(Status.Code.Ok, Status.SubCode.None);
    } catch (final RocksDBException e) {
      return statusToShort(e.getStatus());
    }
  }

  private static short statusToShort(/*@Nullable*/ final Status status) {
    final Status.Code code = status != null && status.getCode() != null
        ? status.getCode()
        : Status.Code.IOError;
    final Status.SubCode subCode = status != null && status.getSubCode() != null
        ? status.getSubCode()
        : Status.SubCode.None;
    return statusToShort(code, subCode);
  }

  private static short statusToShort(final Status.Code code,
      final Status.SubCode subCode) {
    final short result = (short) (code.getValue() << 8);
    return (short)(result | subCode.getValue());
  }

  private native long createNewTraceWriter();
}