Add join_field() Fix function. (#100)

metafacture · Jan 11, 2022 · abca2ef · abca2ef
1 parent db5ce4a
commit abca2ef
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 0 deletions.
diff --git a/metafix/src/main/java/org/metafacture/metafix/FixMethod.java b/metafix/src/main/java/org/metafacture/metafix/FixMethod.java
@@ -293,6 +293,14 @@ public void apply(final Metafix metafix, final Record record, final List<String>
             record.transformFields(params, s -> String.valueOf(s.indexOf(search))); // TODO: multiple
         }
     },
+    join_field {
+        public void apply(final Metafix metafix, final Record record, final List<String> params, final Map<String, String> options) {
+            record.transformField(params.get(0), v -> {
+                final String joinChar = params.size() > 1 ? params.get(1) : "";
+                return v.isArray() ? new Value(v.asArray().stream().map(Value::toString).collect(Collectors.joining(joinChar))) : null;
+            });
+        }
+    },
     lookup {
         public void apply(final Metafix metafix, final Record record, final List<String> params, final Map<String, String> options) {
             final Map<String, String> map;

diff --git a/metafix/src/test/java/org/metafacture/metafix/MetafixMethodTest.java b/metafix/src/test/java/org/metafacture/metafix/MetafixMethodTest.java
@@ -625,6 +625,27 @@ public void shouldGetIndexOfSubstring() {
         );
     }
 
+    @Test
+    public void shouldJoinArrayField() {
+        MetafixTestHelpers.assertFix(streamReceiver, Arrays.asList(
+                "join_field(numbers, '/')"
+            ),
+            i -> {
+                i.startRecord("1");
+                i.literal("numbers", "6");
+                i.literal("numbers", "42");
+                i.literal("numbers", "41");
+                i.literal("numbers", "6");
+                i.endRecord();
+            },
+            o -> {
+                o.get().startRecord("1");
+                o.get().literal("numbers", "6/42/41/6");
+                o.get().endRecord();
+            }
+        );
+    }
+
     @Test
     public void shouldPrependValue() {
         MetafixTestHelpers.assertFix(streamReceiver, Arrays.asList(