Even more atomic router configuration.

[xmlrouter] / impl / src / main / java / org / wamblee / xmlrouter / impl / XMLRouter.java

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 d74c93d09f20a99e09c57480952822c2a1ac4a4c..637440fe4dc0b6e0f1ae4d87fda3676817978302 100644 (file)
--- a/impl/src/main/java/org/wamblee/xmlrouter/impl/XMLRouter.java
+++ b/impl/src/main/java/org/wamblee/xmlrouter/impl/XMLRouter.java
@@ -56,30 +56,38 @@ public class XMLRouter implements Gateway, DestinationRegistry {
     private Clock clock;
     private AtomicLong nextEventId;
 
-    private ExtendedRouterConfig routerConfig;
-    private TransformationPaths transformations;
+    private XMLRouterConfiguration config;
 
     private Map<Id<Destination>, Destination> destinations;
 
-    public XMLRouter(Clock aClock, ExtendedRouterConfig aRouterConfig,
+    public XMLRouter(Clock aClock, XMLRouterConfiguration aConfig,
         EventListener aListener) {
         sequenceNumbers = new AtomicLong(1);
         listener = aListener;
         clock = aClock;
         nextEventId = new AtomicLong(clock.currentTimeMillis());
-        routerConfig = aRouterConfig;
-        transformations = new TransformationPaths();
+        config = aConfig;
         destinations = new LinkedHashMap<Id<Destination>, Destination>();
     }
 
     @Override
     public void publish(String aSource, DOMSource aEvent) {
+        config.startPublishEvent();
+        try {
+            publishImpl(aSource, aEvent);
+        } finally {
+            config.endPublishEvent();
+        }
+    }
+
+    private void publishImpl(String aSource, DOMSource aEvent) {
         long time = clock.currentTimeMillis();
 
-        if (routerConfig.isDirty()) {
-            transformations.replaceTransformations(routerConfig
-                .transformationConfig().map());
-            routerConfig.resetDirty();
+        // TODO dirty flag will become unnecessary in the future.
+        if (config.routerConfig().isDirty()) {
+            config.transformations().replaceTransformations(
+                config.routerConfig().transformationConfig().map());
+            config.routerConfig().resetDirty();
         }
 
         Id<DOMSource> id = new Id<DOMSource>(nextEventId.getAndIncrement());
@@ -124,7 +132,7 @@ public class XMLRouter implements Gateway, DestinationRegistry {
 
         boolean delivered = false;
         Set<String> possibleTargetTypes = new HashSet<String>();
-        possibleTargetTypes.addAll(transformations
+        possibleTargetTypes.addAll(config.transformations()
             .getPossibleTargetTypes(aInputType));
 
         // ask each destination what target types, if any they want to have.
@@ -137,7 +145,7 @@ public class XMLRouter implements Gateway, DestinationRegistry {
             if (!requested.isEmpty()) {
                 // Deliver to the destination.
                 for (String targetType : requested) {
-                    TransformationPath path = transformations.getPath(
+                    TransformationPath path = config.transformations().getPath(
                         aInputType, targetType);
                     List<Transformation> ts = path.getTransformations();
                     int i = 0;
@@ -188,7 +196,8 @@ public class XMLRouter implements Gateway, DestinationRegistry {
 
     private boolean isAllowedByFilters(String aType, DOMSource aEvent) {
         boolean allowed = true;
-        for (Filter filter : routerConfig.filterConfig().map().values()) {
+        for (Filter filter : config.routerConfig().filterConfig().map()
+            .values()) {
             if (!filter.isAllowed(aType, aEvent)) {
                 allowed = false;
             }
@@ -198,8 +207,8 @@ public class XMLRouter implements Gateway, DestinationRegistry {
 
     private List<String> determineDocumentTypes(DOMSource aEvent) {
         List<String> res = new ArrayList<String>();
-        for (DocumentType type : routerConfig.documentTypeConfig().map()
-            .values()) {
+        for (DocumentType type : config.routerConfig().documentTypeConfig()
+            .map().values()) {
             if (type.isInstance(aEvent)) {
                 res.add(type.getName());
             }