Create a Tuscany specific listener manager so that the shutdown hook is cleared properly on runtime shutdown.

git-svn-id: http://svn.us.apache.org/repos/asf/tuscany@798434 13f79535-47bb-0310-9956-ffa450edef68

3 files changed, 172 insertions, 2 deletions

diff --git a/java/sca/modules/binding-ws-axis2/src/main/java/org/apache/tuscany/sca/binding/ws/axis2/Axis2ServiceProvider.java b/java/sca/modules/binding-ws-axis2/src/main/java/org/apache/tuscany/sca/binding/ws/axis2/Axis2ServiceProvider.java
index b018c97344..e6b18f8068 100644
--- a/java/sca/modules/binding-ws-axis2/src/main/java/org/apache/tuscany/sca/binding/ws/axis2/Axis2ServiceProvider.java
+++ b/java/sca/modules/binding-ws-axis2/src/main/java/org/apache/tuscany/sca/binding/ws/axis2/Axis2ServiceProvider.java
@@ -477,6 +477,8 @@ public class Axis2ServiceProvider {
                 servletHost.removeServletMapping(endpointURL);
             }
         }
+        
+        servletHost = null;
 
         if (jmsSender != null)
             jmsSender.stop();
diff --git a/java/sca/modules/binding-ws-axis2/src/main/java/org/apache/tuscany/sca/binding/ws/axis2/Axis2ServiceServlet.java b/java/sca/modules/binding-ws-axis2/src/main/java/org/apache/tuscany/sca/binding/ws/axis2/Axis2ServiceServlet.java
index 746b850ff0..debb2c5764 100644
--- a/java/sca/modules/binding-ws-axis2/src/main/java/org/apache/tuscany/sca/binding/ws/axis2/Axis2ServiceServlet.java
+++ b/java/sca/modules/binding-ws-axis2/src/main/java/org/apache/tuscany/sca/binding/ws/axis2/Axis2ServiceServlet.java
@@ -20,6 +20,7 @@ package org.apache.tuscany.sca.binding.ws.axis2;
 
 import java.io.IOException;
 import java.io.InputStream;
+import java.lang.reflect.Field;
 import java.net.MalformedURLException;
 import java.net.SocketException;
 import java.net.URI;
@@ -29,6 +30,7 @@ import java.util.Enumeration;
 import java.util.Set;
 import java.util.Vector;
 
+import javax.servlet.GenericServlet;
 import javax.servlet.RequestDispatcher;
 import javax.servlet.Servlet;
 import javax.servlet.ServletConfig;
@@ -38,8 +40,11 @@ import javax.servlet.http.HttpServletRequest;
 import javax.servlet.http.HttpServletResponse;
 
 import org.apache.axis2.AxisFault;
+import org.apache.axis2.Constants;
 import org.apache.axis2.addressing.EndpointReference;
 import org.apache.axis2.context.ConfigurationContext;
+import org.apache.axis2.description.TransportInDescription;
+import org.apache.axis2.engine.ListenerManager;
 import org.apache.axis2.transport.http.AxisServlet;
 import org.apache.axis2.transport.http.ListingAgent;
 import org.apache.axis2.transport.http.server.HttpUtils;
