/* * Licensed to ElasticSearch and Shay Banon under one * or more contributor license agreements. See the NOTICE file * distributed with this work for additional information * regarding copyright ownership. ElasticSearch licenses this * file to you under the Apache License, Version 2.0 (the * "License"); you may not use this file except in compliance * with the License. You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, * software distributed under the License is distributed on an * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY * KIND, either express or implied. See the License for the * specific language governing permissions and limitations * under the License. */ package org.elasticsearch.index.query; import org.apache.lucene.index.Term; import org.apache.lucene.search.MultiTermQuery; import org.apache.lucene.search.PrefixQuery; import org.apache.lucene.search.Query; import org.elasticsearch.common.inject.Inject; import org.elasticsearch.common.xcontent.XContentParser; import org.elasticsearch.index.mapper.MapperService; import org.elasticsearch.index.query.support.QueryParsers; import java.io.IOException; import static org.elasticsearch.index.query.support.QueryParsers.wrapSmartNameQuery; /** * */ public class PrefixQueryParser implements QueryParser { public static final String NAME = "prefix"; @Inject public PrefixQueryParser() { } @Override public String[] names() { return new String[]{NAME}; } @Override public Query parse(QueryParseContext parseContext) throws IOException, QueryParsingException { XContentParser parser = parseContext.parser(); XContentParser.Token token = parser.nextToken(); if (token != XContentParser.Token.FIELD_NAME) { throw new QueryParsingException(parseContext.index(), "[prefix] query malformed, no field"); } String fieldName = parser.currentName(); String rewriteMethod = null; String value = null; float boost = 1.0f; token = parser.nextToken(); if (token == XContentParser.Token.START_OBJECT) { String currentFieldName = null; while ((token = parser.nextToken()) != XContentParser.Token.END_OBJECT) { if (token == XContentParser.Token.FIELD_NAME) { currentFieldName = parser.currentName(); } else if (token.isValue()) { if ("prefix".equals(currentFieldName)) { value = parser.text(); } else if ("value".equals(currentFieldName)) { value = parser.text(); } else if ("boost".equals(currentFieldName)) { boost = parser.floatValue(); } else if ("rewrite".equals(currentFieldName)) { rewriteMethod = parser.textOrNull(); } } else { throw new QueryParsingException(parseContext.index(), "[prefix] query does not support [" + currentFieldName + "]"); } } parser.nextToken(); } else { value = parser.text(); parser.nextToken(); } if (value == null) { throw new QueryParsingException(parseContext.index(), "No value specified for prefix query"); } MultiTermQuery.RewriteMethod method = QueryParsers.parseRewriteMethod(rewriteMethod, null); Query query = null; MapperService.SmartNameFieldMappers smartNameFieldMappers = parseContext.smartFieldMappers(fieldName); if (smartNameFieldMappers != null && smartNameFieldMappers.hasMapper()) { if (smartNameFieldMappers.explicitTypeInNameWithDocMapper()) { String[] previousTypes = QueryParseContext.setTypesWithPrevious(new String[]{smartNameFieldMappers.docMapper().type()}); try { query = smartNameFieldMappers.mapper().prefixQuery(value, method, parseContext); } finally { QueryParseContext.setTypes(previousTypes); } } else { query = smartNameFieldMappers.mapper().prefixQuery(value, method, parseContext); } } if (query == null) { PrefixQuery prefixQuery = new PrefixQuery(new Term(fieldName, value)); prefixQuery.setRewriteMethod(method); query = prefixQuery; } query.setBoost(boost); return wrapSmartNameQuery(query, smartNameFieldMappers, parseContext); } }