[WFLY-11186][WFLY-11187] Weld subsystem should declare a capability

tadamski · Apr 26, 2019 · ac53dff · ac53dff
1 parent b0c412a
commit ac53dff
Show file tree

Hide file tree

Showing 67 changed files with 719 additions and 169 deletions.
diff --git a/batch-jberet/pom.xml b/batch-jberet/pom.xml
@@ -79,6 +79,10 @@
             <groupId>${project.groupId}</groupId>
             <artifactId>wildfly-weld-spi</artifactId>
         </dependency>
+        <dependency>
+            <groupId>${project.groupId}</groupId>
+            <artifactId>wildfly-weld-common</artifactId>
+        </dependency>
         <dependency>
             <groupId>org.jboss.weld</groupId>
             <artifactId>weld-core-impl</artifactId>

diff --git a/...rc/main/java/org/wildfly/extension/batch/jberet/deployment/BatchEnvironmentProcessor.java b/...rc/main/java/org/wildfly/extension/batch/jberet/deployment/BatchEnvironmentProcessor.java
@@ -23,8 +23,7 @@
 package org.wildfly.extension.batch.jberet.deployment;
 
 import static org.jboss.as.server.deployment.Attachments.DEPLOYMENT_COMPLETE_SERVICES;
-
-import javax.enterprise.inject.spi.BeanManager;
+import static org.jboss.as.weld.Capabilities.WELD_CAPABILITY_NAME;
 
 import org.jberet.repository.JobRepository;
 import org.jberet.spi.ArtifactFactory;
@@ -34,14 +33,14 @@
 import org.jboss.as.controller.capability.CapabilityServiceSupport;
 import org.jboss.as.ee.structure.DeploymentType;
 import org.jboss.as.ee.structure.DeploymentTypeMarker;
-import org.jboss.as.ee.weld.WeldDeploymentMarker;
 import org.jboss.as.server.Services;
 import org.jboss.as.server.deployment.Attachments;
 import org.jboss.as.server.deployment.DeploymentPhaseContext;
 import org.jboss.as.server.deployment.DeploymentUnit;
 import org.jboss.as.server.deployment.DeploymentUnitProcessingException;
 import org.jboss.as.server.deployment.DeploymentUnitProcessor;
 import org.jboss.as.server.suspend.SuspendController;
+import org.jboss.as.weld.WeldCapability;
 import org.jboss.modules.Module;
 import org.jboss.msc.service.ServiceBuilder;
 import org.jboss.msc.service.ServiceName;
@@ -135,10 +134,12 @@ public void deploy(final DeploymentPhaseContext phaseContext) throws DeploymentU
             final ServiceBuilder<ArtifactFactory> artifactFactoryServiceBuilder = serviceTarget.addService(artifactFactoryServiceName, artifactFactoryService);
 
             // Register the bean manager if this is a CDI deployment
