[FLINK-6899] [state] Create correctly sized state array in NestedMaps…

…StateTable This closes apache#4107.
silenceland · Jun 13, 2017 · 7351ec3 · 7351ec3
1 parent 23c82e3
commit 7351ec3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/flink-runtime/src/main/java/org/apache/flink/runtime/state/heap/NestedMapsStateTable.java b/flink-runtime/src/main/java/org/apache/flink/runtime/state/heap/NestedMapsStateTable.java
@@ -68,7 +68,7 @@ public NestedMapsStateTable(InternalKeyContext<K> keyContext, RegisteredKeyedBac
 		this.keyGroupOffset = keyContext.getKeyGroupRange().getStartKeyGroup();
 
 		@SuppressWarnings("unchecked")
-		Map<N, Map<K, S>>[] state = (Map<N, Map<K, S>>[]) new Map[keyContext.getNumberOfKeyGroups()];
+		Map<N, Map<K, S>>[] state = (Map<N, Map<K, S>>[]) new Map[keyContext.getKeyGroupRange().getNumberOfKeyGroups()];
 		this.state = state;
 	}