WFLY-8697 ModuleContextProcessor doesn't cleanup its Attachments on u…

…ndeploy()
wildfly · May 6, 2017 · bb0a90c · bb0a90c
1 parent a76c5f0
commit bb0a90c
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/ee/src/main/java/org/jboss/as/ee/naming/ModuleContextProcessor.java b/ee/src/main/java/org/jboss/as/ee/naming/ModuleContextProcessor.java
@@ -79,7 +79,7 @@ public void deploy(DeploymentPhaseContext phaseContext) throws DeploymentUnitPro
                 .addInjection(moduleNameBinder.getManagedObjectInjector(), new ValueManagedReferenceFactory(Values.immediateValue(moduleDescription.getModuleName())))
                 .addDependency(moduleContextServiceName, ServiceBasedNamingStore.class, moduleNameBinder.getNamingStoreInjector())
                 .install();
-        deploymentUnit.addToAttachmentList(org.jboss.as.server.deployment.Attachments.JNDI_DEPENDENCIES,moduleNameServiceName);
+        deploymentUnit.addToAttachmentList(org.jboss.as.server.deployment.Attachments.JNDI_DEPENDENCIES, moduleNameServiceName);
 
         deploymentUnit.putAttachment(MODULE_CONTEXT_CONFIG, moduleContextServiceName);
 
@@ -106,7 +106,10 @@ public void deploy(DeploymentPhaseContext phaseContext) throws DeploymentUnitPro
         deploymentUnit.putAttachment(Attachments.JAVA_NAMESPACE_SETUP_ACTION, setupAction);
     }
 
-    public void undeploy(DeploymentUnit context) {
-
+    public void undeploy(DeploymentUnit deploymentUnit) {
+        deploymentUnit.removeAttachment(Attachments.JAVA_NAMESPACE_SETUP_ACTION);
+        deploymentUnit.getAttachmentList(org.jboss.as.ee.component.Attachments.WEB_SETUP_ACTIONS).removeIf(setupAction -> setupAction instanceof JavaNamespaceSetup);
+        deploymentUnit.getAttachmentList(SETUP_ACTIONS).removeIf(setupAction -> setupAction instanceof JavaNamespaceSetup);
+        deploymentUnit.removeAttachment(MODULE_CONTEXT_CONFIG);
     }
 }