io.ebeaninternal.server.core.DiffHelp Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ebean Show documentation
composite of common runtime dependencies for all platforms
There is a newer version: 15.10.0
package io.ebeaninternal.server.core;

import io.ebean.ValuePair;
import io.ebean.bean.EntityBean;
import io.ebeaninternal.server.deploy.BeanDescriptor;
import io.ebeaninternal.server.deploy.BeanProperty;
import io.ebeaninternal.server.deploy.BeanPropertyAssocMany;
import io.ebeaninternal.server.deploy.BeanPropertyAssocOne;

import java.util.Iterator;
import java.util.LinkedHashMap;
import java.util.Map;

/**
 * Helper to perform a diff given two beans of the same type.
 * 
 * This intentionally does not include any OneToMany or ManyToMany properties.
 * 
 */
public class DiffHelp {

  private DiffHelp() {
  }

  /**
   * Return a map of the differences between a and b.
   * 
   * A and B must be of the same type. B can be null, in which case the 'dirty
   * values' of a is returned.
   * 
   * 
   * This intentionally does not include as OneToMany or ManyToMany properties.
   * 
   */
  public static Map diff(Object newBean, Object oldBean, BeanDescriptor desc) {

    if (!(newBean instanceof EntityBean)) {
      throw new IllegalArgumentException("First bean expected to be an enhanced EntityBean? bean:" + newBean);
    }

    if (oldBean != null) {
      if (!(oldBean instanceof EntityBean)) {
        throw new IllegalArgumentException("Second bean expected to be an enhanced EntityBean? bean:" + oldBean);
      }
      if (!newBean.getClass().isAssignableFrom(oldBean.getClass())) {
        throw new IllegalArgumentException("Second bean not assignable to the first bean?");
      }
    }

    if (oldBean == null) {
      return ((EntityBean) newBean)._ebean_getIntercept().getDirtyValues();
    }

    return desc.diff((EntityBean) newBean, (EntityBean) oldBean);
  }


  /**
   * Flattens an existing diff map converting assoc one beans into the associated id changes.
   */
  public static Map flatten(Map values, BeanDescriptor desc) {

    Map flattened = null;

    Iterator> iterator = values.entrySet().iterator();
    while (iterator.hasNext()) {
      Map.Entry entry = iterator.next();
      BeanProperty beanProperty = desc.getBeanProperty(entry.getKey());
      if (beanProperty instanceof BeanPropertyAssocMany) {
        // filter out assoc many bean properties
        iterator.remove();

      } else if (beanProperty instanceof BeanPropertyAssocOne) {
        BeanPropertyAssocOne assoc = (BeanPropertyAssocOne) beanProperty;
        if (!assoc.isEmbedded()) {
          // flatten for assoc one beans
          if (flattened == null) {
            flattened = new LinkedHashMap<>();
          }
          flattenToId(flattened, entry, beanProperty, assoc);
          iterator.remove();
        }
      }
    }

    if (flattened != null) {
      values.putAll(flattened);
    }

    return values;
  }

  private static void flattenToId(Map flattened, Map.Entry entry, BeanProperty beanProperty, BeanPropertyAssocOne assoc) {

    BeanDescriptor oneDesc = assoc.getTargetDescriptor();

    ValuePair value = entry.getValue();
    Object newId = value.getNewValue() == null ? null : oneDesc.getId((EntityBean) value.getNewValue());
    Object oldId = value.getOldValue() == null ? null : oneDesc.getId((EntityBean) value.getOldValue());

    String propName = beanProperty.getName() + "." + oneDesc.getIdProperty().getName();
    flattened.put(propName, new ValuePair(newId, oldId));
  }
}