Minor refactor in GeneratedKeyContextEngine to avoid IndexOutOfBoundE…

…xception (#32936) * Minor refactor in GeneratedKeyContextEngine to avoid IndexOutOfBoundException * fix unit test
apache · Sep 20, 2024 · 474c0f3 · 474c0f3
1 parent 04de462
commit 474c0f3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/...ngsphere/infra/binder/context/segment/insert/keygen/engine/GeneratedKeyContextEngine.java b/...ngsphere/infra/binder/context/segment/insert/keygen/engine/GeneratedKeyContextEngine.java
@@ -98,7 +98,8 @@ private GeneratedKeyContext findGeneratedKey(final Map<String, Integer> insertCo
                 if (params.isEmpty()) {
                     continue;
                 }
-                if (null != params.get(((ParameterMarkerExpressionSegment) expression).getParameterMarkerIndex())) {
+                if (params.size() > ((ParameterMarkerExpressionSegment) expression).getParameterMarkerIndex()
+                        && null != params.get(((ParameterMarkerExpressionSegment) expression).getParameterMarkerIndex())) {
                     result.getGeneratedValues().add((Comparable<?>) params.get(((ParameterMarkerExpressionSegment) expression).getParameterMarkerIndex()));
                 }
             } else if (expression instanceof LiteralExpressionSegment) {