jdk8-mips64-public/jaxws: src/share/jaxws_classes/com/sun/xml/internal/ws/message/jaxb/JAXBMessage.java@9c07ef4934dd (annotated)

src/share/jaxws_classes/com/sun/xml/internal/ws/message/jaxb/JAXBMessage.java@9c07ef4934dd (annotated)

src/share/jaxws_classes/com/sun/xml/internal/ws/message/jaxb/JAXBMessage.java

Thu, 31 Aug 2017 15:18:52 +0800

author: aoqi
date: Thu, 31 Aug 2017 15:18:52 +0800
changeset 637: 9c07ef4934dd
parent 397: b99d7e355d4b
parent 0: 373ffda63c9a
permissions: -rw-r--r--

merge

 /*
  * Copyright (c) 1997, 2013, 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.message.jaxb;
 import com.sun.istack.internal.FragmentContentHandler;
 import com.sun.xml.internal.stream.buffer.MutableXMLStreamBuffer;
 import com.sun.xml.internal.stream.buffer.XMLStreamBuffer;
 import com.sun.xml.internal.stream.buffer.XMLStreamBufferResult;
 import com.sun.xml.internal.ws.api.SOAPVersion;
 import com.sun.xml.internal.ws.api.message.AttachmentSet;
 import com.sun.xml.internal.ws.api.message.Header;
 import com.sun.xml.internal.ws.api.message.HeaderList;
 import com.sun.xml.internal.ws.api.message.Message;
 import com.sun.xml.internal.ws.api.message.MessageHeaders;
 import com.sun.xml.internal.ws.api.message.StreamingSOAP;
 import com.sun.xml.internal.ws.encoding.SOAPBindingCodec;
 import com.sun.xml.internal.ws.message.AbstractMessageImpl;
 import com.sun.xml.internal.ws.message.AttachmentSetImpl;
 import com.sun.xml.internal.ws.message.RootElementSniffer;
 import com.sun.xml.internal.ws.message.stream.StreamMessage;
 import com.sun.xml.internal.ws.spi.db.BindingContext;
 import com.sun.xml.internal.ws.spi.db.BindingContextFactory;
 import com.sun.xml.internal.ws.spi.db.XMLBridge;
 import com.sun.xml.internal.ws.streaming.XMLStreamWriterUtil;
 import com.sun.xml.internal.ws.streaming.XMLStreamReaderUtil;
 import com.sun.xml.internal.ws.streaming.MtomStreamWriter;
 import com.sun.xml.internal.ws.util.xml.XMLReaderComposite;
 import com.sun.xml.internal.ws.util.xml.XMLReaderComposite.ElemInfo;
 import org.xml.sax.ContentHandler;
 import org.xml.sax.ErrorHandler;
 import org.xml.sax.SAXException;
 import javax.xml.bind.JAXBContext;
 import javax.xml.bind.JAXBElement;
 import javax.xml.bind.JAXBException;
 import javax.xml.bind.Marshaller;
 import javax.xml.bind.Unmarshaller;
 import javax.xml.bind.attachment.AttachmentMarshaller;
 import javax.xml.bind.annotation.XmlRootElement;
 import javax.xml.bind.util.JAXBResult;
 import javax.xml.namespace.QName;
 import javax.xml.stream.XMLStreamException;
 import javax.xml.stream.XMLStreamReader;
 import javax.xml.stream.XMLStreamWriter;
 import static javax.xml.stream.XMLStreamConstants.START_DOCUMENT;
 import javax.xml.transform.Source;
 import javax.xml.ws.WebServiceException;
 import java.io.OutputStream;
 import java.util.List;
 /**
  * {@link Message} backed by a JAXB bean.
  *
  * @author Kohsuke Kawaguchi
  */
 public final class JAXBMessage extends AbstractMessageImpl implements StreamingSOAP {
     private MessageHeaders headers;
     /**
      * The JAXB object that represents the payload.
      */
     private final Object jaxbObject;
     private final XMLBridge bridge;
     /**
      * For the use case of a user-supplied JAXB context that is not
      * a known JAXB type, as when creating a Disaptch object with a
      * JAXB object parameter, we will marshal and unmarshal directly with
      * the context object, as there is no Bond available.  In this case,
      * swaRef is not supported.
      */
     private final JAXBContext rawContext;
     /**
      * Lazily sniffed payload element name
      */
     private String nsUri,localName;
     /**
      * If we have the infoset representation for the payload, this field is non-null.
      */
     private XMLStreamBuffer infoset;
     public static Message create(BindingContext context, Object jaxbObject, SOAPVersion soapVersion, MessageHeaders headers, AttachmentSet attachments) {
         if(!context.hasSwaRef()) {
             return new JAXBMessage(context,jaxbObject,soapVersion,headers,attachments);
         }
         // If we have swaRef, then that means we might have attachments.
         // to comply with the packet API, we need to eagerly turn the JAXB object into infoset
         // to correctly find out about attachments.
         try {
             MutableXMLStreamBuffer xsb = new MutableXMLStreamBuffer();
             Marshaller m = context.createMarshaller();
             AttachmentMarshallerImpl am = new AttachmentMarshallerImpl(attachments);
             m.setAttachmentMarshaller(am);
             am.cleanup();
             m.marshal(jaxbObject,xsb.createFromXMLStreamWriter());
             // any way to reuse this XMLStreamBuffer in StreamMessage?
             return new StreamMessage(headers,attachments,xsb.readAsXMLStreamReader(),soapVersion);
         } catch (JAXBException e) {
             throw new WebServiceException(e);
         } catch (XMLStreamException e) {
             throw new WebServiceException(e);
         }
     }
     /**
      * Creates a {@link Message} backed by a JAXB bean.
      *
      * @param context
      *      The JAXBContext to be used for marshalling.
      * @param jaxbObject
      *      The JAXB object that represents the payload. must not be null. This object
      *      must be bound to an element (which means it either is a {@link JAXBElement} or
      *      an instanceof a class with {@link XmlRootElement}).
      * @param soapVersion
      *      The SOAP version of the message. Must not be null.
      */
     public static Message create(BindingContext context, Object jaxbObject, SOAPVersion soapVersion) {
         return create(context,jaxbObject,soapVersion,null,null);
     }
     /** @deprecated */
     public static Message create(JAXBContext context, Object jaxbObject, SOAPVersion soapVersion) {
         return create(BindingContextFactory.create(context),jaxbObject,soapVersion,null,null);
     }
     /**
      * @deprecated
      * For use when creating a Dispatch object with an unknown JAXB implementation
      * for he JAXBContext parameter.
      *
      */
     public static Message createRaw(JAXBContext context, Object jaxbObject, SOAPVersion soapVersion) {
         return new JAXBMessage(context,jaxbObject,soapVersion,null,null);
     }
     private JAXBMessage( BindingContext context, Object jaxbObject, SOAPVersion soapVer, MessageHeaders headers, AttachmentSet attachments ) {
         super(soapVer);
 //        this.bridge = new MarshallerBridge(context);
         this.bridge = context.createFragmentBridge();
         this.rawContext = null;
         this.jaxbObject = jaxbObject;
         this.headers = headers;
         this.attachmentSet = attachments;
     }
     private JAXBMessage( JAXBContext rawContext, Object jaxbObject, SOAPVersion soapVer, MessageHeaders headers, AttachmentSet attachments ) {
         super(soapVer);
 //        this.bridge = new MarshallerBridge(context);
         this.rawContext = rawContext;
         this.bridge = null;
         this.jaxbObject = jaxbObject;
         this.headers = headers;
         this.attachmentSet = attachments;
     }
     /**
      * Creates a {@link Message} backed by a JAXB bean.
      *
      * @param bridge
      *      Specify the payload tag name and how <tt>jaxbObject</tt> is bound.
      * @param jaxbObject
      */
     public static Message create(XMLBridge bridge, Object jaxbObject, SOAPVersion soapVer) {
         if(!bridge.context().hasSwaRef()) {
             return new JAXBMessage(bridge,jaxbObject,soapVer);
         }
         // If we have swaRef, then that means we might have attachments.
         // to comply with the packet API, we need to eagerly turn the JAXB object into infoset
         // to correctly find out about attachments.
         try {
             MutableXMLStreamBuffer xsb = new MutableXMLStreamBuffer();
             AttachmentSetImpl attachments = new AttachmentSetImpl();
             AttachmentMarshallerImpl am = new AttachmentMarshallerImpl(attachments);
             bridge.marshal(jaxbObject,xsb.createFromXMLStreamWriter(), am);
             am.cleanup();
             // any way to reuse this XMLStreamBuffer in StreamMessage?
             return new StreamMessage(null,attachments,xsb.readAsXMLStreamReader(),soapVer);
         } catch (JAXBException e) {
             throw new WebServiceException(e);
         } catch (XMLStreamException e) {
             throw new WebServiceException(e);
         }
     }
     private JAXBMessage(XMLBridge bridge, Object jaxbObject, SOAPVersion soapVer) {
         super(soapVer);
         // TODO: think about a better way to handle BridgeContext
         this.bridge = bridge;
         this.rawContext = null;
         this.jaxbObject = jaxbObject;
         QName tagName = bridge.getTypeInfo().tagName;
         this.nsUri = tagName.getNamespaceURI();
         this.localName = tagName.getLocalPart();
         this.attachmentSet = new AttachmentSetImpl();
     }
     /**
      * Copy constructor.
      */
     public JAXBMessage(JAXBMessage that) {
         super(that);
         this.headers = that.headers;
         if(this.headers!=null)
             this.headers = new HeaderList(this.headers);
         this.attachmentSet = that.attachmentSet;
         this.jaxbObject = that.jaxbObject;
         this.bridge = that.bridge;
         this.rawContext = that.rawContext;
     }
     @Override
     public boolean hasHeaders() {
         return headers!=null && headers.hasHeaders();
     }
     @Override
     public MessageHeaders getHeaders() {
         if(headers==null)
             headers = new HeaderList(getSOAPVersion());
         return headers;
     }
     @Override
     public String getPayloadLocalPart() {
         if(localName==null)
             sniff();
         return localName;
     }
     @Override
     public String getPayloadNamespaceURI() {
         if(nsUri==null)
             sniff();
         return nsUri;
     }
     @Override
     public boolean hasPayload() {
         return true;
     }
     /**
      * Obtains the tag name of the root element.
      */
     private void sniff() {
         RootElementSniffer sniffer = new RootElementSniffer(false);
         try {
                 if (rawContext != null) {
                         Marshaller m = rawContext.createMarshaller();
                         m.setProperty("jaxb.fragment", Boolean.TRUE);
                         m.marshal(jaxbObject,sniffer);
                 } else
                         bridge.marshal(jaxbObject,sniffer,null);
         } catch (JAXBException e) {
             // if it's due to us aborting the processing after the first element,
             // we can safely ignore this exception.
             //
             // if it's due to error in the object, the same error will be reported
             // when the readHeader() method is used, so we don't have to report
             // an error right now.
             nsUri = sniffer.getNsUri();
             localName = sniffer.getLocalName();
         }
     }
     @Override
     public Source readPayloadAsSource() {
         return new JAXBBridgeSource(bridge,jaxbObject);
     }
     @Override
     public <T> T readPayloadAsJAXB(Unmarshaller unmarshaller) throws JAXBException {
         JAXBResult out = new JAXBResult(unmarshaller);
         // since the bridge only produces fragments, we need to fire start/end document.
         try {
             out.getHandler().startDocument();
             if (rawContext != null) {
                 Marshaller m = rawContext.createMarshaller();
                 m.setProperty("jaxb.fragment", Boolean.TRUE);
                 m.marshal(jaxbObject,out);
             } else
                 bridge.marshal(jaxbObject,out);
             out.getHandler().endDocument();
         } catch (SAXException e) {
             throw new JAXBException(e);
         }
         return (T)out.getResult();
     }
     @Override
     public XMLStreamReader readPayload() throws XMLStreamException {
        try {
             if(infoset==null) {
                                 if (rawContext != null) {
                         XMLStreamBufferResult sbr = new XMLStreamBufferResult();
                                         Marshaller m = rawContext.createMarshaller();
                                         m.setProperty("jaxb.fragment", Boolean.TRUE);
                                         m.marshal(jaxbObject, sbr);
                         infoset = sbr.getXMLStreamBuffer();
                                 } else {
                                     MutableXMLStreamBuffer buffer = new MutableXMLStreamBuffer();
                                     writePayloadTo(buffer.createFromXMLStreamWriter());
                                     infoset = buffer;
                                 }
             }
             XMLStreamReader reader = infoset.readAsXMLStreamReader();
             if(reader.getEventType()== START_DOCUMENT)
                 XMLStreamReaderUtil.nextElementContent(reader);
             return reader;
         } catch (JAXBException e) {
            // bug 6449684, spec 4.3.4
            throw new WebServiceException(e);
         }
     }
     /**
      * Writes the payload as SAX events.
      */
     @Override
     protected void writePayloadTo(ContentHandler contentHandler, ErrorHandler errorHandler, boolean fragment) throws SAXException {
         try {
             if(fragment)
                 contentHandler = new FragmentContentHandler(contentHandler);
             AttachmentMarshallerImpl am = new AttachmentMarshallerImpl(attachmentSet);
             if (rawContext != null) {
                 Marshaller m = rawContext.createMarshaller();
                 m.setProperty("jaxb.fragment", Boolean.TRUE);
                 m.setAttachmentMarshaller(am);
                 m.marshal(jaxbObject,contentHandler);
             } else
                 bridge.marshal(jaxbObject,contentHandler, am);
             am.cleanup();
         } catch (JAXBException e) {
             // this is really more helpful but spec compliance
             // errorHandler.fatalError(new SAXParseException(e.getMessage(),NULL_LOCATOR,e));
             // bug 6449684, spec 4.3.4
             throw new WebServiceException(e.getMessage(),e);
         }
     }
     @Override
     public void writePayloadTo(XMLStreamWriter sw) throws XMLStreamException {
         try {
             // MtomCodec sets its own AttachmentMarshaller
             AttachmentMarshaller am = (sw instanceof MtomStreamWriter)
                     ? ((MtomStreamWriter)sw).getAttachmentMarshaller()
                     : new AttachmentMarshallerImpl(attachmentSet);
             // Get the encoding of the writer
             String encoding = XMLStreamWriterUtil.getEncoding(sw);
             // Get output stream and use JAXB UTF-8 writer
             OutputStream os = bridge.supportOutputStream() ? XMLStreamWriterUtil.getOutputStream(sw) : null;
             if (rawContext != null) {
                 Marshaller m = rawContext.createMarshaller();
                 m.setProperty("jaxb.fragment", Boolean.TRUE);
                 m.setAttachmentMarshaller(am);
                 if (os != null) {
                     m.marshal(jaxbObject, os);
                 } else {
                     m.marshal(jaxbObject, sw);
                 }
             } else {
                 if (os != null && encoding != null && encoding.equalsIgnoreCase(SOAPBindingCodec.UTF8_ENCODING)) {
                     bridge.marshal(jaxbObject, os, sw.getNamespaceContext(), am);
                 } else {
                     bridge.marshal(jaxbObject, sw, am);
                 }
             }
             //cleanup() is not needed since JAXB doesn't keep ref to AttachmentMarshaller
             //am.cleanup();
         } catch (JAXBException e) {
             // bug 6449684, spec 4.3.4
             throw new WebServiceException(e);
         }
     }
     @Override
     public Message copy() {
         return new JAXBMessage(this);
     }
     public XMLStreamReader readEnvelope() {
         int base = soapVersion.ordinal()*3;
         this.envelopeTag = DEFAULT_TAGS.get(base);
         this.bodyTag = DEFAULT_TAGS.get(base+2);
         List<XMLStreamReader> hReaders = new java.util.ArrayList<XMLStreamReader>();
         ElemInfo envElem =  new ElemInfo(envelopeTag, null);
         ElemInfo bdyElem =  new ElemInfo(bodyTag, envElem);
         for (Header h : getHeaders().asList()) {
             try {
                 hReaders.add(h.readHeader());
             } catch (XMLStreamException e) {
                 throw new RuntimeException(e);
             }
         }
         XMLStreamReader soapHeader = null;
         if(hReaders.size()>0) {
             headerTag = DEFAULT_TAGS.get(base+1);
             ElemInfo hdrElem = new ElemInfo(headerTag, envElem);
             soapHeader = new XMLReaderComposite(hdrElem, hReaders.toArray(new XMLStreamReader[hReaders.size()]));
         }
         try {
             XMLStreamReader payload= readPayload();
             XMLStreamReader soapBody = new XMLReaderComposite(bdyElem, new XMLStreamReader[]{payload});
             XMLStreamReader[] soapContent = (soapHeader != null) ? new XMLStreamReader[]{soapHeader, soapBody} : new XMLStreamReader[]{soapBody};
             return new XMLReaderComposite(envElem, soapContent);
         } catch (XMLStreamException e) {
             throw new RuntimeException(e);
         }
     }
 }

Mercurial > jdk8-mips64-public > jaxws / annotate

src/share/jaxws_classes/com/sun/xml/internal/ws/message/jaxb/JAXBMessage.java@9c07ef4934dd (annotated)

src/share/jaxws_classes/com/sun/xml/internal/ws/message/jaxb/JAXBMessage.java