Fixed DataBufferUtils.takeUntilByteCount

This commit is contained in:
Arjen Poutsma
2016-04-22 13:01:18 +02:00
parent 52c9b3b235
commit 9ed4bff4a9

View File

@@ -139,10 +139,10 @@ public abstract class DataBufferUtils {
@Override
public void onNext(DataBuffer dataBuffer) {
long currentCount =
this.byteCount.addAndGet(dataBuffer.readableByteCount());
int delta = dataBuffer.readableByteCount();
long currentCount = this.byteCount.addAndGet(delta);
if (currentCount > maxByteCount) {
int size = (int) (currentCount - maxByteCount + 1);
int size = (int) (maxByteCount - currentCount + delta);
ByteBuffer byteBuffer =
(ByteBuffer) dataBuffer.asByteBuffer().limit(size);
DataBuffer partialBuffer =