Mark JAX-RS application classes as un-removable

Fixes: #1914
quarkusio · Apr 8, 2019 · 011553f · 011553f
1 parent 1964a5d
commit 011553f
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/...sy/deployment/src/main/java/io/quarkus/resteasy/deployment/ResteasyScanningProcessor.java b/...sy/deployment/src/main/java/io/quarkus/resteasy/deployment/ResteasyScanningProcessor.java
@@ -26,6 +26,7 @@
 import java.util.List;
 import java.util.Set;
 import java.util.function.Function;
+import java.util.function.Predicate;
 
 import javax.inject.Singleton;
 import javax.servlet.DispatcherType;
@@ -54,6 +55,8 @@
 
 import io.quarkus.arc.deployment.BeanContainerBuildItem;
 import io.quarkus.arc.deployment.BeanDefiningAnnotationBuildItem;
+import io.quarkus.arc.deployment.UnremovableBeanBuildItem;
+import io.quarkus.arc.processor.BeanInfo;
 import io.quarkus.deployment.annotations.BuildProducer;
 import io.quarkus.deployment.annotations.BuildStep;
 import io.quarkus.deployment.annotations.Record;
@@ -204,6 +207,7 @@ SubstrateConfigBuildItem config() {
     @BuildStep
     public void build(
             BuildProducer<FeatureBuildItem> feature,
+            BuildProducer<UnremovableBeanBuildItem> unremovableBeanBuildItem,
             BuildProducer<ReflectiveClassBuildItem> reflectiveClass,
             BuildProducer<ReflectiveHierarchyBuildItem> reflectiveHierarchy,
             BuildProducer<SubstrateProxyDefinitionBuildItem> proxyDefinition,
@@ -340,6 +344,13 @@ public void build(
             servletContextParams.produce(new ServletInitParamBuildItem("resteasy.servlet.mapping.prefix", path));
             if (appClass != null) {
                 servletContextParams.produce(new ServletInitParamBuildItem(JAX_RS_APPLICATION_PARAMETER_NAME, appClass));
+                final String finalAppClass = appClass; // horrible java :(
+                unremovableBeanBuildItem.produce(new UnremovableBeanBuildItem(new Predicate<BeanInfo>() {
+                    @Override
+                    public boolean test(BeanInfo beanInfo) {
+                        return beanInfo.getBeanClass().toString().equals(finalAppClass);
+                    }
+                }));
             }
         } else {
             // no @Application class and no detected @Path resources, bail out