io.pelle.mango.db.util.DBUtil 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.util;

import io.pelle.mango.client.base.db.vos.IHierarchicalVO;
import io.pelle.mango.client.base.vo.IAttributeDescriptor;
import io.pelle.mango.client.base.vo.IBaseEntity;
import io.pelle.mango.client.base.vo.IBaseVO;
import io.pelle.mango.client.base.vo.IVOEntity;
import io.pelle.mango.client.base.vo.query.BaseQuery;
import io.pelle.mango.client.base.vo.query.Entity;
import io.pelle.mango.client.base.vo.query.IBooleanExpression;
import io.pelle.mango.client.base.vo.query.Join;
import io.pelle.mango.client.base.vo.query.SelectQuery;
import io.pelle.mango.db.voquery.VOClassQuery;

import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;

import com.google.common.base.Optional;

/**
 * Various DB utilities
 * 
 * @author pelle
 * 
 */
public final class DBUtil {

	public static Class getQueryClass(BaseQuery baseQuery) {
		try {
			return Class.forName(baseQuery.getFroms().get(0).getClassName());
		} catch (ClassNotFoundException e) {
			return null;
		}
	}

	public static Map, Set> getLoadAssociations(SelectQuery selectQuery) {

		Map, Set> classLoadAssociations = new HashMap<>();

		for (Entity entity : selectQuery.getFroms()) {

			addFirstLevelIBaseVOAttributes(EntityVOMapper.getInstance().getVOClass(entity.getClassName()), classLoadAssociations);

			Set associations = getAssociations(EntityVOMapper.getInstance().getVOClass(entity.getClassName()), classLoadAssociations);

			for (Join join : entity.getJoins()) {
				associations.add(join.getField());

				addJoinAssociations(join, EntityVOMapper.getInstance().getVOClass(entity.getClassName()), classLoadAssociations);
			}

		}

		if (selectQuery.isLoadNaturalKeyReferences()) {
			addNaturalKeyAttributes(classLoadAssociations);
		}

		return classLoadAssociations;
	}

	public static void addJoinAssociations(Join join, Class parentClass, Map, Set> classLoadAssociations) {

		IAttributeDescriptor attributeDescriptor = BeanUtils.getAttributeDescriptor(parentClass, join.getField());

		for (Join join1 : join.getJoins()) {

			getAssociations((Class) attributeDescriptor.getListAttributeType(), classLoadAssociations).add(join1.getField());

			addJoinAssociations(join1, attributeDescriptor.getListAttributeType(), classLoadAssociations);
		}

	}

	public static Set getAssociations(Class clazz, Map, Set> classLoadAssociations) {

		if (!classLoadAssociations.containsKey(clazz)) {
			classLoadAssociations.put(clazz, new HashSet());
		}

		return classLoadAssociations.get(clazz);
	}

	public static void addFirstLevelIBaseVOAttributes(Class voEntityClass, Map, Set> classLoadAssociations) {

		Set associations = getAssociations(voEntityClass, classLoadAssociations);

		for (IAttributeDescriptor attributeDescriptor : BeanUtils.getAttributeDescriptors(voEntityClass)) {
			if (!attributeDescriptor.getAttributeName().equals(IHierarchicalVO.FIELD_PARENT.getAttributeName())
					&& (IVOEntity.class.isAssignableFrom(attributeDescriptor.getListAttributeType()) || List.class.isAssignableFrom(attributeDescriptor.getAttributeType()))) {
				associations.add(attributeDescriptor.getAttributeName());
			}
		}

	}

	public static void addNaturalKeyAttributes(Map, Set> classLoadAssociations) {

		for (Map.Entry, Set> classLoadAssociation : classLoadAssociations.entrySet()) {

			for (String attributeName : classLoadAssociation.getValue()) {

				IAttributeDescriptor attributeDescriptor = VOClassQuery.createQuery(classLoadAssociation.getKey()).attributesDescriptors().byName(attributeName).getSingleResult();

				if (IBaseVO.class.isAssignableFrom(attributeDescriptor.getListAttributeType())) {
					addNaturalKeyAttributes((Class) attributeDescriptor.getListAttributeType(), classLoadAssociations);
				}
			}
		}
	}

