EDIT

/**

* Copyright (C) 2017 EDIT

* European Distributed Institute of Taxonomy

* http://www.e-taxonomy.eu

*

* The contents of this file are subject to the Mozilla Public License Version 1.1

* See LICENSE.TXT at the top of this package for the full license terms.

*/

package eu.etaxonomy.cdm.mock;

import java.util.Collection;

import java.util.HashMap;

import java.util.HashSet;

import java.util.List;

import java.util.Map;

import java.util.Set;

import java.util.UUID;

import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.beans.factory.annotation.Qualifier;

import org.springframework.stereotype.Service;

import org.springframework.transaction.TransactionStatus;

import org.springframework.transaction.annotation.Transactional;

import eu.etaxonomy.cdm.api.application.CdmRepository;

import eu.etaxonomy.cdm.model.common.CdmBase;

import eu.etaxonomy.cdm.model.name.Rank;

import eu.etaxonomy.cdm.model.name.TaxonNameBase;

import eu.etaxonomy.cdm.vaadin.presenter.phycobank.RegistrationDTO;

/**

 * @author a.kohlbecker

 * @since Mar 10, 2017

 *

 */

@Service("registrationServiceMock")

@Transactional

public class RegistrationService {

    /**

     *

     */

    private static final int SIZE = 50; // FIXME test performance with 50 !!!!!

    @Autowired

    @Qualifier("cdmRepository")

    private CdmRepository repo;

    private Map<UUID, Registration> registrationsByUUID = new HashMap<>();

    private Map<String, Registration> registrationsByRegID = new HashMap<>();

    private Map<String, RegistrationDTO> registrationDTOsByRegID = new HashMap<>();

    private Map<Integer, Set<RegistrationDTO>> registrationDTOsByCitationID = new HashMap<>();

    private Collection<CdmBase> cdmEntities = new HashSet<>();

    public RegistrationService() {

    }

    int minTypeDesignationCount = 1;

    protected void init(){

        if(registrationsByUUID.size() == 0){

            TransactionStatus tx = repo.startTransaction(true);

            int pageIndex = 0;

            while(registrationsByUUID.size() < SIZE){

                List<TaxonNameBase> names = repo.getNameService().list(TaxonNameBase.class, 100, pageIndex++, null, null);

                for(TaxonNameBase name : names){

                    if(name != null && name.getRank() != null && name.getRank().isLower(Rank.SUBFAMILY())){

                        if(name.getTypeDesignations().size() > minTypeDesignationCount - 1) {

                            // name

                            Registration nameReg = new Registration();

                            nameReg.setName(name);

                            cdmEntities.add(name);

                            put(nameReg, new RegistrationDTO(nameReg, null));

                            // typedesignation

                            Registration typedesignationReg = new Registration();

                            typedesignationReg.addTypeDesignations(name.getTypeDesignations());

                            cdmEntities.addAll(name.getTypeDesignations());

                            put(typedesignationReg,  new RegistrationDTO(typedesignationReg, name));

                        }

                    }

                }

            }

            repo.commitTransaction(tx);

        }

    }

    /**

     * @param reg

     */

    private void put(Registration reg, RegistrationDTO dto) {

        registrationsByUUID.put(reg.getUuid(), reg);

        registrationsByRegID.put(reg.getSpecificIdentifier(), reg);

        registrationDTOsByRegID.put(reg.getSpecificIdentifier(), dto);

        if(! registrationDTOsByCitationID.containsKey(dto.getCitationID())){

            registrationDTOsByCitationID.put(dto.getCitationID(), new HashSet<RegistrationDTO>());

        }

        registrationDTOsByCitationID.get(dto.getCitationID()).add(dto);

    }

    private void mergeBack(){

        cdmEntities.forEach(e -> repo.getNameService().getSession().merge(e));

    }

    /**

     * {@inheritDoc}

     */

    public Registration load(UUID uuid) {

        init();

        return registrationsByUUID.get(uuid);

    }

    public Collection<Registration> list(){

        init();

        return registrationsByUUID.values();

    }

    public Collection<RegistrationDTO> listDTOs() {

        init();

        return registrationDTOsByRegID.values();

    }

    public Map<Integer, Set<RegistrationDTO>> listDTOsByWorkingSet() {

        init();

        return registrationDTOsByCitationID;

    }

    /**

     * @param registrationID

     * @return

     */

    public Registration loadByRegistrationID(Integer registrationID) {

        init();

        return registrationsByRegID.get(registrationID.toString());

    }

}