@@ -59,13 +64,16 @@ public class Axis2ServiceServlet extends AxisServlet {
     private static final long serialVersionUID = 1L;
     private static final ServletConfig DUMMY_CONFIG = createDummyServletConfig();
     
+    private boolean initCalled = false;
+    
 //JIRA TUSCANY-1561 Port to Axis2 1.3    
     private ConfigurationContext tmpconfigContext;
 
     public void init(ConfigurationContext configContext) {
         this.tmpconfigContext = configContext;
         try {
-            super.init(DUMMY_CONFIG);
+            //super.init(DUMMY_CONFIG);
+        	init(DUMMY_CONFIG);
         } catch (ServletException e) {
             throw new RuntimeException(e);
         }
@@ -85,7 +93,55 @@ public class Axis2ServiceServlet extends AxisServlet {
     public void init(ServletConfig config) throws ServletException {
         ServletContext servletContext = config.getServletContext();
         servletContext.setAttribute(CONFIGURATION_CONTEXT, tmpconfigContext);
-        super.init(config);
+       
+        //super.init(config);
+        
+        // A copy of the init method from the base class because we need to replace the 
+        // version of the ListenerManager that is used so that we can get it's 
+        // shutdown hook removed properly. 
+        
+        // prevent this method from being called more than once per instance
+        if (initCalled == false) {
+	        initCalled = true;
+	        // We can't call super.init() as it will just call the AxisServlet version
+	        // which we are replacing here. So reflect on the base class and
+	        // set the private config field in the base class. 
+	        //super.init(config);
+	        try {
+	        	Field field = GenericServlet.class.getDeclaredField("config");
+	        	field.setAccessible(true);
+	        	field.set(this, config);
+	        } catch (Exception ex){
+	        	ex.printStackTrace();
+	        }
+	        
+	        try {
+	            this.servletConfig = config;
+	            //ServletContext servletContext = servletConfig.getServletContext();
+	            this.configContext =
+	                    (ConfigurationContext) servletContext.getAttribute(CONFIGURATION_CONTEXT);
+	            if(configContext == null){
+	                configContext = initConfigContext(config);
+	                config.getServletContext().setAttribute(CONFIGURATION_CONTEXT, configContext);
+	            }
+	            axisConfiguration = configContext.getAxisConfiguration();
+	
+	            ListenerManager listenerManager = new TuscanyListenerManager();
+	            listenerManager.init(configContext);
+	            TransportInDescription transportInDescription = new TransportInDescription(
+	                    Constants.TRANSPORT_HTTP);
+	            transportInDescription.setReceiver(this);
+	            listenerManager.addListener(transportInDescription, true);
+	            listenerManager.start();
+	            ListenerManager.defaultConfigurationContext = configContext;
+	            super.agent = new ListingAgent(configContext);
+	
+	            initParams();
+	
+	        } catch (Exception e) {
+	            throw new ServletException(e);
+	        }
+        }
     }
 
     /**
@@ -237,9 +293,14 @@ public class Axis2ServiceServlet extends AxisServlet {
     public void destroy() {
         try {
             super.destroy();
+            servletConfig = null;
+            if (tmpconfigContext.getListenerManager() != null){
+            	tmpconfigContext.getListenerManager().destroy();
+            }
         } catch (Exception e) {
             e.printStackTrace();
         }
+    
     }
 
     /**
diff --git a/java/sca/modules/binding-ws-axis2/src/main/java/org/apache/tuscany/sca/binding/ws/axis2/TuscanyListenerManager.java b/java/sca/modules/binding-ws-axis2/src/main/java/org/apache/tuscany/sca/binding/ws/axis2/TuscanyListenerManager.java
new file mode 100644
index 0000000000..7d331aa6c8
--- /dev/null
+++ b/java/sca/modules/binding-ws-axis2/src/main/java/org/apache/tuscany/sca/binding/ws/axis2/TuscanyListenerManager.java
@@ -0,0 +1,107 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ * 
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ * 
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.    
+ */
+
+package org.apache.tuscany.sca.binding.ws.axis2;
+
+import java.lang.reflect.Field;
+import java.util.HashMap;
+import java.util.Iterator;
+
+import org.apache.axis2.AxisFault;
+import org.apache.axis2.context.ConfigurationContext;
+import org.apache.axis2.description.TransportInDescription;
+import org.apache.axis2.engine.ListenerManager;
+import org.apache.axis2.transport.TransportListener;
+import org.apache.commons.logging.Log;
+import org.apache.commons.logging.LogFactory;
+
+/**
+ * A Tuscany specific Axis2 ListenerManager. Created purely as part of
+ * TUSCANY-3149 to unregister the ListenerManager from the runtime
+ * shutown hook
+ */
+public class TuscanyListenerManager extends ListenerManager {
+	private static final Log log = LogFactory.getLog(TuscanyListenerManager.class);
+	
+	private ListenerManagerShutdownThread shutdownThread = null;
+	
+    /**
+     * To start all the transports
+     */
+    public synchronized void start() {
+    	
+    	ConfigurationContext configctx = getConfigctx();
+    	
+    	// very nasty! but this is in order to get someone running who keeps 
+    	// getting perm gen errors. This will all go away when we move up to Axis2 1.5
+    	HashMap startedTransports = null;
+    	
+    	try {
+    		Field field = ListenerManager.class.getDeclaredField("startedTransports");
+    		field.setAccessible(true);
+    		startedTransports = (HashMap)field.get(this);
+    	} catch (Exception ex){
+    		ex.printStackTrace();
+    	}
+
+        for (Iterator transportNames =
+                configctx.getAxisConfiguration().getTransportsIn().values().iterator();
+             transportNames.hasNext();) {
+            try {
+                TransportInDescription transportIn = (TransportInDescription) transportNames.next();
+                TransportListener listener = transportIn.getReceiver();
+                if (listener != null &&
+                    startedTransports.get(transportIn.getName()) == null) {
+                    listener.init(configctx, transportIn);
+                    listener.start();
+                    if (startedTransports.get(transportIn.getName()) == null) {
+                        startedTransports.put(transportIn.getName(), listener);
+                    }
+                }
+            } catch (Exception e) {
+                log.info(e.getMessage(), e);
+            }
+        }
+        shutdownThread = new ListenerManagerShutdownThread(this);
+        Runtime.getRuntime().addShutdownHook(shutdownThread);
+    }
+    
+    public synchronized void stop() throws AxisFault {
+    	super.stop();
+    	Runtime.getRuntime().removeShutdownHook(shutdownThread);
+    }
+    
+    static class ListenerManagerShutdownThread extends Thread {
+        ListenerManager listenerManager;
+
+        public ListenerManagerShutdownThread(ListenerManager listenerManager) {
+            super();
+            this.listenerManager = listenerManager;
+        }
+
+        public void run() {
+            try {                
+            	listenerManager.stop();
+            } catch (AxisFault axisFault) {
+                log.error(axisFault.getMessage(), axisFault);
+            }
+        }
+    }    
+
+}