Add memory usage reporting to MvContains evaluators

ESQL: Track memory in evaluators (elastic#133392) got merged to main at the same as Add MV_CONTAINS function elastic#133099 which caused a compile-error and the merge was reverted. This commit addresses the compile-error.

Author: mjmbischoff

x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvContains.java

@@ -8,6 +8,7 @@
 package org.elasticsearch.xpack.esql.expression.function.scalar.multivalue;
 
 import org.apache.lucene.util.BytesRef;
+import org.apache.lucene.util.RamUsageEstimator;
 import org.elasticsearch.common.io.stream.NamedWriteableRegistry;
 import org.elasticsearch.common.io.stream.StreamInput;
 import org.elasticsearch.compute.data.Block;
@@ -303,6 +304,7 @@ interface ValueExtractor<BlockType extends Block, Type> {
     }
 
     private static final class MvContainsNullEvaluator implements ExpressionEvaluator.Factory {
+        private static final long BASE_RAM_BYTES_USED = RamUsageEstimator.shallowSizeOfInstance(MvContainsNullEvaluator.class);
         private final ExpressionEvaluator.Factory subsetFieldEvaluator;
 
         private MvContainsNullEvaluator(ExpressionEvaluator.Factory subsetFieldEvaluator) {
@@ -322,6 +324,13 @@ public Block eval(Page page) {
                     }
                 }
 
+                @Override
+                public long baseRamBytesUsed() {
+                    long baseRamBytesUsed = BASE_RAM_BYTES_USED;
+                    baseRamBytesUsed += subsetField.baseRamBytesUsed();
+                    return baseRamBytesUsed;
+                }
+
                 @Override
                 public void close() {
                     Releasables.closeExpectNoException(subsetField);
@@ -351,6 +360,7 @@ public String toString() {
      *  TODO extend code generation to handle this case
      */
     public static class MvContainsBooleanEvaluator implements EvalOperator.ExpressionEvaluator {
+        private static final long BASE_RAM_BYTES_USED = RamUsageEstimator.shallowSizeOfInstance(MvContainsBooleanEvaluator.class);
         private final EvalOperator.ExpressionEvaluator field1;
         private final EvalOperator.ExpressionEvaluator field2;
         private final DriverContext driverContext;
@@ -393,6 +403,14 @@ public void close() {
             Releasables.closeExpectNoException(field1, field2);
         }
 
+        @Override
+        public long baseRamBytesUsed() {
+            long baseRamBytesUsed = BASE_RAM_BYTES_USED;
+            baseRamBytesUsed += field1.baseRamBytesUsed();
+            baseRamBytesUsed += field2.baseRamBytesUsed();
+            return baseRamBytesUsed;
+        }
+
         public static class Factory implements EvalOperator.ExpressionEvaluator.Factory {
             private final Source source;
             private final EvalOperator.ExpressionEvaluator.Factory field1;
@@ -431,6 +449,7 @@ public String toString() {
      *  TODO extend code generation to handle this case
      */
     public static class MvContainsBytesRefEvaluator implements EvalOperator.ExpressionEvaluator {
+        private static final long BASE_RAM_BYTES_USED = RamUsageEstimator.shallowSizeOfInstance(MvContainsBytesRefEvaluator.class);
         private final EvalOperator.ExpressionEvaluator field1;
         private final EvalOperator.ExpressionEvaluator field2;
         private final DriverContext driverContext;
@@ -473,6 +492,14 @@ public void close() {
             Releasables.closeExpectNoException(field1, field2);
         }
 
+        @Override
+        public long baseRamBytesUsed() {
+            long baseRamBytesUsed = BASE_RAM_BYTES_USED;
+            baseRamBytesUsed += field1.baseRamBytesUsed();
+            baseRamBytesUsed += field2.baseRamBytesUsed();
+            return baseRamBytesUsed;
+        }
+
         public static class Factory implements EvalOperator.ExpressionEvaluator.Factory {
             private final Source source;
             private final EvalOperator.ExpressionEvaluator.Factory field1;
@@ -511,6 +538,7 @@ public String toString() {
      *  TODO extend code generation to handle this case
      */
     public static class MvContainsDoubleEvaluator implements EvalOperator.ExpressionEvaluator {
+        private static final long BASE_RAM_BYTES_USED = RamUsageEstimator.shallowSizeOfInstance(MvContainsDoubleEvaluator.class);
         private final EvalOperator.ExpressionEvaluator field1;
         private final EvalOperator.ExpressionEvaluator field2;
         private final DriverContext driverContext;
@@ -553,6 +581,14 @@ public void close() {
             Releasables.closeExpectNoException(field1, field2);
         }
 
+        @Override
+        public long baseRamBytesUsed() {
+            long baseRamBytesUsed = BASE_RAM_BYTES_USED;
+            baseRamBytesUsed += field1.baseRamBytesUsed();
+            baseRamBytesUsed += field2.baseRamBytesUsed();
+            return baseRamBytesUsed;
+        }
+
         public static class Factory implements EvalOperator.ExpressionEvaluator.Factory {
             private final Source source;
             private final EvalOperator.ExpressionEvaluator.Factory field1;
@@ -591,6 +627,7 @@ public String toString() {
      *  TODO extend code generation to handle this case
      */
     public static class MvContainsIntEvaluator implements EvalOperator.ExpressionEvaluator {
+        private static final long BASE_RAM_BYTES_USED = RamUsageEstimator.shallowSizeOfInstance(MvContainsIntEvaluator.class);
         private final EvalOperator.ExpressionEvaluator field1;
         private final EvalOperator.ExpressionEvaluator field2;
         private final DriverContext driverContext;
@@ -633,6 +670,14 @@ public void close() {
             Releasables.closeExpectNoException(field1, field2);
         }
 
+        @Override
+        public long baseRamBytesUsed() {
+            long baseRamBytesUsed = BASE_RAM_BYTES_USED;
+            baseRamBytesUsed += field1.baseRamBytesUsed();
+            baseRamBytesUsed += field2.baseRamBytesUsed();
+            return baseRamBytesUsed;
+        }
+
         public static class Factory implements EvalOperator.ExpressionEvaluator.Factory {
             private final Source source;
             private final EvalOperator.ExpressionEvaluator.Factory field1;
@@ -671,6 +716,7 @@ public String toString() {
      *  TODO extend code generation to handle this case
      */
     public static class MvContainsLongEvaluator implements EvalOperator.ExpressionEvaluator {
+        private static final long BASE_RAM_BYTES_USED = RamUsageEstimator.shallowSizeOfInstance(MvContainsLongEvaluator.class);
         private final EvalOperator.ExpressionEvaluator field1;
         private final EvalOperator.ExpressionEvaluator field2;
         private final DriverContext driverContext;
@@ -713,6 +759,14 @@ public void close() {
             Releasables.closeExpectNoException(field1, field2);
         }
 
+        @Override
+        public long baseRamBytesUsed() {
+            long baseRamBytesUsed = BASE_RAM_BYTES_USED;
+            baseRamBytesUsed += field1.baseRamBytesUsed();
+            baseRamBytesUsed += field2.baseRamBytesUsed();
+            return baseRamBytesUsed;
+        }
+
         public static class Factory implements EvalOperator.ExpressionEvaluator.Factory {
             private final Source source;
             private final EvalOperator.ExpressionEvaluator.Factory field1;