X-Git-Url: http://wamblee.org/gitweb/?a=blobdiff_plain;f=impl%2Fsrc%2Fmain%2Fjava%2Forg%2Fwamblee%2Fxmlrouter%2Fimpl%2FXMLRouter.java;fp=impl%2Fsrc%2Fmain%2Fjava%2Forg%2Fwamblee%2Fxmlrouter%2Fimpl%2FXMLRouter.java;h=e86593d274642c3a18500a899e4a9d4b78255103;hb=5b4ee8d862b98d127abb04c646017c184d005838;hp=6b460d1dea203495d7ef258afec33c75e86c71ab;hpb=20807d81708bd33b3b5a4616fadcf3ae91bf9508;p=xmlrouter

diff --git a/impl/src/main/java/org/wamblee/xmlrouter/impl/XMLRouter.java b/impl/src/main/java/org/wamblee/xmlrouter/impl/XMLRouter.java
index 6b460d1..e86593d 100644
--- a/impl/src/main/java/org/wamblee/xmlrouter/impl/XMLRouter.java
+++ b/impl/src/main/java/org/wamblee/xmlrouter/impl/XMLRouter.java
@@ -32,9 +32,9 @@ import javax.xml.transform.dom.DOMSource;
 import org.wamblee.general.Clock;
 import org.wamblee.xml.XMLDocument;
 import org.wamblee.xmlrouter.common.Id;
-import org.wamblee.xmlrouter.config.Config;
 import org.wamblee.xmlrouter.config.DocumentType;
 import org.wamblee.xmlrouter.config.Filter;
+import org.wamblee.xmlrouter.config.RouterConfig;
 import org.wamblee.xmlrouter.config.Transformation;
 import org.wamblee.xmlrouter.listener.EventInfo;
 import org.wamblee.xmlrouter.listener.EventListener;
@@ -44,7 +44,7 @@ import org.wamblee.xmlrouter.subscribe.DestinationRegistry;
 
 // TODO concurrency.
 
-public class XMLRouter implements Config, Gateway, DestinationRegistry {
+public class XMLRouter implements RouterConfig, Gateway, DestinationRegistry {
 
     private static final Logger LOGGER = Logger.getLogger(XMLRouter.class
         .getName());
@@ -53,26 +53,26 @@ public class XMLRouter implements Config, Gateway, DestinationRegistry {
     private Clock clock;
     private AtomicLong nextEventId;
     private AtomicLong sequenceNumbers;
-    private Map<Long, DocumentType> documentTypes;
+    private Map<Id<DocumentType>, DocumentType> documentTypes;
     private Transformations transformations;
-    private Map<Long, Filter> filters;
-    private Map<Long, Destination> destinations;
+    private Map<Id<Filter>, Filter> filters;
+    private Map<Id<Destination>, Destination> destinations;
 
     public XMLRouter(Clock aClock, EventListener aListener) {
         listener = aListener;
         clock = aClock;
         nextEventId = new AtomicLong(clock.currentTimeMillis());
         sequenceNumbers = new AtomicLong(1);
-        documentTypes = new LinkedHashMap<Long, DocumentType>();
+        documentTypes = new LinkedHashMap<Id<DocumentType>, DocumentType>();
         transformations = new Transformations();
-        filters = new LinkedHashMap<Long, Filter>();
-        destinations = new LinkedHashMap<Long, Destination>();
+        filters = new LinkedHashMap<Id<Filter>, Filter>();
+        destinations = new LinkedHashMap<Id<Destination>, Destination>();
     }
 
     @Override
     public Id<DocumentType> addDocumentType(DocumentType aType) {
         long seqno = sequenceNumbers.getAndIncrement();
-        documentTypes.put(seqno, aType);
+        documentTypes.put(new Id<DocumentType>(seqno), aType);
         return new Id<DocumentType>(seqno);
     }
 
@@ -82,8 +82,13 @@ public class XMLRouter implements Config, Gateway, DestinationRegistry {
     }
 
     @Override
-    public Collection<DocumentType> getDocumentTypes() {
-        return Collections.unmodifiableCollection(documentTypes.values());
+    public Collection<Id<DocumentType>> getDocumentTypes() {
+        return Collections.unmodifiableCollection(documentTypes.keySet());
+    }
+
+    @Override
+    public DocumentType getDocumentType(Id<DocumentType> aId) {
+        return documentTypes.get(aId);
     }
 
     @Override
@@ -97,14 +102,19 @@ public class XMLRouter implements Config, Gateway, DestinationRegistry {
     }
 
     @Override
-    public Collection<Transformation> getTransformations() {
+    public Collection<Id<Transformation>> getTransformations() {
         return transformations.getTransformations();
     }
 
+    @Override
+    public Transformation getTransformation(Id<Transformation> aId) {
+        return transformations.getTransformation(aId);
+    }
+
     @Override
     public Id<Filter> addFilter(Filter aFilter) {
         long seqno = sequenceNumbers.getAndIncrement();
-        filters.put(seqno, aFilter);
+        filters.put(new Id<Filter>(seqno), aFilter);
         return new Id<Filter>(seqno);
     }
 
@@ -114,8 +124,13 @@ public class XMLRouter implements Config, Gateway, DestinationRegistry {
     }
 
     @Override
-    public Collection<Filter> getFilters() {
-        return Collections.unmodifiableCollection(filters.values());
+    public Collection<Id<Filter>> getFilters() {
+        return Collections.unmodifiableCollection(filters.keySet());
+    }
+
+    @Override
+    public Filter getFilter(Id<Filter> aId) {
+        return filters.get(aId);
     }
 
     @Override
@@ -168,8 +183,9 @@ public class XMLRouter implements Config, Gateway, DestinationRegistry {
             .getPossibleTargetTypes(aInputType));
 
         // ask each destination what target types, if any they want to have.
-        for (Map.Entry<Long, Destination> entry : destinations.entrySet()) {
-            long destinationId = entry.getKey();
+        for (Map.Entry<Id<Destination>, Destination> entry : destinations
+            .entrySet()) {
+            Id<Destination> destinationId = entry.getKey();
             Destination destination = entry.getValue();
             Collection<String> requested = destination
                 .chooseFromTargetTypes(possibleTargetTypes);
@@ -200,7 +216,7 @@ public class XMLRouter implements Config, Gateway, DestinationRegistry {
                         // all transformations done and all filters still
                         // allow the event.
                         boolean result = destination.receive(transformed);
-                        listener.delivered(aInfo, ts, destinationId,
+                        listener.delivered(aInfo, ts, destinationId.getId(),
                             destination.getName(), result);
                         delivered = delivered || result;
 
@@ -280,13 +296,13 @@ public class XMLRouter implements Config, Gateway, DestinationRegistry {
         notNull("destination", aDestination);
         long seqno = sequenceNumbers.getAndIncrement();
         Id<Destination> id = new Id<Destination>(seqno);
-        destinations.put(seqno, new RobustDestination(id, aDestination));
+        destinations.put(id, new RobustDestination(id, aDestination));
         return id;
     }
 
     @Override
     public void unregisterDestination(Id<Destination> aId) {
-        destinations.remove(aId.getId());
+        destinations.remove(aId);
     }
 
     private void notNull(String aName, Object aValue) {