diff --git a/presto-orc/src/main/java/com/facebook/presto/orc/reader/MapFlatSelectiveStreamReader.java b/presto-orc/src/main/java/com/facebook/presto/orc/reader/MapFlatSelectiveStreamReader.java index 5891e57f89fbc..b8494fa19f79e 100644 --- a/presto-orc/src/main/java/com/facebook/presto/orc/reader/MapFlatSelectiveStreamReader.java +++ b/presto-orc/src/main/java/com/facebook/presto/orc/reader/MapFlatSelectiveStreamReader.java @@ -70,6 +70,7 @@ import static com.facebook.presto.orc.reader.SelectiveStreamReaders.initializeOutputPositions; import static com.facebook.presto.orc.stream.MissingInputStreamSource.missingStreamSource; import static com.google.common.base.Preconditions.checkArgument; +import static com.google.common.base.Preconditions.checkNotNull; import static com.google.common.base.Preconditions.checkState; import static com.google.common.base.Verify.verify; import static com.google.common.collect.ImmutableList.toImmutableList; @@ -411,7 +412,7 @@ private void openRowGroup() presentStream = presentStreamSource.openStream(); for (int i = 0; i < keyCount; i++) { - BooleanInputStream inMapStream = requireNonNull(inMapStreamSources.get(i).openStream(), "missing inMapStream at position " + i); + BooleanInputStream inMapStream = checkNotNull(inMapStreamSources.get(i).openStream(), "missing inMapStream at position %s", i); inMapStreams.add(inMapStream); } @@ -795,6 +796,9 @@ private Block getSliceKeysBlock(List sequenceEncodings) public void startRowGroup(InputStreamSources dataStreamSources) throws IOException { + presentStream = null; + inMapStreams.clear(); + presentStreamSource = dataStreamSources.getInputStreamSource(streamDescriptor, PRESENT, BooleanInputStream.class); for (int i = 0; i < keyCount; i++) { @@ -810,9 +814,6 @@ public void startRowGroup(InputStreamSources dataStreamSources) nestedPositionCounts = ensureCapacity(nestedPositionCounts, keyCount); inMap = ensureCapacity(inMap, keyCount); - presentStream = null; - inMapStreams.clear(); - rowGroupOpen = false; for (SelectiveStreamReader valueStreamReader : valueStreamReaders) {