fix internal reader

apache · Jan 13, 2025 · 2f07b2d · 2f07b2d
1 parent 125fa6a
commit 2f07b2d
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/core/src/main/java/org/apache/iceberg/avro/InternalReader.java b/core/src/main/java/org/apache/iceberg/avro/InternalReader.java
@@ -107,11 +107,15 @@ public ValueReader<?> record(
         return ValueReaders.skipStruct(fieldResults);
       }
 
-      Types.StructType expected = partner.second().asStructType();
-      List<Pair<Integer, ValueReader<?>>> readPlan =
-          ValueReaders.buildReadPlan(expected, record, fieldResults, idToConstant);
+      if (partner.second().isVariantType()) {
+        return ValueReaders.record(fieldResults, record);
+      } else {
+        Types.StructType expected = partner.second().asStructType();
+        List<Pair<Integer, ValueReader<?>>> readPlan =
+            ValueReaders.buildReadPlan(expected, record, fieldResults, idToConstant);
 
-      return structReader(readPlan, partner.first(), expected);
+        return structReader(readPlan, partner.first(), expected);
+      }
     }
 
     private ValueReader<?> structReader(