Added the beginnings of an idea for a bsh data router

JumpMind · Aug 10, 2009 · dd7b88e · dd7b88e
1 parent 669a2d6
commit dd7b88e
Show file tree

Hide file tree

Showing 3 changed files with 57 additions and 0 deletions.
diff --git a/symmetric/pom.xml b/symmetric/pom.xml
@@ -558,6 +558,11 @@
             <artifactId>commons-math</artifactId>
             <version>1.1</version>
         </dependency>
+        <dependency>
+            <groupId>org.beanshell</groupId>
+            <artifactId>bsh</artifactId>
+            <version>2.0b4</version>
+        </dependency>         
         <!-- Should the jetty stuff be split to another project: symmetric-jetty? -->
         <dependency>
             <groupId>org.mortbay.jetty</groupId>

diff --git a/symmetric/src/main/java/org/jumpmind/symmetric/route/AbstractDataRouter.java b/symmetric/src/main/java/org/jumpmind/symmetric/route/AbstractDataRouter.java
@@ -24,13 +24,17 @@
 import java.util.Map;
 import java.util.Set;
 
+import org.apache.commons.logging.Log;
+import org.apache.commons.logging.LogFactory;
 import org.apache.ddlutils.model.Column;
 import org.jumpmind.symmetric.db.IDbDialect;
 import org.jumpmind.symmetric.model.DataMetaData;
 import org.jumpmind.symmetric.model.Node;
 
 public abstract class AbstractDataRouter implements IDataRouter {
 
+    protected final Log logger = LogFactory.getLog(getClass());
+
     private boolean autoRegister = true;
 
     public boolean isAutoRegister() {

diff --git a/symmetric/src/main/java/org/jumpmind/symmetric/route/BshDataRouter.java b/symmetric/src/main/java/org/jumpmind/symmetric/route/BshDataRouter.java
@@ -0,0 +1,48 @@
+package org.jumpmind.symmetric.route;
+
+import java.util.Collection;
+import java.util.Collections;
+import java.util.HashSet;
+import java.util.Set;
+
+import org.jumpmind.symmetric.model.DataMetaData;
+import org.jumpmind.symmetric.model.Node;
+
+import bsh.EvalError;
+import bsh.Interpreter;
+
+/**
+ * In Progress ...
+ * TODO javadoc and unit test
+ */
+public class BshDataRouter extends AbstractDataRouter {
+
+    public void completeBatch(IRouterContext context) {
+    }
+
+    public Collection<String> routeToNodes(IRouterContext context, DataMetaData dataMetaData, Set<Node> nodes,
+            boolean initialLoad) {
+        try {
+            Interpreter interpreter = new Interpreter();
+            interpreter.set("nodes", nodes);
+            // set old and new and cur column values
+            Object value = interpreter.eval(dataMetaData.getTrigger().getRouterExpression());
+            if (value instanceof Boolean && value.equals(Boolean.TRUE)) {
+                return toNodeIds(nodes);
+            } else if (value instanceof Collection<?>) {
+                Collection<?> values = (Collection<?>) value;
+                Set<String> nodeIds = new HashSet<String>(values.size());
+                for (Object v : values) {
+                    if (v != null) {
+                        nodeIds.add(v.toString());
+                    }
+                }
+                return nodeIds;
+            }
+        } catch (EvalError e) {
+            logger.error(e, e);
+        }
+        return Collections.emptySet();
+    }
+
+}