org.hibernate.search.elasticsearch.query.impl.TwoWayFieldBridgeProjection Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of hibernate-search-elasticsearch Show documentation
Hibernate Search backend which has indexing operations forwarded to Elasticsearch
The 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.elasticsearch.query.impl;

import java.util.Map;
import java.util.Set;

import org.apache.lucene.document.Document;
import org.apache.lucene.document.DoubleField;
import org.apache.lucene.document.Field.Store;
import org.apache.lucene.document.StringField;
import org.hibernate.search.bridge.TwoWayFieldBridge;
import org.hibernate.search.bridge.spi.ConversionContext;

import com.google.gson.JsonArray;
import com.google.gson.JsonElement;
import com.google.gson.JsonObject;
import com.google.gson.JsonPrimitive;

class TwoWayFieldBridgeProjection extends FieldProjection {

	private final String absoluteName;
	private final TwoWayFieldBridge bridge;
	private final Set objectFieldNames;
	private final Map primitiveProjections;

	public TwoWayFieldBridgeProjection(String absoluteName,
			TwoWayFieldBridge bridge,
			Set objectFieldNames,
			Map primitiveProjections) {
		this.absoluteName = absoluteName;
		this.bridge = bridge;
		this.objectFieldNames = objectFieldNames;
		this.primitiveProjections = primitiveProjections;
	}

	@Override
	public Object convertHit(JsonObject hit, ConversionContext conversionContext) {
		return convertFieldValue( hit, conversionContext );
	}

	private Object convertFieldValue(JsonObject hit, ConversionContext conversionContext) {
		Document tmp = new Document();

		for ( String objectFieldName : objectFieldNames ) {
			JsonElement jsonValue = extractFieldValue( hit.get( "_source" ).getAsJsonObject(), objectFieldName );
			if ( jsonValue == null || jsonValue.isJsonNull() ) {
				continue;
			}
			JsonObject jsonObject = jsonValue.getAsJsonObject();
			addDocumentFieldsRecursively( tmp, jsonObject, objectFieldName );
		}

		// Add to the document the additional fields created when indexing the value
		for ( PrimitiveProjection subProjection : primitiveProjections.values() ) {
			subProjection.addDocumentField( tmp, hit, conversionContext );
		}

		return conversionContext.twoWayConversionContext( bridge ).get( absoluteName, tmp );
	}

	public void addDocumentFieldsRecursively(Document tmp, JsonElement value, String fieldName) {
		if ( value == null || value.isJsonNull() ) {
			return;
		}

		PrimitiveProjection configuredProjection = primitiveProjections.get( fieldName );
		if ( configuredProjection != null ) {
			// Those projections are handled separately, see the calling method
			return;
		}

		if ( value.isJsonObject() ) {
			JsonObject jsonObject = value.getAsJsonObject();
			for ( Map.Entry entry : jsonObject.entrySet() ) {
				String nestedFieldName = fieldName + "." + entry.getKey();
				JsonElement nestedFieldValue = entry.getValue();
				addDocumentFieldsRecursively( tmp, nestedFieldValue, nestedFieldName );
			}
		}
		else if ( value.isJsonArray() ) {
			JsonArray jsonArray = value.getAsJsonArray();
			for ( JsonElement nestedValue : jsonArray ) {
				addDocumentFieldsRecursively( tmp, nestedValue, fieldName );
			}
		}
		else {
			JsonPrimitive primitive = value.getAsJsonPrimitive();

			if ( primitive.isBoolean() ) {
				tmp.add( new StringField( fieldName, String.valueOf( primitive.getAsBoolean() ), Store.NO ) );
			}
			else if ( primitive.isNumber() ) {
				tmp.add( new DoubleField( fieldName, primitive.getAsDouble(), Store.NO ) );
			}
			else if ( primitive.isString() ) {
				tmp.add( new StringField( fieldName, primitive.getAsString(), Store.NO ) );
			}
			else {
				// TODO HSEARCH-2255 Better raise an exception?
				tmp.add( new StringField( fieldName, primitive.getAsString(), Store.NO ) );
			}
		}
	}
}