Be sure to correctly re-map RowSet column names when writing persiste…

…nt data indexes to Parquet.
lbooker42 · Mar 19, 2024 · c6ff002 · c6ff002
1 parent a7bc940
commit c6ff002
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/extensions/parquet/table/src/main/java/io/deephaven/parquet/table/ParquetTableWriter.java b/extensions/parquet/table/src/main/java/io/deephaven/parquet/table/ParquetTableWriter.java
@@ -148,7 +148,15 @@ static void write(
                         tableInfoBuilder.addDataIndexes(DataIndexInfo.of(
                                 destDirPath.relativize(info.metadataFilePath.toPath()).toString(),
                                 info.parquetColumnNames));
-                        write(indexTable, indexTable.getDefinition(), writeInstructions,
+                        final ParquetInstructions writeInstructionsToUse;
+                        if (INDEX_ROW_SET_COLUMN_NAME.equals(dataIndex.rowSetColumnName())) {
+                            writeInstructionsToUse = writeInstructions;
+                        } else {
+                            writeInstructionsToUse = new ParquetInstructions.Builder(writeInstructions)
+                                    .addColumnNameMapping(INDEX_ROW_SET_COLUMN_NAME, dataIndex.rowSetColumnName())
+                                    .build();
+                        }
+                        write(indexTable, indexTable.getDefinition(), writeInstructionsToUse,
                                 info.destFile.getAbsolutePath(), Collections.emptyMap(), TableInfo.builder());
                     }
                 }