@Override public void allocate() { vector.allocateNew(); }
public void loadData(SerializedField metadata, ArrowBuf buffer) { /* clear the current buffers (if any) */ vector.clear(); /* get the metadata children */ final SerializedField bitsField = metadata.getChild(0); final SerializedField valuesField = metadata.getChild(1); final int valuesLength = buffer.capacity(); vector.allocateNew(metadata.getValueCount()); /* set inner validity buffer */ setValidityBuffer(bitsField); /* load inner value buffer */ vector.valueBuffer.close(); loadDataBuffer(valuesField, buffer.slice(0, valuesLength)); }
private static Pair<SmallIntVector, ResultVerifier> testSmallIntVector(final int startIndexInCurrentOutput, final int startIndexInJob) { SmallIntVector colSmallIntV = new SmallIntVector("colSmallInt", allocator); colSmallIntV.allocateNew(5); colSmallIntV.set(0, 20); colSmallIntV.setNull(1); colSmallIntV.set(2, -2000); colSmallIntV.set(3, 32700); colSmallIntV.set(4, 0); ResultVerifier verifier = new ResultVerifier() { @Override public void verify(DataPOJO output) { int index = startIndexInCurrentOutput; assertEquals(20, ((Integer)output.extractValue("colSmallInt", 0)).intValue()); assertNull(output.extractValue("colSmallInt", 1)); assertEquals(-2000, ((Integer)output.extractValue("colSmallInt", 2)).intValue()); assertEquals(32700, ((Integer)output.extractValue("colSmallInt", 3)).intValue()); assertEquals(0, ((Integer)output.extractValue("colSmallInt", 4)).intValue()); } }; return Pair.of(colSmallIntV, verifier); }