[schema builder] don't generate ref'ed arrays and maps (#545)

linkedin · Jan 26, 2024 · e2849e7 · e2849e7
1 parent 9487a8b
commit e2849e7
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 2 deletions.
diff --git a/...r/src/test/resources/test-projects/classpath-project/input/SchemaWithClasspathImport.avsc b/...r/src/test/resources/test-projects/classpath-project/input/SchemaWithClasspathImport.avsc
@@ -7,6 +7,23 @@
     {
       "name": "date",
       "type": "build.generated.SimpleRecord"
+    },
+    {
+      "name": "unionArrayReference",
+      "type" :[
+        "null",
+        {
+          "type": "array",
+          "items": "build.generated.SimpleRecord"
+        }
+      ]
+    },
+    {
+      "name": "mapReference",
+        "type": {
+            "type": "map",
+            "values": "build.generated.SimpleRecord"
+        }
     }
   ]
 }
diff --git a/avro-codegen/src/main/java/com/linkedin/avroutil1/codegen/SpecificRecordGeneratorUtil.java b/avro-codegen/src/main/java/com/linkedin/avroutil1/codegen/SpecificRecordGeneratorUtil.java
@@ -314,10 +314,16 @@ private static List<AvroNamedSchema> getNestedInternalSchemaListForRecord(AvroRe
           });
           break;
         case MAP:
-          schemaQueue.add(((AvroMapSchema) fieldSchema).getValueSchema());
+          AvroMapSchema mapSchema = (AvroMapSchema) fieldSchema;
+          if (mapSchema.getValueSchemaOrRef().getRef() == null) {
+            schemaQueue.add(mapSchema.getValueSchema());
+          }
           break;
         case ARRAY:
-          schemaQueue.add(((AvroArraySchema) fieldSchema).getValueSchema());
+          AvroArraySchema arraySchema = (AvroArraySchema) fieldSchema;
+          if (arraySchema.getValueSchemaOrRef().getRef() == null) {
+            schemaQueue.add(arraySchema.getValueSchema());
+          }
           break;
       }
     }