org.jvnet.hyperjaxb3.ejb.util.EntityUtils Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of hyperjaxb3-ejb-runtime Show documentation
There is a newer version: 4.0.8
Show newest version
package org.jvnet.hyperjaxb3.ejb.util;

import java.lang.annotation.Annotation;
import java.lang.reflect.AnnotatedElement;
import java.lang.reflect.Field;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.util.Collection;
import java.util.LinkedList;

import jakarta.persistence.EmbeddedId;
import jakarta.persistence.EntityManager;
import jakarta.persistence.Id;

public class EntityUtils {

	private EntityUtils() {
	}

	public static Object getId(Object entity) throws IllegalArgumentException,
			IllegalAccessException, InvocationTargetException {
		return getId(null, entity);

	}

	public static Object getId(EntityManager entityManager, Object entity)
			throws IllegalArgumentException, IllegalAccessException,
			InvocationTargetException {
		// Try annotations
		{

			final Collection atEmbeddedIdFields = getAnnotatedElements(
					entity.getClass().getFields(), EmbeddedId.class);
			final Collection atEmbeddedIdMethods = getAnnotatedElements(
					entity.getClass().getMethods(), EmbeddedId.class);
			final Collection atIdFields = getAnnotatedElements(entity
					.getClass().getFields(), Id.class);
			final Collection atIdMethods = getAnnotatedElements(entity
					.getClass().getMethods(), Id.class);

			if (atEmbeddedIdFields.size() > 0) {
				assert atEmbeddedIdFields.size() == 1 : "More than one field is annotated with @EmbeddedId.";
				final Field field = atEmbeddedIdFields.iterator().next();
				return get(entity, field);
			} else if (atEmbeddedIdMethods.size() > 0) {
				assert atEmbeddedIdMethods.size() == 1 : "More than one method is annotated with @EmbeddedId.";
				final Method method = atEmbeddedIdMethods.iterator().next();
				return get(entity, method);
			} else if (atIdFields.size() == 1) {
				return get(entity, atIdFields.iterator().next());
			} else if (atIdMethods.size() == 1) {
				return get(entity, atIdMethods.iterator().next());
			}
		}
		if (entityManager != null) {

			// Try Hibernate org.hibernate.ejb.AbstractEntityManagerImpl
			try {
				return getIdWithHibernate(entityManager, entity);
			} catch (Exception ignored) {
				// Well, it wasn't Hibernate
			}
		}
		throw new IllegalArgumentException("No id could be found.");
	}

	private static Object getIdWithHibernate(EntityManager entityManager,
			Object entity) throws ClassNotFoundException,
			NoSuchMethodException, IllegalAccessException,
			InvocationTargetException {
		final Class hibernateEntityManagerClass = Class
				.forName("org.hibernate.ejb.AbstractEntityManagerImpl");
		final Class hibernateSessionClass = Class
				.forName("org.hibernate.Session");
		final Method getSession = hibernateEntityManagerClass
				.getMethod("getSession");
		final Method getIdentifier = hibernateSessionClass.getMethod(
				"getIdentifier", Object.class);
		return getIdentifier.invoke(getSession.invoke(entityManager), entity);
	}

	private static  Collection getAnnotatedElements(
			T[] elements, Class annotationType) {
		final Collection annotatedElements = new LinkedList();
		for (final T element : elements) {
			if (element.isAnnotationPresent(annotationType)) {
				annotatedElements.add(element);
			}
		}
		return annotatedElements;
	}

	private static Object get(Object target, Field field)
			throws IllegalArgumentException, IllegalAccessException {
		return field.get(target);
	}

	private static Object get(Object target, Method method)
			throws IllegalArgumentException, IllegalAccessException,
			InvocationTargetException {
		return method.invoke(target, (Object[]) null);
	}
}