-
Notifications
You must be signed in to change notification settings - Fork 24.9k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Two queries for keyword script field (#59527)
This adds the `exits` and `terms` query for the `keyword` style of scripted field.
- Loading branch information
Showing
11 changed files
with
509 additions
and
60 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
91 changes: 91 additions & 0 deletions
91
...main/java/org/elasticsearch/xpack/runtimefields/query/AbstractStringScriptFieldQuery.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,91 @@ | ||
| /* | ||
| * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one | ||
| * or more contributor license agreements. Licensed under the Elastic License; | ||
| * you may not use this file except in compliance with the Elastic License. | ||
| */ | ||
|
|
||
| package org.elasticsearch.xpack.runtimefields.query; | ||
|
|
||
| import org.apache.lucene.index.LeafReaderContext; | ||
| import org.apache.lucene.search.ConstantScoreScorer; | ||
| import org.apache.lucene.search.ConstantScoreWeight; | ||
| import org.apache.lucene.search.DocIdSetIterator; | ||
| import org.apache.lucene.search.IndexSearcher; | ||
| import org.apache.lucene.search.Query; | ||
| import org.apache.lucene.search.ScoreMode; | ||
| import org.apache.lucene.search.Scorer; | ||
| import org.apache.lucene.search.TwoPhaseIterator; | ||
| import org.apache.lucene.search.Weight; | ||
| import org.elasticsearch.xpack.runtimefields.StringScriptFieldScript; | ||
| import org.elasticsearch.xpack.runtimefields.StringScriptFieldScript.LeafFactory; | ||
|
|
||
| import java.io.IOException; | ||
| import java.util.List; | ||
| import java.util.Objects; | ||
|
|
||
| /** | ||
| * Abstract base class for building queries based on {@link StringScriptFieldScript}. | ||
| */ | ||
| abstract class AbstractStringScriptFieldQuery extends Query { | ||
| private final StringScriptFieldScript.LeafFactory leafFactory; | ||
| private final String fieldName; | ||
|
|
||
| AbstractStringScriptFieldQuery(LeafFactory leafFactory, String fieldName) { | ||
| this.leafFactory = Objects.requireNonNull(leafFactory); | ||
| this.fieldName = Objects.requireNonNull(fieldName); | ||
| } | ||
|
|
||
| /** | ||
| * Does the value match this query? | ||
| */ | ||
| public abstract boolean matches(List<String> values); | ||
|
|
||
| @Override | ||
| public final Weight createWeight(IndexSearcher searcher, ScoreMode scoreMode, float boost) throws IOException { | ||
| return new ConstantScoreWeight(this, boost) { | ||
| @Override | ||
| public boolean isCacheable(LeafReaderContext ctx) { | ||
| return false; // scripts aren't really cacheable at this point | ||
| } | ||
|
|
||
| @Override | ||
| public Scorer scorer(LeafReaderContext ctx) throws IOException { | ||
| StringScriptFieldScript script = leafFactory.newInstance(ctx); | ||
| DocIdSetIterator approximation = DocIdSetIterator.all(ctx.reader().maxDoc()); | ||
| TwoPhaseIterator twoPhase = new TwoPhaseIterator(approximation) { | ||
| @Override | ||
| public boolean matches() throws IOException { | ||
| return AbstractStringScriptFieldQuery.this.matches(script.resultsForDoc(approximation().docID())); | ||
| } | ||
|
|
||
| @Override | ||
| public float matchCost() { | ||
| // TODO we don't have a good way of estimating the complexity of the script so we just go with 9000 | ||
| return 9000f; | ||
| } | ||
| }; | ||
| return new ConstantScoreScorer(this, score(), scoreMode, twoPhase); | ||
| } | ||
| }; | ||
| } | ||
|
|
||
| protected final String fieldName() { | ||
| return fieldName; | ||
| } | ||
|
|
||
| @Override | ||
| public int hashCode() { | ||
| // TODO should leafFactory be here? Something about the script probably should be! | ||
| return Objects.hash(getClass(), fieldName); | ||
| } | ||
|
|
||
| @Override | ||
| public boolean equals(Object obj) { | ||
| if (obj == null || getClass() != obj.getClass()) { | ||
| return false; | ||
| } | ||
| // TODO should leafFactory be here? Something about the script probably should be! | ||
| AbstractStringScriptFieldQuery other = (AbstractStringScriptFieldQuery) obj; | ||
| return fieldName.equals(other.fieldName); | ||
| } | ||
| } |
32 changes: 32 additions & 0 deletions
32
...c/main/java/org/elasticsearch/xpack/runtimefields/query/StringScriptFieldExistsQuery.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,32 @@ | ||
| /* | ||
| * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one | ||
| * or more contributor license agreements. Licensed under the Elastic License; | ||
| * you may not use this file except in compliance with the Elastic License. | ||
| */ | ||
|
|
||
| package org.elasticsearch.xpack.runtimefields.query; | ||
|
|
||
| import org.elasticsearch.xpack.runtimefields.StringScriptFieldScript; | ||
|
|
||
| import java.util.List; | ||
|
|
||
| public class StringScriptFieldExistsQuery extends AbstractStringScriptFieldQuery { | ||
| public StringScriptFieldExistsQuery(StringScriptFieldScript.LeafFactory leafFactory, String fieldName) { | ||
| super(leafFactory, fieldName); | ||
| } | ||
|
|
||
| @Override | ||
| public boolean matches(List<String> values) { | ||
| return false == values.isEmpty(); | ||
| } | ||
|
|
||
| @Override | ||
| public final String toString(String field) { | ||
| if (fieldName().contentEquals(field)) { | ||
| return "*"; | ||
| } | ||
| return fieldName() + ":*"; | ||
| } | ||
|
|
||
| // Superclass's equals and hashCode are great for this class | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
69 changes: 69 additions & 0 deletions
69
...rc/main/java/org/elasticsearch/xpack/runtimefields/query/StringScriptFieldTermsQuery.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,69 @@ | ||
| /* | ||
| * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one | ||
| * or more contributor license agreements. Licensed under the Elastic License; | ||
| * you may not use this file except in compliance with the Elastic License. | ||
| */ | ||
|
|
||
| package org.elasticsearch.xpack.runtimefields.query; | ||
|
|
||
| import org.apache.lucene.index.Term; | ||
| import org.apache.lucene.search.QueryVisitor; | ||
| import org.elasticsearch.xpack.runtimefields.StringScriptFieldScript; | ||
|
|
||
| import java.util.List; | ||
| import java.util.Objects; | ||
| import java.util.Set; | ||
|
|
||
| public class StringScriptFieldTermsQuery extends AbstractStringScriptFieldQuery { | ||
| private final Set<String> terms; | ||
|
|
||
| public StringScriptFieldTermsQuery(StringScriptFieldScript.LeafFactory leafFactory, String fieldName, Set<String> terms) { | ||
| super(leafFactory, fieldName); | ||
| this.terms = terms; | ||
| } | ||
|
|
||
| @Override | ||
| public boolean matches(List<String> values) { | ||
| for (String value : values) { | ||
| if (terms.contains(value)) { | ||
| return true; | ||
| } | ||
| } | ||
| return false; | ||
| } | ||
|
|
||
| @Override | ||
| public void visit(QueryVisitor visitor) { | ||
| if (visitor.acceptField(fieldName())) { | ||
| for (String term : terms) { | ||
| visitor.consumeTerms(this, new Term(fieldName(), term)); | ||
| } | ||
| } | ||
| } | ||
|
|
||
| @Override | ||
| public final String toString(String field) { | ||
| if (fieldName().contentEquals(field)) { | ||
| return terms.toString(); | ||
| } | ||
| return fieldName() + ":" + terms; | ||
| } | ||
|
|
||
| @Override | ||
| public int hashCode() { | ||
| return Objects.hash(super.hashCode(), terms); | ||
| } | ||
|
|
||
| @Override | ||
| public boolean equals(Object obj) { | ||
| if (false == super.equals(obj)) { | ||
| return false; | ||
| } | ||
| StringScriptFieldTermsQuery other = (StringScriptFieldTermsQuery) obj; | ||
| return terms.equals(other.terms); | ||
| } | ||
|
|
||
| Set<String> terms() { | ||
| return terms; | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.