package org.bson;
import java.nio.Buffer;
import java.nio.ByteBuffer;
import java.nio.ByteOrder;
import java.util.concurrent.atomic.AtomicInteger;
public class ByteBufNIO implements ByteBuf {
private ByteBuffer buf;
private final AtomicInteger referenceCount = new AtomicInteger(1);
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 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;
}
}