@Override
public SignificanceHeuristic parse(XContentParser parser)
throws IOException, QueryParsingException {
NAMES_FIELD.match(parser.currentName(), ParseField.EMPTY_FLAGS);
String script = null;
String scriptLang;
XContentParser.Token token;
Map<String, Object> params = new HashMap<>();
String currentFieldName = null;
ScriptService.ScriptType scriptType = ScriptService.ScriptType.INLINE;
ScriptParameterParser scriptParameterParser = new ScriptParameterParser();
while ((token = parser.nextToken()) != XContentParser.Token.END_OBJECT) {
if (token.equals(XContentParser.Token.FIELD_NAME)) {
currentFieldName = parser.currentName();
} else if (token == XContentParser.Token.START_OBJECT) {
if ("params".equals(currentFieldName)) {
params = parser.map();
} else {
throw new ElasticsearchParseException(
"unknown object " + currentFieldName + " in script_heuristic");
}
} else if (!scriptParameterParser.token(currentFieldName, token, parser)) {
throw new ElasticsearchParseException(
"unknown field " + currentFieldName + " in script_heuristic");
}
}
ScriptParameterParser.ScriptParameterValue scriptValue =
scriptParameterParser.getDefaultScriptParameterValue();
if (scriptValue != null) {
script = scriptValue.script();
scriptType = scriptValue.scriptType();
}
scriptLang = scriptParameterParser.lang();
if (script == null) {
throw new ElasticsearchParseException("No script found in script_heuristic");
}
ExecutableScript searchScript;
try {
searchScript =
scriptService.executable(
scriptLang, script, scriptType, ScriptContext.Standard.AGGS, params);
} catch (Exception e) {
throw new ElasticsearchParseException("The script [" + script + "] could not be loaded", e);
}
return new ScriptHeuristic(searchScript, scriptLang, script, scriptType, params);
}
@Override
public void parse(XContentParser parser, SearchContext context) throws Exception {
XContentParser.Token token;
String currentFieldName = null;
while ((token = parser.nextToken()) != XContentParser.Token.END_OBJECT) {
if (token == XContentParser.Token.FIELD_NAME) {
currentFieldName = parser.currentName();
} else if (token == XContentParser.Token.START_OBJECT) {
String fieldName = currentFieldName;
ScriptParameterParser scriptParameterParser = new ScriptParameterParser();
String script = null;
ScriptService.ScriptType scriptType = null;
Map<String, Object> params = null;
boolean ignoreException = false;
while ((token = parser.nextToken()) != XContentParser.Token.END_OBJECT) {
if (token == XContentParser.Token.FIELD_NAME) {
currentFieldName = parser.currentName();
} else if (token == XContentParser.Token.START_OBJECT) {
params = parser.map();
} else if (token.isValue()) {
if ("ignore_failure".equals(currentFieldName)) {
ignoreException = parser.booleanValue();
} else {
scriptParameterParser.token(currentFieldName, token, parser);
}
}
}
ScriptParameterValue scriptValue = scriptParameterParser.getDefaultScriptParameterValue();
if (scriptValue != null) {
script = scriptValue.script();
scriptType = scriptValue.scriptType();
}
SearchScript searchScript =
context
.scriptService()
.search(
context.lookup(),
new Script(scriptParameterParser.lang(), script, scriptType, params),
ScriptContext.Standard.SEARCH);
context
.scriptFields()
.add(new ScriptFieldsContext.ScriptField(fieldName, searchScript, ignoreException));
}
}
}
@Override
public AggregatorFactory parse(
String aggregationName, XContentParser parser, SearchContext context) throws IOException {
Script initScript = null;
Script mapScript = null;
Script combineScript = null;
Script reduceScript = null;
Map<String, Object> params = null;
Map<String, Object> reduceParams = null;
XContentParser.Token token;
String currentFieldName = null;
Set<String> scriptParameters = new HashSet<>();
scriptParameters.add(INIT_SCRIPT);
scriptParameters.add(MAP_SCRIPT);
scriptParameters.add(COMBINE_SCRIPT);
scriptParameters.add(REDUCE_SCRIPT);
ScriptParameterParser scriptParameterParser = new ScriptParameterParser(scriptParameters);
while ((token = parser.nextToken()) != XContentParser.Token.END_OBJECT) {
if (token == XContentParser.Token.FIELD_NAME) {
currentFieldName = parser.currentName();
} else if (token == XContentParser.Token.START_OBJECT) {
if (INIT_SCRIPT_FIELD.match(currentFieldName)) {
initScript = Script.parse(parser);
} else if (MAP_SCRIPT_FIELD.match(currentFieldName)) {
mapScript = Script.parse(parser);
} else if (COMBINE_SCRIPT_FIELD.match(currentFieldName)) {
combineScript = Script.parse(parser);
} else if (REDUCE_SCRIPT_FIELD.match(currentFieldName)) {
reduceScript = Script.parse(parser);
} else if (PARAMS_FIELD.match(currentFieldName)) {
params = parser.map();
} else if (REDUCE_PARAMS_FIELD.match(currentFieldName)) {
reduceParams = parser.map();
} else {
throw new SearchParseException(
context,
"Unknown key for a "
+ token
+ " in ["
+ aggregationName
+ "]: ["
+ currentFieldName
+ "].",
parser.getTokenLocation());
}
} else if (token.isValue()) {
if (!scriptParameterParser.token(currentFieldName, token, parser)) {
throw new SearchParseException(
context,
"Unknown key for a "
+ token
+ " in ["
+ aggregationName
+ "]: ["
+ currentFieldName
+ "].",
parser.getTokenLocation());
}
} else {
throw new SearchParseException(
context,
"Unexpected token " + token + " in [" + aggregationName + "].",
parser.getTokenLocation());
}
}
if (initScript
== null) { // Didn't find anything using the new API so try using the old one instead
ScriptParameterValue scriptValue = scriptParameterParser.getScriptParameterValue(INIT_SCRIPT);
if (scriptValue != null) {
initScript =
new Script(
scriptValue.script(),
scriptValue.scriptType(),
scriptParameterParser.lang(),
params);
}
} else if (initScript.getParams() != null) {
throw new SearchParseException(
context,
"init_script params are not supported. Parameters for the init_script must be specified in the params field on the scripted_metric aggregator not inside the init_script object",
parser.getTokenLocation());
}
if (mapScript
== null) { // Didn't find anything using the new API so try using the old one instead
ScriptParameterValue scriptValue = scriptParameterParser.getScriptParameterValue(MAP_SCRIPT);
if (scriptValue != null) {
mapScript =
new Script(
scriptValue.script(),
scriptValue.scriptType(),
scriptParameterParser.lang(),
params);
}
} else if (mapScript.getParams() != null) {
throw new SearchParseException(
context,
"map_script params are not supported. Parameters for the map_script must be specified in the params field on the scripted_metric aggregator not inside the map_script object",
parser.getTokenLocation());
}
if (combineScript
== null) { // Didn't find anything using the new API so try using the old one instead
ScriptParameterValue scriptValue =
scriptParameterParser.getScriptParameterValue(COMBINE_SCRIPT);
if (scriptValue != null) {
combineScript =
new Script(
scriptValue.script(),
scriptValue.scriptType(),
scriptParameterParser.lang(),
params);
}
} else if (combineScript.getParams() != null) {
throw new SearchParseException(
context,
"combine_script params are not supported. Parameters for the combine_script must be specified in the params field on the scripted_metric aggregator not inside the combine_script object",
parser.getTokenLocation());
}
if (reduceScript
== null) { // Didn't find anything using the new API so try using the old one instead
ScriptParameterValue scriptValue =
scriptParameterParser.getScriptParameterValue(REDUCE_SCRIPT);
if (scriptValue != null) {
reduceScript =
new Script(
scriptValue.script(),
scriptValue.scriptType(),
scriptParameterParser.lang(),
reduceParams);
}
}
if (mapScript == null) {
throw new SearchParseException(
context,
"map_script field is required in [" + aggregationName + "].",
parser.getTokenLocation());
}
return new ScriptedMetricAggregator.Factory(
aggregationName, initScript, mapScript, combineScript, reduceScript, params);
}