/
ExpressionDoubleValuesScript.java
88 lines (76 loc) · 3.09 KB
/
ExpressionDoubleValuesScript.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
/*
* 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 and the Server Side Public License, v 1; you may not use this file except
* in compliance with, at your election, the Elastic License 2.0 or the Server
* Side Public License, v 1.
*/
package org.elasticsearch.script.expression;
import org.apache.lucene.expressions.Bindings;
import org.apache.lucene.expressions.Expression;
import org.apache.lucene.search.DoubleValues;
import org.apache.lucene.search.DoubleValuesSource;
import org.apache.lucene.search.Rescorer;
import org.apache.lucene.search.SortField;
import org.elasticsearch.script.DoubleValuesScript;
import java.util.function.Function;
/**
* A factory for a custom compiled {@link Expression} scripts
* <p>
* Instead of an execution result, we return a wrapper to an {@link Expression} object, which
* can be used for all supported double values operations.
*/
public class ExpressionDoubleValuesScript implements DoubleValuesScript.Factory {
private final Expression exprScript;
ExpressionDoubleValuesScript(Expression e) {
this.exprScript = e;
}
@Override
public DoubleValuesScript newInstance() {
return new DoubleValuesScript() {
@Override
public double execute() {
return exprScript.evaluate(new DoubleValues[0]);
}
@Override
public double evaluate(DoubleValues[] functionValues) {
return exprScript.evaluate(functionValues);
}
@Override
public DoubleValuesSource getDoubleValuesSource(Function<String, DoubleValuesSource> sourceProvider) {
return exprScript.getDoubleValuesSource(new Bindings() {
@Override
public DoubleValuesSource getDoubleValuesSource(String name) {
return sourceProvider.apply(name);
}
});
}
@Override
public SortField getSortField(Function<String, DoubleValuesSource> sourceProvider, boolean reverse) {
return exprScript.getSortField(new Bindings() {
@Override
public DoubleValuesSource getDoubleValuesSource(String name) {
return sourceProvider.apply(name);
}
}, reverse);
}
@Override
public Rescorer getRescorer(Function<String, DoubleValuesSource> sourceProvider) {
return exprScript.getRescorer(new Bindings() {
@Override
public DoubleValuesSource getDoubleValuesSource(String name) {
return sourceProvider.apply(name);
}
});
}
@Override
public String sourceText() {
return exprScript.sourceText;
}
@Override
public String[] variables() {
return exprScript.variables;
}
};
}
}