GH-230 Cache processed types in properties to avoid infinite recursio…

…n when going through createEmbeddedTypeDescription (Fix #230)
javalin · Sep 4, 2024 · 7e1ec6d · 7e1ec6d
1 parent 3d15957
commit 7e1ec6d
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 12 deletions.
diff --git a/examples/javalin-gradle-kotlin/src/main/java/io/javalin/openapi/plugin/test/JavalinTest.java b/examples/javalin-gradle-kotlin/src/main/java/io/javalin/openapi/plugin/test/JavalinTest.java
@@ -396,9 +396,9 @@ public static String getStatic() {
         }
 
         // by default nullable fields are not required, but we can force it
-        @OpenApiRequired
-        public String getNullableIsRequired() {
-            return "required";
+        //@OpenApiRequired
+        public JsonSchemaEntity getNullableIsRequired() {
+            return null;
         }
 
     }

diff --git a/...c/main/kotlin/io/javalin/openapi/experimental/processor/generators/TypeSchemaGenerator.kt b/...c/main/kotlin/io/javalin/openapi/experimental/processor/generators/TypeSchemaGenerator.kt
@@ -51,6 +51,10 @@ data class ResultScheme(
 
 class TypeSchemaGenerator(val context: AnnotationProcessorContext) {
 
+    // The cache helps to avoid processing the same property multiple times & prevent infinite recursion
+    // ~ https://github.com/javalin/javalin-openapi/issues/230
+    private val processedProperties = mutableMapOf<Property, ResultScheme>()
+
     fun createTypeSchema(
         type: ClassDefinition,
         inlineRefs: Boolean = false,
@@ -102,15 +106,23 @@ class TypeSchemaGenerator(val context: AnnotationProcessorContext) {
                 val properties = type.findAllProperties(requireNonNulls)
 
                 properties.forEach { property ->
-                    val (propertySchema, refs) = createEmbeddedTypeDescription(
-                        type = property.type,
-                        inlineRefs = inlineRefs,
-                        requiresNonNulls = requireNonNulls,
-                        composition = property.composition,
-                        extra = property.extra
-                    )
-                    propertiesObject.add(property.name, propertySchema)
-                    references.addAll(refs)
+                    val result =
+                        when {
+                            processedProperties.contains(property) ->
+                                processedProperties[property]!!
+                            else ->
+                                createEmbeddedTypeDescription(
+                                    type = property.type,
+                                    inlineRefs = inlineRefs,
+                                    requiresNonNulls = requireNonNulls,
+                                    composition = property.composition,
+                                    extra = property.extra
+                                ).also {
+                                    processedProperties[property] = it
+                                }
+                        }
+                    propertiesObject.add(property.name, result.json)
+                    references.addAll(result.references)
                 }
 
                 if (properties.any { it.required }) {