Mercurial > jdk8-mips64-public > jaxws / file revision

 /*

  * Copyright (c) 1997, 2012, 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;

     }

 }