From e5e1cb5c9fe7fa4d9fdcc15ccd9eb8fccacf3052 Mon Sep 17 00:00:00 2001
From: Brian Stansberry <brian.stansberry@redhat.com>
Date: Mon, 29 Jun 2015 20:06:08 -0500
Subject: [PATCH] [WFCORE-781] Add ability to use capability-provided
 integrations from services and DUPs

---
 .../jboss/as/controller/OperationContext.java |  12 +++
 .../as/controller/OperationContextImpl.java   |  46 ++++++++
 .../ParallelBootOperationContext.java         |   6 ++
 .../jboss/as/controller/ReadOnlyContext.java  |   6 ++
 .../capability/CapabilityServiceSupport.java  | 100 ++++++++++++++++++
 .../management/AuthorizedAddressTest.java     |   6 ++
 .../operations/AbstractOperationTestCase.java |   6 ++
 .../as/server/deployment/Attachments.java     |   8 +-
 .../deployment/DeploymentHandlerUtil.java     |   6 +-
 .../deployment/RootDeploymentUnitService.java |  15 ++-
 .../deployment/SubDeploymentProcessor.java    |   5 +-
 .../deployment/SubDeploymentUnitService.java  |   6 +-
 12 files changed, 212 insertions(+), 10 deletions(-)
 create mode 100644 controller/src/main/java/org/jboss/as/controller/capability/CapabilityServiceSupport.java

diff --git a/controller/src/main/java/org/jboss/as/controller/OperationContext.java b/controller/src/main/java/org/jboss/as/controller/OperationContext.java
index 42252fa0893..5a545abcd3b 100644
--- a/controller/src/main/java/org/jboss/as/controller/OperationContext.java
+++ b/controller/src/main/java/org/jboss/as/controller/OperationContext.java
@@ -30,6 +30,7 @@
 import org.jboss.as.controller.access.Caller;
 import org.jboss.as.controller.access.Environment;
 import org.jboss.as.controller.access.ResourceAuthorization;
+import org.jboss.as.controller.capability.CapabilityServiceSupport;
 import org.jboss.as.controller.capability.RuntimeCapability;
 import org.jboss.as.controller.client.MessageSeverity;
 import org.jboss.as.controller.notification.Notification;
