| /* |
| * Copyright (c) 1997, 2013, 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.istack.internal.Nullable; |
| import com.sun.xml.internal.ws.api.WSBinding; |
| import com.sun.xml.internal.ws.api.handler.MessageHandler; |
| 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.message.Packet; |
| import com.sun.xml.internal.ws.api.model.SEIModel; |
| import com.sun.xml.internal.ws.api.model.wsdl.WSDLPort; |
| import com.sun.xml.internal.ws.api.pipe.Tube; |
| import com.sun.xml.internal.ws.api.pipe.TubeCloner; |
| import com.sun.xml.internal.ws.api.pipe.helper.AbstractFilterTubeImpl; |
| import com.sun.xml.internal.ws.binding.BindingImpl; |
| import com.sun.xml.internal.ws.client.HandlerConfiguration; |
| import com.sun.xml.internal.ws.message.DataHandlerAttachment; |
| |
| import javax.activation.DataHandler; |
| import javax.xml.ws.WebServiceException; |
| import javax.xml.ws.handler.MessageContext; |
| import javax.xml.ws.handler.Handler; |
| import java.util.*; |
| import java.util.Map.Entry; |
| |
| /** |
| * @author Rama Pulavarthi |
| */ |
| public class ClientMessageHandlerTube extends HandlerTube { |
| private SEIModel seiModel; |
| private Set<String> roles; |
| |
| /** |
| * Creates a new instance of MessageHandlerTube |
| */ |
| public ClientMessageHandlerTube(@Nullable SEIModel seiModel, WSBinding binding, WSDLPort port, Tube next) { |
| super(next, port, binding); |
| this.seiModel = seiModel; |
| } |
| |
| /** |
| * Copy constructor for {@link com.sun.xml.internal.ws.api.pipe.Tube#copy(com.sun.xml.internal.ws.api.pipe.TubeCloner)}. |
| */ |
| private ClientMessageHandlerTube(ClientMessageHandlerTube that, TubeCloner cloner) { |
| super(that, cloner); |
| this.seiModel = that.seiModel; |
| } |
| |
| public AbstractFilterTubeImpl copy(TubeCloner cloner) { |
| return new ClientMessageHandlerTube(this, cloner); |
| } |
| |
| 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); |
| } |
| } |
| |
| boolean callHandlersOnRequest(MessageUpdatableContext context, boolean isOneWay) { |
| boolean handlerResult; |
| |
| //Lets copy all the MessageContext.OUTBOUND_ATTACHMENT_PROPERTY to the message |
| Map<String, DataHandler> atts = (Map<String, DataHandler>) context.get(MessageContext.OUTBOUND_MESSAGE_ATTACHMENTS); |
| AttachmentSet attSet = context.packet.getMessage().getAttachments(); |
| for (Entry<String, DataHandler> entry : atts.entrySet()) { |
| String cid = entry.getKey(); |
| 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 closeHandlers(MessageContext mc) { |
| closeClientsideHandlers(mc); |
| |
| } |
| |
| 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<Handler>(); |
| HandlerConfiguration handlerConfig = ((BindingImpl) getBinding()).getHandlerConfig(); |
| List<MessageHandler> msgHandlersSnapShot= handlerConfig.getMessageHandlers(); |
| if (!msgHandlersSnapShot.isEmpty()) { |
| handlers.addAll(msgHandlersSnapShot); |
| roles = new HashSet<String>(); |
| roles.addAll(handlerConfig.getRoles()); |
| processor = new SOAPHandlerProcessor(true, this, getBinding(), handlers); |
| } |
| } |
| } |
| |
| |
| |
| MessageUpdatableContext getContext(Packet p) { |
| MessageHandlerContextImpl context = new MessageHandlerContextImpl(seiModel, getBinding(), port, p, roles); |
| return context; |
| } |
| |
| |
| } |