add classtypedef to annotation value (#183)

micronaut-projects · Nov 11, 2024 · 2a85128 · 2a85128
1 parent b499bdf
commit 2a85128
Show file tree

Hide file tree

Showing 2 changed files with 33 additions and 0 deletions.
diff --git a/sourcegen-generator-java/src/main/java/io/micronaut/sourcegen/JavaPoetSourceGenerator.java b/sourcegen-generator-java/src/main/java/io/micronaut/sourcegen/JavaPoetSourceGenerator.java
@@ -418,6 +418,8 @@ private void addAnnotationValue(AnnotationSpec.Builder builder, String memberNam
             builder.addMember(memberName, "$Lf", value);
         } else if (value instanceof Character) {
             builder.addMember(memberName, "'$L'", Util.characterLiteralWithoutSingleQuotes((char) value));
+        } else if (value instanceof ClassTypeDef typeDef) {
+            builder.addMember(memberName, "$L.class", typeDef.getSimpleName());
         } else {
             builder.addMember(memberName, "$L", value);
         }

diff --git a/...n-generator-java/src/test/java/io/micronaut/sourcegen/javapoet/write/RecordWriteTest.java b/...n-generator-java/src/test/java/io/micronaut/sourcegen/javapoet/write/RecordWriteTest.java
@@ -142,4 +142,35 @@ private String writeRecord(RecordDef recordDef) throws IOException {
         return matcher.group(1);
     }
 
+
+    @Test
+    public void annotationClassValue() throws IOException {
+        RecordDef recordDef = RecordDef.builder("test.$TestRecord")
+            .addAnnotation(AnnotationDef.builder(ClassTypeDef.of("jackson.annotation.JsonSubTypes.Type"))
+                .addMember("value", ClassTypeDef.STRING)
+                .addMember("name", "string")
+                .build())
+            .build();
+        JavaPoetSourceGenerator generator = new JavaPoetSourceGenerator();
+        String result;
+        try (StringWriter writer = new StringWriter()) {
+            generator.write(recordDef, writer);
+            result = writer.toString();
+        }
+
+        var expected = """
+        package test;
+
+        import jackson.annotation.JsonSubTypes;
+
+        @JsonSubTypes.Type(
+            value = String.class,
+            name = "string"
+        )
+        record $TestRecord() {
+        }
+        """;
+        assertEquals(expected.strip(), result.strip());
+    }
+
 }