@@ -448,8 +448,7 @@ public void init() throws Throwable {
448448 // TODO(SPARK-40059): Allow users to include columns named
449449 // FileFormat.ROW_INDEX_TEMPORARY_COLUMN_NAME in their schemas.
450450 long[] rowIndices = FileReader.getRowIndices(blocks);
451- columnReaders[i] =
452- new ArrowRowIndexColumnReader(nonPartitionFields[i], capacity, rowIndices);
451+ columnReaders[i] = new RowIndexColumnReader(nonPartitionFields[i], capacity, rowIndices);
453452 hasRowIndexColumn = true;
454453 missingColumns[i] = true;
455454 } else if (optFileField.isPresent()) {
@@ -474,8 +473,8 @@ public void init() throws Throwable {
474473 + filePath);
475474 }
476475 if (field.isPrimitive()) {
477- ArrowConstantColumnReader reader =
478- new ArrowConstantColumnReader (nonPartitionFields[i], capacity, useDecimal128);
476+ ConstantColumnReader reader =
477+ new ConstantColumnReader (nonPartitionFields[i], capacity, useDecimal128);
479478 columnReaders[i] = reader;
480479 missingColumns[i] = true;
481480 } else {
@@ -493,9 +492,8 @@ public void init() throws Throwable {
493492 for (int i = fields.size(); i < columnReaders.length; i++) {
494493 int fieldIndex = i - fields.size();
495494 StructField field = partitionFields[fieldIndex];
496- ArrowConstantColumnReader reader =
497- new ArrowConstantColumnReader(
498- field, capacity, partitionValues, fieldIndex, useDecimal128);
495+ ConstantColumnReader reader =
496+ new ConstantColumnReader(field, capacity, partitionValues, fieldIndex, useDecimal128);
499497 columnReaders[i] = reader;
500498 }
501499 }
0 commit comments