From b20a2e6b5ee82814011a031d8a8282b0fec3ffe1 Mon Sep 17 00:00:00 2001 From: Salim Achouche Date: Tue, 12 Mar 2019 18:06:43 -0700 Subject: DRILL-7100: Fixed IllegalArgumentException when reading Parquet data --- .../parquet/columnreaders/TestBatchSizingMemoryUtil.java | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) (limited to 'exec/java-exec/src/test/java/org/apache') diff --git a/exec/java-exec/src/test/java/org/apache/drill/exec/store/parquet/columnreaders/TestBatchSizingMemoryUtil.java b/exec/java-exec/src/test/java/org/apache/drill/exec/store/parquet/columnreaders/TestBatchSizingMemoryUtil.java index 896675a3c..4cf150372 100644 --- a/exec/java-exec/src/test/java/org/apache/drill/exec/store/parquet/columnreaders/TestBatchSizingMemoryUtil.java +++ b/exec/java-exec/src/test/java/org/apache/drill/exec/store/parquet/columnreaders/TestBatchSizingMemoryUtil.java @@ -93,9 +93,9 @@ public class TestBatchSizingMemoryUtil extends PhysicalOpUnitTestBase { for (int columnIdx = 0; columnIdx < 3; columnIdx++) { final ColumnMemoryUsageInfo columnInfo = columnMemoryInfo[columnIdx]; - final int remainingBitsCapacity = getRemainingBitsCapacity(columnInfo); - final int remainingOffsetsCapacity = getRemainingOffsetsCapacity(columnInfo); - final int remainingDataCapacity = getRemainingDataCapacity(columnInfo); + final long remainingBitsCapacity = getRemainingBitsCapacity(columnInfo); + final long remainingOffsetsCapacity = getRemainingOffsetsCapacity(columnInfo); + final long remainingDataCapacity = getRemainingDataCapacity(columnInfo); // Test current VV is within quota (since we are not adding new entries) Assert.assertTrue(BatchSizingMemoryUtil.canAddNewData(columnInfo, 0, 0, 0)); @@ -152,15 +152,15 @@ public class TestBatchSizingMemoryUtil extends PhysicalOpUnitTestBase { return result; } - private static int getRemainingBitsCapacity(ColumnMemoryUsageInfo columnInfo) { + private static long getRemainingBitsCapacity(ColumnMemoryUsageInfo columnInfo) { return columnInfo.vectorMemoryUsage.bitsBytesCapacity - columnInfo.vectorMemoryUsage.bitsBytesUsed; } - private static int getRemainingOffsetsCapacity(ColumnMemoryUsageInfo columnInfo) { + private static long getRemainingOffsetsCapacity(ColumnMemoryUsageInfo columnInfo) { return columnInfo.vectorMemoryUsage.offsetsByteCapacity - columnInfo.vectorMemoryUsage.offsetsBytesUsed; } - private static int getRemainingDataCapacity(ColumnMemoryUsageInfo columnInfo) { + private static long getRemainingDataCapacity(ColumnMemoryUsageInfo columnInfo) { return columnInfo.vectorMemoryUsage.dataByteCapacity - columnInfo.vectorMemoryUsage.dataBytesUsed; } -- cgit v1.2.3