AVRO-2254: fix unresolved shema name

apache · Jul 18, 2023 · 8a98795 · 8a98795
1 parent 5e6cec1
commit 8a98795
Show file tree

Hide file tree

Showing 3 changed files with 61 additions and 4 deletions.
diff --git a/lang/java/compiler/src/main/java/org/apache/avro/compiler/idl/SchemaResolver.java b/lang/java/compiler/src/main/java/org/apache/avro/compiler/idl/SchemaResolver.java
@@ -23,6 +23,7 @@
 import java.util.IdentityHashMap;
 import java.util.List;
 import java.util.Map;
+import java.util.concurrent.atomic.AtomicInteger;
 import java.util.function.Function;
 
 import org.apache.avro.Protocol;
@@ -44,6 +45,8 @@ private SchemaResolver() {
 
   private static final String UR_SCHEMA_NS = "org.apache.avro.compiler";
 
+  private static final AtomicInteger COUNTER = new AtomicInteger();
+
   /**
    * Create a schema to represent a "unresolved" schema. (used to represent a
    * schema where the definition is not known at the time) This concept might be
@@ -53,8 +56,8 @@ private SchemaResolver() {
    * @return
    */
   static Schema unresolvedSchema(final String name) {
-    Schema schema = Schema.createRecord(UR_SCHEMA_NAME, "unresolved schema", UR_SCHEMA_NS, false,
-        Collections.EMPTY_LIST);
+    Schema schema = Schema.createRecord(UR_SCHEMA_NAME + '_' + COUNTER.getAndIncrement(), "unresolved schema",
+        UR_SCHEMA_NS, false, Collections.EMPTY_LIST);
     schema.addProp(UR_SCHEMA_ATTR, name);
     return schema;
   }
@@ -66,8 +69,8 @@ static Schema unresolvedSchema(final String name) {
    * @return
    */
   static boolean isUnresolvedSchema(final Schema schema) {
-    return (schema.getType() == Schema.Type.RECORD && schema.getProp(UR_SCHEMA_ATTR) != null
-        && UR_SCHEMA_NAME.equals(schema.getName()) && UR_SCHEMA_NS.equals(schema.getNamespace()));
+    return (schema.getType() == Schema.Type.RECORD && schema.getProp(UR_SCHEMA_ATTR) != null && schema.getName() != null
+        && schema.getName().startsWith(UR_SCHEMA_NAME) && UR_SCHEMA_NS.equals(schema.getNamespace()));
   }
 
   /**

diff --git a/lang/java/compiler/src/test/idl/input/union.avdl b/lang/java/compiler/src/test/idl/input/union.avdl
@@ -0,0 +1,16 @@
+@namespace("org.apache.avro.gen")
+protocol UnionFwd {
+
+	record TestRecord {
+		union {SR1, SR2} unionField;
+	}
+
+	record SR1 {
+		string field;
+	}
+
+	record SR2 {
+		string field;
+	}
+
+}
diff --git a/lang/java/compiler/src/test/idl/output/union.avpr b/lang/java/compiler/src/test/idl/output/union.avpr
@@ -0,0 +1,38 @@
+{
+  "protocol": "UnionFwd",
+  "namespace": "org.apache.avro.gen",
+  "types": [
+    {
+      "type": "record",
+      "name": "TestRecord",
+      "fields": [
+        {
+          "name": "unionField",
+          "type": [
+            {
+              "type": "record",
+              "name": "SR1",
+              "fields": [
+                {
+                  "name": "field",
+                  "type": "string"
+                }
+              ]
+            },
+            {
+              "type": "record",
+              "name": "SR2",
+              "fields": [
+                {
+                  "name": "field",
+                  "type": "string"
+                }
+              ]
+            }
+          ]
+        }
+      ]
+    }
+  ],
+  "messages": {}
+}