[jbossws-commits] JBossWS SVN: r8245 - in stack/native/branches/dlofthouse/JBWS-2304/modules/core/src/main/java/org/jboss: ws/core/soap and 1 other directories.

Author: darran.lofthouse(a)jboss.com Date: 2008-09-26 10:35:42 -0400 (Fri, 26 Sep 2008) New Revision: 8245 Modified: stack/native/branches/dlofthouse/JBWS-2304/modules/core/src/main/java/org/jboss/ws/core/jaxws/handler/MessageContextJAXWS.java stack/native/branches/dlofthouse/JBWS-2304/modules/core/src/main/java/org/jboss/ws/core/soap/MessageContextAssociation.java stack/native/branches/dlofthouse/JBWS-2304/modules/core/src/main/java/org/jboss/ws/core/soap/XMLContent.java stack/native/branches/dlofthouse/JBWS-2304/modules/core/src/main/java/org/jboss/wsf/stack/jbws/RequestHandlerImpl.java Log: Clear ThreadLocals once MessageContextAssociation stack is empty (With exceptions) Modified: stack/native/branches/dlofthouse/JBWS-2304/modules/core/src/main/java/org/jboss/ws/core/jaxws/handler/MessageContextJAXWS.java =================================================================== --- stack/native/branches/dlofthouse/JBWS-2304/modules/core/src/main/java/org/jboss/ws/core/jaxws/handler/MessageContextJAXWS.java 2008-09-26 14:31:08 UTC (rev 8244) +++ stack/native/branches/dlofthouse/JBWS-2304/modules/core/src/main/java/org/jboss/ws/core/jaxws/handler/MessageContextJAXWS.java 2008-09-26 14:35:42 UTC (rev 8245) @@ -106,7 +106,7 @@ if (outbound == null) throw new IllegalStateException("Cannot find property: " + MessageContext.MESSAGE_OUTBOUND_PROPERTY); - MessageContextAssociation.popMessageContext(); + MessageContextAssociation.popMessageContext(false); SOAPMessageContextJAXWS resContext = new SOAPMessageContextJAXWS(reqContext); resContext.setSOAPMessage(null); Modified: stack/native/branches/dlofthouse/JBWS-2304/modules/core/src/main/java/org/jboss/ws/core/soap/MessageContextAssociation.java =================================================================== --- stack/native/branches/dlofthouse/JBWS-2304/modules/core/src/main/java/org/jboss/ws/core/soap/MessageContextAssociation.java 2008-09-26 14:31:08 UTC (rev 8244) +++ stack/native/branches/dlofthouse/JBWS-2304/modules/core/src/main/java/org/jboss/ws/core/soap/MessageContextAssociation.java 2008-09-26 14:35:42 UTC (rev 8245) @@ -26,6 +26,7 @@ import org.jboss.logging.Logger; import org.jboss.ws.core.CommonMessageContext; import org.jboss.ws.core.utils.ThreadLocalAssociation; +import org.jboss.wsf.common.DOMUtils; /** * A thread local association with the current message context @@ -37,11 +38,11 @@ { // provide logging private static Logger log = Logger.getLogger(MessageContextAssociation.class); - public static void pushMessageContext(CommonMessageContext msgContext) { - if(log.isDebugEnabled()) log.debug("pushMessageContext: " + msgContext + " (Thread " +Thread.currentThread().getName()+ ")"); + if (log.isDebugEnabled()) + log.debug("pushMessageContext: " + msgContext + " (Thread " + Thread.currentThread().getName() + ")"); Stack<CommonMessageContext> stack = ThreadLocalAssociation.localMsgContextAssoc().get(); if (stack == null) { @@ -64,13 +65,24 @@ public static CommonMessageContext popMessageContext() { + return popMessageContext(true); + } + + public static CommonMessageContext popMessageContext(boolean clearDOMIfEmpty) + { CommonMessageContext msgContext = null; Stack<CommonMessageContext> stack = ThreadLocalAssociation.localMsgContextAssoc().get(); if (stack != null && stack.isEmpty() == false) { msgContext = stack.pop(); + if (stack.isEmpty() == true && clearDOMIfEmpty == true) + { + DOMUtils.clearThreadLocals(); + } } - if(log.isDebugEnabled()) log.debug("popMessageContext: " + msgContext +" (Thread " +Thread.currentThread().getName()+ ")"); + if (log.isDebugEnabled()) + log.debug("popMessageContext: " + msgContext + " (Thread " + Thread.currentThread().getName() + ")"); return msgContext; } + } Modified: stack/native/branches/dlofthouse/JBWS-2304/modules/core/src/main/java/org/jboss/ws/core/soap/XMLContent.java =================================================================== --- stack/native/branches/dlofthouse/JBWS-2304/modules/core/src/main/java/org/jboss/ws/core/soap/XMLContent.java 2008-09-26 14:31:08 UTC (rev 8244) +++ stack/native/branches/dlofthouse/JBWS-2304/modules/core/src/main/java/org/jboss/ws/core/soap/XMLContent.java 2008-09-26 14:35:42 UTC (rev 8245) @@ -71,6 +71,7 @@ // The well formed XML content of this element. private XMLFragment xmlFragment; + private Document document = DOMUtils.getOwnerDocument(); protected XMLContent(SOAPContentElement container) { @@ -327,10 +328,29 @@ short childType = child.getNodeType(); if (childType == Node.ELEMENT_NODE) { - SOAPElement soapElement = soapFactory.createElement((Element)child); - container.addChildElement(soapElement); - if (Constants.NAME_XOP_INCLUDE.equals(qname) || container.isXOPParameter()) - XOPContext.inlineXOPData(soapElement); + + boolean setOwnerDocument = (DOMUtils.peekOwnerDocument() == null); + + try + { + if (setOwnerDocument) + { + DOMUtils.setOwnerDocument(document); + } + SOAPElement soapElement = soapFactory.createElement((Element)child); + container.addChildElement(soapElement); + if (Constants.NAME_XOP_INCLUDE.equals(qname) || container.isXOPParameter()) + XOPContext.inlineXOPData(soapElement); + + } + finally + { + if (setOwnerDocument) + { + DOMUtils.clearThreadLocals(); + } + } + } else if (childType == Node.TEXT_NODE) { Modified: stack/native/branches/dlofthouse/JBWS-2304/modules/core/src/main/java/org/jboss/wsf/stack/jbws/RequestHandlerImpl.java =================================================================== --- stack/native/branches/dlofthouse/JBWS-2304/modules/core/src/main/java/org/jboss/wsf/stack/jbws/RequestHandlerImpl.java 2008-09-26 14:31:08 UTC (rev 8244) +++ stack/native/branches/dlofthouse/JBWS-2304/modules/core/src/main/java/org/jboss/wsf/stack/jbws/RequestHandlerImpl.java 2008-09-26 14:35:42 UTC (rev 8245) @@ -346,7 +346,6 @@ // clear thread local storage ThreadLocalAssociation.clear(); - DOMUtils.clearThreadLocals(); } }