Skip to content

Commit 7e67a7b

Browse files
committed
minor refactoring
1 parent 6356809 commit 7e67a7b

File tree

5 files changed

+13
-20
lines changed

5 files changed

+13
-20
lines changed

x-pack/plugin/esql/src/internalClusterTest/java/org/elasticsearch/xpack/esql/plugin/ScoreFunctionIT.java renamed to x-pack/plugin/esql/src/internalClusterTest/java/org/elasticsearch/xpack/esql/plugin/ScoreIT.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@
2222
import static org.elasticsearch.test.hamcrest.ElasticsearchAssertions.assertAcked;
2323

2424
//@TestLogging(value = "org.elasticsearch.xpack.esql:TRACE,org.elasticsearch.compute:TRACE", reason = "debug")
25-
public class ScoreFunctionIT extends AbstractEsqlIntegTestCase {
25+
public class ScoreIT extends AbstractEsqlIntegTestCase {
2626

2727
@Before
2828
public void setupIndex() {

x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/expression/function/EsqlFunctionRegistry.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,7 @@
4444
import org.elasticsearch.xpack.esql.expression.function.fulltext.Match;
4545
import org.elasticsearch.xpack.esql.expression.function.fulltext.MultiMatch;
4646
import org.elasticsearch.xpack.esql.expression.function.fulltext.QueryString;
47-
import org.elasticsearch.xpack.esql.expression.function.fulltext.ScoreFunction;
47+
import org.elasticsearch.xpack.esql.expression.function.fulltext.Score;
4848
import org.elasticsearch.xpack.esql.expression.function.fulltext.Term;
4949
import org.elasticsearch.xpack.esql.expression.function.grouping.Bucket;
5050
import org.elasticsearch.xpack.esql.expression.function.grouping.Categorize;
@@ -454,7 +454,7 @@ private static FunctionDefinition[][] snapshotFunctions() {
454454
def(MinOverTime.class, uni(MinOverTime::new), "min_over_time"),
455455
def(AvgOverTime.class, uni(AvgOverTime::new), "avg_over_time"),
456456
def(LastOverTime.class, LastOverTime::withUnresolvedTimestamp, "last_over_time"),
457-
def(ScoreFunction.class, uni(ScoreFunction::new), ScoreFunction.NAME),
457+
def(Score.class, uni(Score::new), Score.NAME),
458458
def(FirstOverTime.class, FirstOverTime::withUnresolvedTimestamp, "first_over_time"),
459459
def(Term.class, bi(Term::new), "term") } };
460460
}

x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/expression/function/fulltext/FullTextFunction.java

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -232,10 +232,7 @@ private static void checkFullTextQueryFunctions(LogicalPlan plan, Failures failu
232232
} else {
233233
// TODO : improve this check, as this is not so nice :(
234234
List<FullTextFunction> scoredFTFs = new ArrayList<>();
235-
plan.forEachExpression(
236-
ScoreFunction.class,
237-
scoreFunction -> { plan.forEachExpression(FullTextFunction.class, scoredFTFs::add); }
238-
);
235+
plan.forEachExpression(Score.class, scoreFunction -> { plan.forEachExpression(FullTextFunction.class, scoredFTFs::add); });
239236
plan.forEachExpression(FullTextFunction.class, ftf -> {
240237
if (scoredFTFs.remove(ftf) == false) {
241238
failures.add(

x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/expression/function/fulltext/FullTextWritables.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ public static List<NamedWriteableRegistry.Entry> getNamedWriteables() {
2828
entries.add(Term.ENTRY);
2929
}
3030
if (EsqlCapabilities.Cap.SCORE_FUNCTION.isEnabled()) {
31-
entries.add(ScoreFunction.ENTRY);
31+
entries.add(Score.ENTRY);
3232
}
3333

3434
return Collections.unmodifiableList(entries);

x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/expression/function/fulltext/ScoreFunction.java renamed to x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/expression/function/fulltext/Score.java

Lines changed: 8 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -34,30 +34,26 @@
3434
* A function to be used to score specific portions of an ES|QL query e.g., in conjunction with
3535
* an {@link org.elasticsearch.xpack.esql.plan.logical.Eval}.
3636
*/
37-
public class ScoreFunction extends Function implements EvaluatorMapper {
37+
public class Score extends Function implements EvaluatorMapper {
3838

39-
public static final NamedWriteableRegistry.Entry ENTRY = new NamedWriteableRegistry.Entry(
40-
Expression.class,
41-
"score",
42-
ScoreFunction::readFrom
43-
);
39+
public static final NamedWriteableRegistry.Entry ENTRY = new NamedWriteableRegistry.Entry(Expression.class, "score", Score::readFrom);
4440

4541
public static final String NAME = "score";
4642

4743
@FunctionInfo(
4844
returnType = "double",
4945
preview = true,
50-
description = "Scores a full text function. Returns scores for all the matching docs.",
46+
description = "Scores a full text function. Returns scores for all the resulting docs.",
5147
examples = { @Example(file = "score-function", tag = "score-single") }
5248
)
53-
public ScoreFunction(
49+
public Score(
5450
Source source,
5551
@Param(name = "query", type = { "keyword", "text" }, description = "full text function.") Expression scorableQuery
5652
) {
5753
this(source, List.of(scorableQuery));
5854
}
5955

60-
protected ScoreFunction(Source source, List<Expression> children) {
56+
protected Score(Source source, List<Expression> children) {
6157
super(source, children);
6258
}
6359

@@ -68,12 +64,12 @@ public DataType dataType() {
6864

6965
@Override
7066
public Expression replaceChildren(List<Expression> newChildren) {
71-
return new ScoreFunction(source(), newChildren);
67+
return new Score(source(), newChildren);
7268
}
7369

7470
@Override
7571
protected NodeInfo<? extends Expression> info() {
76-
return NodeInfo.create(this, ScoreFunction::new, children().getFirst());
72+
return NodeInfo.create(this, Score::new, children().getFirst());
7773
}
7874

7975
@Override
@@ -96,7 +92,7 @@ public void writeTo(StreamOutput out) throws IOException {
9692
private static Expression readFrom(StreamInput in) throws IOException {
9793
Source source = Source.readFrom((PlanStreamInput) in);
9894
Expression query = in.readOptionalNamedWriteable(Expression.class);
99-
return new ScoreFunction(source, query);
95+
return new Score(source, query);
10096
}
10197

10298
private record ScorerEvaluatorFactory(ScoreOperator.ExpressionScorer.Factory scoreFactory)

0 commit comments

Comments
 (0)