support generic records

Author: SentryMan

blackbox-test/src/main/java/org/example/customer/generics/MyGenericHolderRecord.java

@@ -0,0 +1,6 @@
+package org.example.customer.generics;
+
+import io.avaje.jsonb.Json;
+
+@Json
+public record MyGenericHolderRecord<T>(String title, String author, T document) {}

blackbox-test/src/test/java/org/example/customer/generics/MyGenericHolderRecordTest.java

@@ -0,0 +1,78 @@
+package org.example.customer.generics;
+
+import static org.assertj.core.api.Assertions.assertThat;
+
+import java.util.LinkedHashMap;
+
+import org.example.customer.Address;
+import org.junit.jupiter.api.Test;
+
+import io.avaje.jsonb.JsonType;
+import io.avaje.jsonb.Jsonb;
+import io.avaje.jsonb.Types;
+
+class MyGenericHolderRecordTest {
+
+  Jsonb jsonb = Jsonb.builder().build();
+
+  private static MyGenericHolderRecord<Address> createTestData() {
+    return new MyGenericHolderRecord<>("hello", "art", new Address(90L, "one"));
+  }
+
+  @SuppressWarnings({"rawtypes"})
+  @Test
+  void toJson() {
+    final var bean = createTestData();
+
+    final var type = jsonb.type(MyGenericHolderRecord.class);
+
+    final var asJson = type.toJson(bean);
+    assertThat(asJson)
+        .isEqualTo(
+            "{\"title\":\"hello\",\"author\":\"art\",\"document\":{\"id\":90,\"street\":\"one\"}}");
+    assertThat(jsonb.toJson(bean)).isEqualTo(asJson);
+
+    final var pageResult = type.fromJson(asJson);
+    final var document = pageResult.document();
+    // reading via Object means the list contains LinkedHashMap
+    assertThat(document).isInstanceOf(LinkedHashMap.class);
+    final var asMap = (LinkedHashMap) document;
+    assertThat(asMap.get("street")).isEqualTo("one");
+
+    final var view = type.view("author,document(id)");
+    final var partialJson2 = view.toJson(bean);
+    // not supporting partial on the generic object (output includes street)
+    assertThat(partialJson2)
+        .isEqualTo("{\"author\":\"art\",\"document\":{\"id\":90,\"street\":\"one\"}}");
+  }
+
+  @Test
+  void toJson_withGenericParam() {
+    final var bean = createTestData();
+
+    final var jsonb = Jsonb.builder().build();
+    final JsonType<MyGenericHolderRecord<Address>> type =
+        jsonb.type(Types.newParameterizedType(MyGenericHolderRecord.class, Address.class));
+
+    final var asJson = type.toJson(bean);
+    assertThat(asJson)
+        .isEqualTo(
+            "{\"title\":\"hello\",\"author\":\"art\",\"document\":{\"id\":90,\"street\":\"one\"}}");
+    assertThat(jsonb.toJson(bean)).isEqualTo(asJson);
+
+    final var genericResult = type.fromJson(asJson);
+    final var document = genericResult.document();
+
+    assertThat(document.getId()).isEqualTo(90L);
+    assertThat(document.getStreet()).isEqualTo("one");
+
+    final var partial = type.view("author,document(*)");
+    final var partialJson = partial.toJson(bean);
+    assertThat(partialJson)
+        .isEqualTo("{\"author\":\"art\",\"document\":{\"id\":90,\"street\":\"one\"}}");
+
+    final var partial2 = type.view("author,document(id)");
+    final var partialJson2 = partial2.toJson(bean);
+    assertThat(partialJson2).isEqualTo("{\"author\":\"art\",\"document\":{\"id\":90}}");
+  }
+}

jsonb-generator/src/main/java/io/avaje/jsonb/generator/BeanReader.java

@@ -1,14 +1,16 @@
 package io.avaje.jsonb.generator;
 
-import io.avaje.jsonb.Json;
-
-import javax.lang.model.element.Element;
-import javax.lang.model.element.TypeElement;
+import java.lang.reflect.InvocationTargetException;
 import java.util.HashSet;
 import java.util.List;
 import java.util.Set;
 import java.util.TreeSet;
 
+import javax.lang.model.element.Element;
+import javax.lang.model.element.TypeElement;
+
+import io.avaje.jsonb.Json;
+
 class BeanReader {
 
   private final TypeElement beanType;
@@ -25,6 +27,7 @@ class BeanReader {
   private final boolean nonAccessibleField;
   private FieldReader unmappedField;
   private boolean hasRaw;
+  private final boolean isRecord;
 
   BeanReader(TypeElement beanType, ProcessingContext context) {
     this.beanType = beanType;
@@ -40,6 +43,7 @@ class BeanReader {
     this.hasSubTypes = typeReader.hasSubTypes();
     this.allFields = typeReader.allFields();
     this.constructor = typeReader.constructor();
+    this.isRecord = isRecord(beanType);
   }
 
   public BeanReader(TypeElement beanType, TypeElement mixInElement, ProcessingContext context) {
@@ -55,6 +59,21 @@ public BeanReader(TypeElement beanType, TypeElement mixInElement, ProcessingCont
     this.hasSubTypes = typeReader.hasSubTypes();
     this.allFields = typeReader.allFields();
     this.constructor = typeReader.constructor();
+    this.isRecord = isRecord(beanType);
+  }
+
+  boolean isRecord(TypeElement beanType) {
+    try {
+      final List<? extends Element> recordComponents =
+          (List<? extends Element>)
+              TypeElement.class.getMethod("getRecordComponents").invoke(beanType);
+      return !recordComponents.isEmpty();
+    } catch (IllegalAccessException
+        | InvocationTargetException
+        | NoSuchMethodException
+        | SecurityException e) {
+      return false;
+    }
   }
 
   int genericTypeParamsCount() {
@@ -279,9 +298,14 @@ void writeFromJson(Append writer) {
       // default public constructor
       writer.append("    %s _$%s = new %s();", shortName, varName, shortName).eol();
     } else {
-      writer.append("    // variables to read json values into, constructor params don't need _set$ flags").eol();
-      for (FieldReader allField : allFields) {
-        if (allField.includeFromJson()) {
+      writer
+          .append(
+              "    // variables to read json values into, constructor params don't need _set$ flags")
+          .eol();
+      for (final FieldReader allField : allFields) {
+        if (isRecord) {
+          allField.writeFromJsonVariablesRecord(writer);
+        } else if (allField.includeFromJson()) {
           allField.writeFromJsonVariables(writer);
         }
       }

jsonb-generator/src/main/java/io/avaje/jsonb/generator/FieldReader.java

@@ -288,6 +288,13 @@ void writeFromJsonVariables(Append writer) {
     writer.eol();
   }
 
+  void writeFromJsonVariablesRecord(Append writer) {
+    final String type = genericTypeParameter ? "Object" : genericType.shortType();
+    writer.append("    %s _val$%s = %s;", pad(type), fieldName, defaultValue);
+
+    writer.eol();
+  }
+
   private String pad(String value) {
     final int pad = 10 - value.length();
     if (pad < 1) {