Skip to content

Commit 01efd9f

Browse files
committed
Bugfix in ParquetReader to read structs correctly for InMemoryTableParquetReader
Signed-off-by: Arham Chopra <[email protected]>
1 parent b14e2c8 commit 01efd9f

File tree

1 file changed

+10
-2
lines changed

1 file changed

+10
-2
lines changed

cpp/csp/adapters/parquet/ParquetReader.cpp

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -192,9 +192,17 @@ void SingleTableParquetReader::setColumnAdaptersFromCurrentTable()
192192
columnAdapter = createColumnAdapter( *this, *field, getCurFileOrTableName(), &getStructColumnMeta() );
193193
auto &fieldInfo = fieldsInfo[ index ];
194194

195-
for( std::size_t i = 0; i < fieldInfo.m_width; ++i )
195+
if( isArrowIPC() )
196+
{
197+
// Needed for all memory tables
198+
m_neededColumnIndices.push_back( index );
199+
}
200+
else
196201
{
197-
m_neededColumnIndices.push_back( fieldInfo.m_startColumnIndex + i );
202+
for( std::size_t i = 0; i < fieldInfo.m_width; ++i )
203+
{
204+
m_neededColumnIndices.push_back( fieldInfo.m_startColumnIndex + i );
205+
}
198206
}
199207
}
200208
else

0 commit comments

Comments
 (0)