Optimize and minimize bound checks

Motivation:

We should minimize and optimize bound checks as much as possible to get the most out of performance.

Modifications:

- Use bitwise operations to remove branching
- Remove branches when possible

Result:

Better performance for various operations.
This commit is contained in:
Norman Maurer 2015-10-09 23:28:14 +02:00
parent 4403b1a7ea
commit af9dc2c6a6

View File

@ -232,9 +232,13 @@ public abstract class AbstractByteBuf extends ByteBuf {
throw new IllegalArgumentException(String.format(
"minWritableBytes: %d (expected: >= 0)", minWritableBytes));
}
ensureWritable0(minWritableBytes);
return this;
}
private void ensureWritable0(int minWritableBytes) {
if (minWritableBytes <= writableBytes()) {
return this;
return;
}
if (minWritableBytes > maxCapacity - writerIndex) {
@ -248,7 +252,6 @@ public abstract class AbstractByteBuf extends ByteBuf {
// Adjust to the new capacity.
capacity(newCapacity);
return this;
}
@Override
@ -543,21 +546,22 @@ public abstract class AbstractByteBuf extends ByteBuf {
int nLong = length >>> 3;
int nBytes = length & 7;
for (int i = nLong; i > 0; i --) {
setLong(index, 0);
_setLong(index, 0);
index += 8;
}
if (nBytes == 4) {
setInt(index, 0);
_setInt(index, 0);
// Not need to update the index as we not will use it after this.
} else if (nBytes < 4) {
for (int i = nBytes; i > 0; i --) {
setByte(index, (byte) 0);
_setByte(index, (byte) 0);
index ++;
}
} else {
setInt(index, 0);
_setInt(index, 0);
index += 4;
for (int i = nBytes - 4; i > 0; i --) {
setByte(index, (byte) 0);
_setByte(index, (byte) 0);
index ++;
}
}
@ -566,9 +570,9 @@ public abstract class AbstractByteBuf extends ByteBuf {
@Override
public byte readByte() {
checkReadableBytes(1);
checkReadableBytes0(1);
int i = readerIndex;
byte b = getByte(i);
byte b = _getByte(i);
readerIndex = i + 1;
return b;
}
@ -585,7 +589,7 @@ public abstract class AbstractByteBuf extends ByteBuf {
@Override
public short readShort() {
checkReadableBytes(2);
checkReadableBytes0(2);
short v = _getShort(readerIndex);
readerIndex += 2;
return v;
@ -607,7 +611,7 @@ public abstract class AbstractByteBuf extends ByteBuf {
@Override
public int readUnsignedMedium() {
checkReadableBytes(3);
checkReadableBytes0(3);
int v = _getUnsignedMedium(readerIndex);
readerIndex += 3;
return v;
@ -615,7 +619,7 @@ public abstract class AbstractByteBuf extends ByteBuf {
@Override
public int readInt() {
checkReadableBytes(4);
checkReadableBytes0(4);
int v = _getInt(readerIndex);
readerIndex += 4;
return v;
@ -628,7 +632,7 @@ public abstract class AbstractByteBuf extends ByteBuf {
@Override
public long readLong() {
checkReadableBytes(8);
checkReadableBytes0(8);
long v = _getLong(readerIndex);
readerIndex += 8;
return v;
@ -751,7 +755,7 @@ public abstract class AbstractByteBuf extends ByteBuf {
@Override
public ByteBuf writeByte(int value) {
ensureAccessible();
ensureWritable(1);
ensureWritable0(1);
_setByte(writerIndex++, value);
return this;
}
@ -759,7 +763,7 @@ public abstract class AbstractByteBuf extends ByteBuf {
@Override
public ByteBuf writeShort(int value) {
ensureAccessible();
ensureWritable(2);
ensureWritable0(2);
_setShort(writerIndex, value);
writerIndex += 2;
return this;
@ -768,7 +772,7 @@ public abstract class AbstractByteBuf extends ByteBuf {
@Override
public ByteBuf writeMedium(int value) {
ensureAccessible();
ensureWritable(3);
ensureWritable0(3);
_setMedium(writerIndex, value);
writerIndex += 3;
return this;
@ -777,7 +781,7 @@ public abstract class AbstractByteBuf extends ByteBuf {
@Override
public ByteBuf writeInt(int value) {
ensureAccessible();
ensureWritable(4);
ensureWritable0(4);
_setInt(writerIndex, value);
writerIndex += 4;
return this;
@ -786,7 +790,7 @@ public abstract class AbstractByteBuf extends ByteBuf {
@Override
public ByteBuf writeLong(long value) {
ensureAccessible();
ensureWritable(8);
ensureWritable0(8);
_setLong(writerIndex, value);
writerIndex += 8;
return this;
@ -891,25 +895,32 @@ public abstract class AbstractByteBuf extends ByteBuf {
}
ensureWritable(length);
checkIndex(writerIndex, length);
int wIndex = writerIndex;
checkIndex(wIndex, length);
int nLong = length >>> 3;
int nBytes = length & 7;
for (int i = nLong; i > 0; i --) {
writeLong(0);
_setLong(wIndex, 0);
wIndex += 8;
}
if (nBytes == 4) {
writeInt(0);
_setInt(wIndex, 0);
wIndex += 4;
} else if (nBytes < 4) {
for (int i = nBytes; i > 0; i --) {
writeByte((byte) 0);
_setByte(wIndex, (byte) 0);
wIndex++;
}
} else {
writeInt(0);
_setInt(wIndex, 0);
wIndex += 4;
for (int i = nBytes - 4; i > 0; i --) {
writeByte((byte) 0);
_setByte(wIndex, (byte) 0);
wIndex++;
}
}
writerIndex = wIndex;
return this;
}
@ -1117,19 +1128,12 @@ public abstract class AbstractByteBuf extends ByteBuf {
}
protected final void checkIndex(int index) {
ensureAccessible();
if (index < 0 || index >= capacity()) {
throw new IndexOutOfBoundsException(String.format(
"index: %d (expected: range(0, %d))", index, capacity()));
}
checkIndex(index, 1);
}
protected final void checkIndex(int index, int fieldLength) {
ensureAccessible();
if (fieldLength < 0) {
throw new IllegalArgumentException("length: " + fieldLength + " (expected: >= 0)");
}
if (index < 0 || index > capacity() - fieldLength) {
if (isInvalid(index, fieldLength, capacity())) {
throw new IndexOutOfBoundsException(String.format(
"index: %d, length: %d (expected: range(0, %d))", index, fieldLength, capacity()));
}
@ -1137,7 +1141,7 @@ public abstract class AbstractByteBuf extends ByteBuf {
protected final void checkSrcIndex(int index, int length, int srcIndex, int srcCapacity) {
checkIndex(index, length);
if (srcIndex < 0 || srcIndex > srcCapacity - length) {
if (isInvalid(srcIndex, length, srcCapacity)) {
throw new IndexOutOfBoundsException(String.format(
"srcIndex: %d, length: %d (expected: range(0, %d))", srcIndex, length, srcCapacity));
}
@ -1145,22 +1149,30 @@ public abstract class AbstractByteBuf extends ByteBuf {
protected final void checkDstIndex(int index, int length, int dstIndex, int dstCapacity) {
checkIndex(index, length);
if (dstIndex < 0 || dstIndex > dstCapacity - length) {
if (isInvalid(dstIndex, length, dstCapacity)) {
throw new IndexOutOfBoundsException(String.format(
"dstIndex: %d, length: %d (expected: range(0, %d))", dstIndex, length, dstCapacity));
}
}
private static boolean isInvalid(int index, int length, int capacity) {
return (index | length | (index + length) | (capacity - (index + length))) < 0;
}
/**
* Throws an {@link IndexOutOfBoundsException} if the current
* {@linkplain #readableBytes() readable bytes} of this buffer is less
* than the specified value.
*/
protected final void checkReadableBytes(int minimumReadableBytes) {
ensureAccessible();
if (minimumReadableBytes < 0) {
throw new IllegalArgumentException("minimumReadableBytes: " + minimumReadableBytes + " (expected: >= 0)");
}
checkReadableBytes0(minimumReadableBytes);
}
private void checkReadableBytes0(int minimumReadableBytes) {
ensureAccessible();
if (readerIndex > writerIndex - minimumReadableBytes) {
throw new IndexOutOfBoundsException(String.format(
"readerIndex(%d) + length(%d) exceeds writerIndex(%d): %s",