-            if (WeldDeploymentMarker.isPartOfWeldDeployment(deploymentUnit)) {
-                BatchLogger.LOGGER.tracef("Adding BeanManager service dependency for deployment %s", deploymentUnit.getName());
-                artifactFactoryServiceBuilder.addDependency(BatchServiceNames.beanManagerServiceName(deploymentUnit), BeanManager.class,
-                        artifactFactoryService.getBeanManagerInjector());
+            if (support.hasCapability(WELD_CAPABILITY_NAME)) {
+                final WeldCapability api = support.getOptionalCapabilityRuntimeAPI(WELD_CAPABILITY_NAME, WeldCapability.class).get();
+                if (api.isPartOfWeldDeployment(deploymentUnit)) {
+                    BatchLogger.LOGGER.tracef("Adding BeanManager service dependency for deployment %s", deploymentUnit.getName());
+                    api.addBeanManagerService(deploymentUnit, artifactFactoryServiceBuilder, artifactFactoryService.getBeanManagerInjector());
+                }
             }
             artifactFactoryServiceBuilder.install();
             serviceBuilder.addDependency(artifactFactoryServiceName, WildFlyArtifactFactory.class, service.getArtifactFactoryInjector());

diff --git a/bean-validation/pom.xml b/bean-validation/pom.xml
@@ -99,6 +99,10 @@
             <groupId>${project.groupId}</groupId>
             <artifactId>wildfly-naming</artifactId>
         </dependency>
+        <dependency>
+            <groupId>${project.groupId}</groupId>
+            <artifactId>wildfly-weld-common</artifactId>
+        </dependency>
         <dependency>
             <groupId>org.wildfly.security</groupId>
             <artifactId>wildfly-elytron</artifactId>

diff --git a/...alidation/src/main/java/org/jboss/as/ee/beanvalidation/BeanValidationFactoryDeployer.java b/...alidation/src/main/java/org/jboss/as/ee/beanvalidation/BeanValidationFactoryDeployer.java
@@ -21,14 +21,16 @@
  */
 package org.jboss.as.ee.beanvalidation;
 
+import static org.jboss.as.weld.Capabilities.WELD_CAPABILITY_NAME;
+
 import javax.validation.ValidatorFactory;
 
+import org.jboss.as.controller.capability.CapabilityServiceSupport;
 import org.jboss.as.ee.component.ComponentDescription;
 import org.jboss.as.ee.component.ComponentNamingMode;
 import org.jboss.as.ee.component.EEModuleDescription;
 import org.jboss.as.ee.structure.DeploymentType;
 import org.jboss.as.ee.structure.DeploymentTypeMarker;
-import org.jboss.as.ee.weld.WeldDeploymentMarker;
 import org.jboss.as.naming.ServiceBasedNamingStore;
 import org.jboss.as.naming.ValueManagedReferenceFactory;
 import org.jboss.as.naming.deployment.ContextNames;
@@ -38,6 +40,7 @@
 import org.jboss.as.server.deployment.DeploymentUnit;
 import org.jboss.as.server.deployment.DeploymentUnitProcessingException;
 import org.jboss.as.server.deployment.DeploymentUnitProcessor;
+import org.jboss.as.weld.WeldCapability;
 import org.jboss.modules.Module;
 import org.jboss.msc.service.ServiceName;
 import org.jboss.msc.service.ServiceTarget;
@@ -114,7 +117,14 @@ private void bindServices(LazyValidatorFactory factory, ServiceTarget serviceTar
     @Override
     public void undeploy(DeploymentUnit context) {
         ValidatorFactory validatorFactory = context.getAttachment(BeanValidationAttachments.VALIDATOR_FACTORY);
-        if ((validatorFactory != null) && (!WeldDeploymentMarker.isPartOfWeldDeployment(context))) {
+        final CapabilityServiceSupport support = context.getAttachment(Attachments.CAPABILITY_SERVICE_SUPPORT);
+
+        boolean partOfWeldDeployment = false;
+        if (support.hasCapability(WELD_CAPABILITY_NAME)) {
+            partOfWeldDeployment = support.getOptionalCapabilityRuntimeAPI(WELD_CAPABILITY_NAME, WeldCapability.class)
+                    .get().isPartOfWeldDeployment(context);
+        }
+        if (validatorFactory != null && !partOfWeldDeployment) {
             // If the ValidatorFactory is not CDI-enabled, close it here. Otherwise, it's
             // closed via CdiValidatorFactoryService before the Weld service is stopped.
             validatorFactory.close();

diff --git a/ee-security/pom.xml b/ee-security/pom.xml
@@ -47,11 +47,6 @@
             <artifactId>wildfly-ee</artifactId>
         </dependency>
 
-        <dependency>
-            <groupId>${project.groupId}</groupId>
-            <artifactId>wildfly-weld</artifactId>
-        </dependency>
-
         <dependency>
             <groupId>org.jboss.logging</groupId>
             <artifactId>jboss-logging</artifactId>

diff --git a/ee/src/main/java/org/jboss/as/ee/weld/WeldDeploymentMarker.java b/ee/src/main/java/org/jboss/as/ee/weld/WeldDeploymentMarker.java
@@ -29,7 +29,9 @@
  *
  * @author Stuart Douglas
  *
+ * @deprecated Use WeldCapability to get access to the functionality of this class.
  */
+@Deprecated
 public class WeldDeploymentMarker {
 
     private static final AttachmentKey<Boolean> MARKER = AttachmentKey.create(Boolean.class);

diff --git a/feature-pack/src/main/resources/modules/system/layers/base/org/jboss/as/ejb3/main/module.xml b/feature-pack/src/main/resources/modules/system/layers/base/org/jboss/as/ejb3/main/module.xml
@@ -59,7 +59,6 @@
         <module name="org.wildfly.extension.request-controller"/>
         <module name="org.wildfly.discovery"/>
         <module name="org.jboss.as.controller"/>
-        <module name="org.jboss.as.weld"/>
         <!-- So we can access its integration API -->
         <module name="org.jboss.as.connector"/>
         <!-- Need access to org.jboss.as.core.security.ServerSecurityManager -->

diff --git a/...ure-pack/src/main/resources/modules/system/layers/base/org/jboss/as/jaxrs/main/module.xml b/...ure-pack/src/main/resources/modules/system/layers/base/org/jboss/as/jaxrs/main/module.xml
@@ -45,6 +45,8 @@
         <module name="org.wildfly.security.elytron-private"/>
         <module name="org.jboss.as.server"/>
         <module name="org.jboss.as.web-common"/>
+        <!-- Only used if capability org.wildfly.weld is available -->
+        <module name="org.jboss.as.weld.common" optional="true"/>
         <module name="org.wildfly.extension.undertow"/>
         <module name="io.undertow.core"/>
         <module name="io.undertow.servlet"/>

diff --git a/feature-pack/src/main/resources/modules/system/layers/base/org/jboss/as/jpa/main/module.xml b/feature-pack/src/main/resources/modules/system/layers/base/org/jboss/as/jpa/main/module.xml
@@ -70,8 +70,7 @@
         <module name="org.jboss.metadata.common"/>
         <module name="org.jboss.vfs"/>
         <module name="org.jboss.weld.core"/>
-        <!-- Optional; guarded by WeldDeploymentMarker which will not be present
-             if the weld subsystem isn't.-->
+        <!-- Only used if capability org.wildfly.weld is available -->
         <module name="org.jboss.as.weld.common" optional="true"/>
         <module name="org.wildfly.clustering.infinispan.spi"/>
         <module name="org.wildfly.transaction.client"/>

diff --git a/feature-pack/src/main/resources/modules/system/layers/base/org/jboss/as/jsf/main/module.xml b/feature-pack/src/main/resources/modules/system/layers/base/org/jboss/as/jsf/main/module.xml
@@ -44,7 +44,8 @@
         <module name="org.wildfly.security.elytron-private"/>
         <module name="org.jboss.as.server"/>
         <module name="org.jboss.as.web-common"/>
-        <module name="org.jboss.as.weld.common"/>
+        <!-- Only used if capability org.wildfly.weld is available -->
+        <module name="org.jboss.as.weld.common" optional="true"/>
         <module name="org.jboss.jandex"/>
         <module name="org.jboss.logging"/>
         <module name="org.jboss.modules"/>

diff --git a/...ain/resources/modules/system/layers/base/org/jboss/as/weld/beanvalidation/main/module.xml b/...ain/resources/modules/system/layers/base/org/jboss/as/weld/beanvalidation/main/module.xml
@@ -40,6 +40,7 @@
         <module name="org.jboss.as.server"/>
         <module name="org.jboss.as.ee"/>
         <module name="org.wildfly.security.elytron-private"/>
+        <module name="org.jboss.as.controller" />
         <module name="org.jboss.modules"/>
         <module name="org.jboss.msc"/>
         <module name="org.wildfly.extension.bean-validation"/>

diff --git a/feature-pack/src/main/resources/modules/system/layers/base/org/jboss/as/weld/main/module.xml b/feature-pack/src/main/resources/modules/system/layers/base/org/jboss/as/weld/main/module.xml
@@ -28,6 +28,10 @@
         <property name="jboss.api" value="private"/>
     </properties>
 
+    <exports>
+        <exclude path="org/jboss/as/weld/_private"/>
+    </exports>
+
     <resources>
         <artifact name="${org.wildfly:wildfly-weld}"/>
     </resources>

diff --git a/...n/resources/modules/system/layers/base/org/wildfly/extension/batch/jberet/main/module.xml b/...n/resources/modules/system/layers/base/org/wildfly/extension/batch/jberet/main/module.xml
@@ -46,6 +46,8 @@
         <module name="org.jboss.as.naming"/>
         <module name="org.jboss.as.threads"/>
         <module name="org.jboss.as.server"/>
+        <!-- Only used if capability org.wildfly.weld is available -->
+        <module name="org.jboss.as.weld.common" optional="true"/>
         <module name="org.jboss.logging"/>
         <module name="org.jboss.modules"/>
         <module name="org.jboss.msc"/>

diff --git a/...esources/modules/system/layers/base/org/wildfly/extension/bean-validation/main/module.xml b/...esources/modules/system/layers/base/org/wildfly/extension/bean-validation/main/module.xml
@@ -45,6 +45,8 @@
         <module name="org.jboss.as.controller"/>
         <module name="org.jboss.as.ee"/>
         <module name="org.jboss.as.naming" />
+        <!-- Only used if capability org.wildfly.weld is available -->
+        <module name="org.jboss.as.weld.common" optional="true"/>
         <module name="org.wildfly.security.elytron-private"/>
         <module name="org.jboss.as.server" />
         <module name="org.jboss.modules"/>

diff --git a/...in/resources/modules/system/layers/base/org/wildfly/extension/ee-security/main/module.xml b/...in/resources/modules/system/layers/base/org/wildfly/extension/ee-security/main/module.xml
@@ -48,7 +48,6 @@
         <module name="org.glassfish.soteria" />
         <module name="javax.security.enterprise.api" />
         <module name="javax.enterprise.api" />
-        <module name="org.jboss.as.weld.common" />
         <module name="org.jboss.jandex"/>
     </dependencies>
 </module>
diff --git a/...urces/modules/system/layers/base/org/wildfly/extension/messaging-activemq/main/module.xml b/...urces/modules/system/layers/base/org/wildfly/extension/messaging-activemq/main/module.xml
@@ -57,7 +57,8 @@
         <module name="org.wildfly.security.elytron-private"/>
         <module name="org.wildfly.common"/>
         <module name="org.jboss.as.server"/>
-        <module name="org.jboss.as.weld"/>
+        <!-- Only used if capability org.wildfly.weld is available -->
+        <module name="org.jboss.as.weld.common" optional="true"/>
         <module name="org.jboss.jandex"/>
         <module name="org.jboss.metadata.common"/>
         <module name="org.jboss.modules"/>
@@ -74,9 +75,7 @@
         <module name="org.wildfly.naming-client" services="import" />
         <module name="org.jboss.threads"/>
         <module name="org.wildfly.clustering.spi"/>
-        <module name="org.jboss.weld.api" />
         <module name="org.jboss.weld.core"/>
-        <module name="org.jboss.weld.spi" />
         <!-- org.jboss.xnio is required only if http-acceptor are used -->
         <module name="org.jboss.xnio" optional="true" />
         <!-- org.jboss.xnio.netty.netty-xnio-transport is required only if http-acceptor are used -->

diff --git a/...les/system/layers/base/org/wildfly/extension/microprofile/health-smallrye/main/module.xml b/...les/system/layers/base/org/wildfly/extension/microprofile/health-smallrye/main/module.xml
@@ -43,8 +43,7 @@
         <module name="org.jboss.logging"/>
         <module name="org.jboss.vfs"/>
         <module name="org.eclipse.microprofile.health.api"/>
-        <module name="org.jboss.as.weld" />
-        <module name="org.jboss.as.ee" />
+        <module name="org.jboss.as.weld.common" />
         <module name="javax.enterprise.api" />
         <module name="javax.annotation.api" />
     </dependencies>

diff --git a/galleon-pack/src/main/resources/feature_groups/basic-profile.xml b/galleon-pack/src/main/resources/feature_groups/basic-profile.xml
@@ -104,12 +104,12 @@
     <feature spec="subsystem.jsf"/>
     <feature spec="subsystem.jaxrs"/>
     <feature spec="subsystem.jdr"/>
+    <feature spec="subsystem.weld"/>
     <feature spec="subsystem.microprofile-config-smallrye"/>
     <feature spec="subsystem.microprofile-opentracing-smallrye"/>
     <feature spec="subsystem.pojo"/>
     <feature spec="subsystem.sar"/>
     <feature-group name="mail"/>
     <feature-group name="webservices"/>
-    <feature spec="subsystem.weld"/>
     <feature-group name="security-manager"/>
 </feature-group-spec>
diff --git a/galleon-pack/src/main/resources/layers/standalone/observability/layer-spec.xml b/galleon-pack/src/main/resources/layers/standalone/observability/layer-spec.xml
@@ -3,6 +3,7 @@
     <dependencies>
         <layer name="web-server"/>
         <layer name="management"/>
+        <layer name="cdi"/>
     </dependencies>
     <feature-group name="health"/>
     <feature-group name="metrics"/>

diff --git a/jaxrs/pom.xml b/jaxrs/pom.xml
@@ -67,6 +67,11 @@
             <artifactId>wildfly-web-common</artifactId>
         </dependency>
 
+        <dependency>
+            <groupId>${project.groupId}</groupId>
+            <artifactId>wildfly-weld-common</artifactId>
+        </dependency>
+
         <dependency>
             <groupId>${project.groupId}</groupId>
             <artifactId>wildfly-undertow</artifactId>

diff --git a/jaxrs/src/main/java/org/jboss/as/jaxrs/JaxrsSubsystemAdd.java b/jaxrs/src/main/java/org/jboss/as/jaxrs/JaxrsSubsystemAdd.java
@@ -23,8 +23,11 @@
 package org.jboss.as.jaxrs;
 
 
+import static org.jboss.as.weld.Capabilities.WELD_CAPABILITY_NAME;
+
 import org.jboss.as.controller.AbstractBoottimeAddStepHandler;
 import org.jboss.as.controller.OperationContext;
+import org.jboss.as.controller.capability.CapabilityServiceSupport;
 import org.jboss.as.controller.registry.Resource;
 import org.jboss.as.jaxrs.deployment.JaxrsAnnotationProcessor;
 import org.jboss.as.jaxrs.deployment.JaxrsCdiIntegrationProcessor;
@@ -66,8 +69,13 @@ public void execute(DeploymentProcessorTarget processorTarget) {
                 processorTarget.addDeploymentProcessor(JaxrsExtension.SUBSYSTEM_NAME, Phase.DEPENDENCIES, Phase.DEPENDENCIES_JAXRS, new JaxrsDependencyProcessor());
                 processorTarget.addDeploymentProcessor(JaxrsExtension.SUBSYSTEM_NAME, Phase.POST_MODULE, Phase.POST_MODULE_JAXRS_SCANNING, new JaxrsScanningProcessor());
                 processorTarget.addDeploymentProcessor(JaxrsExtension.SUBSYSTEM_NAME, Phase.POST_MODULE, Phase.POST_MODULE_JAXRS_COMPONENT, new JaxrsComponentDeployer());
-                processorTarget.addDeploymentProcessor(JaxrsExtension.SUBSYSTEM_NAME, Phase.POST_MODULE, Phase.POST_MODULE_JAXRS_CDI_INTEGRATION, new JaxrsCdiIntegrationProcessor());
+
+                CapabilityServiceSupport capabilities = context.getCapabilityServiceSupport();
+                if (capabilities.hasCapability(WELD_CAPABILITY_NAME)) {
+                    processorTarget.addDeploymentProcessor(JaxrsExtension.SUBSYSTEM_NAME, Phase.POST_MODULE, Phase.POST_MODULE_JAXRS_CDI_INTEGRATION, new JaxrsCdiIntegrationProcessor());
+                }
                 processorTarget.addDeploymentProcessor(JaxrsExtension.SUBSYSTEM_NAME, Phase.POST_MODULE, Phase.POST_MODULE_JAXRS_METHOD_PARAMETER, new JaxrsMethodParameterProcessor());
+
                 processorTarget.addDeploymentProcessor(JaxrsExtension.SUBSYSTEM_NAME, Phase.INSTALL, Phase.INSTALL_JAXRS_DEPLOYMENT, new JaxrsIntegrationProcessor());
             }
         }, OperationContext.Stage.RUNTIME);

diff --git a/jaxrs/src/main/java/org/jboss/as/jaxrs/deployment/JaxrsCdiIntegrationProcessor.java b/jaxrs/src/main/java/org/jboss/as/jaxrs/deployment/JaxrsCdiIntegrationProcessor.java
@@ -22,23 +22,25 @@
 package org.jboss.as.jaxrs.deployment;
 
 import static org.jboss.as.jaxrs.logging.JaxrsLogger.JAXRS_LOGGER;
+import static org.jboss.as.weld.Capabilities.WELD_CAPABILITY_NAME;
 
+import java.util.ArrayList;
+import java.util.List;
+
+import org.jboss.as.controller.capability.CapabilityServiceSupport;
 import org.jboss.as.ee.structure.DeploymentType;
 import org.jboss.as.ee.structure.DeploymentTypeMarker;
-import org.jboss.as.ee.weld.WeldDeploymentMarker;
 import org.jboss.as.server.deployment.Attachments;
 import org.jboss.as.server.deployment.DeploymentPhaseContext;
 import org.jboss.as.server.deployment.DeploymentUnit;
 import org.jboss.as.server.deployment.DeploymentUnitProcessingException;
 import org.jboss.as.server.deployment.DeploymentUnitProcessor;
 import org.jboss.as.web.common.WarMetaData;
+import org.jboss.as.weld.WeldCapability;
 import org.jboss.metadata.javaee.spec.ParamValueMetaData;
 import org.jboss.metadata.web.jboss.JBossWebMetaData;
 import org.jboss.modules.Module;
 
-import java.util.ArrayList;
-import java.util.List;
-
 /**
  * @author Stuart Douglas
  */
@@ -63,11 +65,15 @@ public void deploy(final DeploymentPhaseContext phaseContext) throws DeploymentU
         final JBossWebMetaData webdata = warMetaData.getMergedJBossWebMetaData();
 
         try {
-            module.getClassLoader().loadClass(CDI_INJECTOR_FACTORY_CLASS);
-            // don't set this param if CDI is not in classpath
-            if (WeldDeploymentMarker.isWeldDeployment(deploymentUnit)) {
-                JAXRS_LOGGER.debug("Found CDI, adding injector factory class");
-                setContextParameter(webdata, "resteasy.injector.factory", CDI_INJECTOR_FACTORY_CLASS);
+            final CapabilityServiceSupport support = deploymentUnit.getAttachment(Attachments.CAPABILITY_SERVICE_SUPPORT);
+            if (support.hasCapability(WELD_CAPABILITY_NAME)) {
+                final WeldCapability api = support.getOptionalCapabilityRuntimeAPI(WELD_CAPABILITY_NAME, WeldCapability.class).get();
+                if (api.isWeldDeployment(deploymentUnit)) {
+                    // don't set this param if CDI is not in classpath
+                    module.getClassLoader().loadClass(CDI_INJECTOR_FACTORY_CLASS);
+                    JAXRS_LOGGER.debug("Found CDI, adding injector factory class");
+                    setContextParameter(webdata, "resteasy.injector.factory", CDI_INJECTOR_FACTORY_CLASS);
+                }
             }
         } catch (ClassNotFoundException ignored) {
         }

diff --git a/jaxrs/src/main/java/org/jboss/as/jaxrs/deployment/JaxrsComponentDeployer.java b/jaxrs/src/main/java/org/jboss/as/jaxrs/deployment/JaxrsComponentDeployer.java
@@ -23,19 +23,21 @@
 package org.jboss.as.jaxrs.deployment;
 
 import static org.jboss.as.jaxrs.logging.JaxrsLogger.JAXRS_LOGGER;
+import static org.jboss.as.weld.Capabilities.WELD_CAPABILITY_NAME;
 
 import java.util.Arrays;
 
+import org.jboss.as.controller.capability.CapabilityServiceSupport;
 import org.jboss.as.ee.component.Attachments;
 import org.jboss.as.ee.component.ComponentDescription;
 import org.jboss.as.ee.component.EEModuleDescription;
 import org.jboss.as.ee.component.ViewDescription;
 import org.jboss.as.ee.managedbean.component.ManagedBeanComponentDescription;
-import org.jboss.as.ee.weld.WeldDeploymentMarker;
 import org.jboss.as.server.deployment.DeploymentPhaseContext;
 import org.jboss.as.server.deployment.DeploymentUnit;
 import org.jboss.as.server.deployment.DeploymentUnitProcessingException;
 import org.jboss.as.server.deployment.DeploymentUnitProcessor;
+import org.jboss.as.weld.WeldCapability;
 import org.jboss.modules.Module;
 import org.jboss.resteasy.util.GetRestful;
 
@@ -81,6 +83,12 @@ public void deploy(DeploymentPhaseContext phaseContext) throws DeploymentUnitPro
 
         final ClassLoader loader = module.getClassLoader();
 
+        final CapabilityServiceSupport support = deploymentUnit.getAttachment(org.jboss.as.server.deployment.Attachments.CAPABILITY_SERVICE_SUPPORT);
+        boolean partOfWeldDeployment = false;
+        if (support.hasCapability(WELD_CAPABILITY_NAME)) {
+            partOfWeldDeployment = support.getOptionalCapabilityRuntimeAPI(WELD_CAPABILITY_NAME, WeldCapability.class).get()
+                    .isPartOfWeldDeployment(deploymentUnit);
+        }
         for (final ComponentDescription component : moduleDescription.getComponentDescriptions()) {
             Class<?> componentClass = null;
             try {
@@ -95,7 +103,7 @@ public void deploy(DeploymentPhaseContext phaseContext) throws DeploymentUnitPro
                     //using SFSB's as JAX-RS endpoints is not recommended, but if people really want to do it they can
 
                     JAXRS_LOGGER.debugf("Stateful session bean %s is being used as a JAX-RS endpoint, this is not recommended", component.getComponentName());
-                    if (WeldDeploymentMarker.isPartOfWeldDeployment(deploymentUnit)) {
+                    if (partOfWeldDeployment) {
                         //if possible just let CDI handle the integration
                         continue;
                     }