@@ -969,6 +970,17 @@ public interface OperationContext extends ExpressionResolver {
      */
     ServiceName getCapabilityServiceName(String capabilityBaseName, String dynamicPart, Class<?> serviceType);
 
+    /**
+     * Gets a support object that allows service implementations installed from this context to
+     * integrate with capabilities.
+     *
+     * @return the support object. Will not return {@code null}
+     *
+     * * @throws java.lang.IllegalStateException if {@link #getCurrentStage() the current stage} is {@link Stage#MODEL}.
+     *                                          Service integration is not supported in the model stage.
+     */
+    CapabilityServiceSupport getCapabilityServiceSupport();
+
     /**
      * Whether normally this operation would require a runtime step. It returns {@code true in the following cases}
      * <ul>
diff --git a/controller/src/main/java/org/jboss/as/controller/OperationContextImpl.java b/controller/src/main/java/org/jboss/as/controller/OperationContextImpl.java
index 0e9d1afeefe..e846a7c582c 100644
--- a/controller/src/main/java/org/jboss/as/controller/OperationContextImpl.java
+++ b/controller/src/main/java/org/jboss/as/controller/OperationContextImpl.java
@@ -77,6 +77,7 @@
 import org.jboss.as.controller.access.TargetAttribute;
 import org.jboss.as.controller.access.TargetResource;
 import org.jboss.as.controller.audit.AuditLogger;
+import org.jboss.as.controller.capability.CapabilityServiceSupport;
 import org.jboss.as.controller.capability.RuntimeCapability;
 import org.jboss.as.controller.capability.registry.CapabilityContext;
 import org.jboss.as.controller.capability.registry.CapabilityId;
@@ -1477,6 +1478,13 @@ private void rejectUserDomainServerUpdates() {
         }
     }
 
+    @Override
+    public CapabilityServiceSupport getCapabilityServiceSupport() {
+        assert isControllingThread();
+        assertCapabilitiesAvailable(currentStage);
+        return new CapabilityServiceSupportImpl(managementModel);
+    }
+
 
     private boolean isModelUpdateRejectionRequired() {
         if (requiresModelUpdateAuthorization == null) {
@@ -2579,4 +2587,42 @@ public boolean equals(Object o) {
             return true;
         }
     }
+
+    private static class CapabilityServiceSupportImpl implements CapabilityServiceSupport {
+        private final ManagementModel managementModel;
+
+        private CapabilityServiceSupportImpl(ManagementModel managementModel) {
+            this.managementModel = managementModel;
+        }
+
+        @Override
+        public <T> T getCapabilityRuntimeAPI(String capabilityName, Class<T> apiType) throws NoSuchCapabilityException {
+            try {
+                return managementModel.getCapabilityRegistry().getCapabilityRuntimeAPI(capabilityName, CapabilityContext.GLOBAL, apiType);
+            } catch (IllegalStateException e) {
+                if (managementModel.getCapabilityRegistry().hasCapability(capabilityName, CapabilityContext.GLOBAL)) {
+                    // Must have been some other problem
+                    throw e;
+                }
+            }
+            throw new NoSuchCapabilityException(capabilityName);
+        }
+
+        @Override
+        public <T> T getCapabilityRuntimeAPI(String capabilityBaseName, String dynamicPart, Class<T> apiType) throws NoSuchCapabilityException {
+            String fullName = RuntimeCapability.buildDynamicCapabilityName(capabilityBaseName, dynamicPart);
+            return getCapabilityRuntimeAPI(fullName, apiType);
+        }
+
+        @Override
+        public ServiceName getCapabilityServiceName(String capabilityName) {
+            return ServiceName.parse(capabilityName);
+        }
+
+        @Override
+        public ServiceName getCapabilityServiceName(String capabilityBaseName, String dynamicPart) {
+            return getCapabilityServiceName(capabilityBaseName).append(dynamicPart);
+        }
+    }
+
 }
diff --git a/controller/src/main/java/org/jboss/as/controller/ParallelBootOperationContext.java b/controller/src/main/java/org/jboss/as/controller/ParallelBootOperationContext.java
index fc7853dbe75..f393ca3107a 100644
--- a/controller/src/main/java/org/jboss/as/controller/ParallelBootOperationContext.java
+++ b/controller/src/main/java/org/jboss/as/controller/ParallelBootOperationContext.java
@@ -30,6 +30,7 @@
 import org.jboss.as.controller.access.AuthorizationResult;
 import org.jboss.as.controller.access.ResourceAuthorization;
 import org.jboss.as.controller.audit.AuditLogger;
+import org.jboss.as.controller.capability.CapabilityServiceSupport;
 import org.jboss.as.controller.capability.RuntimeCapability;
 import org.jboss.as.controller.client.MessageSeverity;
 import org.jboss.as.controller.logging.ControllerLogger;
@@ -399,6 +400,11 @@ public ServiceName getCapabilityServiceName(String capabilityBaseName, String dy
                 serviceType, activeStep);
     }
 
+    @Override
+    public CapabilityServiceSupport getCapabilityServiceSupport() {
+        return primaryContext.getCapabilityServiceSupport();
+    }
+
     @Override
     void releaseStepLocks(Step step) {
         if(lockStep == step) {
diff --git a/controller/src/main/java/org/jboss/as/controller/ReadOnlyContext.java b/controller/src/main/java/org/jboss/as/controller/ReadOnlyContext.java
index c1cdc66f3aa..ae5d8448d4e 100644
--- a/controller/src/main/java/org/jboss/as/controller/ReadOnlyContext.java
+++ b/controller/src/main/java/org/jboss/as/controller/ReadOnlyContext.java
@@ -30,6 +30,7 @@
 import org.jboss.as.controller.access.Action;
 import org.jboss.as.controller.access.AuthorizationResult;
 import org.jboss.as.controller.access.ResourceAuthorization;
+import org.jboss.as.controller.capability.CapabilityServiceSupport;
 import org.jboss.as.controller.capability.RuntimeCapability;
 import org.jboss.as.controller.client.MessageSeverity;
 import org.jboss.as.controller.logging.ControllerLogger;
@@ -422,4 +423,9 @@ public ServiceName getCapabilityServiceName(String capabilityName, Class<?> type
     public ServiceName getCapabilityServiceName(String capabilityBaseName, String dynamicPart, Class<?> serviceType) {
         throw readOnlyContext();
     }
+
+    @Override
+    public CapabilityServiceSupport getCapabilityServiceSupport() {
+        throw readOnlyContext();
+    }
 }
diff --git a/controller/src/main/java/org/jboss/as/controller/capability/CapabilityServiceSupport.java b/controller/src/main/java/org/jboss/as/controller/capability/CapabilityServiceSupport.java
new file mode 100644
index 00000000000..465edd1a7ba
--- /dev/null
+++ b/controller/src/main/java/org/jboss/as/controller/capability/CapabilityServiceSupport.java
@@ -0,0 +1,100 @@
+/*
+ * JBoss, Home of Professional Open Source.
+ * Copyright 2015, Red Hat, Inc., and individual contributors
+ * as indicated by the @author tags. See the copyright.txt file in the
+ * distribution for a full listing of individual contributors.
+ *
+ * This is free software; you can redistribute it and/or modify it
+ * under the terms of the GNU Lesser General Public License as
+ * published by the Free Software Foundation; either version 2.1 of
+ * the License, or (at your option) any later version.
+ *
+ * This software is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this software; if not, write to the Free
+ * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
+ * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
+ */
+
+package org.jboss.as.controller.capability;
+
+import org.jboss.msc.service.ServiceName;
+
+/**
+ * Provides support for capability integration outside the management layer,
+ * in service implementations.
+ * <p>
+ * Note that use of this interface in no way creates a requirement on the
+ * referenced capability by the caller.
+ *
+ * @author Brian Stansberry
+ */
+public interface CapabilityServiceSupport {
+
+    /**
+     * Exception thrown when support for an unregistered capability is requested. This is a checked
+     * exception because {@code CapabilityServiceSupport} is used outside the management layer and
+     * the requirements for capability availability available in {@link org.jboss.as.controller.OperationContext}
+     * are not possible. So callers need to be aware of and handle non-existent capabilities.
+     */
+    class NoSuchCapabilityException extends Exception {
+
+        static final long serialVersionUID = 1L;
+
+        public NoSuchCapabilityException(String message) {
+            super(message);
+        }
+    }
+
+    /**
+     * Gets the runtime API associated with a given capability, if there is one.
+     * @param capabilityName the name of the capability. Cannot be {@code null}
+     * @param apiType class of the java type that exposes the API. Cannot be {@code null}
+     * @param <T> the java type that exposes the API
+     * @return the runtime API. Will not return {@code null}
+     *
+     * @throws NoSuchCapabilityException if no matching capability is registered
+     * @throws java.lang.IllegalArgumentException if the capability does not provide a runtime API
+     * @throws java.lang.ClassCastException if the runtime API exposed by the capability cannot be cast to type {code T}
+     */
+    <T> T getCapabilityRuntimeAPI(String capabilityName, Class<T> apiType) throws NoSuchCapabilityException;
+
+    /**
+     * Gets the runtime API associated with a given {@link RuntimeCapability#isDynamicallyNamed() dynamically named}
+     * capability, if there is one.
+     *
+     * @param capabilityBaseName the base name of the capability. Cannot be {@code null}
+     * @param dynamicPart the dynamic part of the capability name. Cannot be {@code null}
+     * @param apiType class of the java type that exposes the API. Cannot be {@code null}
+     * @param <T> the java type that exposes the API
+     * @return the runtime API. Will not return {@code null}
+     *
+     * @throws NoSuchCapabilityException if no matching capability is registered
+     * @throws java.lang.IllegalArgumentException if the capability does not provide a runtime API
+     * @throws java.lang.ClassCastException if the runtime API exposed by the capability cannot be cast to type {code T}
+     */
+    <T> T getCapabilityRuntimeAPI(String capabilityBaseName, String dynamicPart, Class<T> apiType) throws NoSuchCapabilityException;
+
+    /**
+     * Gets the name of a service associated with a given capability. This method does not confirm that the
+     * capability is currently registered.
+     *
+     * @param capabilityName the name of the capability. Cannot be {@code null}
+     * @return the name of the service. Will not return {@code null}
+     */
+    ServiceName getCapabilityServiceName(String capabilityName);
+
+    /**
+     * Gets the name of a service associated with a given {@link RuntimeCapability#isDynamicallyNamed() dynamically named}
+     * capability. This method does not confirm that the capability is currently registered.
+     *
+     * @param capabilityBaseName the base name of the capability. Cannot be {@code null}
+     * @param dynamicPart the dynamic part of the capability name. Cannot be {@code null}
+     * @return the name of the service. Will not return {@code null}
+     */
+    ServiceName getCapabilityServiceName(String capabilityBaseName, String dynamicPart);
+}
diff --git a/controller/src/test/java/org/jboss/as/controller/access/management/AuthorizedAddressTest.java b/controller/src/test/java/org/jboss/as/controller/access/management/AuthorizedAddressTest.java
index 4f12af9d6b2..8a5b0dff854 100644
--- a/controller/src/test/java/org/jboss/as/controller/access/management/AuthorizedAddressTest.java
+++ b/controller/src/test/java/org/jboss/as/controller/access/management/AuthorizedAddressTest.java
@@ -43,6 +43,7 @@
 import org.jboss.as.controller.access.Caller;
 import org.jboss.as.controller.access.Environment;
 import org.jboss.as.controller.access.ResourceAuthorization;
+import org.jboss.as.controller.capability.CapabilityServiceSupport;
 import org.jboss.as.controller.capability.RuntimeCapability;
 import org.jboss.as.controller.client.MessageSeverity;
 import org.jboss.as.controller.notification.Notification;
@@ -550,6 +551,11 @@ public ServiceName getCapabilityServiceName(String capabilityBaseName, String dy
             throw new UnsupportedOperationException("Not supported yet."); //To change body of generated methods, choose Tools | Templates.
         }
 
+        @Override
+        public CapabilityServiceSupport getCapabilityServiceSupport() {
+            throw new UnsupportedOperationException("Not supported yet."); //To change body of generated methods, choose Tools | Templates.
+        }
+
         @Override
         public void emit(Notification notification) {
             throw new UnsupportedOperationException("Not supported yet."); //To change body of generated methods, choose Tools | Templates.
diff --git a/host-controller/src/test/java/org/jboss/as/domain/controller/operations/AbstractOperationTestCase.java b/host-controller/src/test/java/org/jboss/as/domain/controller/operations/AbstractOperationTestCase.java
index 07b0de8529e..a05ee3d8f3d 100644
--- a/host-controller/src/test/java/org/jboss/as/domain/controller/operations/AbstractOperationTestCase.java
+++ b/host-controller/src/test/java/org/jboss/as/domain/controller/operations/AbstractOperationTestCase.java
@@ -76,6 +76,7 @@
 import org.jboss.as.controller.access.Environment;
 import org.jboss.as.controller.access.ResourceAuthorization;
 import org.jboss.as.controller.access.management.AccessConstraintDefinition;
+import org.jboss.as.controller.capability.CapabilityServiceSupport;
 import org.jboss.as.controller.capability.RuntimeCapability;
 import org.jboss.as.controller.client.MessageSeverity;
 import org.jboss.as.controller.client.OperationAttachments;
@@ -684,6 +685,11 @@ public ServiceName getCapabilityServiceName(String capabilitBaseyName, String dy
             return null;
         }
 
+        @Override
+        public CapabilityServiceSupport getCapabilityServiceSupport() {
+            return null;
+        }
+
         @Override
         public void emit(Notification notification) {
             // no-op
diff --git a/server/src/main/java/org/jboss/as/server/deployment/Attachments.java b/server/src/main/java/org/jboss/as/server/deployment/Attachments.java
index 5fe3e7e86fa..e5c5c461ab4 100644
--- a/server/src/main/java/org/jboss/as/server/deployment/Attachments.java
+++ b/server/src/main/java/org/jboss/as/server/deployment/Attachments.java
@@ -29,6 +29,7 @@
 import java.util.jar.Manifest;
 
 import org.jboss.as.controller.ServiceVerificationHandler;
+import org.jboss.as.controller.capability.CapabilityServiceSupport;
 import org.jboss.as.controller.services.path.PathManager;
 import org.jboss.as.server.deployment.annotation.CompositeIndex;
 import org.jboss.as.server.deployment.module.AdditionalModuleSpecification;
@@ -204,11 +205,14 @@ public final class Attachments {
     public static final AttachmentKey<Map<String, MountedDeploymentOverlay>> DEPLOYMENT_OVERLAY_LOCATIONS = AttachmentKey.create(Map.class);
 
     /**
-     * Support for getting and creating resource models on a deployments resource.
+     * Support for getting and creating resource models on a deployment's resource.
      */
     public static final AttachmentKey<DeploymentResourceSupport> DEPLOYMENT_RESOURCE_SUPPORT = AttachmentKey.create(DeploymentResourceSupport.class);
 
-
+    /**
+     * Support for integrating with services and other runtime API provided by managed capabilities.
+     */
+    public static final AttachmentKey<CapabilityServiceSupport> CAPABILITY_SERVICE_SUPPORT = AttachmentKey.create(CapabilityServiceSupport.class);
 
     //
     // VALIDATE
diff --git a/server/src/main/java/org/jboss/as/server/deployment/DeploymentHandlerUtil.java b/server/src/main/java/org/jboss/as/server/deployment/DeploymentHandlerUtil.java
index 7542701ea59..96c7073472e 100644
--- a/server/src/main/java/org/jboss/as/server/deployment/DeploymentHandlerUtil.java
+++ b/server/src/main/java/org/jboss/as/server/deployment/DeploymentHandlerUtil.java
@@ -133,7 +133,8 @@ public void handleResult(OperationContext.ResultAction resultAction, OperationCo
     }
 
     public static void doDeploy(final OperationContext context, final String deploymentUnitName, final String managementName,
-                                                            final Resource deploymentResource, final ImmutableManagementResourceRegistration registration, final ManagementResourceRegistration mutableRegistration, final AbstractVaultReader vaultReader, final ContentItem... contents) {
+                                final Resource deploymentResource, final ImmutableManagementResourceRegistration registration,
+                                final ManagementResourceRegistration mutableRegistration, final AbstractVaultReader vaultReader, final ContentItem... contents) {
         final ServiceName deploymentUnitServiceName = Services.deploymentUnitName(deploymentUnitName);
 
         final ServiceTarget serviceTarget = context.getServiceTarget();
@@ -149,7 +150,8 @@ public static void doDeploy(final OperationContext context, final String deploym
         }
         DeploymentOverlayIndex overlays = DeploymentOverlayIndex.createDeploymentOverlayIndex(context);
 
-        final RootDeploymentUnitService service = new RootDeploymentUnitService(deploymentUnitName, managementName, null, registration, mutableRegistration, deploymentResource, vaultReader, overlays);
+        final RootDeploymentUnitService service = new RootDeploymentUnitService(deploymentUnitName, managementName, null,
+                registration, mutableRegistration, deploymentResource, context.getCapabilityServiceSupport(), vaultReader, overlays);
         final ServiceController<DeploymentUnit> deploymentUnitController = serviceTarget.addService(deploymentUnitServiceName, service)
                 .addDependency(Services.JBOSS_DEPLOYMENT_CHAINS, DeployerChains.class, service.getDeployerChainsInjector())
                 .addDependency(DeploymentMountProvider.SERVICE_NAME, DeploymentMountProvider.class, service.getServerDeploymentRepositoryInjector())
diff --git a/server/src/main/java/org/jboss/as/server/deployment/RootDeploymentUnitService.java b/server/src/main/java/org/jboss/as/server/deployment/RootDeploymentUnitService.java
index caaab682e0f..2be11273928 100644
--- a/server/src/main/java/org/jboss/as/server/deployment/RootDeploymentUnitService.java
+++ b/server/src/main/java/org/jboss/as/server/deployment/RootDeploymentUnitService.java
@@ -23,6 +23,7 @@
 package org.jboss.as.server.deployment;
 
 import org.jboss.as.controller.ServiceVerificationHandler;
+import org.jboss.as.controller.capability.CapabilityServiceSupport;
 import org.jboss.as.controller.registry.ImmutableManagementResourceRegistration;
 import org.jboss.as.controller.registry.ManagementResourceRegistration;
 import org.jboss.as.controller.registry.Resource;
@@ -48,23 +49,27 @@ final class RootDeploymentUnitService extends AbstractDeploymentUnitService {
     private final DeploymentUnit parent;
     private final ImmutableManagementResourceRegistration registration;
     private final ManagementResourceRegistration mutableRegistration;
-    private Resource resource;
+    private final Resource resource;
+    private final CapabilityServiceSupport capabilityServiceSupport;
     private final AbstractVaultReader vaultReader;
     private final DeploymentOverlayIndex deploymentOverlays;
 
     /**
      * Construct a new instance.
-     *
-     * @param name the deployment unit simple name
+     *  @param name the deployment unit simple name
      * @param managementName the deployment's domain-wide unique name
      * @param parent the parent deployment unit
      * @param registration the registration
      * @param mutableRegistration the mutable registration
      * @param resource the model
+     * @param capabilityServiceSupport support for capability integration
      * @param vaultReader the vault reader
      * @param deploymentOverlays the deployment overlays
      */
-    public RootDeploymentUnitService(final String name, final String managementName, final DeploymentUnit parent, final ImmutableManagementResourceRegistration registration, final ManagementResourceRegistration mutableRegistration, Resource resource, final AbstractVaultReader vaultReader, DeploymentOverlayIndex deploymentOverlays) {
+    public RootDeploymentUnitService(final String name, final String managementName, final DeploymentUnit parent,
+                                     final ImmutableManagementResourceRegistration registration, final ManagementResourceRegistration mutableRegistration,
+                                     final Resource resource, final CapabilityServiceSupport capabilityServiceSupport,
+                                     final AbstractVaultReader vaultReader, DeploymentOverlayIndex deploymentOverlays) {
         assert name != null : "name is null";
         this.name = name;
         this.managementName = managementName;
@@ -72,6 +77,7 @@ public RootDeploymentUnitService(final String name, final String managementName,
         this.registration = registration;
         this.mutableRegistration = mutableRegistration;
         this.resource = resource;
+        this.capabilityServiceSupport = capabilityServiceSupport;
         this.vaultReader = vaultReader;
         this.deploymentOverlays = deploymentOverlays;
     }
@@ -85,6 +91,7 @@ protected DeploymentUnit createAndInitializeDeploymentUnit(final ServiceRegistry
         deploymentUnit.putAttachment(DeploymentResourceSupport.MUTABLE_REGISTRATION_ATTACHMENT, mutableRegistration);
         deploymentUnit.putAttachment(DeploymentResourceSupport.DEPLOYMENT_RESOURCE, resource);
         deploymentUnit.putAttachment(Attachments.DEPLOYMENT_RESOURCE_SUPPORT, new DeploymentResourceSupport(deploymentUnit));
+        deploymentUnit.putAttachment(Attachments.CAPABILITY_SERVICE_SUPPORT, capabilityServiceSupport);
         deploymentUnit.putAttachment(Attachments.VAULT_READER_ATTACHMENT_KEY, vaultReader);
         deploymentUnit.putAttachment(Attachments.DEPLOYMENT_OVERLAY_INDEX, deploymentOverlays);
         deploymentUnit.putAttachment(Attachments.PATH_MANAGER, pathManagerInjector.getValue());
diff --git a/server/src/main/java/org/jboss/as/server/deployment/SubDeploymentProcessor.java b/server/src/main/java/org/jboss/as/server/deployment/SubDeploymentProcessor.java
index 7d2b8685224..3d18b071043 100644
--- a/server/src/main/java/org/jboss/as/server/deployment/SubDeploymentProcessor.java
+++ b/server/src/main/java/org/jboss/as/server/deployment/SubDeploymentProcessor.java
@@ -24,6 +24,7 @@
 
 import java.util.List;
 
+import org.jboss.as.controller.capability.CapabilityServiceSupport;
 import org.jboss.as.controller.registry.ImmutableManagementResourceRegistration;
 import org.jboss.as.controller.registry.ManagementResourceRegistration;
 import org.jboss.as.controller.registry.Resource;
@@ -62,9 +63,11 @@ public void deploy(final DeploymentPhaseContext phaseContext) throws DeploymentU
             final Resource resource = DeploymentResourceSupport.getOrCreateSubDeployment(childRoot.getRootName(), deploymentUnit);
             final ImmutableManagementResourceRegistration registration = deploymentUnit.getAttachment(DeploymentResourceSupport.REGISTRATION_ATTACHMENT);
             final ManagementResourceRegistration mutableRegistration =  deploymentUnit.getAttachment(DeploymentResourceSupport.MUTABLE_REGISTRATION_ATTACHMENT);
+            final CapabilityServiceSupport capabilityServiceSupport = deploymentUnit.getAttachment(Attachments.CAPABILITY_SERVICE_SUPPORT);
             final AbstractVaultReader vaultReader = deploymentUnit.getAttachment(Attachments.VAULT_READER_ATTACHMENT_KEY);
             final PathManager pathManager = deploymentUnit.getAttachment(Attachments.PATH_MANAGER);
-            final SubDeploymentUnitService service = new SubDeploymentUnitService(childRoot, deploymentUnit, registration, mutableRegistration, resource, vaultReader, pathManager);
+            final SubDeploymentUnitService service = new SubDeploymentUnitService(childRoot, deploymentUnit, registration,
+                    mutableRegistration, resource, capabilityServiceSupport, vaultReader, pathManager);
 
             final ServiceName serviceName = Services.deploymentUnitName(deploymentUnit.getName(), childRoot.getRootName());
 
diff --git a/server/src/main/java/org/jboss/as/server/deployment/SubDeploymentUnitService.java b/server/src/main/java/org/jboss/as/server/deployment/SubDeploymentUnitService.java
index 83a2787839e..fa776e25d44 100644
--- a/server/src/main/java/org/jboss/as/server/deployment/SubDeploymentUnitService.java
+++ b/server/src/main/java/org/jboss/as/server/deployment/SubDeploymentUnitService.java
@@ -23,6 +23,7 @@
 package org.jboss.as.server.deployment;
 
 import org.jboss.as.controller.ServiceVerificationHandler;
+import org.jboss.as.controller.capability.CapabilityServiceSupport;
 import org.jboss.as.controller.registry.ImmutableManagementResourceRegistration;
 import org.jboss.as.controller.registry.ManagementResourceRegistration;
 import org.jboss.as.controller.registry.Resource;
@@ -44,10 +45,11 @@ public class SubDeploymentUnitService extends AbstractDeploymentUnitService {
     private final ImmutableManagementResourceRegistration registration;
     private final ManagementResourceRegistration mutableRegistration;
     private Resource resource;
+    private final CapabilityServiceSupport capabilityServiceSupport;
     private final AbstractVaultReader vaultReader;
     private final PathManager pathManager;
 
-    public SubDeploymentUnitService(ResourceRoot deploymentRoot, DeploymentUnit parent, ImmutableManagementResourceRegistration registration, final ManagementResourceRegistration mutableRegistration, Resource resource, final AbstractVaultReader vaultReader, PathManager pathManager) {
+    public SubDeploymentUnitService(ResourceRoot deploymentRoot, DeploymentUnit parent, ImmutableManagementResourceRegistration registration, final ManagementResourceRegistration mutableRegistration, Resource resource, CapabilityServiceSupport capabilityServiceSupport, final AbstractVaultReader vaultReader, PathManager pathManager) {
         this.pathManager = pathManager;
         if (deploymentRoot == null) throw ServerLogger.ROOT_LOGGER.deploymentRootRequired();
         this.deploymentRoot = deploymentRoot;
@@ -56,6 +58,7 @@ public SubDeploymentUnitService(ResourceRoot deploymentRoot, DeploymentUnit pare
         this.registration = registration;
         this.mutableRegistration = mutableRegistration;
         this.resource = resource;
+        this.capabilityServiceSupport = capabilityServiceSupport;
         this.vaultReader = vaultReader;
     }
 
@@ -68,6 +71,7 @@ protected DeploymentUnit createAndInitializeDeploymentUnit(ServiceRegistry regis
         deploymentUnit.putAttachment(DeploymentResourceSupport.MUTABLE_REGISTRATION_ATTACHMENT, mutableRegistration);
         deploymentUnit.putAttachment(DeploymentResourceSupport.DEPLOYMENT_RESOURCE, resource);
         deploymentUnit.putAttachment(Attachments.DEPLOYMENT_RESOURCE_SUPPORT, new DeploymentResourceSupport(deploymentUnit));
+        deploymentUnit.putAttachment(Attachments.CAPABILITY_SERVICE_SUPPORT, capabilityServiceSupport);
         deploymentUnit.putAttachment(Attachments.VAULT_READER_ATTACHMENT_KEY, vaultReader);
         deploymentUnit.putAttachment(Attachments.DEPLOYMENT_OVERLAY_INDEX, parent.getAttachment(Attachments.DEPLOYMENT_OVERLAY_INDEX));
         deploymentUnit.putAttachment(Attachments.PATH_MANAGER, pathManager);