From b9068117177c3b47025f52c00f67938e0c3e4732 Mon Sep 17 00:00:00 2001 From: Amit Hadke Date: Mon, 7 Dec 2015 17:25:34 -0800 Subject: [PATCH] DRILL-4165 Add a precondition for size of merge join record batch. --- .../org/apache/drill/exec/physical/impl/join/JoinStatus.java | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/exec/java-exec/src/main/java/org/apache/drill/exec/physical/impl/join/JoinStatus.java b/exec/java-exec/src/main/java/org/apache/drill/exec/physical/impl/join/JoinStatus.java index e96823611fc..9e31763398f 100644 --- a/exec/java-exec/src/main/java/org/apache/drill/exec/physical/impl/join/JoinStatus.java +++ b/exec/java-exec/src/main/java/org/apache/drill/exec/physical/impl/join/JoinStatus.java @@ -17,6 +17,7 @@ */ package org.apache.drill.exec.physical.impl.join; +import com.google.common.base.Preconditions; import org.apache.drill.exec.physical.impl.join.JoinWorker.JoinOutcome; import org.apache.drill.exec.record.RecordBatch.IterOutcome; import org.apache.drill.exec.record.RecordIterator; @@ -99,7 +100,8 @@ public final void resetOutputPos() { } public final boolean isOutgoingBatchFull() { - return outputPosition >= OUTPUT_BATCH_SIZE; + Preconditions.checkArgument(outputPosition <= OUTPUT_BATCH_SIZE); + return outputPosition == OUTPUT_BATCH_SIZE; } public final void incOutputPos() {