1.1 --- /dev/null Thu Jan 01 00:00:00 1970 +0000
1.2 +++ b/src/share/jaxws_classes/com/sun/xml/internal/ws/handler/SOAPMessageContextImpl.java Wed Apr 27 01:27:09 2016 +0800
1.3 @@ -0,0 +1,131 @@
1.4 +/*
1.5 + * Copyright (c) 1997, 2013, Oracle and/or its affiliates. All rights reserved.
1.6 + * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
1.7 + *
1.8 + * This code is free software; you can redistribute it and/or modify it
1.9 + * under the terms of the GNU General Public License version 2 only, as
1.10 + * published by the Free Software Foundation. Oracle designates this
1.11 + * particular file as subject to the "Classpath" exception as provided
1.12 + * by Oracle in the LICENSE file that accompanied this code.
1.13 + *
1.14 + * This code is distributed in the hope that it will be useful, but WITHOUT
1.15 + * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
1.16 + * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
1.17 + * version 2 for more details (a copy is included in the LICENSE file that
1.18 + * accompanied this code).
1.19 + *
1.20 + * You should have received a copy of the GNU General Public License version
1.21 + * 2 along with this work; if not, write to the Free Software Foundation,
1.22 + * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
1.23 + *
1.24 + * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
1.25 + * or visit www.oracle.com if you need additional information or have any
1.26 + * questions.
1.27 + */
1.28 +
1.29 +package com.sun.xml.internal.ws.handler;
1.30 +import com.sun.xml.internal.ws.api.message.Header;
1.31 +import com.sun.xml.internal.ws.api.message.Message;
1.32 +import com.sun.xml.internal.ws.api.message.Packet;
1.33 +import com.sun.xml.internal.ws.api.message.saaj.SAAJFactory;
1.34 +import com.sun.xml.internal.ws.api.WSBinding;
1.35 +import com.sun.xml.internal.ws.api.SOAPVersion;
1.36 +
1.37 +import javax.xml.bind.JAXBContext;
1.38 +import javax.xml.namespace.QName;
1.39 +import javax.xml.soap.SOAPException;
1.40 +import javax.xml.soap.SOAPMessage;
1.41 +import javax.xml.ws.WebServiceException;
1.42 +import javax.xml.ws.handler.soap.SOAPMessageContext;
1.43 +
1.44 +import java.util.ArrayList;
1.45 +import java.util.Iterator;
1.46 +import java.util.List;
1.47 +import java.util.Set;
1.48 +
1.49 +/**
1.50 + * Implementation of {@link SOAPMessageContext}. This class is used at runtime
1.51 + * to pass to the handlers for processing soap messages.
1.52 + *
1.53 + * @see MessageContextImpl
1.54 + *
1.55 + * @author WS Development Team
1.56 + */
1.57 +public class SOAPMessageContextImpl extends MessageUpdatableContext implements SOAPMessageContext {
1.58 +
1.59 + private Set<String> roles;
1.60 + private SOAPMessage soapMsg = null;
1.61 + private WSBinding binding;
1.62 +
1.63 + public SOAPMessageContextImpl(WSBinding binding, Packet packet,Set<String> roles) {
1.64 + super(packet);
1.65 + this.binding = binding;
1.66 + this.roles = roles;
1.67 + }
1.68 +
1.69 + public SOAPMessage getMessage() {
1.70 + if(soapMsg == null) {
1.71 + try {
1.72 + Message m = packet.getMessage();
1.73 + soapMsg = m != null ? m.readAsSOAPMessage() : null;
1.74 + } catch (SOAPException e) {
1.75 + throw new WebServiceException(e);
1.76 + }
1.77 + }
1.78 + return soapMsg;
1.79 + }
1.80 +
1.81 + public void setMessage(SOAPMessage soapMsg) {
1.82 + try {
1.83 + this.soapMsg = soapMsg;
1.84 + } catch(Exception e) {
1.85 + throw new WebServiceException(e);
1.86 + }
1.87 + }
1.88 +
1.89 + void setPacketMessage(Message newMessage){
1.90 + if(newMessage != null) {
1.91 + packet.setMessage(newMessage);
1.92 + soapMsg = null;
1.93 + }
1.94 + }
1.95 +
1.96 + protected void updateMessage() {
1.97 + //Check if SOAPMessage has changed, if so construct new one,
1.98 + // Packet are handled through MessageContext
1.99 + if(soapMsg != null) {
1.100 + packet.setMessage(SAAJFactory.create(soapMsg));
1.101 + soapMsg = null;
1.102 + }
1.103 + }
1.104 +
1.105 + public Object[] getHeaders(QName header, JAXBContext jaxbContext, boolean allRoles) {
1.106 + SOAPVersion soapVersion = binding.getSOAPVersion();
1.107 +
1.108 + List<Object> beanList = new ArrayList<Object>();
1.109 + try {
1.110 + Iterator<Header> itr = packet.getMessage().getHeaders().getHeaders(header,false);
1.111 + if(allRoles) {
1.112 + while(itr.hasNext()) {
1.113 + beanList.add(itr.next().readAsJAXB(jaxbContext.createUnmarshaller()));
1.114 + }
1.115 + } else {
1.116 + while(itr.hasNext()) {
1.117 + Header soapHeader = itr.next();
1.118 + //Check if the role is one of the roles on this Binding
1.119 + String role = soapHeader.getRole(soapVersion);
1.120 + if(getRoles().contains(role)) {
1.121 + beanList.add(soapHeader.readAsJAXB(jaxbContext.createUnmarshaller()));
1.122 + }
1.123 + }
1.124 + }
1.125 + return beanList.toArray();
1.126 + } catch(Exception e) {
1.127 + throw new WebServiceException(e);
1.128 + }
1.129 + }
1.130 +
1.131 + public Set<String> getRoles() {
1.132 + return roles;
1.133 + }
1.134 +}
