org.lmdbjava.ByteArrayProxy Maven / Gradle / Ivy

Go to download
/*-
 * #%L
 * LmdbJava
 * %%
 * Copyright (C) 2016 - 2020 The LmdbJava Open Source Project
 * %%
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 *      http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 * #L%
 */

package org.lmdbjava;

import static java.util.Objects.requireNonNull;
import static org.lmdbjava.Library.RUNTIME;

import java.util.Arrays;

import jnr.ffi.Pointer;
import jnr.ffi.provider.MemoryManager;

/**
 * Byte array proxy.
 *
 * {@link Env#create(org.lmdbjava.BufferProxy)}.
 */
public final class ByteArrayProxy extends BufferProxy {

  /**
   * The byte array proxy. Guaranteed to never be null.
   */
  public static final BufferProxy PROXY_BA = new ByteArrayProxy();

  private static final MemoryManager MEM_MGR = RUNTIME.getMemoryManager();

  private ByteArrayProxy() {
  }

  /**
   * Lexicographically compare two byte arrays.
   *
   * @param o1 left operand (required)
   * @param o2 right operand (required)
   * @return as specified by {@link Comparable} interface
   */
  public static int compareArrays(final byte[] o1, final byte[] o2) {
    requireNonNull(o1);
    requireNonNull(o2);
    if (o1 == o2) {
      return 0;
    }
    final int minLength = Math.min(o1.length, o2.length);

    for (int i = 0; i < minLength; i++) {
      final int lw = Byte.toUnsignedInt(o1[i]);
      final int rw = Byte.toUnsignedInt(o2[i]);
      final int result = Integer.compareUnsigned(lw, rw);
      if (result != 0) {
        return result;
      }
    }

    return o1.length - o2.length;
  }

  @Override
  protected byte[] allocate() {
    return new byte[0];
  }

  @Override
  protected int compare(final byte[] o1, final byte[] o2) {
    return compareArrays(o1, o2);
  }

  @Override
  protected void deallocate(final byte[] buff) {
    // byte arrays cannot be allocated
  }

  @Override
  protected byte[] getBytes(final byte[] buffer) {
    return Arrays.copyOf(buffer, buffer.length);
  }

  @Override
  protected void in(final byte[] buffer, final Pointer ptr,
                    final long ptrAddr) {
    final Pointer pointer = MEM_MGR.allocateDirect(buffer.length);
    pointer.put(0, buffer, 0, buffer.length);
    ptr.putLong(STRUCT_FIELD_OFFSET_SIZE, buffer.length);
    ptr.putAddress(STRUCT_FIELD_OFFSET_DATA, pointer.address());
  }

  @Override
  protected void in(final byte[] buffer, final int size, final Pointer ptr,
                    final long ptrAddr) {
    // cannot reserve for byte arrays
  }

  @Override
  protected byte[] out(final byte[] buffer, final Pointer ptr,
                       final long ptrAddr) {
    final long addr = ptr.getAddress(STRUCT_FIELD_OFFSET_DATA);
    final int size = (int) ptr.getLong(STRUCT_FIELD_OFFSET_SIZE);
    final Pointer pointer = MEM_MGR.newPointer(addr, size);
    final byte[] bytes = new byte[size];
    pointer.get(0, bytes, 0, size);
    return bytes;
  }
}