Adding GracefulShutdownIntegrationProcessor for letting ws deal with …

…rejecting requests when in suspended mode according to XTS needs.
wildfly · Nov 24, 2016 · c5ab205 · c5ab205
1 parent fd2601e
commit c5ab205
Show file tree

Hide file tree

Showing 2 changed files with 70 additions and 0 deletions.
diff --git a/...rc/main/java/org/jboss/as/webservices/deployers/GracefulShutdownIntegrationProcessor.java b/...rc/main/java/org/jboss/as/webservices/deployers/GracefulShutdownIntegrationProcessor.java
@@ -0,0 +1,68 @@
+/*
+ * JBoss, Home of Professional Open Source.
+ * Copyright 2013, 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.webservices.deployers;
+
+import static org.jboss.as.webservices.util.WSAttachmentKeys.JAXWS_ENDPOINTS_KEY;
+
+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.webservices.metadata.model.JAXWSDeployment;
+import org.jboss.wsf.spi.deployment.WSFServlet;
+import org.wildfly.extension.undertow.deployment.UndertowAttachments;
+
+import io.undertow.predicate.Predicate;
+import io.undertow.server.HttpServerExchange;
+import io.undertow.servlet.handlers.ServletRequestContext;
+
+/**
+ * DUP for telling Undertow to let WS deal with blocking requests to 
+ * serve XTS requirements.
+ * 
+ * @autor <a href="mailto:alessio.soldano@jboss.com">Alessio Soldano</a>
+ *
+ */
+public class GracefulShutdownIntegrationProcessor implements DeploymentUnitProcessor {
+
+    @Override
+    public void deploy(DeploymentPhaseContext phaseContext) throws DeploymentUnitProcessingException {
+        final DeploymentUnit unit = phaseContext.getDeploymentUnit();
+        final JAXWSDeployment wsDeployment = unit.getAttachment(JAXWS_ENDPOINTS_KEY);
+        if (wsDeployment != null) {
+            Predicate predicate = new Predicate() {
+                @Override
+                public boolean resolve(HttpServerExchange exchange) {
+                    ServletRequestContext src = exchange.getAttachment(ServletRequestContext.ATTACHMENT_KEY);
+                    return src.getCurrentServlet().getManagedServlet().getServletInfo().getServletClass().equals(WSFServlet.class);
+                }
+            };
+            unit.addToAttachmentList(UndertowAttachments.ALLOW_REQUEST_WHEN_SUSPENDED, predicate);
+        }
+    }
+
+    @Override
+    public void undeploy(DeploymentUnit context) {
+    }
+
+}
diff --git a/.../server-integration/src/main/java/org/jboss/as/webservices/dmr/WSDeploymentActivator.java b/.../server-integration/src/main/java/org/jboss/as/webservices/dmr/WSDeploymentActivator.java
@@ -27,6 +27,7 @@
 import org.jboss.as.server.DeploymentProcessorTarget;
 import org.jboss.as.server.deployment.Phase;
 import org.jboss.as.webservices.deployers.AspectDeploymentProcessor;
+import org.jboss.as.webservices.deployers.GracefulShutdownIntegrationProcessor;
 import org.jboss.as.webservices.deployers.JBossWebservicesDescriptorDeploymentProcessor;
 import org.jboss.as.webservices.deployers.WSClassVerificationProcessor;
 import org.jboss.as.webservices.deployers.WSDependenciesProcessor;
@@ -74,6 +75,7 @@ static void activate(final DeploymentProcessorTarget processorTarget, final bool
             processorTarget.addDeploymentProcessor(WSExtension.SUBSYSTEM_NAME, Phase.PARSE, Phase.PARSE_JAXWS_HANDLER_CREATE_COMPONENT_DESCRIPTIONS, new WSIntegrationProcessorJAXWS_HANDLER());
             processorTarget.addDeploymentProcessor(WSExtension.SUBSYSTEM_NAME, Phase.STRUCTURE, Phase.STRUCTURE_WS_SERVICES_DEPS, new WSServiceDependenciesProcessor());
             processorTarget.addDeploymentProcessor(WSExtension.SUBSYSTEM_NAME, Phase.POST_MODULE, Phase.POST_MODULE_WS_VERIFICATION, new WSClassVerificationProcessor());
+            processorTarget.addDeploymentProcessor(WSExtension.SUBSYSTEM_NAME, Phase.POST_MODULE, Phase.POST_MODULE_WS_VERIFICATION, new GracefulShutdownIntegrationProcessor());
             processorTarget.addDeploymentProcessor(WSExtension.SUBSYSTEM_NAME, Phase.INSTALL, Phase.INSTALL_WS_UNIVERSAL_META_DATA_MODEL, new WSModelDeploymentProcessor());
             addDeploymentProcessors(processorTarget, Phase.INSTALL, Phase.INSTALL_WS_DEPLOYMENT_ASPECTS);
         }