MatchQueryTranslatorImpl.java

/**
 * Copyright (c) 2000-present Liferay, Inc. All rights reserved.
 *
 * This library is free software; you can redistribute it and/or modify it under
 * the terms of the GNU Lesser General Public License as published by the Free
 * Software Foundation; either version 2.1 of the License, or (at your option)
 * any later version.
 *
 * This library is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
 * FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
 * details.
 */

package com.liferay.portal.search.elasticsearch.internal.query;

import com.liferay.portal.kernel.search.generic.MatchQuery;
import com.liferay.portal.kernel.util.StringPool;
import com.liferay.portal.kernel.util.Validator;
import com.liferay.portal.search.elasticsearch.query.MatchQueryTranslator;

import org.elasticsearch.common.unit.Fuzziness;
import org.elasticsearch.index.query.MatchQueryBuilder;
import org.elasticsearch.index.query.QueryBuilder;
import org.elasticsearch.index.query.QueryBuilders;

import org.osgi.service.component.annotations.Component;

/**
 * @author Michael C. Han
 */
@Component(immediate = true, service = MatchQueryTranslator.class)
public class MatchQueryTranslatorImpl
	extends BaseMatchQueryTranslatorImpl implements MatchQueryTranslator {

	@Override
	public QueryBuilder translate(MatchQuery matchQuery) {
		String value = matchQuery.getValue();

		MatchQuery.Type matchQueryType = matchQuery.getType();

		if (value.startsWith(StringPool.QUOTE) &&
			value.endsWith(StringPool.QUOTE)) {

			value = value.substring(1, value.length() - 1);

			if (value.endsWith(StringPool.STAR)) {
				value = value.substring(0, value.length() - 1);

				matchQueryType = MatchQuery.Type.PHRASE_PREFIX;
			}
			else {
				matchQueryType = MatchQuery.Type.PHRASE;
			}
		}

		MatchQueryBuilder matchQueryBuilder = QueryBuilders.matchQuery(
			matchQuery.getField(), value);

		if (Validator.isNotNull(matchQuery.getAnalyzer())) {
			matchQueryBuilder.analyzer(matchQuery.getAnalyzer());
		}

		if (matchQuery.getCutOffFrequency() != null) {
			matchQueryBuilder.cutoffFrequency(matchQuery.getCutOffFrequency());
		}

		if (matchQuery.getFuzziness() != null) {
			matchQueryBuilder.fuzziness(
				Fuzziness.build(matchQuery.getFuzziness()));
		}

		if (matchQuery.getFuzzyRewriteMethod() != null) {
			String matchQueryFuzzyRewrite = translate(
				matchQuery.getFuzzyRewriteMethod());

			matchQueryBuilder.fuzzyRewrite(matchQueryFuzzyRewrite);
		}

		if (matchQuery.getMaxExpansions() != null) {
			matchQueryBuilder.maxExpansions(matchQuery.getMaxExpansions());
		}

		if (Validator.isNotNull(matchQuery.getMinShouldMatch())) {
			matchQueryBuilder.minimumShouldMatch(
				matchQuery.getMinShouldMatch());
		}

		if (matchQuery.getOperator() != null) {
			MatchQueryBuilder.Operator operator = translate(
				matchQuery.getOperator());

			matchQueryBuilder.operator(operator);
		}

		if (matchQuery.getPrefixLength() != null) {
			matchQueryBuilder.prefixLength(matchQuery.getPrefixLength());
		}

		if (matchQuery.getSlop() != null) {
			matchQueryBuilder.slop(matchQuery.getSlop());
		}

		if (matchQueryType != null) {
			MatchQueryBuilder.Type matchQueryBuilderType = translate(
				matchQueryType);

			matchQueryBuilder.type(matchQueryBuilderType);
		}

		if (matchQuery.getZeroTermsQuery() != null) {
			MatchQueryBuilder.ZeroTermsQuery matchQueryBuilderZeroTermsQuery =
				translate(matchQuery.getZeroTermsQuery());

			matchQueryBuilder.zeroTermsQuery(matchQueryBuilderZeroTermsQuery);
		}

		if (!matchQuery.isDefaultBoost()) {
			matchQueryBuilder.boost(matchQuery.getBoost());
		}

		if (matchQuery.isFuzzyTranspositions() != null) {
			matchQueryBuilder.fuzzyTranspositions(
				matchQuery.isFuzzyTranspositions());
		}

		if (matchQuery.isLenient() != null) {
			matchQueryBuilder.setLenient(matchQuery.isLenient());
		}

		return matchQueryBuilder;
	}

	protected MatchQueryBuilder.Type translate(MatchQuery.Type matchQueryType) {
		if (matchQueryType == MatchQuery.Type.BOOLEAN) {
			return MatchQueryBuilder.Type.BOOLEAN;
		}
		else if (matchQueryType == MatchQuery.Type.PHRASE) {
			return MatchQueryBuilder.Type.PHRASE;
		}
		else if (matchQueryType == MatchQuery.Type.PHRASE_PREFIX) {
			return MatchQueryBuilder.Type.PHRASE_PREFIX;
		}

		throw new IllegalArgumentException(
			"Invalid match query type: " + matchQueryType);
	}

}