org.swiftboot.data.model.id.IdPopulator Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of swiftboot-data Show documentation
Data access component based on Spring-Data-JPA
There is a newer version: 2.4.8
package org.swiftboot.data.model.id;

import org.apache.commons.lang3.StringUtils;
import org.apache.commons.lang3.reflect.FieldUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.swiftboot.data.model.entity.IdPersistable;
import org.swiftboot.util.BeanUtils;

import javax.annotation.Resource;
import javax.persistence.ManyToOne;
import javax.persistence.OneToMany;
import javax.persistence.OneToOne;
import java.lang.reflect.Field;
import java.util.List;

/**
 * Populator for generating ID and populating ID to entity and all it's one-to-one and one-to-many entities.
 *
 * @author swiftech
 * @since 2.0.1
 */
public class IdPopulator {

    private static final Logger log = LoggerFactory.getLogger(IdPopulator.class);

    @Resource
    private IdGenerator idGenerator;

    /**
     * Populate id to entity and all it's related entities.
     *
     * @param idEntity
     * @param attachParent whether attach parent entity.
     */
    public void populate(IdPersistable idEntity, boolean attachParent) {
//        log.trace(String.format("Try to populate entity '%s' with id", idEntity.getClass().getSimpleName()));
        if (StringUtils.isBlank(idEntity.getId())) {
            String id = idGenerator.generate(idEntity);
            if (log.isTraceEnabled()) log.trace(String.format("Generate ID for %s: %s", idEntity, id));
            idEntity.setId(id);
        }

        this.tryToPopulateOneToOneEntities(idEntity, attachParent);
        this.tryToPopulateOneToManyEntities(idEntity, attachParent);
    }

    /**
     * Try to set id of one to one relations.
     *
     * @param parentEntity
     * @param attachParent whether attach parent entity.
     */
    private void tryToPopulateOneToOneEntities(IdPersistable parentEntity, boolean attachParent) {
        List subObjectList = FieldUtils.getFieldsListWithAnnotation(parentEntity.getClass(), OneToOne.class);
        for (Field subObjectField : subObjectList) {
            Object relEntity = BeanUtils.forceGetProperty(parentEntity, subObjectField);
            if (relEntity == null) continue;
            if (log.isTraceEnabled()) log.trace(String.format("Try to populate one-to-one sub-entity %s with id", subObjectField.getName()));
            populate((IdPersistable) relEntity, attachParent);
        }
    }

    /**
     * Try to set id of one to many relations.
     *
     * @param parentEntity
     * @param attachParent whether attach parent entity.
     */
    private void tryToPopulateOneToManyEntities(IdPersistable parentEntity, boolean attachParent) {
        List subObjectList = FieldUtils.getFieldsListWithAnnotation(parentEntity.getClass(), OneToMany.class);
        for (Field subObjectField : subObjectList) {
            Object relEntities = BeanUtils.forceGetProperty(parentEntity, subObjectField);
            if (relEntities == null) {
                if (log.isDebugEnabled()) log.debug("@OneToMany annotated field is null");
                continue;
            }
            if (log.isTraceEnabled()) log.trace(String.format("Try to populate one-to-many sub-entity '%s' with id", subObjectField.getName()));
            for (Object subEntity : ((Iterable) relEntities)) {
                if (!(subEntity instanceof IdPersistable)) {
                    continue;
                }
                populate((IdPersistable) subEntity, attachParent); // - 递归 -
                // 反向处理 ManyToOne 的属性
                List m2oFields = FieldUtils.getFieldsListWithAnnotation(subEntity.getClass(), ManyToOne.class);
                for (Field m2oField : m2oFields) {
                    if (m2oField.getType() == parentEntity.getClass()) {
                        Object o = BeanUtils.forceGetProperty(subEntity, m2oField);
                        if (o == null) {
                            if (attachParent) {
                                if (log.isDebugEnabled()) log.debug(String.format("Attach parent entity %s to sub entity %s", parentEntity, subEntity));
                                BeanUtils.forceSetProperty(subEntity, m2oField, parentEntity);
                            }
                        }
                        else {
                            if (log.isTraceEnabled()) log.trace(String.format("Parent entity %s has already been attached sub entity %s", o, subEntity));
                        }
                        // log.debug(String.valueOf(BeanUtils.forceGetProperty(subEntity, m2oField)));
//                        log.trace(String.format("parentEntity %s- %s", parentEntity, parentEntity.hashCode()));
//                        log.trace(String.format("assigned parent entity %s - %s", o, o == null ? "" : o.hashCode()));
//                        log.trace(String.valueOf(parentEntity.equals(o)));
                    }
                }
            }
        }
    }

}