io.pelle.mango.db.copy.BaseCopyBean 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.copy;

import io.pelle.mango.client.base.vo.IBaseVO;
import io.pelle.mango.client.base.vo.IChangeTracker;
import io.pelle.mango.client.base.vo.IHasMetadata;
import io.pelle.mango.client.base.vo.IVOEntity;
import io.pelle.mango.client.base.vo.IVOEntityMetadata;

import java.lang.reflect.Field;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;

import org.apache.commons.beanutils.ConstructorUtils;

public abstract class BaseCopyBean {

	@SuppressWarnings({ "rawtypes", "unchecked" })
	public static Object getAnnotation(Class clazz, String fieldName, Class annotationClass) {
		Field field = getDeclaredField(clazz, fieldName);

		if (field != null) {
			return field.getAnnotation(annotationClass);
		}
		{
			return null;
		}
	}

	public static Field getDeclaredField(Class clazz, String fieldName) {
		Field[] fields = clazz.getDeclaredFields();

		for (Field field : fields) {
			if (field.getName().equals(fieldName)) {
				return field;
			}
		}

		return null;
	}

	public static boolean hasAnnotation(Class clazz, String fieldName, Class annotationClass) {
		return getAnnotation(clazz, fieldName, annotationClass) != null;
	}

	private final List fieldCopyHandlers = new ArrayList();

	public Object copyObject(Object sourceObject, Class destClass, boolean clearChanges) {
		return copyObject(sourceObject, destClass, new HashMap(), null, new ArrayList(), clearChanges);
	}

	public Object copyObject(Object sourceObject, Class destClass) {
		return copyObject(sourceObject, destClass, true);
	}

	public Object copyObject(Object sourceObject, Class destClass, String[] attributesToOmit) {
		return copyObject(sourceObject, destClass, new HashMap(), null, Arrays.asList(attributesToOmit));
	}

	public Object copyObject(Object sourceObject, Class destClass, Map, Set> loadAssociations, boolean clearChanges) {
		return copyObject(sourceObject, destClass, loadAssociations, clearChanges, new String[0]);
	}

	public Object copyObject(Object sourceObject, Class destClass, Map, Set> classLoadAssociations, boolean clearChanges, String[] attributesToOmit) {
		return copyObject(sourceObject, destClass, new HashMap(), classLoadAssociations, Arrays.asList(attributesToOmit), clearChanges);
	}

	public Object copyObject(Object sourceObject) {
		return copyObject(sourceObject, getMappedTargetType(sourceObject.getClass()), new HashMap(), null, new ArrayList());
	}

	public void addFieldCopyHandler(IFieldCopyHandler fieldCopyHandler) {
		this.fieldCopyHandlers.add(fieldCopyHandler);
	}

	private Object copyObject(Object sourceObject, Class destClass, Map visited, Map, Set> classLoadAssociations, List attributesToOmit) {
		return copyObject(sourceObject, destClass, visited, classLoadAssociations, attributesToOmit, false);
	}

	private Object copyObject(Object sourceObject, Class destClass, Map visited, Map, Set> classLoadAssociations, List attributesToOmit, boolean clearChanges) {

		Set loadAssociations = null;

		if (classLoadAssociations != null) {

			loadAssociations = new HashSet();

			if (classLoadAssociations.containsKey(sourceObject.getClass())) {
				loadAssociations = classLoadAssociations.get(sourceObject.getClass());
			}

			if (classLoadAssociations.containsKey(destClass)) {
				loadAssociations = classLoadAssociations.get(destClass);
			}
		}

		if (sourceObject == null) {
			return null;
		}

		Object targetObject = null;
		try {
			targetObject = ConstructorUtils.invokeConstructor(destClass, new Object[] {});
		} catch (Exception e) {
			throw new RuntimeException(String.format("error invoking constructor for '%s'", destClass.getName()), e);
		}

		visited.put(sourceObject, targetObject);

		IVOEntityMetadata targetMetadata = null;
		IChangeTracker sourceMetadata = null;

		if (targetObject instanceof IHasMetadata) {
			targetMetadata = ((IHasMetadata) targetObject).getMetadata();
		}

		if (sourceObject instanceof IHasMetadata) {
			sourceMetadata = ((IHasMetadata) sourceObject).getMetadata();
		}

		for (ObjectFieldDescriptor fieldDescriptor : new ObjectFieldIterator(sourceObject, targetObject)) {

			if (fieldDescriptor.getSourceType() == null || fieldDescriptor.getTargetType() == null) {
				continue;
			}

			if (!fieldDescriptor.sourceHasReadMethod() || !fieldDescriptor.targetHasWriteMethod()) {
				continue;
			}

			if (attributesToOmit.contains(fieldDescriptor.getFieldName())) {
				continue;
			}

			if (loadAssociations != null && fieldDescriptor.sourceTypeIsReference() && !loadAssociations.contains(fieldDescriptor.getFieldName())) {
				continue;
			}

			Object sourceValue = null;
			Object targetValue = null;

			if (sourceObject instanceof IBaseVO) {

				IBaseVO voEntity = (IBaseVO) sourceObject;

				// boolean isReference = fieldDescriptor.;
				boolean isLoaded = voEntity.getMetadata().isLoaded(fieldDescriptor.getFieldName());

				if (voEntity.isNew() || isLoaded) {
					sourceValue = fieldDescriptor.getSourceValue(sourceObject);
					targetValue = fieldDescriptor.getTargetValue(targetObject);
				}
			} else {

				if (targetMetadata != null) {
					targetMetadata.setLoaded(fieldDescriptor.getFieldName());
				}

				sourceValue = fieldDescriptor.getSourceValue(sourceObject);
				targetValue = fieldDescriptor.getTargetValue(targetObject);
			}

			if (sourceValue == null) {
				continue;
			}

			// collection
			if (List.class.isAssignableFrom(fieldDescriptor.getSourceType()) && List.class.isAssignableFrom(fieldDescriptor.getTargetType())) {

				List sourceList = (List) sourceValue;
				@SuppressWarnings("unchecked")
				List