cdm-pesi/src/main/java/eu/etaxonomy/cdm/io/pesi/erms/ErmsLinkImport.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.pesi.erms;
  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.io.common.IOValidator;
  23 import eu.etaxonomy.cdm.io.common.mapping.DbImportExtensionCreationMapper;
  24 import eu.etaxonomy.cdm.io.common.mapping.DbImportMapping;
  25 import eu.etaxonomy.cdm.io.common.mapping.DbNotYetImplementedMapper;
  26 import eu.etaxonomy.cdm.io.pesi.erms.validation.ErmsLinkImportValidator;
  27 import eu.etaxonomy.cdm.model.common.CdmBase;
  28 import eu.etaxonomy.cdm.model.common.Extension;
  29 import eu.etaxonomy.cdm.model.common.ExtensionType;
  30 import eu.etaxonomy.cdm.model.taxon.TaxonBase;
  31
  32
  33 /**
  34  * @author a.mueller
  35  * @created 20.02.2010
  36  * @version 1.0
  37  */
  38 @Component
  39 public class ErmsLinkImport  extends ErmsImportBase<TaxonBase> {
  40         private static final Logger logger = Logger.getLogger(ErmsLinkImport.class);
  41
  42         private DbImportMapping mapping;
  43
  44
  45         private int modCount = 10000;
  46         private static final String pluralString = "links";
  47         private static final String dbTableName = "links";
  48         private static final Class cdmTargetClass = Extension.class;
  49
  50         public ErmsLinkImport(){
  51                 super(pluralString, dbTableName, cdmTargetClass);
  52         }
  53
  54
  55         /* (non-Javadoc)
  56          * @see eu.etaxonomy.cdm.io.berlinModel.in.BerlinModelImportBase#getRecordQuery(eu.etaxonomy.cdm.io.berlinModel.in.BerlinModelImportConfigurator)
  57          */
  58         @Override
  59         protected String getRecordQuery(ErmsImportConfigurator config) {
  60                 String strRecordQuery =
  61                         " SELECT * " +
  62                         " FROM links " +
  63                         " WHERE ( links.id IN (" + ID_LIST_TOKEN + ") )";
  64                 return strRecordQuery;
  65         }
  66
  67         /* (non-Javadoc)
  68          * @see eu.etaxonomy.cdm.io.erms.ErmsImportBase#getMapping()
  69          */
  70         protected DbImportMapping getMapping() {
  71                 if (mapping == null){
  72                         mapping = new DbImportMapping();
  73                         ExtensionType extensionType = ExtensionType.URL();
  74                         //TODO do we need to add to TaxonNameBase too?
  75                         mapping.addMapper(DbImportExtensionCreationMapper.NewInstance("tu_id", ErmsTaxonImport.TAXON_NAMESPACE, "link_url", "id", extensionType));
  76                         //not yet implemented
  77                         mapping.addMapper(DbNotYetImplementedMapper.NewInstance("link_text"));  //maybe implement as a second extension ?? but this is ambigous!
  78                         mapping.addMapper(DbNotYetImplementedMapper.NewInstance("link_fn"));
  79                         mapping.addMapper(DbNotYetImplementedMapper.NewInstance("note"));
  80                 }
  81                 return mapping;
  82         }
  83
  84
  85         /* (non-Javadoc)
  86          * @see eu.etaxonomy.cdm.io.berlinModel.in.IPartitionedIO#getRelatedObjectsForPartition(java.sql.ResultSet)
  87          */
  88         public Map<Object, Map<String, ? extends CdmBase>> getRelatedObjectsForPartition(ResultSet rs) {
  89                 String nameSpace;
  90                 Class cdmClass;
  91                 Set<String> idSet;
  92                 Map<Object, Map<String, ? extends CdmBase>> result = new HashMap<Object, Map<String, ? extends CdmBase>>();
  93
  94                 try{
  95                         Set<String> taxonIdSet = new HashSet<String>();
  96                         Set<String> languageIdSet = new HashSet<String>();
  97                         while (rs.next()){
  98                                 handleForeignKey(rs, taxonIdSet, "tu_id");
  99                         }
 100
 101                         //taxon map
 102                         nameSpace = TAXON_NAMESPACE;
 103                         cdmClass = TaxonBase.class;
 104                         idSet = taxonIdSet;
 105                         Map<String, TaxonBase> taxonMap = (Map<String, TaxonBase>)getCommonService().getSourcedObjectsByIdInSource(cdmClass, idSet, nameSpace);
 106                         result.put(nameSpace, taxonMap);
 107
 108                 } catch (SQLException e) {
 109                         throw new RuntimeException(e);
 110                 }
 111                 return result;
 112         }
 113
 114         /* (non-Javadoc)
 115          * @see eu.etaxonomy.cdm.io.common.CdmIoBase#doCheck(eu.etaxonomy.cdm.io.common.IImportConfigurator)
 116          */
 117         @Override
 118         protected boolean doCheck(ErmsImportState state){
 119                 IOValidator<ErmsImportState> validator = new ErmsLinkImportValidator();
 120                 return validator.validate(state);
 121         }
 122
 123
 124         /* (non-Javadoc)
 125          * @see eu.etaxonomy.cdm.io.common.CdmIoBase#isIgnore(eu.etaxonomy.cdm.io.common.IImportConfigurator)
 126          */
 127         protected boolean isIgnore(ErmsImportState state){
 128                 return ! state.getConfig().isDoLinks();
 129         }
 130
 131
 132
 133
 134
 135 }