io.beanmapper.spring.web.MergePair Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of beanmapper-spring Show documentation

Spring support for the Bean Mapper

There is a newer version: 6.0.0

package io.beanmapper.spring.web;

import io.beanmapper.BeanMapper;

public class MergePair {

    private T beforeMerge;
    private T afterMerge;
    private final BeanMapper beanMapper;
    private final EntityFinder entityFinder;
    private final Class entityClass;
    private final MergedForm annotation;

    public MergePair(BeanMapper beanMapper, EntityFinder entityFinder, Class entityClass, MergedForm annotation) {
        this.beanMapper = beanMapper;
        this.entityFinder = entityFinder;
        this.entityClass = entityClass;
        this.annotation = annotation;
    }

    public void initNew(Object source) {
        setAfterMerge(beanMapper.map(source, targetEntityClass()));
    }

    public void merge(Object source, Long id) {
        T target = (T)findSource(id);
        setAfterMerge(beanMapper.map(source, target));
    }

    public T getBeforeMerge() {
        return beforeMerge;
    }

    public T getAfterMerge() {
        return afterMerge;
    }

    public Object result() {
        return isMergePair() ? this : getAfterMerge();
    }

    private void createBeforeMerge(Object source) {
        setBeforeMerge(beanMapper.wrapConfig().build().map(source, targetEntityClass()));
    }

    private Object findSource(Long id) {
        Object entity = entityFinder.find(id, targetEntityClass());
        if (isMergePair()) {
            createBeforeMerge(entity);
        }
        return entity;
    }

    private boolean isMergePair() {
        return entityClass.equals(MergePair.class);
    }

    private Class targetEntityClass() {
        return isMergePair() ? (Class)annotation.mergePairClass() : (Class)entityClass;
    }

    private void setBeforeMerge(T beforeMerge) {
        this.beforeMerge = beforeMerge;
    }

    private void setAfterMerge(T afterMerge) {
        this.afterMerge = afterMerge;
    }

}