Skip to content

Commit ac4ac21

Browse files
committed
new_with_options
1 parent 925d86f commit ac4ac21

File tree

1 file changed

+32
-12
lines changed
  • parquet/src/arrow/async_reader

1 file changed

+32
-12
lines changed

parquet/src/arrow/async_reader/mod.rs

Lines changed: 32 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -1278,9 +1278,14 @@ mod tests {
12781278
};
12791279

12801280
let options = ArrowReaderOptions::new().with_page_index(true);
1281-
let builder = ParquetRecordBatchStreamBuilder::new_with_options(async_reader, options)
1282-
.await
1283-
.unwrap();
1281+
let builder = ParquetRecordBatchStreamBuilder::new_with_options(
1282+
async_reader,
1283+
options,
1284+
#[cfg(feature = "encryption")]
1285+
None,
1286+
)
1287+
.await
1288+
.unwrap();
12841289

12851290
// The builder should have page and offset indexes loaded now
12861291
let metadata_with_index = builder.metadata();
@@ -1392,9 +1397,14 @@ mod tests {
13921397
};
13931398

13941399
let options = ArrowReaderOptions::new().with_page_index(true);
1395-
let builder = ParquetRecordBatchStreamBuilder::new_with_options(async_reader, options)
1396-
.await
1397-
.unwrap();
1400+
let builder = ParquetRecordBatchStreamBuilder::new_with_options(
1401+
async_reader,
1402+
options,
1403+
#[cfg(feature = "encryption")]
1404+
None,
1405+
)
1406+
.await
1407+
.unwrap();
13981408

13991409
let selection = RowSelection::from(vec![
14001410
RowSelector::skip(21), // Skip first page
@@ -1475,9 +1485,14 @@ mod tests {
14751485
};
14761486

14771487
let options = ArrowReaderOptions::new().with_page_index(true);
1478-
let builder = ParquetRecordBatchStreamBuilder::new_with_options(async_reader, options)
1479-
.await
1480-
.unwrap();
1488+
let builder = ParquetRecordBatchStreamBuilder::new_with_options(
1489+
async_reader,
1490+
options,
1491+
#[cfg(feature = "encryption")]
1492+
None,
1493+
)
1494+
.await
1495+
.unwrap();
14811496

14821497
let col_idx: usize = rand.gen_range(0..13);
14831498
let mask = ProjectionMask::leaves(builder.parquet_schema(), vec![col_idx]);
@@ -1546,9 +1561,14 @@ mod tests {
15461561
};
15471562

15481563
let options = ArrowReaderOptions::new().with_page_index(true);
1549-
let builder = ParquetRecordBatchStreamBuilder::new_with_options(async_reader, options)
1550-
.await
1551-
.unwrap();
1564+
let builder = ParquetRecordBatchStreamBuilder::new_with_options(
1565+
async_reader,
1566+
options,
1567+
#[cfg(feature = "encryption")]
1568+
None,
1569+
)
1570+
.await
1571+
.unwrap();
15521572

15531573
let col_idx: usize = rand.gen_range(0..13);
15541574
let mask = ProjectionMask::leaves(builder.parquet_schema(), vec![col_idx]);

0 commit comments

Comments
 (0)