001/*
002 * (C) Copyright 2019 Nuxeo (http://nuxeo.com/) and others.
003 *
004 * Licensed under the Apache License, Version 2.0 (the "License");
005 * you may not use this file except in compliance with the License.
006 * You may obtain a copy of the License at
007 *
008 *     http://www.apache.org/licenses/LICENSE-2.0
009 *
010 * Unless required by applicable law or agreed to in writing, software
011 * distributed under the License is distributed on an "AS IS" BASIS,
012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
013 * See the License for the specific language governing permissions and
014 * limitations under the License.
015 *
016 * Contributors:
017 *     Salem Aouana
018 */
019
020package org.nuxeo.elasticsearch.hint;
021
022import java.util.Map;
023import java.util.stream.Collectors;
024
025import org.elasticsearch.index.query.QueryBuilder;
026import org.elasticsearch.index.query.QueryBuilders;
027import org.nuxeo.ecm.core.query.sql.model.EsHint;
028import org.nuxeo.elasticsearch.api.ESHintQueryBuilder;
029
030/**
031 * The implementation of {@link ESHintQueryBuilder} for the <strong>"multi_match"</strong> Elasticsearch hint operator.
032 *
033 * @since 11.1
034 */
035public class MultiMatchESHintQueryBuilder implements ESHintQueryBuilder {
036
037    /**
038     * {@inheritDoc}
039     * <p>
040     *
041     * @return {@link org.elasticsearch.index.query.MultiMatchQueryBuilder}
042     */
043    @Override
044    public QueryBuilder make(EsHint hint, String fieldName, Object value) {
045        // multiMatchQuery requires at least 1 field on creation, so we set them twice, in case there's a field boost
046        Map<String, Float> fields = hint.getIndex()
047                                        .stream()
048                                        .collect(Collectors.toMap(EsHint.FieldHint::getField,
049                                                EsHint.FieldHint::getBoost));
050        return QueryBuilders.multiMatchQuery(value, hint.getIndexFieldNames()).fields(fields).analyzer(hint.analyzer);
051    }
052}