Skip to content

Commit

Permalink
Fix bug
Browse files Browse the repository at this point in the history
  • Loading branch information
nik9000 committed Sep 30, 2024
1 parent 1c93400 commit 99bcb6a
Show file tree
Hide file tree
Showing 18 changed files with 36 additions and 17 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -390,7 +390,7 @@ private MethodSpec addRawVector(boolean masked) {
builder.addParameter(BOOLEAN_VECTOR, "mask");
}

if (stateTypeHasSeen) {
if (stateTypeHasSeen && masked == false) {
builder.addStatement("state.seen(true)");
}
if (valuesIsBytesRef) {
Expand All @@ -403,6 +403,9 @@ private MethodSpec addRawVector(boolean masked) {
if (masked) {
builder.beginControlFlow("if (mask.getBoolean(i) == false)").addStatement("continue").endControlFlow();
}
if (stateTypeHasSeen && masked) {
builder.addStatement("state.seen(true)");
}
combineRawInput(builder, "vector");
}
builder.endControlFlow();
Expand Down

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

Original file line number Diff line number Diff line change
Expand Up @@ -9,9 +9,12 @@

import org.elasticsearch.compute.data.Block;
import org.elasticsearch.compute.data.BlockFactory;
import org.elasticsearch.compute.data.BooleanVector;
import org.elasticsearch.compute.data.Page;
import org.elasticsearch.compute.operator.EvalOperator;

import static org.elasticsearch.test.ESTestCase.randomBoolean;

/**
* An {@link EvalOperator.ExpressionEvaluator} that evaluates to a constant boolean value.
*/
Expand All @@ -22,7 +25,15 @@ public static EvalOperator.ExpressionEvaluator.Factory factory(boolean value) {

@Override
public Block eval(Page page) {
return factory.newConstantBooleanVector(value, page.getPositionCount()).asBlock();
if (randomBoolean()) {
return factory.newConstantBooleanVector(value, page.getPositionCount()).asBlock();
}
try (BooleanVector.Builder builder = factory.newBooleanVectorFixedBuilder(page.getPositionCount())) {
for (int p = 0; p < page.getPositionCount(); p++) {
builder.appendBoolean(value);
}
return builder.build().asBlock();
}
}

@Override
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -190,7 +190,7 @@ public void testAllFiltered() {
assertOutputFromEmpty(results.get(0).getBlock(0));
}

public final void testNoneFiltered() {
public void testNoneFiltered() {
Operator.OperatorFactory factory = simpleWithMode(
AggregatorMode.SINGLE,
agg -> new FilteredAggregatorFunctionSupplier(agg, ConstantBooleanExpressionEvaluator.factory(true))
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -94,6 +94,11 @@ public void checkUnclosed() {
assertThat(unclosed, empty());
}

@Override
public void testNoneFiltered() {
assumeFalse("can't double filter. tests already filter.", true);
}

@Override
public void testAllFiltered() {
assumeFalse("can't double filter. tests already filter.", true);
Expand Down

0 comments on commit 99bcb6a

Please sign in to comment.