01: package org.mdarad.framework.util.xml;
02:
03: import java.util.Collection;
04: import java.util.HashMap;
05: import java.util.Map;
06: import java.util.Set;
07:
08: import org.dataisland.primitives.bean.Entity;
09:
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,
15: String marshallEntityMethodName) {
16: String key = entity.getClass().getName();
17: if (entityMap.containsKey(key)) {
18: EntitySet entitySet = (EntitySet) entityMap.get(key);
19: entitySet.add(entity);
20: } else {
21: EntitySet entitySet = new EntitySet(xmlFacadeClassName,
22: marshallEntityMethodName);
23: entitySet.add(entity);
24: entityMap.put(key, entitySet);
25: }
26: }
27:
28: public boolean containsEntity(Entity entity) {
29: String key = entity.getClass().getName();
30: if (entityMap.containsKey(key)) {
31: EntitySet entitySet = (EntitySet) entityMap.get(key);
32: if (entitySet.contains(entity)) {
33: return true;
34: }
35: }
36: return false;
37: }
38:
39: public EntitySet getEntitySet(String key) {
40: return (EntitySet) entityMap.get(key);
41: }
42:
43: public int size() {
44: return entityMap.size();
45: }
46:
47: public void clear() {
48: entityMap.clear();
49: }
50:
51: public boolean isEmpty() {
52: return entityMap.isEmpty();
53: }
54:
55: public Collection values() {
56: return entityMap.values();
57: }
58:
59: public Set keySet() {
60: return entityMap.keySet();
61: }
62: }
|