Optimized fastserde enum field deserialization (#500)

BUG=FI-13182 Closes #302 Optimize enum field deserialization #499
linkedin · Jul 14, 2023 · ebfd273 · ebfd273
1 parent accc9ed
commit ebfd273
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/fastserde/avro-fastserde/src/main/java/com/linkedin/avro/fastserde/SchemaAssistant.java b/fastserde/avro-fastserde/src/main/java/com/linkedin/avro/fastserde/SchemaAssistant.java
@@ -10,6 +10,7 @@
 import com.linkedin.avro.fastserde.primitive.PrimitiveFloatArrayList;
 import com.linkedin.avro.fastserde.primitive.PrimitiveIntArrayList;
 import com.linkedin.avro.fastserde.primitive.PrimitiveLongArrayList;
+import com.linkedin.avroutil1.Enums;
 import com.linkedin.avroutil1.compatibility.AvroCompatibilityHelper;
 import com.sun.codemodel.JBlock;
 import com.sun.codemodel.JClass;
@@ -463,7 +464,8 @@ public JExpression getEnumValueByIndex(Schema enumSchema, JExpression indexExpr,
             .arg(getSchemaExpr.invoke("getEnumSymbols").invoke("get").arg(indexExpr));
       }
     } else {
-      return codeModel.ref(AvroCompatibilityHelper.getSchemaFullName(enumSchema)).staticInvoke("values").component(indexExpr);
+      JClass enumClass = codeModel.ref(AvroCompatibilityHelper.getSchemaFullName(enumSchema));
+      return codeModel.ref(Enums.class).staticInvoke("getConstant").arg(enumClass.dotclass()).arg(indexExpr);
     }
   }