Correct publish flag import for BM import
[cdmlib-apps.git] / app-import / src / main / java / eu / etaxonomy / cdm / io / globis / GlobisImportState.java
index e92441cc349ad39fb7878bc040ae0912baed6318..5325ab51c145a9fe41ede4d0cfb2d9f8e0290b18 100644 (file)
 package eu.etaxonomy.cdm.io.globis;\r
 \r
 import java.util.HashMap;\r
+import java.util.HashSet;\r
 import java.util.Map;\r
+import java.util.Set;\r
 \r
 import org.apache.log4j.Logger;\r
 \r
 import eu.etaxonomy.cdm.io.common.DbImportStateBase;\r
+import eu.etaxonomy.cdm.model.agent.AgentBase;\r
+import eu.etaxonomy.cdm.model.agent.Person;\r
+import eu.etaxonomy.cdm.model.agent.Team;\r
 import eu.etaxonomy.cdm.model.common.DefinedTermBase;\r
 import eu.etaxonomy.cdm.model.common.MarkerType;\r
 import eu.etaxonomy.cdm.model.common.User;\r
@@ -30,15 +35,18 @@ public class GlobisImportState extends DbImportStateBase<GlobisImportConfigurato
        @SuppressWarnings("unused")\r
        private static final Logger logger = Logger.getLogger(GlobisImportState.class);\r
 \r
-       Map<String, DefinedTermBase> dbCdmDefTermMap = new HashMap<String, DefinedTermBase>();\r
+       private Map<String, DefinedTermBase> dbCdmDefTermMap = new HashMap<String, DefinedTermBase>();\r
        \r
-       Map<String, User> usernameMap = new HashMap<String, User>();\r
+       private Map<String, User> usernameMap = new HashMap<String, User>();\r
        \r
        private Map<Integer, Map<Integer,Rank>> rankMap;\r
+\r
+       private Map<String, Person> personMap = new HashMap<String, Person>();\r
+       \r
+       private Map<String, Team> teamMap = new HashMap<String, Team>();\r
+       \r
+       \r
        \r
-       /* (non-Javadoc)\r
-        * @see eu.etaxonomy.cdm.io.common.IoStateBase#initialize(eu.etaxonomy.cdm.io.common.IoConfiguratorBase)\r
-        */\r
        @Override\r
        public void initialize(GlobisImportConfigurator config) {\r
 //             super(config);\r
@@ -107,5 +115,30 @@ public class GlobisImportState extends DbImportStateBase<GlobisImportConfigurato
                return result;\r
        }\r
 \r
-    \r
+       public void putPerson(String str, Person person){\r
+               personMap.put(str, person);\r
+       }\r
+       \r
+\r
+       public Person getPerson(String str){\r
+               return personMap.get(str);\r
+       }\r
+\r
+\r
+       public void putTeam(String str, Team team){\r
+               teamMap.put(str, team);\r
+       }\r
+       \r
+\r
+       public Team getTeam(String str){\r
+               return teamMap.get(str);\r
+       }\r
+       \r
+       public Set<AgentBase> getAgents(){\r
+               Set<AgentBase> result = new HashSet<AgentBase>();\r
+               result.addAll(personMap.values());\r
+               result.addAll(teamMap.values());\r
+               return result;\r
+       }\r
+       \r
 }\r