summaryrefslogtreecommitdiff
path: root/core/src/main/java/org/elasticsearch/search/suggest/phrase/PhraseSuggestionBuilder.java
diff options
context:
space:
mode:
Diffstat (limited to 'core/src/main/java/org/elasticsearch/search/suggest/phrase/PhraseSuggestionBuilder.java')
-rw-r--r--core/src/main/java/org/elasticsearch/search/suggest/phrase/PhraseSuggestionBuilder.java152
1 files changed, 72 insertions, 80 deletions
diff --git a/core/src/main/java/org/elasticsearch/search/suggest/phrase/PhraseSuggestionBuilder.java b/core/src/main/java/org/elasticsearch/search/suggest/phrase/PhraseSuggestionBuilder.java
index a4793dfbda..299c57d8d2 100644
--- a/core/src/main/java/org/elasticsearch/search/suggest/phrase/PhraseSuggestionBuilder.java
+++ b/core/src/main/java/org/elasticsearch/search/suggest/phrase/PhraseSuggestionBuilder.java
@@ -60,8 +60,6 @@ public class PhraseSuggestionBuilder extends SuggestionBuilder<PhraseSuggestionB
private static final String SUGGESTION_NAME = "phrase";
- public static final PhraseSuggestionBuilder PROTOTYPE = new PhraseSuggestionBuilder("_na_");
-
protected static final ParseField MAXERRORS_FIELD = new ParseField("max_errors");
protected static final ParseField RWE_LIKELIHOOD_FIELD = new ParseField("real_word_error_likelihood");
protected static final ParseField SEPARATOR_FIELD = new ParseField("separator");
@@ -122,6 +120,76 @@ public class PhraseSuggestionBuilder extends SuggestionBuilder<PhraseSuggestionB
}
/**
+ * Read from a stream.
+ */
+ PhraseSuggestionBuilder(StreamInput in) throws IOException {
+ super(in);
+ maxErrors = in.readFloat();
+ realWordErrorLikelihood = in.readFloat();
+ confidence = in.readFloat();
+ gramSize = in.readOptionalVInt();
+ if (in.readBoolean()) {
+ model = in.readPhraseSuggestionSmoothingModel();
+ }
+ forceUnigrams = in.readBoolean();
+ tokenLimit = in.readVInt();
+ preTag = in.readOptionalString();
+ postTag = in.readOptionalString();
+ separator = in.readString();
+ if (in.readBoolean()) {
+ collateQuery = Template.readTemplate(in);
+ }
+ collateParams = in.readMap();
+ collatePrune = in.readOptionalBoolean();
+ int generatorsEntries = in.readVInt();
+ for (int i = 0; i < generatorsEntries; i++) {
+ String type = in.readString();
+ int numberOfGenerators = in.readVInt();
+ List<CandidateGenerator> generatorsList = new ArrayList<>(numberOfGenerators);
+ for (int g = 0; g < numberOfGenerators; g++) {
+ DirectCandidateGeneratorBuilder generator = new DirectCandidateGeneratorBuilder(in);
+ generatorsList.add(generator);
+ }
+ generators.put(type, generatorsList);
+ }
+ }
+
+ @Override
+ public void doWriteTo(StreamOutput out) throws IOException {
+ out.writeFloat(maxErrors);
+ out.writeFloat(realWordErrorLikelihood);
+ out.writeFloat(confidence);
+ out.writeOptionalVInt(gramSize);
+ boolean hasModel = model != null;
+ out.writeBoolean(hasModel);
+ if (hasModel) {
+ out.writePhraseSuggestionSmoothingModel(model);
+ }
+ out.writeBoolean(forceUnigrams);
+ out.writeVInt(tokenLimit);
+ out.writeOptionalString(preTag);
+ out.writeOptionalString(postTag);
+ out.writeString(separator);
+ if (collateQuery != null) {
+ out.writeBoolean(true);
+ collateQuery.writeTo(out);
+ } else {
+ out.writeBoolean(false);
+ }
+ out.writeMap(collateParams);
+ out.writeOptionalBoolean(collatePrune);
+ out.writeVInt(this.generators.size());
+ for (Entry<String, List<CandidateGenerator>> entry : this.generators.entrySet()) {
+ out.writeString(entry.getKey());
+ List<CandidateGenerator> generatorsList = entry.getValue();
+ out.writeVInt(generatorsList.size());
+ for (CandidateGenerator generator : generatorsList) {
+ generator.writeTo(out);
+ }
+ }
+ }
+
+ /**
* Sets the gram size for the n-gram model used for this suggester. The
* default value is <tt>1</tt> corresponding to <tt>unigrams</tt>. Use
* <tt>2</tt> for <tt>bigrams</tt> and <tt>3</tt> for <tt>trigrams</tt>.
@@ -422,8 +490,7 @@ public class PhraseSuggestionBuilder extends SuggestionBuilder<PhraseSuggestionB
return builder;
}
- @Override
- protected PhraseSuggestionBuilder innerFromXContent(QueryParseContext parseContext) throws IOException {
+ static PhraseSuggestionBuilder innerFromXContent(QueryParseContext parseContext) throws IOException {
XContentParser parser = parseContext.parser();
PhraseSuggestionBuilder tmpSuggestion = new PhraseSuggestionBuilder("_na_");
ParseFieldMatcher parseFieldMatcher = parseContext.parseFieldMatcher();
@@ -464,7 +531,7 @@ public class PhraseSuggestionBuilder extends SuggestionBuilder<PhraseSuggestionB
if (parseFieldMatcher.match(currentFieldName, DirectCandidateGeneratorBuilder.DIRECT_GENERATOR_FIELD)) {
// for now we only have a single type of generators
while ((token = parser.nextToken()) == Token.START_OBJECT) {
- tmpSuggestion.addCandidateGenerator(DirectCandidateGeneratorBuilder.PROTOTYPE.fromXContent(parseContext));
+ tmpSuggestion.addCandidateGenerator(DirectCandidateGeneratorBuilder.fromXContent(parseContext));
}
} else {
throw new ParsingException(parser.getTokenLocation(),
@@ -578,10 +645,6 @@ public class PhraseSuggestionBuilder extends SuggestionBuilder<PhraseSuggestionB
suggestionContext.setCollatePrune(this.collatePrune);
}
- if (suggestionContext.model() == null) {
- suggestionContext.setModel(StupidBackoffScorer.FACTORY);
- }
-
if (this.gramSize == null || suggestionContext.generators().isEmpty()) {
final ShingleTokenFilterFactory.Factory shingleFilterFactory = SuggestUtils
.getShingleFilterFactory(suggestionContext.getAnalyzer());
@@ -624,75 +687,6 @@ public class PhraseSuggestionBuilder extends SuggestionBuilder<PhraseSuggestionB
}
@Override
- public void doWriteTo(StreamOutput out) throws IOException {
- out.writeFloat(maxErrors);
- out.writeFloat(realWordErrorLikelihood);
- out.writeFloat(confidence);
- out.writeOptionalVInt(gramSize);
- boolean hasModel = model != null;
- out.writeBoolean(hasModel);
- if (hasModel) {
- out.writePhraseSuggestionSmoothingModel(model);
- }
- out.writeBoolean(forceUnigrams);
- out.writeVInt(tokenLimit);
- out.writeOptionalString(preTag);
- out.writeOptionalString(postTag);
- out.writeString(separator);
- if (collateQuery != null) {
- out.writeBoolean(true);
- collateQuery.writeTo(out);
- } else {
- out.writeBoolean(false);
- }
- out.writeMap(collateParams);
- out.writeOptionalBoolean(collatePrune);
- out.writeVInt(this.generators.size());
- for (Entry<String, List<CandidateGenerator>> entry : this.generators.entrySet()) {
- out.writeString(entry.getKey());
- List<CandidateGenerator> generatorsList = entry.getValue();
- out.writeVInt(generatorsList.size());
- for (CandidateGenerator generator : generatorsList) {
- generator.writeTo(out);
- }
- }
- }
-
- @Override
- public PhraseSuggestionBuilder doReadFrom(StreamInput in, String field) throws IOException {
- PhraseSuggestionBuilder builder = new PhraseSuggestionBuilder(field);
- builder.maxErrors = in.readFloat();
- builder.realWordErrorLikelihood = in.readFloat();
- builder.confidence = in.readFloat();
- builder.gramSize = in.readOptionalVInt();
- if (in.readBoolean()) {
- builder.model = in.readPhraseSuggestionSmoothingModel();
- }
- builder.forceUnigrams = in.readBoolean();
- builder.tokenLimit = in.readVInt();
- builder.preTag = in.readOptionalString();
- builder.postTag = in.readOptionalString();
- builder.separator = in.readString();
- if (in.readBoolean()) {
- builder.collateQuery = Template.readTemplate(in);
- }
- builder.collateParams = in.readMap();
- builder.collatePrune = in.readOptionalBoolean();
- int generatorsEntries = in.readVInt();
- for (int i = 0; i < generatorsEntries; i++) {
- String type = in.readString();
- int numberOfGenerators = in.readVInt();
- List<CandidateGenerator> generatorsList = new ArrayList<>(numberOfGenerators);
- for (int g = 0; g < numberOfGenerators; g++) {
- DirectCandidateGeneratorBuilder generator = DirectCandidateGeneratorBuilder.PROTOTYPE.readFrom(in);
- generatorsList.add(generator);
- }
- builder.generators.put(type, generatorsList);
- }
- return builder;
- }
-
- @Override
protected boolean doEquals(PhraseSuggestionBuilder other) {
return Objects.equals(maxErrors, other.maxErrors) &&
Objects.equals(separator, other.separator) &&
@@ -723,8 +717,6 @@ public class PhraseSuggestionBuilder extends SuggestionBuilder<PhraseSuggestionB
public interface CandidateGenerator extends Writeable<CandidateGenerator>, ToXContent {
String getType();
- CandidateGenerator fromXContent(QueryParseContext parseContext) throws IOException;
-
PhraseSuggestionContext.DirectCandidateGenerator build(MapperService mapperService) throws IOException;
}
}
generated by cgit v1.2.3 (git 2.25.1) at 2024-07-13 06:26:01 +0000