System.out.println(String.format("Took %f s to connect", (float)(C-B) / 1E9));
System.out.println(String.format("Took %f s to run query", (float)(D-C) / 1E9));
//List<QueryResultBatch> results = client.runQuery(UserProtos.QueryType.PHYSICAL, Files.toString(FileUtils.getResourceAsFile("/parquet_scan_union_screen_physical.json"), Charsets.UTF_8));
int count = 0;
// RecordBatchLoader batchLoader = new RecordBatchLoader(new BootStrapContext(config).getAllocator());
RecordBatchLoader batchLoader = new RecordBatchLoader(bit1.getContext().getAllocator());
byte[] bytes;
int batchCounter = 1;
int columnValCounter = 0;
int i = 0;
FieldInfo currentField;
HashMap<String, Integer> valuesChecked = new HashMap();
for(QueryResultBatch b : results){
count += b.getHeader().getRowCount();
boolean schemaChanged = batchLoader.load(b.getHeader().getDef(), b.getData());
int recordCount = 0;
// print headers.
if (schemaChanged) {
} // do not believe any change is needed for when the schema changes, with the current mock scan use case
for (VectorWrapper vw : batchLoader) {
ValueVector vv = vw.getValueVector();
currentField = props.fields.get(vv.getField().getName());
if (VERBOSE_DEBUG){
System.out.println("\n" + (String) currentField.name);
}
if ( ! valuesChecked.containsKey(vv.getField().getName())){
valuesChecked.put(vv.getField().getName(), 0);
columnValCounter = 0;
} else {
columnValCounter = valuesChecked.get(vv.getField().getName());
}
for (int j = 0; j < vv.getAccessor().getValueCount(); j++) {
if (VERBOSE_DEBUG){
System.out.print(vv.getAccessor().getObject(j) + ", " + (j % 25 == 0 ? "\n batch:" + batchCounter + " v:" + j + " - " : ""));
}
assertField(vv, j, currentField.type,
currentField.values[columnValCounter % 3], currentField.name + "/");
columnValCounter++;
}
if (VERBOSE_DEBUG){
System.out.println("\n" + vv.getAccessor().getValueCount());
}
valuesChecked.remove(vv.getField().getName());
valuesChecked.put(vv.getField().getName(), columnValCounter);
}
if (VERBOSE_DEBUG){
for (i = 0; i < batchLoader.getRecordCount(); i++) {
recordCount++;
if (i % 50 == 0){
System.out.println();
for (VectorWrapper vw : batchLoader) {
ValueVector v = vw.getValueVector();