org.molgenis.data.AbstractSystemEntityFactory Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of molgenis-data Show documentation
There is a newer version: 8.4.5
package org.molgenis.data;

import static java.util.Objects.requireNonNull;

import java.lang.reflect.Constructor;
import java.lang.reflect.InvocationTargetException;
import org.molgenis.data.meta.SystemEntityType;
import org.molgenis.data.meta.model.EntityType;
import org.molgenis.data.populate.EntityPopulator;

/**
 * Entity factory base class
 *
 * @param  entity type
 * @param  entity meta data type
 * @param  entity id type
 */
public abstract class AbstractSystemEntityFactory
    implements EntityFactory {
  private final Class entityClass;
  private final Constructor entityConstructorWithEntity;
  private final Constructor entityConstructorWithEntityType;
  private final M systemEntityType;
  private final EntityPopulator entityPopulator;

  /**
   * Constructs a new entity factory that creates entities of the given type, meta data type and id
   * type
   *
   * @param entityClass entity type
   * @param systemEntityType entity meta data type
   * @param entityPopulator entity populator
   */
  protected AbstractSystemEntityFactory(
      Class entityClass, M systemEntityType, EntityPopulator entityPopulator) {
    this.entityClass = requireNonNull(entityClass);

    // determining constructors at creation time validates that required constructors exist on
    // start-up
    this.entityConstructorWithEntity = getConstructorEntity(entityClass);
    this.entityConstructorWithEntityType = getConstructorEntityType(entityClass);
    this.systemEntityType = systemEntityType;
    this.entityPopulator = requireNonNull(entityPopulator);
  }

  public M getEntityType() {
    return systemEntityType;
  }

  @Override
  public String getEntityTypeId() {
    return systemEntityType.getId();
  }

  @Override
  public E create() {
    E entity;
    try {
      entity = entityConstructorWithEntityType.newInstance(systemEntityType);
    } catch (InstantiationException | IllegalAccessException | InvocationTargetException e) {
      throw new EntityConstructionException(e);
    }
    entityPopulator.populate(entity);
    return entity;
  }

  @Override
  public E create(P id) {
    E entity = create();
    entity.setIdValue(id);
    return entity;
  }

  @SuppressWarnings("unchecked")
  @Override
  public E create(Entity entity) {
    if (entity == null) {
      return null;
    }
    if (entity.getClass().equals(entityClass)) {
      return (E) entity;
    }

    try {
      return entityConstructorWithEntity.newInstance(entity);
    } catch (InstantiationException | IllegalAccessException | InvocationTargetException e) {
      throw new EntityConstructionException(e);
    }
  }

  private Constructor getConstructorEntity(Class entityClass) {
    try {
      return entityClass.getConstructor(Entity.class);
    } catch (NoSuchMethodException e) {
      String message =
          String.format(
              "[%s] is missing the required constructor [public %s(%s)]",
              entityClass.getName(), entityClass.getSimpleName(), Entity.class.getSimpleName());
      throw new EntityConstructionException(message, e);
    }
  }

  private Constructor getConstructorEntityType(Class entityClass) {
    try {
      return entityClass.getConstructor(EntityType.class);
    } catch (NoSuchMethodException e) {
      String message =
          String.format(
              "[%s] is missing the required constructor [public %s(%s)]",
              entityClass.getName(), entityClass.getSimpleName(), EntityType.class.getSimpleName());
      throw new EntityConstructionException(message, e);
    }
  }
}