1 package org.mdarad.framework.util.xml; 2 3 import java.util.Collection ; 4 import java.util.HashMap ; 5 import java.util.Map ; 6 import java.util.Set ; 7 8 import org.dataisland.primitives.bean.Entity; 9 10 public class EntityMap { 11 private static final long serialVersionUID = -4144379532451194078L; 12 private Map entityMap = new HashMap ();; 13 14 public void putEntity(Entity entity, String xmlFacadeClassName, String marshallEntityMethodName) { 15 String key = entity.getClass().getName(); 16 if(entityMap.containsKey(key)) { 17 EntitySet entitySet = (EntitySet) entityMap.get(key); 18 entitySet.add(entity); 19 } else { 20 EntitySet entitySet = new EntitySet(xmlFacadeClassName, marshallEntityMethodName); 21 entitySet.add(entity); 22 entityMap.put(key, entitySet); 23 } 24 } 25 26 public boolean containsEntity(Entity entity) { 27 String key = entity.getClass().getName(); 28 if(entityMap.containsKey(key)) { 29 EntitySet entitySet = (EntitySet) entityMap.get(key); 30 if(entitySet.contains(entity)) { 31 return true; 32 } 33 } 34 return false; 35 } 36 37 public EntitySet getEntitySet(String key) { 38 return (EntitySet) entityMap.get(key); 39 } 40 41 public int size() { 42 return entityMap.size(); 43 } 44 45 public void clear() { 46 entityMap.clear(); 47 } 48 49 public boolean isEmpty() { 50 return entityMap.isEmpty(); 51 } 52 53 public Collection values() { 54 return entityMap.values(); 55 } 56 57 public Set keySet() { 58 return entityMap.keySet(); 59 } 60 } 61 | Popular Tags |