src/share/jaxws_classes/com/sun/xml/internal/ws/api/addressing/EPRHeader.java

Tue, 09 Apr 2013 14:51:13 +0100

author
alanb
date
Tue, 09 Apr 2013 14:51:13 +0100
changeset 368
0989ad8c0860
parent 286
f50545b5e2f1
child 637
9c07ef4934dd
permissions
-rw-r--r--

8010393: Update JAX-WS RI to 2.2.9-b12941
Reviewed-by: alanb, erikj
Contributed-by: miroslav.kos@oracle.com, martin.grebac@oracle.com

ohair@286 1 /*
alanb@368 2 * Copyright (c) 1997, 2013, Oracle and/or its affiliates. All rights reserved.
ohair@286 3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
ohair@286 4 *
ohair@286 5 * This code is free software; you can redistribute it and/or modify it
ohair@286 6 * under the terms of the GNU General Public License version 2 only, as
ohair@286 7 * published by the Free Software Foundation. Oracle designates this
ohair@286 8 * particular file as subject to the "Classpath" exception as provided
ohair@286 9 * by Oracle in the LICENSE file that accompanied this code.
ohair@286 10 *
ohair@286 11 * This code is distributed in the hope that it will be useful, but WITHOUT
ohair@286 12 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
ohair@286 13 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
ohair@286 14 * version 2 for more details (a copy is included in the LICENSE file that
ohair@286 15 * accompanied this code).
ohair@286 16 *
ohair@286 17 * You should have received a copy of the GNU General Public License version
ohair@286 18 * 2 along with this work; if not, write to the Free Software Foundation,
ohair@286 19 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
ohair@286 20 *
ohair@286 21 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
ohair@286 22 * or visit www.oracle.com if you need additional information or have any
ohair@286 23 * questions.
ohair@286 24 */
ohair@286 25
ohair@286 26 package com.sun.xml.internal.ws.api.addressing;
ohair@286 27
alanb@368 28 import java.io.ByteArrayInputStream;
alanb@368 29 import java.io.ByteArrayOutputStream;
alanb@368 30 import java.io.Writer;
alanb@368 31
ohair@286 32 import com.sun.istack.internal.NotNull;
ohair@286 33 import com.sun.istack.internal.Nullable;
ohair@286 34 import com.sun.xml.internal.ws.message.AbstractHeaderImpl;
ohair@286 35 import com.sun.xml.internal.ws.util.xml.XmlUtil;
alanb@368 36
alanb@368 37 import org.w3c.dom.Document;
alanb@368 38 import org.w3c.dom.Node;
ohair@286 39 import org.xml.sax.ContentHandler;
ohair@286 40 import org.xml.sax.ErrorHandler;
ohair@286 41 import org.xml.sax.SAXException;
ohair@286 42
ohair@286 43 import javax.xml.namespace.QName;
alanb@368 44 import javax.xml.parsers.DocumentBuilderFactory;
ohair@286 45 import javax.xml.soap.SOAPException;
ohair@286 46 import javax.xml.soap.SOAPMessage;
ohair@286 47 import javax.xml.soap.SOAPHeader;
alanb@368 48 import javax.xml.stream.XMLOutputFactory;
ohair@286 49 import javax.xml.stream.XMLStreamConstants;
ohair@286 50 import javax.xml.stream.XMLStreamException;
ohair@286 51 import javax.xml.stream.XMLStreamReader;
ohair@286 52 import javax.xml.stream.XMLStreamWriter;
ohair@286 53 import javax.xml.transform.Transformer;
ohair@286 54 import javax.xml.transform.dom.DOMResult;
ohair@286 55
ohair@286 56 /**
ohair@286 57 * Used to represent outbound endpoint reference header,
ohair@286 58 * such as <ReplyTo> and <FaultTo>.
ohair@286 59 * Used from {@link WSEndpointReference}.
ohair@286 60 *
ohair@286 61 * @author Kohsuke Kawaguchi
ohair@286 62 * @see WSEndpointReference
ohair@286 63 */
ohair@286 64 final class EPRHeader extends AbstractHeaderImpl {
ohair@286 65
ohair@286 66 private final String nsUri,localName;
ohair@286 67 private final WSEndpointReference epr;
ohair@286 68
ohair@286 69 EPRHeader(QName tagName, WSEndpointReference epr) {
ohair@286 70 this.nsUri = tagName.getNamespaceURI();
ohair@286 71 this.localName = tagName.getLocalPart();
ohair@286 72 this.epr = epr;
ohair@286 73 }
ohair@286 74
ohair@286 75 public @NotNull String getNamespaceURI() {
ohair@286 76 return nsUri;
ohair@286 77 }
ohair@286 78
ohair@286 79 public @NotNull String getLocalPart() {
ohair@286 80 return localName;
ohair@286 81 }
ohair@286 82
ohair@286 83 @Nullable
ohair@286 84 public String getAttribute(@NotNull String nsUri, @NotNull String localName) {
ohair@286 85 try {
ohair@286 86 XMLStreamReader sr = epr.read("EndpointReference"/*doesn't matter*/);
ohair@286 87 while(sr.getEventType()!= XMLStreamConstants.START_ELEMENT)
ohair@286 88 sr.next();
ohair@286 89
ohair@286 90 return sr.getAttributeValue(nsUri,localName);
ohair@286 91 } catch (XMLStreamException e) {
ohair@286 92 // since we are reading from buffer, this can't happen.
ohair@286 93 throw new AssertionError(e);
ohair@286 94 }
ohair@286 95 }
ohair@286 96
ohair@286 97 public XMLStreamReader readHeader() throws XMLStreamException {
ohair@286 98 return epr.read(localName);
ohair@286 99 }
ohair@286 100
ohair@286 101 public void writeTo(XMLStreamWriter w) throws XMLStreamException {
ohair@286 102 epr.writeTo(localName, w);
ohair@286 103 }
ohair@286 104
ohair@286 105 public void writeTo(SOAPMessage saaj) throws SOAPException {
ohair@286 106 try {
ohair@286 107 // TODO what about in-scope namespaces
ohair@286 108 // Not very efficient consider implementing a stream buffer
ohair@286 109 // processor that produces a DOM node from the buffer.
ohair@286 110 Transformer t = XmlUtil.newTransformer();
ohair@286 111 SOAPHeader header = saaj.getSOAPHeader();
ohair@286 112 if (header == null)
ohair@286 113 header = saaj.getSOAPPart().getEnvelope().addHeader();
alanb@368 114 // TODO workaround for oracle xdk bug 16555545, when this bug is fixed the line below can be
alanb@368 115 // uncommented and all lines below, except the catch block, can be removed.
alanb@368 116 // t.transform(epr.asSource(localName), new DOMResult(header));
alanb@368 117 ByteArrayOutputStream baos = new ByteArrayOutputStream();
alanb@368 118 XMLStreamWriter w = XMLOutputFactory.newFactory().createXMLStreamWriter(baos);
alanb@368 119 epr.writeTo(localName, w);
alanb@368 120 w.flush();
alanb@368 121 ByteArrayInputStream bais = new ByteArrayInputStream(baos.toByteArray());
alanb@368 122 DocumentBuilderFactory fac = DocumentBuilderFactory.newInstance();
alanb@368 123 fac.setNamespaceAware(true);
alanb@368 124 Node eprNode = fac.newDocumentBuilder().parse(bais).getDocumentElement();
alanb@368 125 Node eprNodeToAdd = header.getOwnerDocument().importNode(eprNode, true);
alanb@368 126 header.appendChild(eprNodeToAdd);
ohair@286 127 } catch (Exception e) {
ohair@286 128 throw new SOAPException(e);
ohair@286 129 }
ohair@286 130 }
ohair@286 131
ohair@286 132 public void writeTo(ContentHandler contentHandler, ErrorHandler errorHandler) throws SAXException {
ohair@286 133 epr.writeTo(localName,contentHandler,errorHandler,true);
ohair@286 134 }
ohair@286 135 }

mercurial