0001601: Add Map-based methods to TransformData so one can more easil…

…y lookup target transform data
JumpMind · Feb 24, 2014 · 5896f8c · 5896f8c
1 parent 834760f
commit 5896f8c
Showing 1 changed file with 28 additions and 14 deletions.
diff --git a/symmetric-io/src/main/java/org/jumpmind/symmetric/io/data/transform/TransformedData.java b/symmetric-io/src/main/java/org/jumpmind/symmetric/io/data/transform/TransformedData.java
@@ -119,19 +119,16 @@ public void put(TransformColumn column, String columnValue, boolean recordAsKey)
         columnValues.put(column.getTargetColumnName(), columnValue);
     }
 
-    protected List<String> retrieve(
-            Map<TransformColumn.IncludeOnType, LinkedHashMap<String, String>> source,
-            boolean getColumnNames) {
+
+    protected Map<String, String> retrieve(
+            Map<TransformColumn.IncludeOnType, LinkedHashMap<String, String>> source) {
 
-        List<String> list = new ArrayList<String>(source == null ? 0 : source.size());
+        Map<String, String> list = new LinkedHashMap<String, String>(source == null ? 0
+                : source.size());
         if (source != null) {
             LinkedHashMap<String, String> values = source.get(IncludeOnType.ALL);
             if (values != null) {
-                if (getColumnNames) {
-                    list.addAll(values.keySet());
-                } else {
-                    list.addAll(values.values());
-                }
+                list.putAll(values);
             }
 
             IncludeOnType type = IncludeOnType.DELETE;
@@ -143,16 +140,33 @@ protected List<String> retrieve(
 
             values = source.get(type);
             if (values != null) {
-                if (getColumnNames) {
-                    list.addAll(values.keySet());
-                } else {
-                    list.addAll(values.values());
-                }
+                list.putAll(values);
             }
         }
         return list;
     }
+
+    protected List<String> retrieve(
+            Map<TransformColumn.IncludeOnType, LinkedHashMap<String, String>> source,
+            boolean getColumnNames) {
+        Map<String, String> values = retrieve(source);
+        if (getColumnNames) {
+            return new ArrayList<String>(values.keySet());
+        } else {
+            return new ArrayList<String>(values.values());
+        }
+    }
+
+
+    public Map<String, String> getTargetKeyValues() {
+        return retrieve(keysBy);
+    }
+
+    public Map<String, String> getTargetValues() {
+        return retrieve(columnsBy);
+    }
 
+
     public String[] getKeyNames() {
 
         List<String> list = retrieve(keysBy, true);