EDIT

/**

 * Copyright (C) 2007 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.io.redlist.gefaesspflanzen;

import java.sql.ResultSet;

import java.sql.SQLException;

import java.util.HashMap;

import java.util.HashSet;

import java.util.Map;

import java.util.Set;

import org.apache.log4j.Logger;

import org.springframework.stereotype.Component;

import eu.etaxonomy.cdm.io.common.DbImportBase;

import eu.etaxonomy.cdm.io.common.IPartitionedIO;

import eu.etaxonomy.cdm.io.common.ImportHelper;

import eu.etaxonomy.cdm.io.common.ResultSetPartitioner;

import eu.etaxonomy.cdm.io.common.mapping.UndefinedTransformerMethodException;

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

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

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

import eu.etaxonomy.cdm.model.taxon.Taxon;

import eu.etaxonomy.cdm.model.taxon.TaxonBase;

import eu.etaxonomy.cdm.strategy.parser.NonViralNameParserImpl;

/**

 * @author a.mueller

 * @created 05.01.2016

 */

@Component

@SuppressWarnings("serial")

public class RedListGefaesspflanzenImportTaxa extends DbImportBase<RedListGefaesspflanzenImportState, RedListGefaesspflanzenImportConfigurator> {

    private static final Logger logger = Logger.getLogger(RedListGefaesspflanzenImportTaxa.class);

    private static final String tableName = "Rote Liste Gefäßpflanzen";

    private static final String pluralString = "taxa";

    private static final String TAXON_NAMESPACE = "taxon";

    public RedListGefaesspflanzenImportTaxa() {

        super(tableName, pluralString);

    }

    @Override

    protected String getIdQuery(RedListGefaesspflanzenImportState state) {

        return "SELECT NAMNR FROM V_TAXATLAS_D20_EXPORT t "

//                + " ORDER BY NAMNR"

                ;

    }

    @Override

    protected String getRecordQuery(RedListGefaesspflanzenImportConfigurator config) {

        String result = " SELECT * "

                + " FROM V_TAXATLAS_D20_EXPORT t "

                + " WHERE t.NAMNR IN (@IDSET)";

        result = result.replace("@IDSET", IPartitionedIO.ID_LIST_TOKEN);

        return result;

    }

    @Override

    protected void doInvoke(RedListGefaesspflanzenImportState state) {

        super.doInvoke(state);

    }

    @Override

    public boolean doPartition(ResultSetPartitioner partitioner, RedListGefaesspflanzenImportState state) {

        ResultSet rs = partitioner.getResultSet();

        Set<TaxonBase> taxaToSave = new HashSet<>();

        try {

            while (rs.next()){

                taxaToSave = makeSingleTaxon(state, rs);

            }

        } catch (SQLException e) {

            e.printStackTrace();

        }

        getTaxonService().saveOrUpdate(taxaToSave);

        return true;

    }

    private Set<TaxonBase> makeSingleTaxon(RedListGefaesspflanzenImportState state, ResultSet rs)

            throws SQLException {

        Set<TaxonBase> taxaToSave = new HashSet<>();

        long id = rs.getLong("NAMNR");

        String taxonNameString = rs.getString("TAXNAME");

        NonViralNameParserImpl parser = NonViralNameParserImpl.NewInstance();

        NonViralName<?> name = parser.parseFullName(taxonNameString);

        TaxonBase taxon = Taxon.NewInstance(name, null);

        taxaToSave.add(taxon);

        //id

        ImportHelper.setOriginalSource(taxon, state.getTransactionalSourceReference(), id, TAXON_NAMESPACE);

        ImportHelper.setOriginalSource(name, state.getTransactionalSourceReference(), id, TAXON_NAMESPACE);

        return taxaToSave;

    }

    private Rank makeRank(RedListGefaesspflanzenImportState state, String rankStr) {

        Rank rank = null;

        try {

            rank = state.getTransformer().getRankByKey(rankStr);

        } catch (UndefinedTransformerMethodException e) {

            e.printStackTrace();

        }

        return rank;

    }

    @Override

    public Map<Object, Map<String, ? extends CdmBase>> getRelatedObjectsForPartition(ResultSet rs,

            RedListGefaesspflanzenImportState state) {

        Map<Object, Map<String, ? extends CdmBase>> result = new HashMap<>();

//        Map<String, TeamOrPersonBase<?>> authorMap = new HashMap<>();

//        Set<String> authorSet = new HashSet<>();

//        Set<String> referenceIdSet = new HashSet<String>();

//

//        try {

//            while (rs.next()){

//                String authorStr = rs.getString("tax_author_name");

//                authorSet.add(authorStr);

//                handleForeignKey(rs, referenceIdSet, "tax_document");

//            }

//        } catch (SQLException e) {

//            e.printStackTrace();

//        }

//

//        //Authors

//        Set<UUID> uuidSet = new HashSet<>();

//        for (String authorStr : authorSet){

//            UUID uuid = state.getAuthorUuid(authorStr);

//            uuidSet.add(uuid);

//        }

//        List<TeamOrPersonBase<?>> authors = (List)getAgentService().find(uuidSet);

//        Map<UUID, TeamOrPersonBase<?>> authorUuidMap = new HashMap<>();

//        for (TeamOrPersonBase<?> author : authors){

//            authorUuidMap.put(author.getUuid(), author);

//        }

//

//        for (String authorStr : authorSet){

//            UUID uuid = state.getAuthorUuid(authorStr);

//            TeamOrPersonBase<?> author = authorUuidMap.get(uuid);

//            authorMap.put(authorStr, author);

//        }

//        result.put(AUTHOR_NAMESPACE, authorMap);

//

//        //reference map

//        String nameSpace = REFERENCE_NAMESPACE;

//        Class<?> cdmClass = Reference.class;

//        Set<String> idSet = referenceIdSet;

//        Map<String, Reference<?>> referenceMap = (Map<String, Reference<?>>)getCommonService().getSourcedObjectsByIdInSource(cdmClass, idSet, nameSpace);

//        result.put(nameSpace, referenceMap);

//

//        //secundum

//        UUID secUuid = state.getConfig().getSecUuid();

//        Reference<?> secRef = getReferenceService().find(secUuid);

//        referenceMap.put(secUuid.toString(), secRef);

        return result;

    }

    @Override

    protected boolean doCheck(RedListGefaesspflanzenImportState state) {

        return false;

    }

    @Override

    protected boolean isIgnore(RedListGefaesspflanzenImportState state) {

        return false;

    }

}