Added SlicedAbstractByteBuf that can provide fast-path for _get* and _set* methods

Motivation:

SlicedByteBuf can be used for any ByteBuf implementations and so can not do any optimizations that could be done
when AbstractByteBuf is sliced.

Modifications:

- Add SlicedAbstractByteBuf that can eliminate range and reference count checks for _get* and _set* methods.

Result:

Faster SlicedByteBuf implementations for AbstractByteBuf sub-classes.
This commit is contained in:
Norman Maurer 2015-10-08 14:49:31 +02:00
parent b8c73a4806
commit 291674262c
4 changed files with 192 additions and 28 deletions

View File

@ -954,7 +954,7 @@ public abstract class AbstractByteBuf extends ByteBuf {
@Override
public ByteBuf slice(int index, int length) {
return new SlicedByteBuf(this, index, length);
return new SlicedAbstractByteBuf(this, index, length);
}
@Override

View File

@ -0,0 +1,81 @@
/*
* Copyright 2015 The Netty Project
*
* The Netty Project licenses this file to you 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 io.netty.buffer;
/**
* A special {@link SlicedByteBuf} that can make optimizations because it knows the sliced buffer is of type
* {@link AbstractByteBuf}.
*/
final class SlicedAbstractByteBuf extends SlicedByteBuf {
SlicedAbstractByteBuf(AbstractByteBuf buffer, int index, int length) {
super(buffer, index, length);
}
@Override
protected byte _getByte(int index) {
return unwrap0()._getByte(idx(index));
}
@Override
protected short _getShort(int index) {
return unwrap0()._getShort(idx(index));
}
@Override
protected int _getUnsignedMedium(int index) {
return unwrap0()._getUnsignedMedium(idx(index));
}
@Override
protected int _getInt(int index) {
return unwrap0()._getInt(idx(index));
}
@Override
protected long _getLong(int index) {
return unwrap0()._getLong(idx(index));
}
@Override
protected void _setByte(int index, int value) {
unwrap0()._setByte(idx(index), value);
}
@Override
protected void _setShort(int index, int value) {
unwrap0()._setShort(idx(index), value);
}
@Override
protected void _setMedium(int index, int value) {
unwrap0()._setMedium(idx(index), value);
}
@Override
protected void _setInt(int index, int value) {
unwrap0()._setInt(idx(index), value);
}
@Override
protected void _setLong(int index, long value) {
unwrap0()._setLong(idx(index), value);
}
private AbstractByteBuf unwrap0() {
return (AbstractByteBuf) unwrap();
}
}

View File

@ -99,7 +99,7 @@ public class SlicedByteBuf extends AbstractDerivedByteBuf {
@Override
public int arrayOffset() {
return buffer.arrayOffset() + adjustment;
return idx(buffer.arrayOffset());
}
@Override
@ -114,27 +114,27 @@ public class SlicedByteBuf extends AbstractDerivedByteBuf {
@Override
protected byte _getByte(int index) {
return buffer.getByte(index + adjustment);
return buffer.getByte(idx(index));
}
@Override
protected short _getShort(int index) {
return buffer.getShort(index + adjustment);
return buffer.getShort(idx(index));
}
@Override
protected int _getUnsignedMedium(int index) {
return buffer.getUnsignedMedium(index + adjustment);
return buffer.getUnsignedMedium(idx(index));
}
@Override
protected int _getInt(int index) {
return buffer.getInt(index + adjustment);
return buffer.getInt(idx(index));
}
@Override
protected long _getLong(int index) {
return buffer.getLong(index + adjustment);
return buffer.getLong(idx(index));
}
@Override
@ -147,7 +147,7 @@ public class SlicedByteBuf extends AbstractDerivedByteBuf {
@Override
public ByteBuf copy(int index, int length) {
checkIndex(index, length);
return buffer.copy(index + adjustment, length);
return buffer.copy(idx(index), length);
}
@Override
@ -156,99 +156,99 @@ public class SlicedByteBuf extends AbstractDerivedByteBuf {
if (length == 0) {
return Unpooled.EMPTY_BUFFER;
}
return buffer.slice(index + adjustment, length);
return buffer.slice(idx(index), length);
}
@Override
public ByteBuf getBytes(int index, ByteBuf dst, int dstIndex, int length) {
checkIndex(index, length);
buffer.getBytes(index + adjustment, dst, dstIndex, length);
buffer.getBytes(idx(index), dst, dstIndex, length);
return this;
}
@Override
public ByteBuf getBytes(int index, byte[] dst, int dstIndex, int length) {
checkIndex(index, length);
buffer.getBytes(index + adjustment, dst, dstIndex, length);
buffer.getBytes(idx(index), dst, dstIndex, length);
return this;
}
@Override
public ByteBuf getBytes(int index, ByteBuffer dst) {
checkIndex(index, dst.remaining());
buffer.getBytes(index + adjustment, dst);
buffer.getBytes(idx(index), dst);
return this;
}
@Override
protected void _setByte(int index, int value) {
buffer.setByte(index + adjustment, value);
buffer.setByte(idx(index), value);
}
@Override
protected void _setShort(int index, int value) {
buffer.setShort(index + adjustment, value);
buffer.setShort(idx(index), value);
}
@Override
protected void _setMedium(int index, int value) {
buffer.setMedium(index + adjustment, value);
buffer.setMedium(idx(index), value);
}
@Override
protected void _setInt(int index, int value) {
buffer.setInt(index + adjustment, value);
buffer.setInt(idx(index), value);
}
@Override
protected void _setLong(int index, long value) {
buffer.setLong(index + adjustment, value);
buffer.setLong(idx(index), value);
}
@Override
public ByteBuf setBytes(int index, byte[] src, int srcIndex, int length) {
checkIndex(index, length);
buffer.setBytes(index + adjustment, src, srcIndex, length);
buffer.setBytes(idx(index), src, srcIndex, length);
return this;
}
@Override
public ByteBuf setBytes(int index, ByteBuf src, int srcIndex, int length) {
checkIndex(index, length);
buffer.setBytes(index + adjustment, src, srcIndex, length);
buffer.setBytes(idx(index), src, srcIndex, length);
return this;
}
@Override
public ByteBuf setBytes(int index, ByteBuffer src) {
checkIndex(index, src.remaining());
buffer.setBytes(index + adjustment, src);
buffer.setBytes(idx(index), src);
return this;
}
@Override
public ByteBuf getBytes(int index, OutputStream out, int length) throws IOException {
checkIndex(index, length);
buffer.getBytes(index + adjustment, out, length);
buffer.getBytes(idx(index), out, length);
return this;
}
@Override
public int getBytes(int index, GatheringByteChannel out, int length) throws IOException {
checkIndex(index, length);
return buffer.getBytes(index + adjustment, out, length);
return buffer.getBytes(idx(index), out, length);
}
@Override
public int setBytes(int index, InputStream in, int length) throws IOException {
checkIndex(index, length);
return buffer.setBytes(index + adjustment, in, length);
return buffer.setBytes(idx(index), in, length);
}
@Override
public int setBytes(int index, ScatteringByteChannel in, int length) throws IOException {
checkIndex(index, length);
return buffer.setBytes(index + adjustment, in, length);
return buffer.setBytes(idx(index), in, length);
}
@Override
@ -259,13 +259,13 @@ public class SlicedByteBuf extends AbstractDerivedByteBuf {
@Override
public ByteBuffer nioBuffer(int index, int length) {
checkIndex(index, length);
return buffer.nioBuffer(index + adjustment, length);
return buffer.nioBuffer(idx(index), length);
}
@Override
public ByteBuffer[] nioBuffers(int index, int length) {
checkIndex(index, length);
return buffer.nioBuffers(index + adjustment, length);
return buffer.nioBuffers(idx(index), length);
}
@Override
@ -276,7 +276,7 @@ public class SlicedByteBuf extends AbstractDerivedByteBuf {
@Override
public int forEachByte(int index, int length, ByteBufProcessor processor) {
int ret = buffer.forEachByte(index + adjustment, length, processor);
int ret = buffer.forEachByte(idx(index), length, processor);
if (ret >= adjustment) {
return ret - adjustment;
} else {
@ -286,11 +286,18 @@ public class SlicedByteBuf extends AbstractDerivedByteBuf {
@Override
public int forEachByteDesc(int index, int length, ByteBufProcessor processor) {
int ret = buffer.forEachByteDesc(index + adjustment, length, processor);
int ret = buffer.forEachByteDesc(idx(index), length, processor);
if (ret >= adjustment) {
return ret - adjustment;
} else {
return -1;
}
}
/**
* Returns the index with the needed adjustment.
*/
final int idx(int index) {
return index + adjustment;
}
}

View File

@ -0,0 +1,76 @@
/*
* Copyright 2015 The Netty Project
*
* The Netty Project licenses this file to you 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 io.netty.microbench.buffer;
import io.netty.buffer.ByteBuf;
import io.netty.buffer.ByteBufUtil;
import io.netty.buffer.SlicedByteBuf;
import io.netty.buffer.Unpooled;
import io.netty.microbench.util.AbstractMicrobenchmark;
import org.openjdk.jmh.annotations.Benchmark;
import org.openjdk.jmh.annotations.Measurement;
import org.openjdk.jmh.annotations.Scope;
import org.openjdk.jmh.annotations.Setup;
import org.openjdk.jmh.annotations.State;
import org.openjdk.jmh.annotations.TearDown;
import org.openjdk.jmh.annotations.Warmup;
@State(Scope.Benchmark)
@Warmup(iterations = 10)
@Measurement(iterations = 25)
public class SlicedByteBufBenchmark extends AbstractMicrobenchmark {
private ByteBuf slicedByteBuf;
private ByteBuf slicedAbstractByteBuf;
private String ascii;
@Setup
public void setup() {
// Use buffer sizes that will also allow to write UTF-8 without grow the buffer
ByteBuf buffer = Unpooled.buffer(512).retain();
slicedByteBuf = new SlicedByteBuf(buffer, 0, 256);
slicedAbstractByteBuf = buffer.slice(0, 256);
if (slicedByteBuf.getClass() == slicedAbstractByteBuf.getClass()) {
throw new IllegalStateException();
}
StringBuilder asciiSequence = new StringBuilder(128);
for (int i = 0; i < 128; i++) {
asciiSequence.append('a');
}
ascii = asciiSequence.toString();
}
@TearDown
public void tearDown() {
slicedByteBuf.release();
slicedAbstractByteBuf.release();
}
@Benchmark
public void writeAsciiStringSlice() {
slicedByteBuf.resetWriterIndex();
ByteBufUtil.writeAscii(slicedByteBuf, ascii);
}
@Benchmark
public void writeAsciiStringSliceAbstract() {
slicedAbstractByteBuf.resetWriterIndex();
ByteBufUtil.writeAscii(slicedAbstractByteBuf, ascii);
}
}