Fix generic record construction for fields with fieldName annotation

zio · Jun 27, 2024 · 53f6093 · 53f6093
1 parent e3ede68
commit 53f6093
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/zio-schema-derivation/shared/src/main/scala-2/zio/schema/DeriveSchema.scala b/zio-schema-derivation/shared/src/main/scala-2/zio/schema/DeriveSchema.scala
@@ -330,9 +330,10 @@ object DeriveSchema {
               }
           }
           val fromMap = {
-            val casts = fieldTypes.map { termSymbol =>
+            val casts = fieldTypes.zip(fieldAnnotations).map { case (termSymbol, annotations) =>
+              val newName = getFieldName(annotations).getOrElse(termSymbol.name.toString.trim)
               q"""
-                 try m.apply(${termSymbol.name.toString.trim}).asInstanceOf[${termSymbol.typeSignature}]
+                 try m.apply(${newName}).asInstanceOf[${termSymbol.typeSignature}]
                  catch {
                    case _: ClassCastException => throw new RuntimeException("Field " + ${termSymbol.name.toString.trim} + " has invalid type")
                    case _: Throwable  => throw new RuntimeException("Field " + ${termSymbol.name.toString.trim} + " is missing")

diff --git a/zio-schema-json/shared/src/main/scala/zio/schema/codec/JsonCodec.scala b/zio-schema-json/shared/src/main/scala/zio/schema/codec/JsonCodec.scala
@@ -814,7 +814,7 @@ object JsonCodec {
           if (Lexer.firstField(trace, in)) {
             while ({
               val field = Lexer.string(trace, in).toString
-              structure.find(_.name == field) match {
+              structure.find(f => f.name == field || f.annotations.collectFirst{case fieldName(name) => name}.contains(field)) match {
                 case Some(Schema.Field(label, schema, _, _, _, _)) =>
                   val trace_ = JsonError.ObjectAccess(label) :: trace
                   Lexer.char(trace_, in, ':')

diff --git a/zio-schema-json/shared/src/test/scala-2/zio/schema/codec/JsonCodecSpec.scala b/zio-schema-json/shared/src/test/scala-2/zio/schema/codec/JsonCodecSpec.scala
@@ -1906,7 +1906,7 @@ object JsonCodecSpec extends ZIOSpecDefault {
     f20: Option[String] = None,
     f21: Option[String] = None,
     f22: Option[String] = None,
-    f23: Option[String] = None
+    @fieldName("$f23") f23: Option[String] = None
   )
 
   object RecordExample {