org.fastnate.generator.context.EmbeddedProperty Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of fastnate-generator Show documentation
The Fastnate Offline SQL Generator
There is a newer version: 1.5.0
package org.fastnate.generator.context;

import java.util.ArrayList;
import java.util.Collection;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.TreeMap;

import javax.persistence.Access;
import javax.persistence.AssociationOverride;
import javax.persistence.AttributeOverride;
import javax.persistence.Column;
import javax.persistence.Embeddable;
import javax.persistence.EmbeddedId;

import org.fastnate.generator.statements.EntityStatement;
import org.fastnate.generator.statements.InsertStatement;

import lombok.Getter;

/**
 * Handling of properties defined in {@link Embeddable}s.
 *
 * @author Tobias Liefke
 *
 * @param 
 *            The type of the container class (the entity)
 * @param 
 *            The type of the embeddable object
 */
@Getter
public class EmbeddedProperty extends Property {

	/**
	 * The properties within the embedded object.
	 */
	private final Map> embeddedProperties = new TreeMap<>();

	private final boolean id;

	/**
	 * Instantiates a new embedded property.
	 *
	 * @param entityClass
	 *            the class of the entity
	 * @param attribute
	 *            the attribute that contains the embedded object
	 */
	public EmbeddedProperty(final EntityClass entityClass, final AttributeAccessor attribute) {
		super(attribute);

		this.id = attribute.hasAnnotation(EmbeddedId.class);

		final Class type = attribute.getType();
		if (!type.isAnnotationPresent(Embeddable.class)) {
			throw new IllegalArgumentException(attribute + " does reference " + type + " which is not embeddable.");
		}

		// Determine the access style
		final AccessStyle accessStyle;
		final Access accessType = type.getAnnotation(Access.class);
		if (accessType != null) {
			accessStyle = AccessStyle.getStyle(accessType.value());
		} else {
			accessStyle = attribute.getAccessStyle();
		}

		final Map attributeOverrides = EntityClass
				.getAttributeOverrides(attribute.getElement());
		final Map accociationOverrides = EntityClass
				.getAccociationOverrides(attribute.getElement());
		for (final AttributeAccessor field : accessStyle.getDeclaredAttributes(type)) {
			final AttributeOverride attrOveride = attributeOverrides.get(field.getName());
			final Property property = entityClass.buildProperty(field,
					attrOveride != null ? attrOveride.column() : field.getAnnotation(Column.class),
					accociationOverrides.get(field.getName()));
			if (property != null) {
				this.embeddedProperties.put(field.getName(), property);
			}
		}
	}

	@Override
	public void addInsertExpression(final E entity, final InsertStatement statement) {
		final T value = getValue(entity);
		if (value != null) {
			for (final Property property : this.embeddedProperties.values()) {
				property.addInsertExpression(value, statement);
			}
		} else {
			failIfRequired();
		}
	}

	@Override
	public List buildAdditionalStatements(final E entity) {
		final T value = getValue(entity);
		final List result = new ArrayList<>();
		for (final Property property : this.embeddedProperties.values()) {
			result.addAll(property.buildAdditionalStatements(value));
		}

		return result;
	}

	@Override
	public Collection findReferencedEntities(final E entity) {
		final T value = getValue(entity);
		final Set