fr.ird.observe.spi.PersistenceModelHelperInitializerSupport Maven / Gradle / Ivy

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

/*-
 * #%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 fr.ird.observe.dto.IdHelper;
import fr.ird.observe.dto.data.DataDto;
import fr.ird.observe.dto.reference.DataDtoReference;
import fr.ird.observe.dto.reference.ReferentialDtoReference;
import fr.ird.observe.dto.referential.ReferentialDto;
import fr.ird.observe.entities.ObserveDataEntity;
import fr.ird.observe.entities.referentiel.ObserveReferentialEntity;
import fr.ird.observe.spi.context.DataDtoEntityContext;
import fr.ird.observe.spi.context.ReferentialDtoEntityContext;
import fr.ird.observe.spi.map.DtoToEntityClassMap;
import fr.ird.observe.spi.map.EntityToDtoClassMap;
import fr.ird.observe.spi.map.ImmutableDtoMap;
import fr.ird.observe.spi.map.ImmutableEntityMap;

public abstract class PersistenceModelHelperInitializerSupport {

    private ImmutableEntityMap referentialEntityContexts;
    private ImmutableEntityMap dataEntityContexts;
    private ImmutableDtoMap referentialDtoContexts;
    private ImmutableDtoMap dataDtoContexts;

    private EntityToDtoClassMap entityToDtoClassMapping;
    private DtoToEntityClassMap dtoToEntityClassMapping;

    private final ImmutableEntityMap.Builder referentialEntityContextsBuilder = ImmutableEntityMap.builder();
    private final ImmutableEntityMap.Builder dataEntityContextsBuilder = ImmutableEntityMap.builder();

    private final ImmutableDtoMap.Builder referentialDtoContextsBuilder = ImmutableDtoMap.builder();
    private final ImmutableDtoMap.Builder dataDtoContextsBuilder = ImmutableDtoMap.builder();

    private final ImmutableEntityMap.Builder entityToDtoClassMappingBuilder = ImmutableEntityMap.builder();
    private final ImmutableDtoMap.Builder dtoToEntityClassMappingBuilder = ImmutableDtoMap.builder();

    private static PersistenceModelHelperInitializerSupport INSTANCE;

    public static PersistenceModelHelperInitializerSupport get() {
        if (INSTANCE == null) {
            INSTANCE = DtoModelHelperInitializerSupport.getService(PersistenceModelHelperInitializerSupport.class);
        }
        return INSTANCE;
    }

    public void end() {
        this.dataEntityContexts = dataEntityContextsBuilder.build();
        this.referentialEntityContexts = referentialEntityContextsBuilder.build();
        this.referentialDtoContexts = referentialDtoContextsBuilder.build();
        this.dataDtoContexts = dataDtoContextsBuilder.build();

        this.entityToDtoClassMapping = new EntityToDtoClassMap(entityToDtoClassMappingBuilder.build());
        this.dtoToEntityClassMapping = new DtoToEntityClassMap(dtoToEntityClassMappingBuilder.build());

    }

    , E extends ObserveReferentialEntity> void registerReferentialContext(ReferentialDtoEntityContext context) {
        referentialEntityContextsBuilder.put(context.toEntityType(), context);
        referentialDtoContextsBuilder.put(context.toDtoType(), context);
    }

    , E extends ObserveDataEntity> void registerDataContext(DataDtoEntityContext context) {
        //FIXME Only register if main type
        Class entityType = context.toEntityType();

        Class dtoType = context.toDtoType();
        dataDtoContextsBuilder.put(dtoType, context);
        if (IdHelper.equalsCleanId(entityType, dtoType)) {
            // main dto, keep this link only since when coming from entity, we only can bind to main dto
            dataEntityContextsBuilder.put(entityType, context);
        }
    }

    ImmutableEntityMap getReferentialEntityContexts() {
        return referentialEntityContexts;
    }

    ImmutableEntityMap getDataEntityContexts() {
        return dataEntityContexts;
    }

    ImmutableDtoMap getReferentialDtoContexts() {
        return referentialDtoContexts;
    }

    ImmutableDtoMap getDataDtoContexts() {
        return dataDtoContexts;
    }

}