Wed, 27 Apr 2016 01:27:09 +0800
Initial load
http://hg.openjdk.java.net/jdk8u/jdk8u/jaxws/
changeset: 657:d47a47f961ee
tag: jdk8u25-b17
1 /*
2 * Copyright (c) 1997, 2013, Oracle and/or its affiliates. All rights reserved.
3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
4 *
5 * This code is free software; you can redistribute it and/or modify it
6 * under the terms of the GNU General Public License version 2 only, as
7 * published by the Free Software Foundation. Oracle designates this
8 * particular file as subject to the "Classpath" exception as provided
9 * by Oracle in the LICENSE file that accompanied this code.
10 *
11 * This code is distributed in the hope that it will be useful, but WITHOUT
12 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
13 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
14 * version 2 for more details (a copy is included in the LICENSE file that
15 * accompanied this code).
16 *
17 * You should have received a copy of the GNU General Public License version
18 * 2 along with this work; if not, write to the Free Software Foundation,
19 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
20 *
21 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
22 * or visit www.oracle.com if you need additional information or have any
23 * questions.
24 */
26 package com.sun.xml.internal.ws.handler;
27 import com.sun.xml.internal.ws.api.message.Header;
28 import com.sun.xml.internal.ws.api.message.Message;
29 import com.sun.xml.internal.ws.api.message.Packet;
30 import com.sun.xml.internal.ws.api.message.saaj.SAAJFactory;
31 import com.sun.xml.internal.ws.api.WSBinding;
32 import com.sun.xml.internal.ws.api.SOAPVersion;
34 import javax.xml.bind.JAXBContext;
35 import javax.xml.namespace.QName;
36 import javax.xml.soap.SOAPException;
37 import javax.xml.soap.SOAPMessage;
38 import javax.xml.ws.WebServiceException;
39 import javax.xml.ws.handler.soap.SOAPMessageContext;
41 import java.util.ArrayList;
42 import java.util.Iterator;
43 import java.util.List;
44 import java.util.Set;
46 /**
47 * Implementation of {@link SOAPMessageContext}. This class is used at runtime
48 * to pass to the handlers for processing soap messages.
49 *
50 * @see MessageContextImpl
51 *
52 * @author WS Development Team
53 */
54 public class SOAPMessageContextImpl extends MessageUpdatableContext implements SOAPMessageContext {
56 private Set<String> roles;
57 private SOAPMessage soapMsg = null;
58 private WSBinding binding;
60 public SOAPMessageContextImpl(WSBinding binding, Packet packet,Set<String> roles) {
61 super(packet);
62 this.binding = binding;
63 this.roles = roles;
64 }
66 public SOAPMessage getMessage() {
67 if(soapMsg == null) {
68 try {
69 Message m = packet.getMessage();
70 soapMsg = m != null ? m.readAsSOAPMessage() : null;
71 } catch (SOAPException e) {
72 throw new WebServiceException(e);
73 }
74 }
75 return soapMsg;
76 }
78 public void setMessage(SOAPMessage soapMsg) {
79 try {
80 this.soapMsg = soapMsg;
81 } catch(Exception e) {
82 throw new WebServiceException(e);
83 }
84 }
86 void setPacketMessage(Message newMessage){
87 if(newMessage != null) {
88 packet.setMessage(newMessage);
89 soapMsg = null;
90 }
91 }
93 protected void updateMessage() {
94 //Check if SOAPMessage has changed, if so construct new one,
95 // Packet are handled through MessageContext
96 if(soapMsg != null) {
97 packet.setMessage(SAAJFactory.create(soapMsg));
98 soapMsg = null;
99 }
100 }
102 public Object[] getHeaders(QName header, JAXBContext jaxbContext, boolean allRoles) {
103 SOAPVersion soapVersion = binding.getSOAPVersion();
105 List<Object> beanList = new ArrayList<Object>();
106 try {
107 Iterator<Header> itr = packet.getMessage().getHeaders().getHeaders(header,false);
108 if(allRoles) {
109 while(itr.hasNext()) {
110 beanList.add(itr.next().readAsJAXB(jaxbContext.createUnmarshaller()));
111 }
112 } else {
113 while(itr.hasNext()) {
114 Header soapHeader = itr.next();
115 //Check if the role is one of the roles on this Binding
116 String role = soapHeader.getRole(soapVersion);
117 if(getRoles().contains(role)) {
118 beanList.add(soapHeader.readAsJAXB(jaxbContext.createUnmarshaller()));
119 }
120 }
121 }
122 return beanList.toArray();
123 } catch(Exception e) {
124 throw new WebServiceException(e);
125 }
126 }
128 public Set<String> getRoles() {
129 return roles;
130 }
131 }