Merge pull request #6 from ktown4u/feature/printWithInclusions

Feature/ 원하는 필드만 포함하는 printWithInclusions method 추가

Author: JinuCheon

lib/src/main/java/com/ktown4u/utils/YamlPrinter.java

@@ -12,6 +12,8 @@
 import com.fasterxml.jackson.databind.ser.impl.SimpleFilterProvider;
 import com.fasterxml.jackson.dataformat.yaml.YAMLFactory;
 
+import java.util.Arrays;
+
 public enum YamlPrinter {
     ;
     private static final ObjectMapper mapper;
@@ -38,6 +40,19 @@ public static String printWithExclusions(final Object object, final String... fi
         return writeValueAsString(writer, object);
     }
 
+    public static String printWithInclusions(final Object object, final String... fieldPathToInclude) {
+        final SimpleBeanPropertyFilter filter = SimpleBeanPropertyFilter.filterOutAllExcept(splitAndFlatten(fieldPathToInclude));
+        final FilterProvider filterProvider = new SimpleFilterProvider().addFilter("PropertyFilter", filter);
+        final ObjectWriter writer = mapper.writer(filterProvider);
+        return writeValueAsString(writer, object);
+    }
+
+    private static String[] splitAndFlatten(final String[] fieldNamesToInclude) {
+        return Arrays.stream(fieldNamesToInclude)
+                .flatMap(s -> Arrays.stream(s.split("\\.")))
+                .toArray(String[]::new);
+    }
+
     private static String writeValueAsString(final ObjectWriter writer, final Object object) {
         try {
             return writer.writeValueAsString(object);

lib/src/test/java/com/ktown4u/utils/YamlPrinterTest.java

@@ -48,4 +48,13 @@ void printWithExclusions() {
 
         Approvals.verify(result);
     }
+
+    @Test
+    @DisplayName("printWithInclusions - 원하는 필드만 포함하고 YAML 포멧 문자열로 반환.")
+    void printWithInclusions() {
+        final String[] filedPathToInclude = {"lineItems.product.description", "lineItems.product.price"};
+        final String result = YamlPrinter.printWithInclusions(order, filedPathToInclude);
+
+        Approvals.verify(result);
+    }
 }