io.pelle.mango.db.voquery.EntityClassQuery Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mango-db Show documentation

mango web framework

The newest version!

package io.pelle.mango.db.voquery;

import java.lang.reflect.Field;
import java.util.ArrayList;
import java.util.List;

import javax.persistence.CollectionTable;
import javax.persistence.ElementCollection;
import javax.persistence.JoinTable;
import javax.persistence.OneToMany;

import io.pelle.mango.client.base.vo.IBaseEntity;

public class EntityClassQuery {

	private Class entityClass;

	public EntityClassQuery(Class entityClass) {
		this.entityClass = entityClass;
	}

	public static EntityClassQuery createQuery(Class voEntityClass) {
		return new EntityClassQuery(voEntityClass);
	}

	public static EntityClassQuery createQuery(IBaseEntity entity) {
		return new EntityClassQuery(entity.getClass());
	}

	public List getElementCollections() {

		List result = new ArrayList<>();

		for (Field field : entityClass.getDeclaredFields()) {
			if (field.isAnnotationPresent(ElementCollection.class) && field.isAnnotationPresent(CollectionTable.class)) {
				String tableName = field.getAnnotation(CollectionTable.class).name();
				result.add(tableName);
			}
		}

		return result;
	}

	public List getOneToManyJoinTables() {

		List result = new ArrayList<>();

		for (Field field : entityClass.getDeclaredFields()) {
			if (field.isAnnotationPresent(JoinTable.class) && field.isAnnotationPresent(OneToMany.class)) {
				String tableName = field.getAnnotation(JoinTable.class).name();
				result.add(tableName);
			}
		}

		return result;
	}

}