[FLINK-15549] Fix Integer overflow in ResettableIterator. (#10903)

The ResettableIterator had a data overflow problem if the number of elements in a single input exceeded Integer.MAX_VALUE.
apache · Jan 21, 2020 · 90ad2e6 · 90ad2e6
1 parent d44848c
commit 90ad2e6
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/...c/main/java/org/apache/flink/runtime/operators/resettable/SpillingResettableIterator.java b/...c/main/java/org/apache/flink/runtime/operators/resettable/SpillingResettableIterator.java
@@ -58,9 +58,9 @@ public class SpillingResettableIterator<T> implements ResettableIterator<T> {
 
 	protected final TypeSerializer<T> serializer;
 
-	private int elementCount;
+	private long elementCount;
 
-	private int currentElementNum;
+	private long currentElementNum;
 
 	protected final SpillingBuffer buffer;
 

diff --git a/...rg/apache/flink/runtime/operators/resettable/SpillingResettableMutableObjectIterator.java b/...rg/apache/flink/runtime/operators/resettable/SpillingResettableMutableObjectIterator.java
@@ -53,9 +53,9 @@ public class SpillingResettableMutableObjectIterator<T> implements ResettableMut
 
 	protected final TypeSerializer<T> serializer;
 
-	private int elementCount;
+	private long elementCount;
 
-	private int currentElementNum;
+	private long currentElementNum;
 
 	protected final SpillingBuffer buffer;