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

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

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

merge

aoqi@0 1 /*
aoqi@0 2 * Copyright (c) 1997, 2013, Oracle and/or its affiliates. All rights reserved.
aoqi@0 3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
aoqi@0 4 *
aoqi@0 5 * This code is free software; you can redistribute it and/or modify it
aoqi@0 6 * under the terms of the GNU General Public License version 2 only, as
aoqi@0 7 * published by the Free Software Foundation. Oracle designates this
aoqi@0 8 * particular file as subject to the "Classpath" exception as provided
aoqi@0 9 * by Oracle in the LICENSE file that accompanied this code.
aoqi@0 10 *
aoqi@0 11 * This code is distributed in the hope that it will be useful, but WITHOUT
aoqi@0 12 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
aoqi@0 13 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
aoqi@0 14 * version 2 for more details (a copy is included in the LICENSE file that
aoqi@0 15 * accompanied this code).
aoqi@0 16 *
aoqi@0 17 * You should have received a copy of the GNU General Public License version
aoqi@0 18 * 2 along with this work; if not, write to the Free Software Foundation,
aoqi@0 19 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
aoqi@0 20 *
aoqi@0 21 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
aoqi@0 22 * or visit www.oracle.com if you need additional information or have any
aoqi@0 23 * questions.
aoqi@0 24 */
aoqi@0 25
aoqi@0 26 package com.sun.xml.internal.ws.message;
aoqi@0 27
aoqi@0 28 import com.sun.istack.internal.FragmentContentHandler;
aoqi@0 29 import com.sun.xml.internal.bind.api.Bridge;
aoqi@0 30 import com.sun.xml.internal.bind.unmarshaller.DOMScanner;
aoqi@0 31 import com.sun.xml.internal.ws.api.SOAPVersion;
aoqi@0 32 import com.sun.xml.internal.ws.api.message.HeaderList;
aoqi@0 33 import com.sun.xml.internal.ws.api.message.Message;
aoqi@0 34 import com.sun.xml.internal.ws.api.message.AttachmentSet;
aoqi@0 35 import com.sun.xml.internal.ws.api.message.MessageHeaders;
aoqi@0 36 import com.sun.xml.internal.ws.streaming.DOMStreamReader;
aoqi@0 37 import com.sun.xml.internal.ws.util.DOMUtil;
aoqi@0 38 import org.w3c.dom.Element;
aoqi@0 39 import org.xml.sax.ContentHandler;
aoqi@0 40 import org.xml.sax.ErrorHandler;
aoqi@0 41 import org.xml.sax.SAXException;
aoqi@0 42
aoqi@0 43 import javax.xml.bind.JAXBException;
aoqi@0 44 import javax.xml.bind.Unmarshaller;
aoqi@0 45 import javax.xml.stream.XMLStreamException;
aoqi@0 46 import javax.xml.stream.XMLStreamReader;
aoqi@0 47 import javax.xml.stream.XMLStreamWriter;
aoqi@0 48 import javax.xml.transform.Source;
aoqi@0 49 import javax.xml.transform.dom.DOMSource;
aoqi@0 50 import javax.xml.ws.WebServiceException;
aoqi@0 51
aoqi@0 52 /**
aoqi@0 53 * {@link Message} backed by a DOM {@link Element} that represents the payload.
aoqi@0 54 *
aoqi@0 55 * @author Kohsuke Kawaguchi
aoqi@0 56 */
aoqi@0 57 public final class DOMMessage extends AbstractMessageImpl {
aoqi@0 58 private MessageHeaders headers;
aoqi@0 59 private final Element payload;
aoqi@0 60
aoqi@0 61 public DOMMessage(SOAPVersion ver, Element payload) {
aoqi@0 62 this(ver,null,payload);
aoqi@0 63 }
aoqi@0 64
aoqi@0 65 public DOMMessage(SOAPVersion ver, MessageHeaders headers, Element payload) {
aoqi@0 66 this(ver,headers,payload,null);
aoqi@0 67 }
aoqi@0 68
aoqi@0 69 public DOMMessage(SOAPVersion ver, MessageHeaders headers, Element payload, AttachmentSet attachments) {
aoqi@0 70 super(ver);
aoqi@0 71 this.headers = headers;
aoqi@0 72 this.payload = payload;
aoqi@0 73 this.attachmentSet = attachments;
aoqi@0 74 assert payload!=null;
aoqi@0 75 }
aoqi@0 76 /**
aoqi@0 77 * This constructor is a convenience and called by the {@link #copy}
aoqi@0 78 */
aoqi@0 79 private DOMMessage(DOMMessage that) {
aoqi@0 80 super(that);
aoqi@0 81 this.headers = HeaderList.copy(that.headers);
aoqi@0 82 this.payload = that.payload;
aoqi@0 83 }
aoqi@0 84
aoqi@0 85 public boolean hasHeaders() {
aoqi@0 86 return getHeaders().hasHeaders();
aoqi@0 87 }
aoqi@0 88
aoqi@0 89 public MessageHeaders getHeaders() {
aoqi@0 90 if (headers == null)
aoqi@0 91 headers = new HeaderList(getSOAPVersion());
aoqi@0 92
aoqi@0 93 return headers;
aoqi@0 94 }
aoqi@0 95
aoqi@0 96 public String getPayloadLocalPart() {
aoqi@0 97 return payload.getLocalName();
aoqi@0 98 }
aoqi@0 99
aoqi@0 100 public String getPayloadNamespaceURI() {
aoqi@0 101 return payload.getNamespaceURI();
aoqi@0 102 }
aoqi@0 103
aoqi@0 104 public boolean hasPayload() {
aoqi@0 105 return true;
aoqi@0 106 }
aoqi@0 107
aoqi@0 108 public Source readPayloadAsSource() {
aoqi@0 109 return new DOMSource(payload);
aoqi@0 110 }
aoqi@0 111
aoqi@0 112 public <T> T readPayloadAsJAXB(Unmarshaller unmarshaller) throws JAXBException {
aoqi@0 113 if(hasAttachments())
aoqi@0 114 unmarshaller.setAttachmentUnmarshaller(new AttachmentUnmarshallerImpl(getAttachments()));
aoqi@0 115 try {
aoqi@0 116 return (T)unmarshaller.unmarshal(payload);
aoqi@0 117 } finally{
aoqi@0 118 unmarshaller.setAttachmentUnmarshaller(null);
aoqi@0 119 }
aoqi@0 120 }
aoqi@0 121 /** @deprecated */
aoqi@0 122 public <T> T readPayloadAsJAXB(Bridge<T> bridge) throws JAXBException {
aoqi@0 123 return bridge.unmarshal(payload,
aoqi@0 124 hasAttachments()? new AttachmentUnmarshallerImpl(getAttachments()) : null);
aoqi@0 125 }
aoqi@0 126
aoqi@0 127 public XMLStreamReader readPayload() throws XMLStreamException {
aoqi@0 128 DOMStreamReader dss = new DOMStreamReader();
aoqi@0 129 dss.setCurrentNode(payload);
aoqi@0 130 dss.nextTag();
aoqi@0 131 assert dss.getEventType()==XMLStreamReader.START_ELEMENT;
aoqi@0 132 return dss;
aoqi@0 133 }
aoqi@0 134
aoqi@0 135 public void writePayloadTo(XMLStreamWriter sw) {
aoqi@0 136 try {
aoqi@0 137 if (payload != null)
aoqi@0 138 DOMUtil.serializeNode(payload, sw);
aoqi@0 139 } catch (XMLStreamException e) {
aoqi@0 140 throw new WebServiceException(e);
aoqi@0 141 }
aoqi@0 142 }
aoqi@0 143
aoqi@0 144 protected void writePayloadTo(ContentHandler contentHandler, ErrorHandler errorHandler, boolean fragment) throws SAXException {
aoqi@0 145 if(fragment)
aoqi@0 146 contentHandler = new FragmentContentHandler(contentHandler);
aoqi@0 147 DOMScanner ds = new DOMScanner();
aoqi@0 148 ds.setContentHandler(contentHandler);
aoqi@0 149 ds.scan(payload);
aoqi@0 150 }
aoqi@0 151
aoqi@0 152 public Message copy() {
aoqi@0 153 return new DOMMessage(this);
aoqi@0 154 }
aoqi@0 155
aoqi@0 156 }

mercurial