/* * Copyright 2005-2006 Sun Microsystems, Inc. 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. Sun designates this * particular file as subject to the "Classpath" exception as provided * by Sun 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 Sun Microsystems, Inc., 4150 Network Circle, Santa Clara, * CA 95054 USA or visit www.sun.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.AttachmentSet; import com.sun.xml.internal.ws.api.message.Attachment; 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.binding.BindingImpl; import com.sun.xml.internal.ws.message.DataHandlerAttachment; import javax.xml.ws.handler.LogicalHandler; import javax.xml.ws.handler.MessageContext; import javax.xml.ws.WebServiceException; import javax.activation.DataHandler; import java.util.List; import java.util.ArrayList; import java.util.Map; /** * * @author WS Development Team */ public class ServerLogicalHandlerTube extends HandlerTube { private WSBinding binding; private List<LogicalHandler> logicalHandlers; /** * Creates a new instance of LogicalHandlerTube */ public ServerLogicalHandlerTube(WSBinding binding, WSDLPort port, Tube next) { super(next, port); this.binding = binding; setUpProcessorOnce(); } /** * This constructor is used on client-side where, SOAPHandlerTube is created * first and then a LogicalHandlerTube is created with a handler to that * SOAPHandlerTube. * With this handle, LogicalHandlerTube can call * SOAPHandlerTube.closeHandlers() */ public ServerLogicalHandlerTube(WSBinding binding, Tube next, HandlerTube cousinTube) { super(next, cousinTube); this.binding = binding; setUpProcessorOnce(); } /** * Copy constructor for {@link com.sun.xml.internal.ws.api.pipe.Tube#copy(com.sun.xml.internal.ws.api.pipe.TubeCloner)}. */ private ServerLogicalHandlerTube(ServerLogicalHandlerTube that, TubeCloner cloner) { super(that, cloner); this.binding = that.binding; setUpProcessorOnce(); } boolean isHandlerChainEmpty() { return logicalHandlers.isEmpty(); } /** * Close SOAPHandlers first and then LogicalHandlers on Client * Close LogicalHandlers first and then SOAPHandlers on Server */ public void close(MessageContext msgContext) { if (binding.getSOAPVersion() != null) { //SOAPHandlerTube will drive the closing of LogicalHandlerTube } else { if (processor != null) closeLogicalHandlers(msgContext); } } /** * This is called from cousinTube. * Close this Tube's handlers. */ public void closeCall(MessageContext msgContext) { closeLogicalHandlers(msgContext); } //TODO: private void closeLogicalHandlers(MessageContext msgContext) { if (processor == null) return; if (remedyActionTaken) { //Close only invoked handlers in the chain //SERVER-SIDE processor.closeHandlers(msgContext, processor.getIndex(), logicalHandlers.size() - 1); processor.setIndex(-1); //reset remedyActionTaken remedyActionTaken = false; } else { //Close all handlers in the chain //SERVER-SIDE processor.closeHandlers(msgContext, 0, logicalHandlers.size() - 1); } } public AbstractFilterTubeImpl copy(TubeCloner cloner) { return new ServerLogicalHandlerTube(this, cloner); } private void setUpProcessorOnce() { logicalHandlers = new ArrayList<LogicalHandler>(); List<LogicalHandler> logicalSnapShot= ((BindingImpl) binding).getHandlerConfig().getLogicalHandlers(); if (!logicalSnapShot.isEmpty()) { logicalHandlers.addAll(logicalSnapShot); if (binding.getSOAPVersion() == null) { processor = new XMLHandlerProcessor(this, binding, logicalHandlers); } else { processor = new SOAPHandlerProcessor(false, this, binding, logicalHandlers); } } } void setUpProcessor() { // Do nothing, Processor is setup in the constructor. } MessageUpdatableContext getContext(Packet packet) { return new LogicalMessageContextImpl(binding, packet); } boolean callHandlersOnRequest(MessageUpdatableContext context, boolean isOneWay) { boolean handlerResult; try { //SERVER-SIDE handlerResult = processor.callHandlersRequest(HandlerProcessor.Direction.INBOUND, context, !isOneWay); } catch (RuntimeException re) { remedyActionTaken = true; throw re; } if (!handlerResult) { remedyActionTaken = true; } return handlerResult; } void callHandlersOnResponse(MessageUpdatableContext context, boolean handleFault) { //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 = packet.getMessage().getAttachments(); for(String cid : atts.keySet()){ Attachment att = new DataHandlerAttachment(cid, atts.get(cid)); attSet.add(att); } try { //SERVER-SIDE processor.callHandlersResponse(HandlerProcessor.Direction.OUTBOUND, context, handleFault); } catch (WebServiceException wse) { //no rewrapping throw wse; } catch (RuntimeException re) { throw re; } } }