	public static void addNaturalKeyAttributes(Class voEntityClass, Map, Set> classLoadAssociations) {

		for (IAttributeDescriptor naturalKey : VOClassQuery.createQuery(voEntityClass).attributesDescriptors().naturalKeys()) {

			if (IBaseVO.class.isAssignableFrom(naturalKey.getListAttributeType())) {
				getAssociations(voEntityClass, classLoadAssociations).add(naturalKey.getAttributeName());
				addNaturalKeyAttributes((Class) naturalKey.getListAttributeType(), classLoadAssociations);
			}

		}

	}

	public static IBaseEntity convertVOToEntityClass(IBaseVO baseVO) {
		Class entityClass = EntityVOMapper.getInstance().getMappedEntityClass(baseVO.getClass());
		return (IBaseEntity) CopyBean.getInstance().copyObject(baseVO, entityClass, false);
	}

	@SuppressWarnings("unchecked")
	public static  T convertEntityToVO(IBaseEntity baseEntity) {
		Class voClass = EntityVOMapper.getInstance().getMappedVOClass(baseEntity.getClass());
		return (T) CopyBean.getInstance().copyObject(baseEntity, voClass);
	}

	public static IBaseVO convertEntityToVO(IBaseEntity baseEntity, Map, Set> loadAssociations) {
		Class voClass = EntityVOMapper.getInstance().getMappedVOClass(baseEntity.getClass());
		return (IBaseVO) CopyBean.getInstance().copyObject(baseEntity, voClass, loadAssociations, true);
	}

	@SuppressWarnings({ "unchecked", "rawtypes" })
	public static String mapToString(Map map) {

		String lineSeperator = System.getProperty("line.separator");

		if (map == null) {
			return "";
		}

		StringBuffer stringBuffer = new StringBuffer();

		Set s = map.entrySet();

		for (Map.Entry elem : s) {

			Object key = elem.getKey();
			Object value = elem.getValue();

			stringBuffer.append(key.toString());
			stringBuffer.append("=");
			stringBuffer.append(value == null ? "" : value.toString());
			stringBuffer.append(lineSeperator);

		}

		return stringBuffer.toString();
	}

	public static  SelectQuery getNaturalKeyQuery(Class voEntityClass, String naturalKey) {

		SelectQuery selectQuery = SelectQuery.selectFrom(voEntityClass);

		populateNaturalKeyQuery(voEntityClass, selectQuery, null, naturalKey);

		return selectQuery;
	}

	public static  void populateNaturalKeyQuery(Class voEntityClass, SelectQuery selectQuery, IAttributeDescriptor parentAttribute, String naturalKey) {

		for (IAttributeDescriptor naturalKeyAttributeDescriptor : VOClassQuery.createQuery(voEntityClass).attributesDescriptors().naturalKeys()) {

			if (IBaseVO.class.isAssignableFrom(naturalKeyAttributeDescriptor.getListAttributeType())) {

				IAttributeDescriptor attributeDescriptor = null;

				if (parentAttribute != null) {
					attributeDescriptor = parentAttribute.path(naturalKeyAttributeDescriptor);
				} else {
					attributeDescriptor = naturalKeyAttributeDescriptor;
				}
				populateNaturalKeyQuery(naturalKeyAttributeDescriptor.getListAttributeType(), selectQuery, attributeDescriptor, naturalKey);
			} else {
				
				Optional expression = null;
				
				if (parentAttribute == null) {
					expression = naturalKeyAttributeDescriptor.search(naturalKey);
				} else {
					expression = parentAttribute.path(naturalKeyAttributeDescriptor).search(naturalKey);
				}
				
				if (expression.isPresent()) {
					selectQuery.addWhereOr(expression.get());
				}
			}
		}

	}

	private DBUtil() {
	}

}