aoqi@0: /* aoqi@0: * Copyright (c) 1997, 2013, Oracle and/or its affiliates. All rights reserved. aoqi@0: * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. aoqi@0: * aoqi@0: * This code is free software; you can redistribute it and/or modify it aoqi@0: * under the terms of the GNU General Public License version 2 only, as aoqi@0: * published by the Free Software Foundation. Oracle designates this aoqi@0: * particular file as subject to the "Classpath" exception as provided aoqi@0: * by Oracle in the LICENSE file that accompanied this code. aoqi@0: * aoqi@0: * This code is distributed in the hope that it will be useful, but WITHOUT aoqi@0: * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or aoqi@0: * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License aoqi@0: * version 2 for more details (a copy is included in the LICENSE file that aoqi@0: * accompanied this code). aoqi@0: * aoqi@0: * You should have received a copy of the GNU General Public License version aoqi@0: * 2 along with this work; if not, write to the Free Software Foundation, aoqi@0: * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. aoqi@0: * aoqi@0: * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA aoqi@0: * or visit www.oracle.com if you need additional information or have any aoqi@0: * questions. aoqi@0: */ aoqi@0: aoqi@0: package com.sun.xml.internal.ws.message; aoqi@0: aoqi@0: import com.sun.istack.internal.FragmentContentHandler; aoqi@0: import com.sun.xml.internal.bind.api.Bridge; aoqi@0: import com.sun.xml.internal.bind.unmarshaller.DOMScanner; aoqi@0: import com.sun.xml.internal.ws.api.SOAPVersion; aoqi@0: import com.sun.xml.internal.ws.api.message.HeaderList; aoqi@0: import com.sun.xml.internal.ws.api.message.Message; aoqi@0: import com.sun.xml.internal.ws.api.message.AttachmentSet; aoqi@0: import com.sun.xml.internal.ws.api.message.MessageHeaders; aoqi@0: import com.sun.xml.internal.ws.streaming.DOMStreamReader; aoqi@0: import com.sun.xml.internal.ws.util.DOMUtil; aoqi@0: import org.w3c.dom.Element; aoqi@0: import org.xml.sax.ContentHandler; aoqi@0: import org.xml.sax.ErrorHandler; aoqi@0: import org.xml.sax.SAXException; aoqi@0: aoqi@0: import javax.xml.bind.JAXBException; aoqi@0: import javax.xml.bind.Unmarshaller; aoqi@0: import javax.xml.stream.XMLStreamException; aoqi@0: import javax.xml.stream.XMLStreamReader; aoqi@0: import javax.xml.stream.XMLStreamWriter; aoqi@0: import javax.xml.transform.Source; aoqi@0: import javax.xml.transform.dom.DOMSource; aoqi@0: import javax.xml.ws.WebServiceException; aoqi@0: aoqi@0: /** aoqi@0: * {@link Message} backed by a DOM {@link Element} that represents the payload. aoqi@0: * aoqi@0: * @author Kohsuke Kawaguchi aoqi@0: */ aoqi@0: public final class DOMMessage extends AbstractMessageImpl { aoqi@0: private MessageHeaders headers; aoqi@0: private final Element payload; aoqi@0: aoqi@0: public DOMMessage(SOAPVersion ver, Element payload) { aoqi@0: this(ver,null,payload); aoqi@0: } aoqi@0: aoqi@0: public DOMMessage(SOAPVersion ver, MessageHeaders headers, Element payload) { aoqi@0: this(ver,headers,payload,null); aoqi@0: } aoqi@0: aoqi@0: public DOMMessage(SOAPVersion ver, MessageHeaders headers, Element payload, AttachmentSet attachments) { aoqi@0: super(ver); aoqi@0: this.headers = headers; aoqi@0: this.payload = payload; aoqi@0: this.attachmentSet = attachments; aoqi@0: assert payload!=null; aoqi@0: } aoqi@0: /** aoqi@0: * This constructor is a convenience and called by the {@link #copy} aoqi@0: */ aoqi@0: private DOMMessage(DOMMessage that) { aoqi@0: super(that); aoqi@0: this.headers = HeaderList.copy(that.headers); aoqi@0: this.payload = that.payload; aoqi@0: } aoqi@0: aoqi@0: public boolean hasHeaders() { aoqi@0: return getHeaders().hasHeaders(); aoqi@0: } aoqi@0: aoqi@0: public MessageHeaders getHeaders() { aoqi@0: if (headers == null) aoqi@0: headers = new HeaderList(getSOAPVersion()); aoqi@0: aoqi@0: return headers; aoqi@0: } aoqi@0: aoqi@0: public String getPayloadLocalPart() { aoqi@0: return payload.getLocalName(); aoqi@0: } aoqi@0: aoqi@0: public String getPayloadNamespaceURI() { aoqi@0: return payload.getNamespaceURI(); aoqi@0: } aoqi@0: aoqi@0: public boolean hasPayload() { aoqi@0: return true; aoqi@0: } aoqi@0: aoqi@0: public Source readPayloadAsSource() { aoqi@0: return new DOMSource(payload); aoqi@0: } aoqi@0: aoqi@0: public T readPayloadAsJAXB(Unmarshaller unmarshaller) throws JAXBException { aoqi@0: if(hasAttachments()) aoqi@0: unmarshaller.setAttachmentUnmarshaller(new AttachmentUnmarshallerImpl(getAttachments())); aoqi@0: try { aoqi@0: return (T)unmarshaller.unmarshal(payload); aoqi@0: } finally{ aoqi@0: unmarshaller.setAttachmentUnmarshaller(null); aoqi@0: } aoqi@0: } aoqi@0: /** @deprecated */ aoqi@0: public T readPayloadAsJAXB(Bridge bridge) throws JAXBException { aoqi@0: return bridge.unmarshal(payload, aoqi@0: hasAttachments()? new AttachmentUnmarshallerImpl(getAttachments()) : null); aoqi@0: } aoqi@0: aoqi@0: public XMLStreamReader readPayload() throws XMLStreamException { aoqi@0: DOMStreamReader dss = new DOMStreamReader(); aoqi@0: dss.setCurrentNode(payload); aoqi@0: dss.nextTag(); aoqi@0: assert dss.getEventType()==XMLStreamReader.START_ELEMENT; aoqi@0: return dss; aoqi@0: } aoqi@0: aoqi@0: public void writePayloadTo(XMLStreamWriter sw) { aoqi@0: try { aoqi@0: if (payload != null) aoqi@0: DOMUtil.serializeNode(payload, sw); aoqi@0: } catch (XMLStreamException e) { aoqi@0: throw new WebServiceException(e); aoqi@0: } aoqi@0: } aoqi@0: aoqi@0: protected void writePayloadTo(ContentHandler contentHandler, ErrorHandler errorHandler, boolean fragment) throws SAXException { aoqi@0: if(fragment) aoqi@0: contentHandler = new FragmentContentHandler(contentHandler); aoqi@0: DOMScanner ds = new DOMScanner(); aoqi@0: ds.setContentHandler(contentHandler); aoqi@0: ds.scan(payload); aoqi@0: } aoqi@0: aoqi@0: public Message copy() { aoqi@0: return new DOMMessage(this); aoqi@0: } aoqi@0: aoqi@0: }