public void retainBuffers(final int increment) { for (final ArrowBuf buf : buffers) { buf.retain(increment); } }
public RawFragmentBatch(FragmentRecordBatch header, ArrowBuf body, AckSender sender) { this.header = header; this.sender = sender; this.body = body; if (body != null) { body.retain(1); } // ACK has been sent when the batch was spilled ackSent.set(sender == null); }
public void setBuffer(ArrowBuf bufferHandle) { /* clear the existing buffer */ clear(); this.buffer = bufferHandle; buffer.retain(1); }
public void loadDataBuffer(SerializedField metadata, ArrowBuf buffer) { final int actualLength = metadata.getBufferLength(); final int valueCount = metadata.getValueCount(); final int expectedLength = valueCount * 8; assert actualLength == expectedLength : String.format("Expected to load %d bytes but actually loaded %d bytes in data buffer", expectedLength, actualLength); vector.valueBuffer = buffer.slice(0, actualLength); vector.valueBuffer.retain(1); vector.valueBuffer.writerIndex(actualLength); } }
public void loadDataBuffer(SerializedField metadata, ArrowBuf buffer) { final int actualLength = metadata.getBufferLength(); final int valueCount = metadata.getValueCount(); final int expectedLength = valueCount * 4; assert actualLength == expectedLength : String.format("Expected to load %d bytes but actually loaded %d bytes in data buffer", expectedLength, actualLength); vector.valueBuffer = buffer.slice(0, actualLength); vector.valueBuffer.retain(1); vector.valueBuffer.writerIndex(actualLength); } }
public void loadDataBuffer(SerializedField metadata, ArrowBuf buffer) { final int actualLength = metadata.getBufferLength(); final int valueCount = metadata.getValueCount(); final int expectedLength = valueCount * 4; assert actualLength == expectedLength : String.format("Expected to load %d bytes but actually loaded %d bytes in data buffer", expectedLength, actualLength); vector.valueBuffer = buffer.slice(0, actualLength); vector.valueBuffer.retain(1); vector.valueBuffer.writerIndex(actualLength); } }
protected void loadOffsetBuffer(SerializedField metadata, ArrowBuf buffer) { final int valueCount = metadata.getValueCount(); final int actualLength = metadata.getBufferLength(); final int expectedLength = valueCount * BaseRepeatedValueVector.OFFSET_WIDTH; assert actualLength == expectedLength : String.format("Expected to load %d bytes in offset buffer but actually loaded %d bytes", expectedLength, actualLength); vector.offsetBuffer = buffer.slice(0, actualLength); vector.offsetBuffer.writerIndex(actualLength); vector.offsetBuffer.retain(1); }
public void loadDataBuffer(SerializedField metadata, ArrowBuf buffer) { final int actualLength = metadata.getBufferLength(); final int valueCount = metadata.getValueCount(); final int expectedLength = valueCount * -1; assert actualLength == expectedLength : String.format("Expected to load %d bytes but actually loaded %d bytes in data buffer", expectedLength, actualLength); vector.valueBuffer = buffer.slice(0, actualLength); vector.valueBuffer.retain(1); vector.valueBuffer.writerIndex(actualLength); } }
public void loadDataBuffer(SerializedField metadata, ArrowBuf buffer) { final int actualLength = metadata.getBufferLength(); final int valueCount = metadata.getValueCount(); final int expectedLength = valueCount * 2; assert actualLength == expectedLength : String.format("Expected to load %d bytes but actually loaded %d bytes in data buffer", expectedLength, actualLength); vector.valueBuffer = buffer.slice(0, actualLength); vector.valueBuffer.retain(1); vector.valueBuffer.writerIndex(actualLength); } }
public void loadDataBuffer(SerializedField metadata, ArrowBuf buffer) { final int actualLength = metadata.getBufferLength(); final int valueCount = metadata.getValueCount(); final int expectedLength = valueCount * 8; assert actualLength == expectedLength : String.format("Expected to load %d bytes but actually loaded %d bytes in data buffer", expectedLength, actualLength); vector.valueBuffer = buffer.slice(0, actualLength); vector.valueBuffer.retain(1); vector.valueBuffer.writerIndex(actualLength); } }
private void loadValidityBuffer(SerializedField metadata, ArrowBuf buffer) { final int valueCount = metadata.getValueCount(); final int actualLength = metadata.getBufferLength(); final int expectedLength = getValidityBufferSizeFromCount(valueCount); assert expectedLength == actualLength: String.format("Expected to load %d bytes but actually loaded %d bytes in validity buffer", expectedLength, actualLength); vector.validityBuffer = buffer.slice(0, actualLength); vector.validityBuffer.writerIndex(actualLength); vector.validityBuffer.retain(1); }
public void loadDataBuffer(SerializedField metadata, ArrowBuf buffer) { final int actualLength = metadata.getBufferLength(); final int valueCount = metadata.getValueCount(); final int expectedLength = getValidityBufferSizeFromCount(valueCount); assert actualLength == expectedLength : String.format("Expected to load %d bytes but actually loaded %d bytes in data buffer", expectedLength, actualLength); vector.valueBuffer = buffer.slice(0, actualLength); vector.valueBuffer.retain(1); vector.valueBuffer.writerIndex(actualLength); } }
private void loadValidityBuffer(SerializedField metadata, ArrowBuf buffer) { final int valueCount = metadata.getValueCount(); final int actualLength = metadata.getBufferLength(); final int expectedLength = getValidityBufferSizeFromCount(valueCount); assert expectedLength == actualLength: String.format("Expected to load %d bytes but actually loaded %d bytes in validity buffer", expectedLength, actualLength); vector.validityBuffer = buffer.slice(0, actualLength); vector.validityBuffer.writerIndex(actualLength); vector.validityBuffer.retain(1); }
private void loadValidityBuffer(SerializedField metadata, ArrowBuf buffer) { final int valueCount = metadata.getValueCount(); final int actualLength = metadata.getBufferLength(); final int expectedLength = getValidityBufferSizeFromCount(valueCount); assert expectedLength == actualLength: String.format("Expected to load %d bytes but actually loaded %d bytes in validity buffer", expectedLength, actualLength); vector.validityBuffer = buffer.slice(0, actualLength); vector.validityBuffer.writerIndex(actualLength); vector.validityBuffer.retain(1); }
private void loadValidityBuffer(SerializedField metadata, ArrowBuf buffer) { final int valueCount = metadata.getValueCount(); final int actualLength = metadata.getBufferLength(); final int expectedLength = getValidityBufferSizeFromCount(valueCount); assert expectedLength == actualLength: String.format("Expected to load %d bytes but actually loaded %d bytes in validity buffer", expectedLength, actualLength); vector.validityBuffer = buffer.slice(0, actualLength); vector.validityBuffer.writerIndex(actualLength); vector.validityBuffer.retain(1); }
private void loadValidityBuffer(SerializedField metadata, ArrowBuf buffer) { final int valueCount = metadata.getValueCount(); final int actualLength = metadata.getBufferLength(); final int expectedLength = getValidityBufferSizeFromCount(valueCount); assert expectedLength == actualLength: String.format("Expected to load %d bytes but actually loaded %d bytes in validity buffer", expectedLength, actualLength); vector.validityBuffer = buffer.slice(0, actualLength); vector.validityBuffer.writerIndex(actualLength); vector.validityBuffer.retain(1); }
private void loadValidityBuffer(SerializedField metadata, ArrowBuf buffer) { final int valueCount = metadata.getValueCount(); final int actualLength = metadata.getBufferLength(); final int expectedLength = getValidityBufferSizeFromCount(valueCount); assert expectedLength == actualLength: String.format("Expected to load %d bytes but actually loaded %d bytes in validity buffer", expectedLength, actualLength); vector.validityBuffer = buffer.slice(0, actualLength); vector.validityBuffer.writerIndex(actualLength); vector.validityBuffer.retain(1); }
private void loadValidityBuffer(SerializedField metadata, ArrowBuf buffer) { final int valueCount = metadata.getValueCount(); final int actualLength = metadata.getBufferLength(); final int expectedLength = getValidityBufferSizeFromCount(valueCount); assert expectedLength == actualLength: String.format("Expected to load %d bytes but actually loaded %d bytes in validity buffer", expectedLength, actualLength); vector.validityBuffer = buffer.slice(0, actualLength); vector.validityBuffer.writerIndex(actualLength); vector.validityBuffer.retain(1); }
public void load(SerializedField metadata, ArrowBuf buffer) { Preconditions.checkArgument(vector.name.equals(metadata.getNamePart().getName()), "The field %s doesn't match the provided metadata %s.", vector.name, metadata); final int valueCount = metadata.getValueCount(); final int expectedLength = vector.getValidityBufferSizeFromCount(valueCount); final int actualLength = metadata.getBufferLength(); assert expectedLength == actualLength: "expected and actual buffer sizes do not match"; vector.clear(); vector.valueBuffer = buffer.slice(0, actualLength); vector.valueBuffer.writerIndex(actualLength); vector.valueBuffer.retain(); vector.valueCount = valueCount; }
@Override public SentResponseMessage<QueryId> handle(NodeEndpoint request, ArrowBuf dBody) throws RpcException { if (!request.equals(expectedD)) { throw new RpcException("Objects not same."); } assertEqualsRpc(r1B, dBody); random2.retain(); return new SentResponseMessage<>(expectedQ, random2); } });