Skip to content

Commit c3baa17

Browse files
committed
Request channels in aggregator supplier methods and factory
1 parent 82e1496 commit c3baa17

File tree

86 files changed

+396
-515
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

86 files changed

+396
-515
lines changed

x-pack/plugin/esql/compute/gen/src/main/java/org/elasticsearch/compute/gen/AggregatorFunctionSupplierImplementer.java

Lines changed: 4 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -68,7 +68,6 @@ public AggregatorFunctionSupplierImplementer(
6868
createParameters.addAll(groupingAggregatorImplementer.createParameters());
6969
}
7070
this.createParameters = new ArrayList<>(createParameters);
71-
this.createParameters.add(0, new Parameter(LIST_INTEGER, "channels"));
7271

7372
this.implementation = ClassName.get(
7473
elements.getPackageOf(declarationType).toString(),
@@ -132,10 +131,7 @@ private MethodSpec nonGroupingIntermediateStateDesc() {
132131
return builder.build();
133132
}
134133

135-
builder.addStatement(
136-
"return $T.intermediateStateDesc()",
137-
aggregatorImplementer.implementation()
138-
);
134+
builder.addStatement("return $T.intermediateStateDesc()", aggregatorImplementer.implementation());
139135

140136
return builder.build();
141137
}
@@ -145,10 +141,7 @@ private MethodSpec groupingIntermediateStateDesc() {
145141
builder.addAnnotation(Override.class).addModifiers(Modifier.PUBLIC);
146142
builder.returns(LIST_AGG_FUNC_DESC);
147143

148-
builder.addStatement(
149-
"return $T.intermediateStateDesc()",
150-
groupingAggregatorImplementer.implementation()
151-
);
144+
builder.addStatement("return $T.intermediateStateDesc()", groupingAggregatorImplementer.implementation());
152145

153146
return builder.build();
154147
}
@@ -157,6 +150,7 @@ private MethodSpec aggregator() {
157150
MethodSpec.Builder builder = MethodSpec.methodBuilder("aggregator");
158151
builder.addAnnotation(Override.class).addModifiers(Modifier.PUBLIC);
159152
builder.addParameter(DRIVER_CONTEXT, "driverContext");
153+
builder.addParameter(LIST_INTEGER, "channels");
160154

161155
if (aggregatorImplementer == null) {
162156
builder.returns(Types.AGGREGATOR_FUNCTION);
@@ -183,14 +177,14 @@ private MethodSpec aggregator() {
183177
).collect(Collectors.joining(", "))
184178
);
185179

186-
187180
return builder.build();
188181
}
189182

190183
private MethodSpec groupingAggregator() {
191184
MethodSpec.Builder builder = MethodSpec.methodBuilder("groupingAggregator");
192185
builder.addAnnotation(Override.class).addModifiers(Modifier.PUBLIC);
193186
builder.addParameter(DRIVER_CONTEXT, "driverContext");
187+
builder.addParameter(LIST_INTEGER, "channels");
194188
builder.returns(groupingAggregatorImplementer.implementation());
195189

196190
if (hasWarnings) {

x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/CountDistinctBooleanAggregatorFunctionSupplier.java

Lines changed: 4 additions & 6 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/CountDistinctBytesRefAggregatorFunctionSupplier.java

Lines changed: 4 additions & 6 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/CountDistinctDoubleAggregatorFunctionSupplier.java

Lines changed: 4 additions & 6 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/CountDistinctFloatAggregatorFunctionSupplier.java

Lines changed: 4 additions & 6 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/CountDistinctIntAggregatorFunctionSupplier.java

Lines changed: 5 additions & 7 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/CountDistinctLongAggregatorFunctionSupplier.java

Lines changed: 5 additions & 7 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MaxBooleanAggregatorFunctionSupplier.java

Lines changed: 5 additions & 6 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MaxBytesRefAggregatorFunctionSupplier.java

Lines changed: 5 additions & 6 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MaxDoubleAggregatorFunctionSupplier.java

Lines changed: 5 additions & 6 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

0 commit comments

Comments
 (0)