fr.ird.observe.spi.context.DataDtoEntityContext Maven / Gradle / Ivy

Go to download
package fr.ird.observe.spi.context;

/*-
 * #%L
 * ObServe Toolkit :: Common Persistence
 * %%
 * Copyright (C) 2008 - 2017 IRD, Ultreia.io
 * %%
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as
 * published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public
 * License along with this program.  If not, see
 * .
 * #L%
 */

import com.google.common.collect.ImmutableSet;
import fr.ird.observe.binder.data.DataDtoReferenceBinder;
import fr.ird.observe.binder.data.DataEntityDtoBinderSupport;
import fr.ird.observe.binder.data.DataEntityReferenceBinderSupport;
import fr.ird.observe.dto.DataNotFoundException;
import fr.ird.observe.dto.data.DataDto;
import fr.ird.observe.dto.form.Form;
import fr.ird.observe.dto.reference.DataDtoReference;
import fr.ird.observe.dto.reference.DataDtoReferenceSet;
import fr.ird.observe.dto.reference.DtoReferenceDefinition;
import fr.ird.observe.dto.referential.ReferentialLocale;
import fr.ird.observe.entities.ObserveDataEntity;
import fr.ird.observe.spi.DtoModelHelper;
import java.util.Collection;
import java.util.function.Consumer;
import java.util.function.Function;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.nuiton.topia.persistence.TopiaDao;
import org.nuiton.topia.persistence.TopiaNoResultException;
import org.nuiton.topia.persistence.TopiaPersistenceContext;

/**
 * Created by tchemit on 03/09/17.
 *
 * @author Tony Chemit - [email protected]
 */
@SuppressWarnings("unchecked")
public class DataDtoEntityContext, E extends ObserveDataEntity, T extends TopiaDao> {

    /** Logger. */
    private static final Log log = LogFactory.getLog(DataDtoEntityContext.class);

    private final DataDtoContext dtoContext;
    private final Class entityType;
    private final DataEntityDtoBinderSupport entityBinder;
    private final DataEntityReferenceBinderSupport entityReferenceBinder;
    private final Function daoFunction;

    public DataDtoEntityContext(DataEntityDtoBinderSupport entityBinder, DataEntityReferenceBinderSupport entityReferenceBinder, Function daoFunction) {

        Class dtoType = entityBinder.getDtoType();
        DataDtoContext dtoContext = DtoModelHelper.fromDataDto(dtoType);
        if (dtoContext == null) {
            dtoContext = new DataDtoContext<>(dtoType);
        }
        this.dtoContext = dtoContext;
        this.entityType = entityBinder.getEntityType();
        this.entityBinder = entityBinder;
        this.entityReferenceBinder = entityReferenceBinder;
        this.daoFunction = daoFunction;
    }

    public Class toEntityType() {
        return entityType;
    }

    public DataEntityDtoBinderSupport toEntityBinder() {
        return entityBinder;
    }

    public Class toDtoType() {
        return dtoContext.toDtoType();
    }

    public Class toReferenceType() {
        return dtoContext.toReferenceType();
    }

    public DataDtoReferenceBinder toReferenceBinder() {
        return dtoContext.toReferenceBinder();
    }

    public DtoReferenceDefinition toReferenceDefinition() {
        return dtoContext.toReferenceDefinition();
    }

    public R toReference(ReferentialLocale referentialLocale, D dto) {
        return dtoContext.toReference(referentialLocale, dto);
    }

    //FIXME Il faut voir si on a besoin de spécifier le dto, ça semble fonctionner sans, donc à supprimer après tests
    @SuppressWarnings("unused")
    public D toDto(ReferentialLocale referentialLocale, Class dtoType, E entity) {
        D dto = entityBinder.newDto();
        entityBinder.copyToDto(referentialLocale, entity, dto);
        return dto;
    }

    public void copyToEntity(ReferentialLocale referentialLocale, D dto, E entity) {
        entityBinder.copyToEntity(referentialLocale, dto, entity);
    }

    public DataEntityReferenceBinderSupport toEntityReferenceBinder() {
        return entityReferenceBinder;
    }


    public T getDao(TopiaPersistenceContext persistenceContext) {
        return daoFunction.apply(persistenceContext);
    }

    public boolean existsEntity(TopiaPersistenceContext persistenceContext, String id) {
        return getDao(persistenceContext).forTopiaIdEquals(id).exists();
    }

    public E loadEntity(TopiaPersistenceContext persistenceContext, String id) {
        log.debug("Load entity with id: " + id);
        try {
            return getDao(persistenceContext).forTopiaIdEquals(id).findUnique();
        } catch (TopiaNoResultException e) {
            throw new DataNotFoundException(toDtoType(), id);
        }
    }

    public R loadEntityToDataReferenceDto(TopiaPersistenceContext persistenceContext, ReferentialLocale referentialLocale, String id) {

        E entity = loadEntity(persistenceContext, id);

        DataEntityReferenceBinderSupport dataBinderFromDto = toEntityReferenceBinder();
        return dataBinderFromDto.toReference(referentialLocale, entity);

    }

    public Form dataEntityToForm(ReferentialLocale referentialLocale, E entity) {

        D dto = toDto(referentialLocale, toDtoType(), entity);
        return Form.newFormDto(toDtoType(), dto);

    }

    public E loadOrCreateEntityFromDataDto(TopiaPersistenceContext persistenceContext, D dto) {

        E entity;
        if (dto.isPersisted()) {
            entity = loadEntity(persistenceContext, dto.getId());
        } else {
            DataEntityDtoBinderSupport binder = toEntityBinder();
            entity = binder.newEntity();
        }
        return entity;
    }

    public void copyDataDtoToEntity(D dto, E entity, ReferentialLocale referentialLocale) {

        copyToEntity(referentialLocale, dto, entity);

    }

    public void deleteEntity(TopiaPersistenceContext persistenceContext, Iterable ids, Consumer> extraCallback) {
        T dao = getDao(persistenceContext);
        for (String id : ids) {
            E entity = dao.forTopiaIdEquals(id).findUniqueOrNull();
            if (entity == null) {
                throw new DataNotFoundException(toDtoType(), id);
            }
            dao.delete(entity);
        }

        if (extraCallback != null) {
            extraCallback.accept(toEntityType());
        }

    }

    public D loadEntityToDataDto(TopiaPersistenceContext persistenceContext, String id, ReferentialLocale referentialLocale) {

        E entity = loadEntity(persistenceContext, id);
        DataEntityDtoBinderSupport binder = toEntityBinder();
        return binder.toDto(referentialLocale, entity);

    }

    public DataDtoReferenceSet toDataReferenceSet(ReferentialLocale referentialLocale, Collection entities) {

        DataEntityReferenceBinderSupport binder = toEntityReferenceBinder();

        ImmutableSet.Builder references = ImmutableSet.builder();
        for (E entity : entities) {

            R reference = binder.toReference(referentialLocale, entity);
            references.add(reference);

        }

        return DataDtoReferenceSet.of(toReferenceType(), references.build());

    }


    public R toReference(ReferentialLocale referentialLocale, E entity) {
        DataEntityReferenceBinderSupport binder = toEntityReferenceBinder();
        return binder.toReference(referentialLocale, entity);
    }

    public E newEntity() {
        DataEntityDtoBinderSupport binder = toEntityBinder();
        return binder.newEntity();
    }
}