Add custom field value resolver

ballerina-platform · Nov 4, 2022 · 034e9b4 · 034e9b4
1 parent 94324f6
commit 034e9b4
Showing 1 changed file with 26 additions and 1 deletion.
diff --git a/...openapi-to-ballerina-generator/src/main/java/org/ballerinalang/openapi/CodeGenerator.java b/...openapi-to-ballerina-generator/src/main/java/org/ballerinalang/openapi/CodeGenerator.java
@@ -42,6 +42,7 @@
 import java.io.IOException;
 import java.io.PrintStream;
 import java.io.PrintWriter;
+import java.lang.reflect.AccessibleObject;
 import java.nio.file.Files;
 import java.nio.file.Path;
 import java.nio.file.Paths;
@@ -50,8 +51,10 @@
 import java.util.Iterator;
 import java.util.List;
 import java.util.Locale;
+import java.util.Set;
 import java.util.regex.Matcher;
 import java.util.regex.Pattern;
+import java.util.stream.Collectors;
 
 import static org.ballerinalang.openapi.model.GenSrcFile.GenFileType;
 import static org.ballerinalang.openapi.utils.GeneratorConstants.GenType.GEN_CLIENT;
@@ -414,8 +417,30 @@ private List<GenSrcFile> generateBallerinaService(BallerinaOpenApiType api) thro
      */
     private String getContent(BallerinaOpenApiType object, String templateDir, String templateName) throws IOException {
         Template template = compileTemplate(templateDir, templateName);
+        FieldValueResolver fieldValueResolver = new FieldValueResolver() {
+
+            @Override
+            protected Set<FieldWrapper> members(
+                    Class<?> clazz) {
+                Set members = super.members(clazz);
+                return (Set<FieldWrapper>) members.stream()
+                        .filter(fw -> isValidField((FieldWrapper) fw))
+                        .collect(Collectors.toSet());
+            }
+
+            boolean isValidField(
+                    FieldWrapper fw) {
+                if (fw instanceof AccessibleObject) {
+                    if (isUseSetAccessible(fw)) {
+                        return true;
+                    }
+                    return false;
+                }
+                return true;
+            }
+        };
         Context context = Context.newBuilder(object)
-                .resolver(MapValueResolver.INSTANCE, JavaBeanValueResolver.INSTANCE, FieldValueResolver.INSTANCE)
+                .resolver(MapValueResolver.INSTANCE, JavaBeanValueResolver.INSTANCE, fieldValueResolver)
                 .build();
         return template.apply(context);
     }