com.devonfw.module.jpa.dataaccess.impl.data.GenericRevisionedRepositoryImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of devon4j-jpa-spring-data Show documentation
Spring-data infrastructure.
There is a newer version: 2023.01.001
Show newest version
package com.devonfw.module.jpa.dataaccess.impl.data;

import java.io.Serializable;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.stream.Collectors;

import javax.persistence.EntityManager;

import org.hibernate.envers.AuditReader;
import org.hibernate.envers.AuditReaderFactory;
import org.springframework.data.jpa.repository.support.JpaEntityInformation;

import com.devonfw.module.basic.common.api.RevisionMetadata;
import com.devonfw.module.basic.common.api.entity.RevisionedEntity;
import com.devonfw.module.jpa.dataaccess.api.AdvancedRevisionEntity;
import com.devonfw.module.jpa.dataaccess.api.JpaHelper;
import com.devonfw.module.jpa.dataaccess.api.QueryUtil;
import com.devonfw.module.jpa.dataaccess.api.data.GenericRevisionedRepository;
import com.devonfw.module.jpa.dataaccess.impl.LazyRevisionMetadata;
import com.querydsl.core.alias.Alias;
import com.querydsl.jpa.impl.JPAQuery;

/**
 * Implementation of {@link GenericRevisionedRepository}.
 *
 * @param  generic type of the managed {@link #getEntityClass() entity}.
 * @param  generic type of the {@link com.devonfw.module.basic.common.api.entity.PersistenceEntity#getId() primary
 *        key} of the entity.
 *
 * @since 3.0.0
 */
public class GenericRevisionedRepositoryImpl extends GenericRepositoryImpl
    implements GenericRevisionedRepository {

  /**
   * The constructor.
   *
   * @param entityInformation the {@link JpaEntityInformation}.
   * @param entityManager the JPA {@link EntityManager}.
   */
  public GenericRevisionedRepositoryImpl(JpaEntityInformation entityInformation, EntityManager entityManager) {

    super(entityInformation, entityManager);
  }

  @Override
  public E find(ID id, Number revision) {

    AuditReader auditReader = AuditReaderFactory.get(this.entityManager);
    E entity = auditReader.find(this.entityInformation.getJavaType(), id, revision);
    if (entity instanceof RevisionedEntity) {
      ((RevisionedEntity) entity).setRevision(revision);
    }
    return entity;
  }

  @SuppressWarnings("unchecked")
  @Override
  public List getRevisionHistoryMetadata(ID id, boolean lazy) {

    AuditReader auditReader = AuditReaderFactory.get(this.entityManager);
    List revisionList = auditReader.getRevisions(getEntityClass(), id);
    if (revisionList.isEmpty()) {
      return Collections.emptyList();
    }
    if (lazy) {
      List result = new ArrayList<>(revisionList.size());
      for (Number revision : revisionList) {
        Long revisionLong = Long.valueOf(revision.longValue());
        result.add(new LazyRevisionMetadata(this.entityManager, revisionLong));
      }
      return result;
    } else {
      AdvancedRevisionEntity rev = Alias.alias(AdvancedRevisionEntity.class);
      JPAQuery query = new JPAQuery(this.entityManager)
          .from(Alias.$(rev));
      @SuppressWarnings("rawtypes")
      List revList = revisionList;
      QueryUtil.get().whereIn(query, Alias.$(rev.getId()), (List) revList);
      query.orderBy(Alias.$(rev.getId()).asc());
      List resultList = query.fetch();
      return resultList.stream().map(x -> JpaHelper.asRevisionMetaData(x)).collect(Collectors.toList());
    }
  }

  @Override
  public RevisionMetadata getLastRevisionHistoryMetadata(ID id) {

    AuditReader auditReader = AuditReaderFactory.get(this.entityManager);
    List revisionList = auditReader.getRevisions(getEntityClass(), id);
    if (revisionList.isEmpty()) {
      return null;
    }
    Number lastRevision = revisionList.get(revisionList.size() - 1);
    AdvancedRevisionEntity revisionEntity = this.entityManager.find(AdvancedRevisionEntity.class, lastRevision);
    if (revisionEntity == null) {
      throw new IllegalStateException("Could not find AdvancedRevisionEntity for ID '" + id + "'.");
    }
    return JpaHelper.asRevisionMetaData(revisionEntity);
  }

}