1.1 --- /dev/null Thu Jan 01 00:00:00 1970 +0000 1.2 +++ b/src/share/jaxws_classes/com/sun/xml/internal/ws/message/jaxb/JAXBHeader.java Wed Apr 27 01:27:09 2016 +0800 1.3 @@ -0,0 +1,212 @@ 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.message.jaxb; 1.30 + 1.31 +import com.sun.istack.internal.NotNull; 1.32 +import com.sun.istack.internal.XMLStreamException2; 1.33 +import com.sun.xml.internal.bind.api.Bridge; 1.34 +import com.sun.xml.internal.stream.buffer.MutableXMLStreamBuffer; 1.35 +import com.sun.xml.internal.stream.buffer.XMLStreamBuffer; 1.36 +import com.sun.xml.internal.stream.buffer.XMLStreamBufferResult; 1.37 +import com.sun.xml.internal.ws.api.message.Header; 1.38 +import com.sun.xml.internal.ws.encoding.SOAPBindingCodec; 1.39 +import com.sun.xml.internal.ws.message.AbstractHeaderImpl; 1.40 +import com.sun.xml.internal.ws.message.RootElementSniffer; 1.41 +import com.sun.xml.internal.ws.spi.db.BindingContext; 1.42 +import com.sun.xml.internal.ws.spi.db.XMLBridge; 1.43 +import com.sun.xml.internal.ws.streaming.XMLStreamWriterUtil; 1.44 +import org.xml.sax.Attributes; 1.45 +import org.xml.sax.ContentHandler; 1.46 +import org.xml.sax.ErrorHandler; 1.47 +import org.xml.sax.SAXException; 1.48 +import org.xml.sax.SAXParseException; 1.49 + 1.50 +import javax.xml.bind.JAXBElement; 1.51 +import javax.xml.bind.JAXBException; 1.52 +import javax.xml.bind.Unmarshaller; 1.53 +import javax.xml.bind.util.JAXBResult; 1.54 +import javax.xml.namespace.QName; 1.55 +import javax.xml.soap.SOAPException; 1.56 +import javax.xml.soap.SOAPMessage; 1.57 +import javax.xml.soap.SOAPHeader; 1.58 +import javax.xml.stream.XMLStreamException; 1.59 +import javax.xml.stream.XMLStreamReader; 1.60 +import javax.xml.stream.XMLStreamWriter; 1.61 +import java.io.OutputStream; 1.62 + 1.63 +/** 1.64 + * {@link Header} whose physical data representation is a JAXB bean. 1.65 + * 1.66 + * @author Kohsuke Kawaguchi 1.67 + */ 1.68 +public final class JAXBHeader extends AbstractHeaderImpl { 1.69 + 1.70 + /** 1.71 + * The JAXB object that represents the header. 1.72 + */ 1.73 + private final Object jaxbObject; 1.74 + 1.75 + private final XMLBridge bridge; 1.76 + 1.77 + // information about this header. lazily obtained. 1.78 + private String nsUri; 1.79 + private String localName; 1.80 + private Attributes atts; 1.81 + 1.82 + /** 1.83 + * Once the header is turned into infoset, 1.84 + * this buffer keeps it. 1.85 + */ 1.86 + private XMLStreamBuffer infoset; 1.87 + 1.88 + public JAXBHeader(BindingContext context, Object jaxbObject) { 1.89 + this.jaxbObject = jaxbObject; 1.90 +// this.bridge = new MarshallerBridge(context); 1.91 + this.bridge = context.createFragmentBridge(); 1.92 + 1.93 + if (jaxbObject instanceof JAXBElement) { 1.94 + JAXBElement e = (JAXBElement) jaxbObject; 1.95 + this.nsUri = e.getName().getNamespaceURI(); 1.96 + this.localName = e.getName().getLocalPart(); 1.97 + } 1.98 + } 1.99 + 1.100 + public JAXBHeader(XMLBridge bridge, Object jaxbObject) { 1.101 + this.jaxbObject = jaxbObject; 1.102 + this.bridge = bridge; 1.103 + 1.104 + QName tagName = bridge.getTypeInfo().tagName; 1.105 + this.nsUri = tagName.getNamespaceURI(); 1.106 + this.localName = tagName.getLocalPart(); 1.107 + } 1.108 + 1.109 + /** 1.110 + * Lazily parse the first element to obtain attribute values on it. 1.111 + */ 1.112 + private void parse() { 1.113 + RootElementSniffer sniffer = new RootElementSniffer(); 1.114 + try { 1.115 + bridge.marshal(jaxbObject,sniffer,null); 1.116 + } catch (JAXBException e) { 1.117 + // if it's due to us aborting the processing after the first element, 1.118 + // we can safely ignore this exception. 1.119 + // 1.120 + // if it's due to error in the object, the same error will be reported 1.121 + // when the readHeader() method is used, so we don't have to report 1.122 + // an error right now. 1.123 + nsUri = sniffer.getNsUri(); 1.124 + localName = sniffer.getLocalName(); 1.125 + atts = sniffer.getAttributes(); 1.126 + } 1.127 + } 1.128 + 1.129 + 1.130 + public @NotNull String getNamespaceURI() { 1.131 + if(nsUri==null) 1.132 + parse(); 1.133 + return nsUri; 1.134 + } 1.135 + 1.136 + public @NotNull String getLocalPart() { 1.137 + if(localName==null) 1.138 + parse(); 1.139 + return localName; 1.140 + } 1.141 + 1.142 + public String getAttribute(String nsUri, String localName) { 1.143 + if(atts==null) 1.144 + parse(); 1.145 + return atts.getValue(nsUri,localName); 1.146 + } 1.147 + 1.148 + public XMLStreamReader readHeader() throws XMLStreamException { 1.149 + if(infoset==null) { 1.150 + MutableXMLStreamBuffer buffer = new MutableXMLStreamBuffer(); 1.151 + writeTo(buffer.createFromXMLStreamWriter()); 1.152 + infoset = buffer; 1.153 + } 1.154 + return infoset.readAsXMLStreamReader(); 1.155 + } 1.156 + 1.157 + public <T> T readAsJAXB(Unmarshaller unmarshaller) throws JAXBException { 1.158 + try { 1.159 + JAXBResult r = new JAXBResult(unmarshaller); 1.160 + // bridge marshals a fragment, so we need to add start/endDocument by ourselves 1.161 + r.getHandler().startDocument(); 1.162 + bridge.marshal(jaxbObject,r); 1.163 + r.getHandler().endDocument(); 1.164 + return (T)r.getResult(); 1.165 + } catch (SAXException e) { 1.166 + throw new JAXBException(e); 1.167 + } 1.168 + } 1.169 + /** @deprecated */ 1.170 + public <T> T readAsJAXB(Bridge<T> bridge) throws JAXBException { 1.171 + return bridge.unmarshal(new JAXBBridgeSource(this.bridge,jaxbObject)); 1.172 + } 1.173 + 1.174 + public <T> T readAsJAXB(XMLBridge<T> bond) throws JAXBException { 1.175 + return bond.unmarshal(new JAXBBridgeSource(this.bridge,jaxbObject),null); 1.176 + } 1.177 + 1.178 + public void writeTo(XMLStreamWriter sw) throws XMLStreamException { 1.179 + try { 1.180 + // Get the encoding of the writer 1.181 + String encoding = XMLStreamWriterUtil.getEncoding(sw); 1.182 + 1.183 + // Get output stream and use JAXB UTF-8 writer 1.184 + OutputStream os = bridge.supportOutputStream() ? XMLStreamWriterUtil.getOutputStream(sw) : null; 1.185 + if (os != null && encoding != null && encoding.equalsIgnoreCase(SOAPBindingCodec.UTF8_ENCODING)) { 1.186 + bridge.marshal(jaxbObject, os, sw.getNamespaceContext(), null); 1.187 + } else { 1.188 + bridge.marshal(jaxbObject,sw, null); 1.189 + } 1.190 + } catch (JAXBException e) { 1.191 + throw new XMLStreamException2(e); 1.192 + } 1.193 + } 1.194 + 1.195 + public void writeTo(SOAPMessage saaj) throws SOAPException { 1.196 + try { 1.197 + SOAPHeader header = saaj.getSOAPHeader(); 1.198 + if (header == null) 1.199 + header = saaj.getSOAPPart().getEnvelope().addHeader(); 1.200 + bridge.marshal(jaxbObject,header); 1.201 + } catch (JAXBException e) { 1.202 + throw new SOAPException(e); 1.203 + } 1.204 + } 1.205 + 1.206 + public void writeTo(ContentHandler contentHandler, ErrorHandler errorHandler) throws SAXException { 1.207 + try { 1.208 + bridge.marshal(jaxbObject,contentHandler,null); 1.209 + } catch (JAXBException e) { 1.210 + SAXParseException x = new SAXParseException(e.getMessage(),null,null,-1,-1,e); 1.211 + errorHandler.fatalError(x); 1.212 + throw x; 1.213 + } 1.214 + } 1.215 +}