Introduce LambdaCapturingTypeBuildItem

Author: gastaldi

core/deployment/src/main/java/io/quarkus/deployment/builditem/nativeimage/LambdaCapturingTypeBuildItem.java

@@ -0,0 +1,23 @@
+package io.quarkus.deployment.builditem.nativeimage;
+
+import io.quarkus.builder.item.MultiBuildItem;
+
+/**
+ * Used to register a lambda capturing type in native mode
+ */
+public final class LambdaCapturingTypeBuildItem extends MultiBuildItem {
+
+    private final String className;
+
+    public LambdaCapturingTypeBuildItem(Class<?> lambdaCapturingType) {
+        this.className = lambdaCapturingType.getName();
+    }
+
+    public LambdaCapturingTypeBuildItem(String className) {
+        this.className = className;
+    }
+
+    public String getClassName() {
+        return className;
+    }
+}

core/deployment/src/main/java/io/quarkus/deployment/steps/NativeImageAutoFeatureStep.java

@@ -28,6 +28,7 @@
 import io.quarkus.deployment.builditem.GeneratedResourceBuildItem;
 import io.quarkus.deployment.builditem.nativeimage.ForceNonWeakReflectiveClassBuildItem;
 import io.quarkus.deployment.builditem.nativeimage.JniRuntimeAccessBuildItem;
+import io.quarkus.deployment.builditem.nativeimage.LambdaCapturingTypeBuildItem;
 import io.quarkus.deployment.builditem.nativeimage.NativeImageProxyDefinitionBuildItem;
 import io.quarkus.deployment.builditem.nativeimage.NativeImageResourceBuildItem;
 import io.quarkus.deployment.builditem.nativeimage.NativeImageResourceBundleBuildItem;
@@ -72,6 +73,17 @@ public class NativeImageAutoFeatureStep {
     private static final MethodDescriptor RERUN_INITIALIZATION = ofMethod(
             "org.graalvm.nativeimage.impl.RuntimeClassInitializationSupport",
             "rerunInitialization", void.class, Class.class, String.class);
+
+    private static final MethodDescriptor CONFIGURATION_ALWAYS_TRUE = ofMethod(
+            "org.graalvm.nativeimage.impl.ConfigurationCondition",
+            "alwaysTrue", "org.graalvm.nativeimage.impl.ConfigurationCondition");
+
+    private static final MethodDescriptor REGISTER_LAMBDA_CAPTURING_CLASS = ofMethod(
+            "org.graalvm.nativeimage.impl.RuntimeSerializationSupport",
+            "registerLambdaCapturingClass", void.class,
+            "org.graalvm.nativeimage.impl.ConfigurationCondition",
+            String.class);
+
     private static final MethodDescriptor LOOKUP_METHOD = ofMethod(
             "com.oracle.svm.util.ReflectionUtil",
             "lookupMethod", Method.class, Class.class, String.class, Class[].class);
@@ -120,7 +132,8 @@ void generateFeature(BuildProducer<GeneratedNativeImageClassBuildItem> nativeIma
             List<ForceNonWeakReflectiveClassBuildItem> nonWeakReflectiveClassBuildItems,
             List<ServiceProviderBuildItem> serviceProviderBuildItems,
             List<UnsafeAccessedFieldBuildItem> unsafeAccessedFields,
-            List<JniRuntimeAccessBuildItem> jniRuntimeAccessibleClasses) {
+            List<JniRuntimeAccessBuildItem> jniRuntimeAccessibleClasses,
+            List<LambdaCapturingTypeBuildItem> lambdaCapturingTypeBuildItems) {
         ClassCreator file = new ClassCreator(new ClassOutput() {
             @Override
             public void write(String s, byte[] bytes) {
@@ -299,6 +312,18 @@ public void write(String s, byte[] bytes) {
                     overallCatch.load(i.serviceDescriptorFile()));
         }
 
+        // Register Lambda Capturing Types
+        if (!lambdaCapturingTypeBuildItems.isEmpty()) {
+            ResultHandle runtimeSerializationSupportSingleton = overallCatch.invokeStaticMethod(IMAGE_SINGLETONS_LOOKUP,
+                    overallCatch.loadClassFromTCCL("org.graalvm.nativeimage.impl.RuntimeSerializationSupport"));
+            ResultHandle configAlwaysTrue = overallCatch.invokeStaticMethod(CONFIGURATION_ALWAYS_TRUE);
+            for (LambdaCapturingTypeBuildItem i : lambdaCapturingTypeBuildItems) {
+                overallCatch.invokeInterfaceMethod(REGISTER_LAMBDA_CAPTURING_CLASS, runtimeSerializationSupportSingleton,
+                        configAlwaysTrue,
+                        overallCatch.load(i.getClassName()));
+            }
+        }
+
         if (!resourceBundles.isEmpty()) {
             AssignableResultHandle registerMethod = overallCatch.createVariable(Method.class);
             AssignableResultHandle locClass = overallCatch.createVariable(Class.class);