app-import/src/main/java/eu/etaxonomy/cdm/io/redlist/gefaesspflanzen/RedListGefaesspflanzenImportNames.java

   1 /**
   2  * Copyright (C) 2007 EDIT
   3  * European Distributed Institute of Taxonomy
   4  * http://www.e-taxonomy.eu
   5  *
   6  * The contents of this file are subject to the Mozilla Public License Version 1.1
   7  * See LICENSE.TXT at the top of this package for the full license terms.
   8  */
   9
  10 package eu.etaxonomy.cdm.io.redlist.gefaesspflanzen;
  11
  12 import java.sql.ResultSet;
  13 import java.sql.SQLException;
  14 import java.util.HashMap;
  15 import java.util.HashSet;
  16 import java.util.Map;
  17 import java.util.Set;
  18
  19 import org.apache.log4j.Logger;
  20 import org.springframework.stereotype.Component;
  21
  22 import eu.etaxonomy.cdm.common.CdmUtils;
  23 import eu.etaxonomy.cdm.io.common.DbImportBase;
  24 import eu.etaxonomy.cdm.io.common.IPartitionedIO;
  25 import eu.etaxonomy.cdm.io.common.ImportHelper;
  26 import eu.etaxonomy.cdm.io.common.ResultSetPartitioner;
  27 import eu.etaxonomy.cdm.io.common.mapping.UndefinedTransformerMethodException;
  28 import eu.etaxonomy.cdm.model.agent.AgentBase;
  29 import eu.etaxonomy.cdm.model.agent.TeamOrPersonBase;
  30 import eu.etaxonomy.cdm.model.common.CdmBase;
  31 import eu.etaxonomy.cdm.model.name.BotanicalName;
  32 import eu.etaxonomy.cdm.model.name.Rank;
  33 import eu.etaxonomy.cdm.model.name.TaxonNameBase;
  34 import eu.etaxonomy.cdm.model.taxon.Synonym;
  35 import eu.etaxonomy.cdm.model.taxon.Taxon;
  36 import eu.etaxonomy.cdm.model.taxon.TaxonBase;
  37
  38 /**
  39  *
  40  * @author pplitzner
  41  * @date Mar 1, 2016
  42  *
  43  */
  44
  45 @Component
  46 @SuppressWarnings("serial")
  47 public class RedListGefaesspflanzenImportNames extends DbImportBase<RedListGefaesspflanzenImportState, RedListGefaesspflanzenImportConfigurator> {
  48
  49     private static final String EX = " ex ";
  50
  51     private static final Logger logger = Logger.getLogger(RedListGefaesspflanzenImportNames.class);
  52
  53     private static final String tableName = "Rote Liste Gefäßpflanzen";
  54
  55     private static final String pluralString = "names";
  56
  57     public RedListGefaesspflanzenImportNames() {
  58         super(tableName, pluralString);
  59     }
  60
  61     @Override
  62     protected String getIdQuery(RedListGefaesspflanzenImportState state) {
  63         return "SELECT NAMNR "
  64                 + "FROM V_TAXATLAS_D20_EXPORT t "
  65                 + " ORDER BY NAMNR";
  66     }
  67
  68     @Override
  69     protected String getRecordQuery(RedListGefaesspflanzenImportConfigurator config) {
  70         String result = " SELECT * "
  71                 + " FROM V_TAXATLAS_D20_EXPORT t "
  72                 + " WHERE t.NAMNR IN (@IDSET)";
  73         result = result.replace("@IDSET", IPartitionedIO.ID_LIST_TOKEN);
  74         return result;
  75     }
  76
  77     @Override
  78     protected void doInvoke(RedListGefaesspflanzenImportState state) {
  79         super.doInvoke(state);
  80     }
  81
  82
  83     @Override
  84     public boolean doPartition(ResultSetPartitioner partitioner, RedListGefaesspflanzenImportState state) {
  85         ResultSet rs = partitioner.getResultSet();
  86         Set<TaxonNameBase> namesToSave = new HashSet<TaxonNameBase>();
  87         Set<TaxonBase> taxaToSave = new HashSet<TaxonBase>();
  88         try {
  89             while (rs.next()){
  90                 makeSingleNameAndTaxon(state, rs, namesToSave, taxaToSave);
  91
  92             }
  93         } catch (SQLException e) {
  94             e.printStackTrace();
  95         }
  96
  97         getNameService().saveOrUpdate(namesToSave);
  98         getTaxonService().saveOrUpdate(taxaToSave);
  99         return true;
 100     }
 101
 102     private void makeSingleNameAndTaxon(RedListGefaesspflanzenImportState state, ResultSet rs, Set<TaxonNameBase> namesToSave, Set<TaxonBase> taxaToSave)
 103             throws SQLException {
 104         long id = rs.getLong("NAMNR");
 105         String taxNameString = rs.getString("TAXNAME");
 106         String gueltString = rs.getString("GUELT");
 107         String rangString = rs.getString("RANG");
 108         String ep1String = rs.getString("EPI1");
 109         String ep2String = rs.getString("EPI2");
 110         String ep3String = rs.getString("EPI3");
 111         String nomZusatzString = rs.getString("NOM_ZUSATZ");
 112         String zusatzString = rs.getString("ZUSATZ");
 113         String authorKombString = rs.getString("AUTOR_KOMB");
 114         String authorBasiString = rs.getString("AUTOR_BASI");
 115
 116         if(CdmUtils.isBlank(taxNameString) && CdmUtils.isBlank(ep1String)){
 117             logger.error("NAMNR: "+id+" No name found!");
 118         }
 119
 120         Rank rank = makeRank(state, rangString);
 121         if(rank==null){
 122             logger.error("NAMNR: "+id+" Rank could not be resolved.");
 123         }
 124         BotanicalName name = BotanicalName.NewInstance(rank);
 125
 126         //ep1 should always be present
 127         if(CdmUtils.isBlank(ep1String)){
 128             logger.error("NAMNR: "+id+" EPI1 is empty!");
 129         }
 130         name.setGenusOrUninomial(ep1String);
 131         if(!CdmUtils.isBlank(ep2String)){
 132             name.setSpecificEpithet(ep2String);
 133         }
 134         if(!CdmUtils.isBlank(ep3String)){
 135             if(rank==Rank.SUBSPECIES() ||
 136                     rank==Rank.VARIETY()){
 137                 name.setInfraSpecificEpithet(ep3String);
 138             }
 139         }
 140
 141         //--- AUTHORS ---
 142         //combination author
 143         if(authorKombString.contains(EX)){
 144             //TODO: what happens with multiple ex authors??
 145             String[] kombSplit = authorKombString.split(EX);
 146             if(kombSplit.length!=2){
 147                 logger.error("NAMNR: "+id+" Multiple ex combination authors found");
 148             }
 149             for (int i = 0; i < kombSplit.length; i++) {
 150                 if(i==0){
 151                     //first author is ex author
 152                     TeamOrPersonBase authorKomb = (TeamOrPersonBase) state.getRelatedObject(Namespace.AUTHOR_NAMESPACE, kombSplit[i]);
 153                     name.setExCombinationAuthorship(authorKomb);
 154                 }
 155                 else{
 156                     TeamOrPersonBase authorKomb = (TeamOrPersonBase) state.getRelatedObject(Namespace.AUTHOR_NAMESPACE, kombSplit[i]);
 157                     name.setCombinationAuthorship(authorKomb);
 158                 }
 159             }
 160         }
 161         else if(authorKombString.trim().equals("auct.")){
 162             logger.warn("NAMNR: "+id+" AUCT information in AUTOR_KOMB column");
 163         }
 164         else if(!CdmUtils.isBlank(authorKombString)){
 165             TeamOrPersonBase authorKomb = (TeamOrPersonBase) state.getRelatedObject(Namespace.AUTHOR_NAMESPACE, authorKombString);
 166             name.setCombinationAuthorship(authorKomb);
 167         }
 168         //basionym author
 169         if(authorBasiString.contains(EX)){
 170             String[] basiSplit = authorBasiString.split(EX);
 171             for (int i = 0; i < basiSplit.length; i++) {
 172                 if(basiSplit.length!=2){
 173                     logger.error("NAMNR: "+id+" Multiple ex basionymn authors found");
 174                 }
 175                 if(i==0){
 176                     TeamOrPersonBase authorBasi= (TeamOrPersonBase) state.getRelatedObject(Namespace.AUTHOR_NAMESPACE, basiSplit[i]);
 177                     name.setExBasionymAuthorship(authorBasi);
 178                 }
 179                 else{
 180                     TeamOrPersonBase authorBasi= (TeamOrPersonBase) state.getRelatedObject(Namespace.AUTHOR_NAMESPACE, basiSplit[i]);
 181                     name.setBasionymAuthorship(authorBasi);
 182                 }
 183             }
 184         }
 185         else if(authorBasiString.trim().equals("auct.")){
 186             name.setAppendedPhrase(authorBasiString);
 187         }
 188         else if(!CdmUtils.isBlank(authorBasiString)){
 189             //this seems to be a convention in the source database: When there is only a single author then only the "AUTOR_BASI" column is used
 190             TeamOrPersonBase authorBasi= (TeamOrPersonBase) state.getRelatedObject(Namespace.AUTHOR_NAMESPACE, authorBasiString);
 191             if(CdmUtils.isBlank(authorKombString)){
 192                 name.setCombinationAuthorship(authorBasi);
 193             }
 194             else{
 195                 name.setBasionymAuthorship(authorBasi);
 196             }
 197         }
 198
 199         //check authorship consistency
 200         String authorString = rs.getString("AUTOR");
 201         String authorshipCache = name.getAuthorshipCache();
 202
 203         if(!CdmUtils.isBlank(zusatzString)){
 204             authorString = authorString.replace(", "+zusatzString, "");
 205         }
 206 //        if(CdmUtils.isBlank(authorKombString) && !CdmUtils.isBlank(authorBasiString)){
 207 //            authorString = "("+authorString+")";
 208 //        }
 209         if(!authorString.equals(authorshipCache)){
 210             logger.warn("NAMNR: "+id+" Authorship inconsistent! name.authorhshipCache <-> Column AUTOR: "+authorshipCache+" <-> "+authorString);
 211         }
 212
 213         //id
 214         ImportHelper.setOriginalSource(name, state.getTransactionalSourceReference(), id, Namespace.NAME_NAMESPACE);
 215         state.getNameMap().put(id, name.getUuid());
 216
 217         namesToSave.add(name);
 218
 219         //---TAXON---
 220         TaxonBase taxonBase = null;
 221         if(gueltString.equals("1")){
 222             taxonBase = Taxon.NewInstance(name, null);
 223         }
 224         else if(gueltString.equals("x") || gueltString.equals("b")){
 225             taxonBase = Synonym.NewInstance(name, null);
 226         }
 227         if(taxonBase==null){
 228             logger.error("NAMNR: "+id+" Taxon for name "+name+" could not be created.");
 229             return;
 230         }
 231
 232         taxaToSave.add(taxonBase);
 233
 234         //id
 235         ImportHelper.setOriginalSource(taxonBase, state.getTransactionalSourceReference(), id, Namespace.TAXON_NAMESPACE);
 236         state.getTaxonMap().put(id, taxonBase.getUuid());
 237     }
 238
 239     private Rank makeRank(RedListGefaesspflanzenImportState state, String rankStr) {
 240         Rank rank = null;
 241         try {
 242             rank = state.getTransformer().getRankByKey(rankStr);
 243         } catch (UndefinedTransformerMethodException e) {
 244             e.printStackTrace();
 245         }
 246         if(rank==null){
 247             logger.error(rankStr+" could not be associated to a known rank.");
 248         }
 249         return rank;
 250     }
 251
 252
 253
 254     @Override
 255     public Map<Object, Map<String, ? extends CdmBase>> getRelatedObjectsForPartition(ResultSet rs,
 256             RedListGefaesspflanzenImportState state) {
 257         Map<Object, Map<String, ? extends CdmBase>> result = new HashMap<>();
 258         Map<String, AgentBase<?>> authorMap = new HashMap<String, AgentBase<?>>();
 259
 260         try {
 261             while (rs.next()){
 262                 String authorKombString = rs.getString("AUTOR_KOMB");
 263
 264                 if(authorKombString.contains(EX)){
 265                     String[] kombSplit = authorKombString.split(EX);
 266                     for (int i = 0; i < kombSplit.length; i++) {
 267                         if(!authorMap.containsKey(kombSplit[i])){
 268                             authorMap.put(kombSplit[i], getAgentService().load(state.getAuthorMap().get(kombSplit[i])));
 269                         }
 270                     }
 271                 }
 272                 else if(!CdmUtils.isBlank(authorKombString) && !authorMap.containsKey(authorKombString)){
 273                     authorMap.put(authorKombString, getAgentService().load(state.getAuthorMap().get(authorKombString)));
 274                 }
 275
 276                 String authorBasiString = rs.getString("AUTOR_BASI");
 277                 //basionym author
 278                 if(authorBasiString.contains(EX)){
 279                     String[] basiSplit = authorBasiString.split(EX);
 280                     for (int i = 0; i < basiSplit.length; i++) {
 281                         if(!authorMap.containsKey(basiSplit[i])){
 282                             authorMap.put(basiSplit[i], getAgentService().load(state.getAuthorMap().get(basiSplit[i])));
 283                         }
 284                     }
 285                 }
 286                 else if(!CdmUtils.isBlank(authorBasiString) && !authorMap.containsKey(authorBasiString)){
 287                     authorMap.put(authorBasiString, getAgentService().load(state.getAuthorMap().get(authorBasiString)));
 288                 }
 289             }
 290         } catch (SQLException e) {
 291             e.printStackTrace();
 292         }
 293         result.put(Namespace.AUTHOR_NAMESPACE, authorMap);
 294
 295         return result;
 296     }
 297
 298     @Override
 299     protected boolean doCheck(RedListGefaesspflanzenImportState state) {
 300         return false;
 301     }
 302
 303     @Override
 304     protected boolean isIgnore(RedListGefaesspflanzenImportState state) {
 305         return false;
 306     }
 307
 308 }