Skip to content

Commit bc27466

Browse files
CR: Make BucketSelectorPipelineAggregator agnositc to lang
1 parent 05e6559 commit bc27466

File tree

1 file changed

+6
-18
lines changed

1 file changed

+6
-18
lines changed

server/src/main/java/org/elasticsearch/search/aggregations/pipeline/bucketselector/BucketSelectorPipelineAggregator.java

Lines changed: 6 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,6 @@
2323
import org.elasticsearch.common.io.stream.StreamInput;
2424
import org.elasticsearch.common.io.stream.StreamOutput;
2525
import org.elasticsearch.script.BucketAggregationSelectorScript;
26-
import org.elasticsearch.script.BucketAggregationScript;
2726
import org.elasticsearch.script.Script;
2827
import org.elasticsearch.search.aggregations.InternalAggregation;
2928
import org.elasticsearch.search.aggregations.InternalAggregation.ReduceContext;
@@ -84,23 +83,12 @@ public InternalAggregation reduce(InternalAggregation aggregation, ReduceContext
8483
List<? extends InternalMultiBucketAggregation.InternalBucket> buckets = originalAgg.getBuckets();
8584

8685
List<InternalMultiBucketAggregation.InternalBucket> newBuckets = new ArrayList<>();
87-
if ("expression".equals(script.getLang())) {
88-
BucketAggregationScript.Factory factory =
89-
reduceContext.scriptService().compile(script, BucketAggregationScript.CONTEXT);
90-
BucketAggregationScript executableScript = factory.newInstance();
91-
for (InternalMultiBucketAggregation.InternalBucket bucket : buckets) {
92-
if (executableScript.execute(scriptArgs(originalAgg, bucket)) == 1.0) {
93-
newBuckets.add(bucket);
94-
}
95-
}
96-
} else {
97-
BucketAggregationSelectorScript.Factory factory =
98-
reduceContext.scriptService().compile(script, BucketAggregationSelectorScript.CONTEXT);
99-
BucketAggregationSelectorScript executableScript = factory.newInstance();
100-
for (InternalMultiBucketAggregation.InternalBucket bucket : buckets) {
101-
if (executableScript.execute(scriptArgs(originalAgg, bucket))) {
102-
newBuckets.add(bucket);
103-
}
86+
BucketAggregationSelectorScript.Factory factory =
87+
reduceContext.scriptService().compile(script, BucketAggregationSelectorScript.CONTEXT);
88+
BucketAggregationSelectorScript executableScript = factory.newInstance();
89+
for (InternalMultiBucketAggregation.InternalBucket bucket : buckets) {
90+
if (executableScript.execute(scriptArgs(originalAgg, bucket))) {
91+
newBuckets.add(bucket);
10492
}
10593
}
10694
return originalAgg.create(newBuckets);

0 commit comments

Comments
 (0)