fauna europaea import fix

[cdmlib.git] / cdmlib-io / src / main / java / eu / etaxonomy / cdm / io / common / ImportStateBase.java

diff --git a/cdmlib-io/src/main/java/eu/etaxonomy/cdm/io/common/ImportStateBase.java b/cdmlib-io/src/main/java/eu/etaxonomy/cdm/io/common/ImportStateBase.java
index aaf6a2a5697582029721aac06482bffd30def5af..555c0419fb36b21e24e80c1b11480d8a0d7eab06 100644 (file)
--- a/cdmlib-io/src/main/java/eu/etaxonomy/cdm/io/common/ImportStateBase.java
+++ b/cdmlib-io/src/main/java/eu/etaxonomy/cdm/io/common/ImportStateBase.java
@@ -12,6 +12,7 @@ package eu.etaxonomy.cdm.io.common;
 \r
 import java.util.HashMap;\r
 import java.util.Map;\r
+import java.util.UUID;\r
 \r
 import org.apache.log4j.Logger;\r
 \r
@@ -37,6 +38,9 @@ public abstract class ImportStateBase<CONFIG extends ImportConfiguratorBase> ext
        \r
        private Map<ReferenceBase,TaxonomicTree> treeMap = new HashMap<ReferenceBase,TaxonomicTree>();\r
 \r
+       private Map<ReferenceBase,UUID> treeUuidMap = new HashMap<ReferenceBase,UUID>();\r
+\r
+       \r
        protected ImportStateBase(CONFIG config){\r
                this.config = config;\r
                stores.put(ICdmIO.USER_STORE, new MapWrapper<User>(service));\r
@@ -93,6 +97,23 @@ public abstract class ImportStateBase<CONFIG extends ImportConfiguratorBase> ext
        }\r
        \r
        public int countTrees(){\r
-               return treeMap.size();\r
+               return treeUuidMap.size();\r
+       }\r
+       \r
+       /**\r
+        * @return the treeUuid\r
+        */\r
+       public UUID getTreeUuid(ReferenceBase ref) {\r
+               return treeUuidMap.get(ref);\r
+       }\r
+\r
+       public void putTreeUuid(ReferenceBase ref, TaxonomicTree tree) {\r
+               if (tree != null){\r
+                       this.treeUuidMap.put(ref, tree.getUuid());\r
+               }\r
+       }\r
+       \r
+       public int countTreeUuids(){\r
+               return treeUuidMap.size();\r
        }\r
 }\r