org.molgenis.semanticmapper.mapping.model.MappingTarget Maven / Gradle / Ivy

Go to download
package org.molgenis.semanticmapper.mapping.model;

import com.google.common.collect.Lists;
import java.util.Collection;
import java.util.LinkedHashMap;
import java.util.Map;
import java.util.Set;
import java.util.stream.Collectors;
import org.molgenis.data.meta.model.EntityType;
import org.molgenis.data.util.EntityTypeUtils;

public class MappingTarget {
  private String identifier;
  private final EntityType target;
  /** Maps source name to entityMapping for that source to the target */
  private Map entityMappings;

  /** Creates a new empty {@link MappingTarget} */
  public MappingTarget(EntityType target) {
    this.identifier = null;
    this.target = target;
    this.entityMappings = new LinkedHashMap<>();
  }

  /** Creates a new instance of MappingTarget. */
  public MappingTarget(
      String identifier, EntityType target, Collection entityMappings) {
    this.identifier = identifier;
    this.target = target;
    this.entityMappings = new LinkedHashMap<>();
    for (EntityMapping mapping : entityMappings) {
      this.entityMappings.put(mapping.getName(), mapping);
    }
  }

  public String getIdentifier() {
    return identifier;
  }

  public void removeIdentifiers() {
    identifier = null;
    entityMappings.values().forEach(EntityMapping::removeIdentifiers);
  }

  public EntityType getTarget() {
    return target;
  }

  public EntityMapping getMappingForSource(String source) {
    return entityMappings.get(source);
  }

  public Set getMissingTargetAttributeNames() {
    return entityMappings.values().stream()
        .flatMap(EntityMapping::getMissingTargetAttributeNames)
        .collect(Collectors.toSet());
  }

  /**
   * Adds a new {@link EntityMapping} to this target for a certain source.
   *
   * @param source {@link EntityType} for the source entity that is mapped to this target
   * @return the newly created empty {@link EntityMapping}
   */
  public EntityMapping addSource(EntityType source) {
    if (entityMappings.containsKey(source.getId())) {
      throw new IllegalStateException("Mapping already present for source " + source.getId());
    }
    EntityMapping result = new EntityMapping(source, target);
    entityMappings.put(source.getId(), result);
    return result;
  }

  @Override
  public int hashCode() {
    final int prime = 31;
    int result = 1;
    result = prime * result + ((entityMappings == null) ? 0 : entityMappings.hashCode());
    result = prime * result + ((identifier == null) ? 0 : identifier.hashCode());
    result = prime * result + ((target == null) ? 0 : target.hashCode());
    return result;
  }

  @Override
  public boolean equals(Object obj) {
    if (this == obj) return true;
    if (obj == null) return false;
    if (getClass() != obj.getClass()) return false;
    MappingTarget other = (MappingTarget) obj;
    if (entityMappings == null) {
      if (other.entityMappings != null) return false;
    } else if (!entityMappings.equals(other.entityMappings)) return false;
    if (identifier == null) {
      if (other.identifier != null) return false;
    } else if (!identifier.equals(other.identifier)) return false;
    if (target == null) {
      if (other.target != null) return false;
    } else if (!target.equals(other.target)) return false;
    return true;
  }

  public void setIdentifier(String identifier) {
    this.identifier = identifier;
  }

  @Override
  public String toString() {
    return "MappingTarget [identifier="
        + identifier
        + ", target="
        + target
        + ", entityMappings="
        + entityMappings
        + "]";
  }

  public String getName() {
    return target.getId();
  }

  public Collection getEntityMappings() {
    return Lists.newArrayList(entityMappings.values());
  }

  public void removeSource(String source) {
    entityMappings.remove(source);
  }

  public boolean hasMappingFor(String name) {
    return entityMappings.containsKey(name);
  }

  public boolean hasSelfReferences() {
    return EntityTypeUtils.hasSelfReferences(getTarget());
  }
}