ByteBufNIO.java

/*
 * Copyright 2008-present MongoDB, Inc.
 *
 * 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.
 */

package org.bson;

import java.nio.Buffer;
import java.nio.ByteBuffer;
import java.nio.ByteOrder;
import java.util.concurrent.atomic.AtomicInteger;

/**
 * Implementation of {@code ByteBuf} which simply wraps an NIO {@code ByteBuffer} and forwards all calls to it.
 *
 * @since 3.0
 */
public class ByteBufNIO implements ByteBuf {
    private ByteBuffer buf;
    private final AtomicInteger referenceCount = new AtomicInteger(1);

    /**
     * Creates a new instance.
     *
     * @param buf the {@code ByteBuffer} to wrap.
     */
    public ByteBufNIO(final ByteBuffer buf) {
        this.buf = buf.order(ByteOrder.LITTLE_ENDIAN);
    }

    @Override
    public int getReferenceCount() {
        return referenceCount.get();
    }

    @Override
    public ByteBufNIO retain() {
        if (referenceCount.incrementAndGet() == 1) {
            referenceCount.decrementAndGet();
            throw new IllegalStateException("Attempted to increment the reference count when it is already 0");
        }
        return this;
    }

    @Override
    public void release() {
        if (referenceCount.decrementAndGet() < 0) {
            referenceCount.incrementAndGet();
            throw new IllegalStateException("Attempted to decrement the reference count below 0");
        }
        if (referenceCount.get() == 0) {
            buf = null;
        }
    }

    @Override
    public int capacity() {
        return buf.capacity();
    }

    @Override
    public ByteBuf put(final int index, final byte b) {
        buf.put(index, b);
        return this;
    }

    @Override
    public int remaining() {
        return buf.remaining();
    }

    @Override
    public ByteBuf put(final byte[] src, final int offset, final int length) {
        buf.put(src, offset, length);
        return this;
    }

    @Override
    public boolean hasRemaining() {
        return buf.hasRemaining();
    }

    @Override
    public ByteBuf put(final byte b) {
        buf.put(b);
        return this;
    }

    @Override
    public ByteBuf flip() {
        ((Buffer) buf).flip();
        return this;
    }

    @Override
    public byte[] array() {
        return buf.array();
    }

    @Override
    public boolean hasArray() {
        return buf.hasArray();
    }

    @Override
    public int arrayOffset() {
        return buf.arrayOffset();
    }

    @Override
    public int limit() {
        return buf.limit();
    }

    @Override
    public ByteBuf position(final int newPosition) {
        ((Buffer) buf).position(newPosition);
        return this;
    }

    @Override
    public ByteBuf clear() {
        ((Buffer) buf).clear();
        return this;
    }

    @Override
    public ByteBuf order(final ByteOrder byteOrder) {
        buf.order(byteOrder);
        return this;
    }

    @Override
    public byte get() {
        return buf.get();
    }

    @Override
    public byte get(final int index) {
        return buf.get(index);
    }

    @Override
    public ByteBuf get(final byte[] bytes) {
        buf.get(bytes);
        return this;
    }

    @Override
    public ByteBuf get(final int index, final byte[] bytes) {
        return get(index, bytes, 0, bytes.length);
    }

    @Override
    public ByteBuf get(final byte[] bytes, final int offset, final int length) {
        buf.get(bytes, offset, length);
        return this;
    }

    @Override
    public ByteBuf get(final int index, final byte[] bytes, final int offset, final int length) {
        for (int i = 0; i < length; i++) {
            bytes[offset + i] = buf.get(index + i);
        }
        return this;
    }

    @Override
    public long getLong() {
        return buf.getLong();
    }

    @Override
    public long getLong(final int index) {
        return buf.getLong(index);
    }

    @Override
    public double getDouble() {
        return buf.getDouble();
    }

    @Override
    public double getDouble(final int index) {
        return buf.getDouble(index);
    }

    @Override
    public int getInt() {
        return buf.getInt();
    }

    @Override
    public int getInt(final int index) {
        return buf.getInt(index);
    }

    @Override
    public int position() {
        return buf.position();
    }

    @Override
    public ByteBuf limit(final int newLimit) {
        ((Buffer) buf).limit(newLimit);
        return this;
    }

    @Override
    public ByteBuf asReadOnly() {
        return new ByteBufNIO(buf.asReadOnlyBuffer());
    }

    @Override
    public ByteBuf duplicate() {
        return new ByteBufNIO(buf.duplicate());
    }

    @Override
    public ByteBuffer asNIO() {
        return buf;
    }
}