org.hibernate.search.query.engine.impl.EntityInfoImpl Maven / Gradle / Ivy

Go to download
/*
 * 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.query.engine.impl;

import java.io.Serializable;

import org.hibernate.search.engine.ProjectionConstants;
import org.hibernate.search.query.engine.spi.EntityInfo;
import org.hibernate.search.spi.IndexedTypeIdentifier;
import org.hibernate.search.util.logging.impl.Log;
import org.hibernate.search.util.logging.impl.LoggerFactory;
import java.lang.invoke.MethodHandles;

/**
 * Wrapper class describing the loading of an element.
 *
 * @author Emmanuel Bernard
 * @author Hardy Ferentschik
 */
public class EntityInfoImpl implements EntityInfo {

	private static final Log log = LoggerFactory.make( MethodHandles.lookup() );

	/**
	 * The entity type.
	 */
	private final IndexedTypeIdentifier type;

	/**
	 * The document id.
	 */
	private final Serializable id;

	/**
	 * The name of the document id property.
	 */
	private final String idName;

	/**
	 * Array of projected values. {@code null} in case there are no projections.
	 */
	private final Object[] projection;

	/**
	 * The entity instance returned in a search.
	 *
	 * @see ProjectionConstants#THIS
	 */
	private Object entityInstance;

	public EntityInfoImpl(IndexedTypeIdentifier type, String idName, Serializable id, Object[] projection) {
		this.type = type;
		this.idName = idName;
		this.id = id;
		this.projection = projection;
	}

	@Override
	public IndexedTypeIdentifier getType() {
		return type;
	}

	@Override
	public Serializable getId() {
		if ( id == null ) {
			/*
			 * Only throw an exception here, not in the constructor,
			 * because in some cases we don't need the identifier at all (e.g. projections).
			 */
			throw log.incompleteEntityInfo( type, id );
		}
		return id;
	}

	@Override
	public String getIdName() {
		return idName;
	}

	@Override
	public Object[] getProjection() {
		return projection;
	}

	@Override
	public Object getEntityInstance() {
		return entityInstance;
	}

	@Override
	public void populateWithEntityInstance(Object entity) {
		this.entityInstance = entity;
		for ( int i = 0; i < projection.length; i++ ) {
			if ( projection[i] == ENTITY_PLACEHOLDER ) {
				projection[i] = entity;
			}
		}
	}

	@Override
	public String toString() {
		return "EntityInfoImpl{" +
				"idName='" + idName + '\'' +
				", id=" + id +
				", clazz=" + type +
				'}';
	}

}