|
|
@@ -1,742 +0,0 @@
|
|
|
-/*
|
|
|
- * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
|
|
|
- * or more contributor license agreements. Licensed under the Elastic License
|
|
|
- * 2.0; you may not use this file except in compliance with the Elastic License
|
|
|
- * 2.0.
|
|
|
- */
|
|
|
-
|
|
|
-package org.elasticsearch.xpack.esql.expression.function.scalar.multivalue;
|
|
|
-
|
|
|
-import org.apache.lucene.util.BytesRef;
|
|
|
-import org.elasticsearch.common.io.stream.NamedWriteableRegistry;
|
|
|
-import org.elasticsearch.common.io.stream.StreamInput;
|
|
|
-import org.elasticsearch.compute.data.Block;
|
|
|
-import org.elasticsearch.compute.data.BooleanBlock;
|
|
|
-import org.elasticsearch.compute.data.BytesRefBlock;
|
|
|
-import org.elasticsearch.compute.data.DoubleBlock;
|
|
|
-import org.elasticsearch.compute.data.ElementType;
|
|
|
-import org.elasticsearch.compute.data.IntBlock;
|
|
|
-import org.elasticsearch.compute.data.LongBlock;
|
|
|
-import org.elasticsearch.compute.data.Page;
|
|
|
-import org.elasticsearch.compute.operator.DriverContext;
|
|
|
-import org.elasticsearch.compute.operator.EvalOperator;
|
|
|
-import org.elasticsearch.compute.operator.EvalOperator.ExpressionEvaluator;
|
|
|
-import org.elasticsearch.core.Releasables;
|
|
|
-import org.elasticsearch.xpack.esql.EsqlIllegalArgumentException;
|
|
|
-import org.elasticsearch.xpack.esql.core.expression.Expression;
|
|
|
-import org.elasticsearch.xpack.esql.core.expression.FoldContext;
|
|
|
-import org.elasticsearch.xpack.esql.core.expression.Nullability;
|
|
|
-import org.elasticsearch.xpack.esql.core.expression.function.scalar.BinaryScalarFunction;
|
|
|
-import org.elasticsearch.xpack.esql.core.tree.NodeInfo;
|
|
|
-import org.elasticsearch.xpack.esql.core.tree.Source;
|
|
|
-import org.elasticsearch.xpack.esql.core.type.DataType;
|
|
|
-import org.elasticsearch.xpack.esql.evaluator.mapper.EvaluatorMapper;
|
|
|
-import org.elasticsearch.xpack.esql.expression.function.Example;
|
|
|
-import org.elasticsearch.xpack.esql.expression.function.FunctionAppliesTo;
|
|
|
-import org.elasticsearch.xpack.esql.expression.function.FunctionAppliesToLifecycle;
|
|
|
-import org.elasticsearch.xpack.esql.expression.function.FunctionInfo;
|
|
|
-import org.elasticsearch.xpack.esql.expression.function.Param;
|
|
|
-import org.elasticsearch.xpack.esql.planner.PlannerUtils;
|
|
|
-
|
|
|
-import java.io.IOException;
|
|
|
-
|
|
|
-import static org.elasticsearch.xpack.esql.core.expression.TypeResolutions.ParamOrdinal.FIRST;
|
|
|
-import static org.elasticsearch.xpack.esql.core.expression.TypeResolutions.ParamOrdinal.SECOND;
|
|
|
-import static org.elasticsearch.xpack.esql.core.expression.TypeResolutions.isRepresentableExceptCounters;
|
|
|
-import static org.elasticsearch.xpack.esql.core.expression.TypeResolutions.isType;
|
|
|
-
|
|
|
-/**
|
|
|
- * Function that takes two multivalued expressions and checks if values of one expression are all present(equals) in the other.
|
|
|
- * <p>
|
|
|
- * Given Set A = {"a","b","c"} and Set B = {"b","c"}, the relationship between first (row) and second (column) arguments is:
|
|
|
- * <ul>
|
|
|
- * <li>A, B ⇒ true (A ⊆ B)</li>
|
|
|
- * <li>B, A ⇒ false (A ⊈ B)</li>
|
|
|
- * <li>A, A ⇒ true (A ≡ A)</li>
|
|
|
- * <li>B, B ⇒ true (B ≡ B)</li>
|
|
|
- * <li>A, null ⇒ true (B ⊆ ∅)</li>
|
|
|
- * <li>null, A ⇒ false (∅ ⊈ B)</li>
|
|
|
- * <li>B, null ⇒ true (B ⊆ ∅)</li>
|
|
|
- * <li>null, B ⇒ false (∅ ⊈ B)</li>
|
|
|
- * <li>null, null ⇒ true (∅ ≡ ∅)</li>
|
|
|
- * </ul>
|
|
|
- */
|
|
|
-public class MvContains extends BinaryScalarFunction implements EvaluatorMapper {
|
|
|
- public static final NamedWriteableRegistry.Entry ENTRY = new NamedWriteableRegistry.Entry(
|
|
|
- Expression.class,
|
|
|
- "MvContains",
|
|
|
- MvContains::new
|
|
|
- );
|
|
|
-
|
|
|
- @FunctionInfo(
|
|
|
- returnType = "boolean",
|
|
|
- description = "Checks if all values yielded by the second multivalue expression are present in the values yielded by "
|
|
|
- + "the first multivalue expression. Returns a boolean. Null values are treated as an empty set.",
|
|
|
- examples = {
|
|
|
- @Example(file = "string", tag = "mv_contains"),
|
|
|
- @Example(file = "string", tag = "mv_contains_bothsides"),
|
|
|
- @Example(file = "string", tag = "mv_contains_where"), },
|
|
|
- appliesTo = { @FunctionAppliesTo(lifeCycle = FunctionAppliesToLifecycle.PREVIEW, version = "9.2.0") }
|
|
|
- )
|
|
|
- public MvContains(
|
|
|
- Source source,
|
|
|
- @Param(
|
|
|
- name = "superset",
|
|
|
- type = {
|
|
|
- "boolean",
|
|
|
- "cartesian_point",
|
|
|
- "cartesian_shape",
|
|
|
- "date",
|
|
|
- "date_nanos",
|
|
|
- "double",
|
|
|
- "geo_point",
|
|
|
- "geo_shape",
|
|
|
- "integer",
|
|
|
- "ip",
|
|
|
- "keyword",
|
|
|
- "long",
|
|
|
- "text",
|
|
|
- "unsigned_long",
|
|
|
- "version" },
|
|
|
- description = "Multivalue expression."
|
|
|
- ) Expression superset,
|
|
|
- @Param(
|
|
|
- name = "subset",
|
|
|
- type = {
|
|
|
- "boolean",
|
|
|
- "cartesian_point",
|
|
|
- "cartesian_shape",
|
|
|
- "date",
|
|
|
- "date_nanos",
|
|
|
- "double",
|
|
|
- "geo_point",
|
|
|
- "geo_shape",
|
|
|
- "integer",
|
|
|
- "ip",
|
|
|
- "keyword",
|
|
|
- "long",
|
|
|
- "text",
|
|
|
- "unsigned_long",
|
|
|
- "version" },
|
|
|
- description = "Multivalue expression."
|
|
|
- ) Expression subset
|
|
|
- ) {
|
|
|
- super(source, superset, subset);
|
|
|
- }
|
|
|
-
|
|
|
- private MvContains(StreamInput in) throws IOException {
|
|
|
- super(in);
|
|
|
- }
|
|
|
-
|
|
|
- @Override
|
|
|
- public String getWriteableName() {
|
|
|
- return ENTRY.name;
|
|
|
- }
|
|
|
-
|
|
|
- @Override
|
|
|
- protected TypeResolution resolveType() {
|
|
|
- if (childrenResolved() == false) {
|
|
|
- return new TypeResolution("Unresolved children");
|
|
|
- }
|
|
|
-
|
|
|
- TypeResolution resolution = isRepresentableExceptCounters(left(), sourceText(), FIRST);
|
|
|
- if (resolution.unresolved()) {
|
|
|
- return resolution;
|
|
|
- }
|
|
|
- if (left().dataType() == DataType.NULL) {
|
|
|
- return isRepresentableExceptCounters(right(), sourceText(), SECOND);
|
|
|
- }
|
|
|
- return isType(right(), t -> t.noText() == left().dataType().noText(), sourceText(), SECOND, left().dataType().noText().typeName());
|
|
|
- }
|
|
|
-
|
|
|
- @Override
|
|
|
- public DataType dataType() {
|
|
|
- return DataType.BOOLEAN;
|
|
|
- }
|
|
|
-
|
|
|
- @Override
|
|
|
- public Nullability nullable() {
|
|
|
- return Nullability.FALSE;
|
|
|
- }
|
|
|
-
|
|
|
- @Override
|
|
|
- protected MvContains replaceChildren(Expression newLeft, Expression newRight) {
|
|
|
- return new MvContains(source(), newLeft, newRight);
|
|
|
- }
|
|
|
-
|
|
|
- @Override
|
|
|
- protected NodeInfo<? extends Expression> info() {
|
|
|
- return NodeInfo.create(this, MvContains::new, left(), right());
|
|
|
- }
|
|
|
-
|
|
|
- @Override
|
|
|
- public Object fold(FoldContext ctx) {
|
|
|
- return EvaluatorMapper.super.fold(source(), ctx);
|
|
|
- }
|
|
|
-
|
|
|
- @Override
|
|
|
- public ExpressionEvaluator.Factory toEvaluator(ToEvaluator toEvaluator) {
|
|
|
- var supersetType = PlannerUtils.toElementType(left().dataType());
|
|
|
- var subsetType = PlannerUtils.toElementType(right().dataType());
|
|
|
- if (supersetType != ElementType.NULL && subsetType != ElementType.NULL && supersetType != subsetType) {
|
|
|
- throw new EsqlIllegalArgumentException(
|
|
|
- "Incompatible data types for MvContains, superset type({}) value({}) and subset type({}) value({}) don't match.",
|
|
|
- supersetType,
|
|
|
- left(),
|
|
|
- subsetType,
|
|
|
- right()
|
|
|
- );
|
|
|
- }
|
|
|
- if (supersetType == ElementType.NULL || subsetType == ElementType.NULL) {
|
|
|
- return new MvContainsNullEvaluator(toEvaluator.apply(right()));
|
|
|
- }
|
|
|
- return switch (supersetType) {
|
|
|
- case BOOLEAN -> new MvContainsBooleanEvaluator.Factory(source(), toEvaluator.apply(left()), toEvaluator.apply(right()));
|
|
|
- case BYTES_REF -> new MvContainsBytesRefEvaluator.Factory(source(), toEvaluator.apply(left()), toEvaluator.apply(right()));
|
|
|
- case DOUBLE -> new MvContainsDoubleEvaluator.Factory(source(), toEvaluator.apply(left()), toEvaluator.apply(right()));
|
|
|
- case INT -> new MvContainsIntEvaluator.Factory(source(), toEvaluator.apply(left()), toEvaluator.apply(right()));
|
|
|
- case LONG -> new MvContainsLongEvaluator.Factory(source(), toEvaluator.apply(left()), toEvaluator.apply(right()));
|
|
|
- default -> throw EsqlIllegalArgumentException.illegalDataType(dataType());
|
|
|
- };
|
|
|
- }
|
|
|
-
|
|
|
- // @Evaluator(extraName = "Int") see end of file.
|
|
|
- static void process(BooleanBlock.Builder builder, int position, IntBlock field1, IntBlock field2) {
|
|
|
- appendTo(builder, containsAll(field1, field2, position, IntBlock::getInt));
|
|
|
- }
|
|
|
-
|
|
|
- // @Evaluator(extraName = "Boolean") see end of file.
|
|
|
- static void process(BooleanBlock.Builder builder, int position, BooleanBlock field1, BooleanBlock field2) {
|
|
|
- appendTo(builder, containsAll(field1, field2, position, BooleanBlock::getBoolean));
|
|
|
- }
|
|
|
-
|
|
|
- // @Evaluator(extraName = "Long") see end of file.
|
|
|
- static void process(BooleanBlock.Builder builder, int position, LongBlock field1, LongBlock field2) {
|
|
|
- appendTo(builder, containsAll(field1, field2, position, LongBlock::getLong));
|
|
|
- }
|
|
|
-
|
|
|
- // @Evaluator(extraName = "Double") see end of file.
|
|
|
- static void process(BooleanBlock.Builder builder, int position, DoubleBlock field1, DoubleBlock field2) {
|
|
|
- appendTo(builder, containsAll(field1, field2, position, DoubleBlock::getDouble));
|
|
|
- }
|
|
|
-
|
|
|
- // @Evaluator(extraName = "BytesRef") see end of file.
|
|
|
- static void process(BooleanBlock.Builder builder, int position, BytesRefBlock field1, BytesRefBlock field2) {
|
|
|
- appendTo(builder, containsAll(field1, field2, position, (block, index) -> {
|
|
|
- var ref = new BytesRef();
|
|
|
- // we pass in a reference, but sometimes we only get a return value, see ConstantBytesRefVector.getBytesRef
|
|
|
- ref = block.getBytesRef(index, ref);
|
|
|
- // pass empty ref as null
|
|
|
- if (ref.length == 0) {
|
|
|
- return null;
|
|
|
- }
|
|
|
- return ref;
|
|
|
- }));
|
|
|
- }
|
|
|
-
|
|
|
- static void appendTo(BooleanBlock.Builder builder, Boolean bool) {
|
|
|
- if (bool == null) {
|
|
|
- builder.appendNull();
|
|
|
- } else {
|
|
|
- builder.beginPositionEntry().appendBoolean(bool).endPositionEntry();
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- /**
|
|
|
- * A block is considered a subset if the superset contains values that test equal for all the values in the subset, independent of
|
|
|
- * order. Duplicates are ignored in the sense that for each duplicate in the subset, we will search/match against the first/any value
|
|
|
- * in the superset.
|
|
|
- *
|
|
|
- * @param superset block to check against
|
|
|
- * @param subset block containing values that should be present in the other block.
|
|
|
- * @return {@code true} if the given blocks are a superset and subset to each other, {@code false} if not.
|
|
|
- */
|
|
|
- static <BlockType extends Block, Type> Boolean containsAll(
|
|
|
- BlockType superset,
|
|
|
- BlockType subset,
|
|
|
- final int position,
|
|
|
- ValueExtractor<BlockType, Type> valueExtractor
|
|
|
- ) {
|
|
|
- if (superset == subset) {
|
|
|
- return true;
|
|
|
- }
|
|
|
- if (subset.areAllValuesNull()) {
|
|
|
- return true;
|
|
|
- }
|
|
|
-
|
|
|
- final var subsetCount = subset.getValueCount(position);
|
|
|
- final var startIndex = subset.getFirstValueIndex(position);
|
|
|
- for (int subsetIndex = startIndex; subsetIndex < startIndex + subsetCount; subsetIndex++) {
|
|
|
- var value = valueExtractor.extractValue(subset, subsetIndex);
|
|
|
- if (hasValue(superset, position, value, valueExtractor) == false) {
|
|
|
- return false;
|
|
|
- }
|
|
|
- }
|
|
|
- return true;
|
|
|
- }
|
|
|
-
|
|
|
- /**
|
|
|
- * Check if the block has the value at any of it's positions
|
|
|
- * @param superset Block to search
|
|
|
- * @param value to search for
|
|
|
- * @return true if the supplied long value is in the supplied Block
|
|
|
- */
|
|
|
- static <BlockType extends Block, Type> boolean hasValue(
|
|
|
- BlockType superset,
|
|
|
- final int position,
|
|
|
- Type value,
|
|
|
- ValueExtractor<BlockType, Type> valueExtractor
|
|
|
- ) {
|
|
|
- final var supersetCount = superset.getValueCount(position);
|
|
|
- final var startIndex = superset.getFirstValueIndex(position);
|
|
|
- for (int supersetIndex = startIndex; supersetIndex < startIndex + supersetCount; supersetIndex++) {
|
|
|
- var element = valueExtractor.extractValue(superset, supersetIndex);
|
|
|
- if (element != null && element.equals(value)) {
|
|
|
- return true;
|
|
|
- }
|
|
|
- }
|
|
|
- return false;
|
|
|
- }
|
|
|
-
|
|
|
- interface ValueExtractor<BlockType extends Block, Type> {
|
|
|
- Type extractValue(BlockType block, int position);
|
|
|
- }
|
|
|
-
|
|
|
- private static final class MvContainsNullEvaluator implements ExpressionEvaluator.Factory {
|
|
|
- private final ExpressionEvaluator.Factory subsetFieldEvaluator;
|
|
|
-
|
|
|
- private MvContainsNullEvaluator(ExpressionEvaluator.Factory subsetFieldEvaluator) {
|
|
|
- this.subsetFieldEvaluator = subsetFieldEvaluator;
|
|
|
- }
|
|
|
-
|
|
|
- @Override
|
|
|
- public ExpressionEvaluator get(DriverContext context) {
|
|
|
- return new ExpressionEvaluator() {
|
|
|
- final ExpressionEvaluator subsetField = subsetFieldEvaluator.get(context);
|
|
|
-
|
|
|
- @Override
|
|
|
- public Block eval(Page page) {
|
|
|
- try (Block block = subsetField.eval(page)) {
|
|
|
- var position = page.getPositionCount();
|
|
|
- return context.blockFactory().newConstantBooleanBlockWith(block.isNull(position), position);
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- @Override
|
|
|
- public void close() {
|
|
|
- Releasables.closeExpectNoException(subsetField);
|
|
|
- }
|
|
|
-
|
|
|
- @Override
|
|
|
- public String toString() {
|
|
|
- return "MvContainsNullEvaluator[" + "subsetField=" + subsetFieldEvaluator + "]";
|
|
|
- }
|
|
|
- };
|
|
|
- }
|
|
|
-
|
|
|
- @Override
|
|
|
- public String toString() {
|
|
|
- return "MvContainsNullEvaluator[" + "subsetField=" + subsetFieldEvaluator + "]";
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- /**
|
|
|
- * Currently {@code EvaluatorImplementer} generates:
|
|
|
- * if (allBlocksAreNulls) {
|
|
|
- * result.appendNull();
|
|
|
- * continue position;
|
|
|
- * }
|
|
|
- * when all params are null, this violates our contract of always returning a boolean.
|
|
|
- * It should probably also generate the warnings method conditionally - omitted here.
|
|
|
- * TODO extend code generation to handle this case
|
|
|
- */
|
|
|
- public static class MvContainsBooleanEvaluator implements EvalOperator.ExpressionEvaluator {
|
|
|
- private final EvalOperator.ExpressionEvaluator field1;
|
|
|
- private final EvalOperator.ExpressionEvaluator field2;
|
|
|
- private final DriverContext driverContext;
|
|
|
-
|
|
|
- public MvContainsBooleanEvaluator(
|
|
|
- EvalOperator.ExpressionEvaluator field1,
|
|
|
- EvalOperator.ExpressionEvaluator field2,
|
|
|
- DriverContext driverContext
|
|
|
- ) {
|
|
|
- this.field1 = field1;
|
|
|
- this.field2 = field2;
|
|
|
- this.driverContext = driverContext;
|
|
|
- }
|
|
|
-
|
|
|
- @Override
|
|
|
- public Block eval(Page page) {
|
|
|
- try (BooleanBlock field1Block = (BooleanBlock) field1.eval(page)) {
|
|
|
- try (BooleanBlock field2Block = (BooleanBlock) field2.eval(page)) {
|
|
|
- return eval(page.getPositionCount(), field1Block, field2Block);
|
|
|
- }
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- public BooleanBlock eval(int positionCount, BooleanBlock field1Block, BooleanBlock field2Block) {
|
|
|
- try (BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) {
|
|
|
- for (int p = 0; p < positionCount; p++) {
|
|
|
- MvContains.process(result, p, field1Block, field2Block);
|
|
|
- }
|
|
|
- return result.build();
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- @Override
|
|
|
- public String toString() {
|
|
|
- return "MvContainsBooleanEvaluator[" + "field1=" + field1 + ", field2=" + field2 + "]";
|
|
|
- }
|
|
|
-
|
|
|
- @Override
|
|
|
- public void close() {
|
|
|
- Releasables.closeExpectNoException(field1, field2);
|
|
|
- }
|
|
|
-
|
|
|
- public static class Factory implements EvalOperator.ExpressionEvaluator.Factory {
|
|
|
- private final Source source;
|
|
|
- private final EvalOperator.ExpressionEvaluator.Factory field1;
|
|
|
- private final EvalOperator.ExpressionEvaluator.Factory field2;
|
|
|
-
|
|
|
- public Factory(
|
|
|
- Source source,
|
|
|
- EvalOperator.ExpressionEvaluator.Factory field1,
|
|
|
- EvalOperator.ExpressionEvaluator.Factory field2
|
|
|
- ) {
|
|
|
- this.source = source;
|
|
|
- this.field1 = field1;
|
|
|
- this.field2 = field2;
|
|
|
- }
|
|
|
-
|
|
|
- @Override
|
|
|
- public MvContainsBooleanEvaluator get(DriverContext context) {
|
|
|
- return new MvContainsBooleanEvaluator(field1.get(context), field2.get(context), context);
|
|
|
- }
|
|
|
-
|
|
|
- @Override
|
|
|
- public String toString() {
|
|
|
- return "MvContainsBooleanEvaluator[" + "field1=" + field1 + ", field2=" + field2 + "]";
|
|
|
- }
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- /**
|
|
|
- * Currently {@code EvaluatorImplementer} generates:
|
|
|
- * if (allBlocksAreNulls) {
|
|
|
- * result.appendNull();
|
|
|
- * continue position;
|
|
|
- * }
|
|
|
- * when all params are null, this violates our contract of always returning a boolean.
|
|
|
- * It should probably also generate the warnings method conditionally - omitted here.
|
|
|
- * TODO extend code generation to handle this case
|
|
|
- */
|
|
|
- public static class MvContainsBytesRefEvaluator implements EvalOperator.ExpressionEvaluator {
|
|
|
- private final EvalOperator.ExpressionEvaluator field1;
|
|
|
- private final EvalOperator.ExpressionEvaluator field2;
|
|
|
- private final DriverContext driverContext;
|
|
|
-
|
|
|
- public MvContainsBytesRefEvaluator(
|
|
|
- EvalOperator.ExpressionEvaluator field1,
|
|
|
- EvalOperator.ExpressionEvaluator field2,
|
|
|
- DriverContext driverContext
|
|
|
- ) {
|
|
|
- this.field1 = field1;
|
|
|
- this.field2 = field2;
|
|
|
- this.driverContext = driverContext;
|
|
|
- }
|
|
|
-
|
|
|
- @Override
|
|
|
- public Block eval(Page page) {
|
|
|
- try (BytesRefBlock field1Block = (BytesRefBlock) field1.eval(page)) {
|
|
|
- try (BytesRefBlock field2Block = (BytesRefBlock) field2.eval(page)) {
|
|
|
- return eval(page.getPositionCount(), field1Block, field2Block);
|
|
|
- }
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- public BooleanBlock eval(int positionCount, BytesRefBlock field1Block, BytesRefBlock field2Block) {
|
|
|
- try (BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) {
|
|
|
- for (int p = 0; p < positionCount; p++) {
|
|
|
- MvContains.process(result, p, field1Block, field2Block);
|
|
|
- }
|
|
|
- return result.build();
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- @Override
|
|
|
- public String toString() {
|
|
|
- return "MvContainsBytesRefEvaluator[" + "field1=" + field1 + ", field2=" + field2 + "]";
|
|
|
- }
|
|
|
-
|
|
|
- @Override
|
|
|
- public void close() {
|
|
|
- Releasables.closeExpectNoException(field1, field2);
|
|
|
- }
|
|
|
-
|
|
|
- public static class Factory implements EvalOperator.ExpressionEvaluator.Factory {
|
|
|
- private final Source source;
|
|
|
- private final EvalOperator.ExpressionEvaluator.Factory field1;
|
|
|
- private final EvalOperator.ExpressionEvaluator.Factory field2;
|
|
|
-
|
|
|
- public Factory(
|
|
|
- Source source,
|
|
|
- EvalOperator.ExpressionEvaluator.Factory field1,
|
|
|
- EvalOperator.ExpressionEvaluator.Factory field2
|
|
|
- ) {
|
|
|
- this.source = source;
|
|
|
- this.field1 = field1;
|
|
|
- this.field2 = field2;
|
|
|
- }
|
|
|
-
|
|
|
- @Override
|
|
|
- public MvContainsBytesRefEvaluator get(DriverContext context) {
|
|
|
- return new MvContainsBytesRefEvaluator(field1.get(context), field2.get(context), context);
|
|
|
- }
|
|
|
-
|
|
|
- @Override
|
|
|
- public String toString() {
|
|
|
- return "MvContainsBytesRefEvaluator[" + "field1=" + field1 + ", field2=" + field2 + "]";
|
|
|
- }
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- /**
|
|
|
- * Currently {@code EvaluatorImplementer} generates:
|
|
|
- * if (allBlocksAreNulls) {
|
|
|
- * result.appendNull();
|
|
|
- * continue position;
|
|
|
- * }
|
|
|
- * when all params are null, this violates our contract of always returning a boolean.
|
|
|
- * It should probably also generate the warnings method conditionally - omitted here.
|
|
|
- * TODO extend code generation to handle this case
|
|
|
- */
|
|
|
- public static class MvContainsDoubleEvaluator implements EvalOperator.ExpressionEvaluator {
|
|
|
- private final EvalOperator.ExpressionEvaluator field1;
|
|
|
- private final EvalOperator.ExpressionEvaluator field2;
|
|
|
- private final DriverContext driverContext;
|
|
|
-
|
|
|
- public MvContainsDoubleEvaluator(
|
|
|
- EvalOperator.ExpressionEvaluator field1,
|
|
|
- EvalOperator.ExpressionEvaluator field2,
|
|
|
- DriverContext driverContext
|
|
|
- ) {
|
|
|
- this.field1 = field1;
|
|
|
- this.field2 = field2;
|
|
|
- this.driverContext = driverContext;
|
|
|
- }
|
|
|
-
|
|
|
- @Override
|
|
|
- public Block eval(Page page) {
|
|
|
- try (DoubleBlock field1Block = (DoubleBlock) field1.eval(page)) {
|
|
|
- try (DoubleBlock field2Block = (DoubleBlock) field2.eval(page)) {
|
|
|
- return eval(page.getPositionCount(), field1Block, field2Block);
|
|
|
- }
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- public BooleanBlock eval(int positionCount, DoubleBlock field1Block, DoubleBlock field2Block) {
|
|
|
- try (BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) {
|
|
|
- for (int p = 0; p < positionCount; p++) {
|
|
|
- MvContains.process(result, p, field1Block, field2Block);
|
|
|
- }
|
|
|
- return result.build();
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- @Override
|
|
|
- public String toString() {
|
|
|
- return "MvContainsDoubleEvaluator[" + "field1=" + field1 + ", field2=" + field2 + "]";
|
|
|
- }
|
|
|
-
|
|
|
- @Override
|
|
|
- public void close() {
|
|
|
- Releasables.closeExpectNoException(field1, field2);
|
|
|
- }
|
|
|
-
|
|
|
- public static class Factory implements EvalOperator.ExpressionEvaluator.Factory {
|
|
|
- private final Source source;
|
|
|
- private final EvalOperator.ExpressionEvaluator.Factory field1;
|
|
|
- private final EvalOperator.ExpressionEvaluator.Factory field2;
|
|
|
-
|
|
|
- public Factory(
|
|
|
- Source source,
|
|
|
- EvalOperator.ExpressionEvaluator.Factory field1,
|
|
|
- EvalOperator.ExpressionEvaluator.Factory field2
|
|
|
- ) {
|
|
|
- this.source = source;
|
|
|
- this.field1 = field1;
|
|
|
- this.field2 = field2;
|
|
|
- }
|
|
|
-
|
|
|
- @Override
|
|
|
- public MvContainsDoubleEvaluator get(DriverContext context) {
|
|
|
- return new MvContainsDoubleEvaluator(field1.get(context), field2.get(context), context);
|
|
|
- }
|
|
|
-
|
|
|
- @Override
|
|
|
- public String toString() {
|
|
|
- return "MvContainsDoubleEvaluator[" + "field1=" + field1 + ", field2=" + field2 + "]";
|
|
|
- }
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- /**
|
|
|
- * Currently {@code EvaluatorImplementer} generates:
|
|
|
- * if (allBlocksAreNulls) {
|
|
|
- * result.appendNull();
|
|
|
- * continue position;
|
|
|
- * }
|
|
|
- * when all params are null, this violates our contract of always returning a boolean.
|
|
|
- * It should probably also generate the warnings method conditionally - omitted here.
|
|
|
- * TODO extend code generation to handle this case
|
|
|
- */
|
|
|
- public static class MvContainsIntEvaluator implements EvalOperator.ExpressionEvaluator {
|
|
|
- private final EvalOperator.ExpressionEvaluator field1;
|
|
|
- private final EvalOperator.ExpressionEvaluator field2;
|
|
|
- private final DriverContext driverContext;
|
|
|
-
|
|
|
- public MvContainsIntEvaluator(
|
|
|
- EvalOperator.ExpressionEvaluator field1,
|
|
|
- EvalOperator.ExpressionEvaluator field2,
|
|
|
- DriverContext driverContext
|
|
|
- ) {
|
|
|
- this.field1 = field1;
|
|
|
- this.field2 = field2;
|
|
|
- this.driverContext = driverContext;
|
|
|
- }
|
|
|
-
|
|
|
- @Override
|
|
|
- public Block eval(Page page) {
|
|
|
- try (IntBlock field1Block = (IntBlock) field1.eval(page)) {
|
|
|
- try (IntBlock field2Block = (IntBlock) field2.eval(page)) {
|
|
|
- return eval(page.getPositionCount(), field1Block, field2Block);
|
|
|
- }
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- public BooleanBlock eval(int positionCount, IntBlock field1Block, IntBlock field2Block) {
|
|
|
- try (BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) {
|
|
|
- for (int p = 0; p < positionCount; p++) {
|
|
|
- MvContains.process(result, p, field1Block, field2Block);
|
|
|
- }
|
|
|
- return result.build();
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- @Override
|
|
|
- public String toString() {
|
|
|
- return "MvContainsIntEvaluator[" + "field1=" + field1 + ", field2=" + field2 + "]";
|
|
|
- }
|
|
|
-
|
|
|
- @Override
|
|
|
- public void close() {
|
|
|
- Releasables.closeExpectNoException(field1, field2);
|
|
|
- }
|
|
|
-
|
|
|
- public static class Factory implements EvalOperator.ExpressionEvaluator.Factory {
|
|
|
- private final Source source;
|
|
|
- private final EvalOperator.ExpressionEvaluator.Factory field1;
|
|
|
- private final EvalOperator.ExpressionEvaluator.Factory field2;
|
|
|
-
|
|
|
- public Factory(
|
|
|
- Source source,
|
|
|
- EvalOperator.ExpressionEvaluator.Factory field1,
|
|
|
- EvalOperator.ExpressionEvaluator.Factory field2
|
|
|
- ) {
|
|
|
- this.source = source;
|
|
|
- this.field1 = field1;
|
|
|
- this.field2 = field2;
|
|
|
- }
|
|
|
-
|
|
|
- @Override
|
|
|
- public MvContainsIntEvaluator get(DriverContext context) {
|
|
|
- return new MvContainsIntEvaluator(field1.get(context), field2.get(context), context);
|
|
|
- }
|
|
|
-
|
|
|
- @Override
|
|
|
- public String toString() {
|
|
|
- return "MvContainsIntEvaluator[" + "field1=" + field1 + ", field2=" + field2 + "]";
|
|
|
- }
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- /**
|
|
|
- * Currently {@code EvaluatorImplementer} generates:
|
|
|
- * if (allBlocksAreNulls) {
|
|
|
- * result.appendNull();
|
|
|
- * continue position;
|
|
|
- * }
|
|
|
- * when all params are null, this violates our contract of always returning a boolean.
|
|
|
- * It should probably also generate the warnings method conditionally - omitted here.
|
|
|
- * TODO extend code generation to handle this case
|
|
|
- */
|
|
|
- public static class MvContainsLongEvaluator implements EvalOperator.ExpressionEvaluator {
|
|
|
- private final EvalOperator.ExpressionEvaluator field1;
|
|
|
- private final EvalOperator.ExpressionEvaluator field2;
|
|
|
- private final DriverContext driverContext;
|
|
|
-
|
|
|
- public MvContainsLongEvaluator(
|
|
|
- EvalOperator.ExpressionEvaluator field1,
|
|
|
- EvalOperator.ExpressionEvaluator field2,
|
|
|
- DriverContext driverContext
|
|
|
- ) {
|
|
|
- this.field1 = field1;
|
|
|
- this.field2 = field2;
|
|
|
- this.driverContext = driverContext;
|
|
|
- }
|
|
|
-
|
|
|
- @Override
|
|
|
- public Block eval(Page page) {
|
|
|
- try (LongBlock field1Block = (LongBlock) field1.eval(page)) {
|
|
|
- try (LongBlock field2Block = (LongBlock) field2.eval(page)) {
|
|
|
- return eval(page.getPositionCount(), field1Block, field2Block);
|
|
|
- }
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- public BooleanBlock eval(int positionCount, LongBlock field1Block, LongBlock field2Block) {
|
|
|
- try (BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) {
|
|
|
- for (int p = 0; p < positionCount; p++) {
|
|
|
- MvContains.process(result, p, field1Block, field2Block);
|
|
|
- }
|
|
|
- return result.build();
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- @Override
|
|
|
- public String toString() {
|
|
|
- return "MvContainsLongEvaluator[" + "field1=" + field1 + ", field2=" + field2 + "]";
|
|
|
- }
|
|
|
-
|
|
|
- @Override
|
|
|
- public void close() {
|
|
|
- Releasables.closeExpectNoException(field1, field2);
|
|
|
- }
|
|
|
-
|
|
|
- public static class Factory implements EvalOperator.ExpressionEvaluator.Factory {
|
|
|
- private final Source source;
|
|
|
- private final EvalOperator.ExpressionEvaluator.Factory field1;
|
|
|
- private final EvalOperator.ExpressionEvaluator.Factory field2;
|
|
|
-
|
|
|
- public Factory(
|
|
|
- Source source,
|
|
|
- EvalOperator.ExpressionEvaluator.Factory field1,
|
|
|
- EvalOperator.ExpressionEvaluator.Factory field2
|
|
|
- ) {
|
|
|
- this.source = source;
|
|
|
- this.field1 = field1;
|
|
|
- this.field2 = field2;
|
|
|
- }
|
|
|
-
|
|
|
- @Override
|
|
|
- public MvContainsLongEvaluator get(DriverContext context) {
|
|
|
- return new MvContainsLongEvaluator(field1.get(context), field2.get(context), context);
|
|
|
- }
|
|
|
-
|
|
|
- @Override
|
|
|
- public String toString() {
|
|
|
- return "MvContainsLongEvaluator[" + "field1=" + field1 + ", field2=" + field2 + "]";
|
|
|
- }
|
|
|
- }
|
|
|
- }
|
|
|
-}
|
David Turner
