diff -r 88b85470e72c -r f50545b5e2f1 src/share/jaxws_classes/com/sun/xml/internal/ws/handler/ClientSOAPHandlerTube.java --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/src/share/jaxws_classes/com/sun/xml/internal/ws/handler/ClientSOAPHandlerTube.java Tue Mar 06 16:09:35 2012 -0800 @@ -0,0 +1,160 @@ +/* + * Copyright (c) 1997, 2010, Oracle and/or its affiliates. All rights reserved. + * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. + * + * This code is free software; you can redistribute it and/or modify it + * under the terms of the GNU General Public License version 2 only, as + * published by the Free Software Foundation. Oracle designates this + * particular file as subject to the "Classpath" exception as provided + * by Oracle in the LICENSE file that accompanied this code. + * + * This code 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 General Public License + * version 2 for more details (a copy is included in the LICENSE file that + * accompanied this code). + * + * You should have received a copy of the GNU General Public License version + * 2 along with this work; if not, write to the Free Software Foundation, + * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. + * + * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA + * or visit www.oracle.com if you need additional information or have any + * questions. + */ + +package com.sun.xml.internal.ws.handler; + +import com.sun.xml.internal.ws.api.WSBinding; +import com.sun.xml.internal.ws.api.message.Packet; +import com.sun.xml.internal.ws.api.message.Attachment; +import com.sun.xml.internal.ws.api.message.AttachmentSet; +import com.sun.xml.internal.ws.api.pipe.TubeCloner; +import com.sun.xml.internal.ws.api.pipe.Tube; +import com.sun.xml.internal.ws.api.pipe.helper.AbstractFilterTubeImpl; +import com.sun.xml.internal.ws.api.model.wsdl.WSDLPort; +import com.sun.xml.internal.ws.client.HandlerConfiguration; +import com.sun.xml.internal.ws.binding.BindingImpl; +import com.sun.xml.internal.ws.message.DataHandlerAttachment; + +import javax.xml.ws.handler.soap.SOAPHandler; +import javax.xml.ws.handler.MessageContext; +import javax.xml.ws.handler.Handler; +import javax.xml.ws.WebServiceException; +import javax.activation.DataHandler; +import java.util.*; + +/** + * + * @author WS Development Team + */ +public class ClientSOAPHandlerTube extends HandlerTube { + + private Set roles; + + /** + * Creates a new instance of SOAPHandlerTube + */ + public ClientSOAPHandlerTube(WSBinding binding, WSDLPort port, Tube next) { + super(next, port, binding); + if (binding.getSOAPVersion() != null) { + // SOAPHandlerTube should n't be used for bindings other than SOAP. + // TODO: throw Exception + } + } + + // Handle to LogicalHandlerTube means its used on SERVER-SIDE + + /** + * This constructor is used on client-side where, LogicalHandlerTube is created + * first and then a SOAPHandlerTube is created with a handler to that + * LogicalHandlerTube. + * With this handle, SOAPHandlerTube can call LogicalHandlerTube.closeHandlers() + */ + public ClientSOAPHandlerTube(WSBinding binding, Tube next, HandlerTube cousinTube) { + super(next, cousinTube, binding); + } + + /** + * Copy constructor for {@link com.sun.xml.internal.ws.api.pipe.Tube#copy(com.sun.xml.internal.ws.api.pipe.TubeCloner)}. + */ + private ClientSOAPHandlerTube(ClientSOAPHandlerTube that, TubeCloner cloner) { + super(that, cloner); + } + + public AbstractFilterTubeImpl copy(TubeCloner cloner) { + return new ClientSOAPHandlerTube(this, cloner); + } + + void setUpProcessor() { + if (handlers == null) { + // Take a snapshot, User may change chain after invocation, Same chain + // should be used for the entire MEP + handlers = new ArrayList(); + HandlerConfiguration handlerConfig = ((BindingImpl) getBinding()).getHandlerConfig(); + List soapSnapShot= handlerConfig.getSoapHandlers(); + if (!soapSnapShot.isEmpty()) { + handlers.addAll(soapSnapShot); + roles = new HashSet(); + roles.addAll(handlerConfig.getRoles()); + processor = new SOAPHandlerProcessor(true, this, getBinding(), handlers); + } + } + } + + MessageUpdatableContext getContext(Packet packet) { + SOAPMessageContextImpl context = new SOAPMessageContextImpl(getBinding(), packet,roles); + return context; + } + + boolean callHandlersOnRequest(MessageUpdatableContext context, boolean isOneWay) { + + boolean handlerResult; + //Lets copy all the MessageContext.OUTBOUND_ATTACHMENT_PROPERTY to the message + Map atts = (Map) context.get(MessageContext.OUTBOUND_MESSAGE_ATTACHMENTS); + AttachmentSet attSet = context.packet.getMessage().getAttachments(); + for(String cid : atts.keySet()){ + if (attSet.get(cid) == null) { // Otherwise we would be adding attachments twice + Attachment att = new DataHandlerAttachment(cid, atts.get(cid)); + attSet.add(att); + } + } + + try { + //CLIENT-SIDE + handlerResult = processor.callHandlersRequest(HandlerProcessor.Direction.OUTBOUND, context, !isOneWay); + } catch (WebServiceException wse) { + remedyActionTaken = true; + //no rewrapping + throw wse; + } catch (RuntimeException re) { + remedyActionTaken = true; + + throw new WebServiceException(re); + + } + if (!handlerResult) { + remedyActionTaken = true; + } + return handlerResult; + } + + void callHandlersOnResponse(MessageUpdatableContext context, boolean handleFault) { + try { + + //CLIENT-SIDE + processor.callHandlersResponse(HandlerProcessor.Direction.INBOUND, context, handleFault); + + } catch (WebServiceException wse) { + //no rewrapping + throw wse; + } catch (RuntimeException re) { + throw new WebServiceException(re); + } + } + + void closeHandlers(MessageContext mc) { + closeClientsideHandlers(mc); + + } +}