Attempt to make some operations cheaper

Author: idegtiarenko

x-pack/plugin/esql-core/src/main/java/org/elasticsearch/xpack/esql/core/tree/Node.java

@@ -67,7 +67,11 @@ public List<T> children() {
     @SuppressWarnings("unchecked")
     public void forEachDown(Consumer<? super T> action) {
         action.accept((T) this);
-        children().forEach(c -> c.forEachDown(action));
+        // please do not refactor it to a for-each loop
+        // to avoid allocating iterator that performs concurrent modification checks
+        for (int c = 0; c < children.size(); c++) {
+            children.get(c).forEachDown(action);
+        }
     }
 
     @SuppressWarnings("unchecked")
@@ -81,7 +85,11 @@ public <E extends T> void forEachDown(Class<E> typeToken, Consumer<? super E> ac
 
     @SuppressWarnings("unchecked")
     public void forEachUp(Consumer<? super T> action) {
-        children().forEach(c -> c.forEachUp(action));
+        // please do not refactor it to a for-each loop
+        // to avoid allocating iterator that performs concurrent modification checks
+        for (int c = 0; c < children.size(); c++) {
+            children.get(c).forEachUp(action);
+        }
         action.accept((T) this);
     }
 

x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/plan/QueryPlan.java

@@ -140,11 +140,13 @@ private static Object doTransformExpression(Object arg, Function<Expression, ? e
         // preserving the type information is hacky and weird (a lot of context needs to be passed around and the lambda itself
         // has no type info so it's difficult to have automatic checking without having base classes).
 
-        if (arg instanceof Collection<?> c) {
+        if (arg instanceof List<?> c) {
             List<Object> transformed = null;
             boolean hasChanged = false;
-            int i = 0;
-            for (Object e : c) {
+            // please do not refactor it to a for-each loop
+            // to avoid allocating iterator that performs concurrent modification checks
+            for (int i = 0; i < c.size(); i++) {
+                var e = c.get(i);
                 Object next = doTransformExpression(e, traversal);
                 if (e.equals(next) == false) {
                     if (hasChanged == false) {
@@ -153,7 +155,6 @@ private static Object doTransformExpression(Object arg, Function<Expression, ? e
                     }
                     transformed.set(i, next);
                 }
-                i++;
             }
 
             return hasChanged ? transformed : arg;