com.elefana.search.query.DisMaxQuery Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of elefana-core Show documentation
null
The newest version!
/*******************************************************************************
 * Copyright 2018 Viridian Software Limited
 *
 * Licensed 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 com.elefana.search.query;

import java.util.ArrayList;
import java.util.List;

import com.elefana.api.exception.ElefanaException;
import com.elefana.api.indices.IndexTemplate;
import com.jsoniter.ValueType;
import com.jsoniter.any.Any;

public class DisMaxQuery extends Query {
	private static final String KEY_QUERIES = "queries";
	
	private final List queries = new ArrayList();
	
	public DisMaxQuery(Any queryContext) throws ElefanaException {
		super();
		
		if(!queryContext.get(KEY_QUERIES).valueType().equals(ValueType.INVALID)) {
			for(Any query : queryContext.get(KEY_QUERIES).asList()) {
				queries.add(QueryParser.parseQuery(query));
			}
		}
	}
	
	@Override
	public boolean isMatchAllQuery() {
		for(Query query : queries) {
			if(!query.isMatchAllQuery()) {
				return false;
			}
		}
		return true;
	}

	@Override
	public String toSqlWhereClause(IndexTemplate indexTemplate) {
		//TODO: Handle scoring/sorting
		StringBuilder result = new StringBuilder();
		for(int i = 0; i < queries.size(); i++) {
			if(i > 0) {
				result.append(" OR ");
			}
			result.append('(');
			result.append(queries.get(i).toSqlWhereClause(indexTemplate));
			result.append(')');
		}
		return result.toString();
	}

}