org.hibernate.search.backend.elasticsearch.ElasticsearchQueries Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of hibernate-search-backend-elasticsearch Show documentation

Hibernate Search backend which has indexing operations forwarded to Elasticsearch

There is a newer version: 5.6.0.Alpha3

Show newest version

/*
 * Hibernate Search, full-text search for your domain model
 *
 * License: GNU Lesser General Public License (LGPL), version 2.1 or later
 * See the lgpl.txt file in the root directory or .
 */
package org.hibernate.search.backend.elasticsearch;

import org.hibernate.search.backend.elasticsearch.impl.ElasticsearchHSQueryImpl;
import org.hibernate.search.engine.integration.impl.ExtendedSearchIntegrator;
import org.hibernate.search.query.engine.spi.HSQuery;
import org.hibernate.search.query.engine.spi.QueryDescriptor;

import com.google.gson.JsonObject;
import com.google.gson.JsonParser;

/**
 * Creates queries to be used with the Elasticsearch backend.
 *
 * @author Gunnar Morling
 */
public class ElasticsearchQueries {

	private ElasticsearchQueries() {
	}

	/**
	 * Creates an Elasticsearch query from the given JSON query representation. See the official
	 * documentation for the complete query syntax.
	 */
	public static QueryDescriptor fromJson(String jsonQuery) {
		// TODO Parse + Re-render using Gson for now to leverage single quote support
		jsonQuery = new JsonParser().parse( jsonQuery ).toString();

		return new ElasticsearchJsonQuery( jsonQuery );
	}

	/**
	 * Creates an Elasticsearch query from the given Query String Query, as e.g. to be used with the "q" parameter in
	 * the Elasticsearch API. See the official
	 * documentation for a description of the query syntax.
	 */
	public static QueryDescriptor fromQueryString(String queryStringQuery) {
		// Payload looks like so:
		// { "query" : { "query_string" : { "query" : "abstract:Hibernate" } } }

		JsonObject query = new JsonObject();
		query.addProperty( "query", queryStringQuery );

		JsonObject queryString = new JsonObject();
		queryString.add( "query_string", query );

		JsonObject queryObject = new JsonObject();
		queryObject.add( "query", queryString );

		return new ElasticsearchJsonQuery( queryObject.toString() );
	}

	private static class ElasticsearchJsonQuery implements QueryDescriptor {

		private final String jsonQuery;

		public ElasticsearchJsonQuery(String jsonQuery) {
			this.jsonQuery = jsonQuery;
		}

		@Override
		public HSQuery createHSQuery(ExtendedSearchIntegrator extendedIntegrator) {
			return new ElasticsearchHSQueryImpl( jsonQuery, extendedIntegrator );
		}

		@Override
		public String toString() {
			return jsonQuery;
		}
	}
}