src/share/jaxws_classes/com/sun/xml/internal/bind/v2/runtime/output/XMLStreamWriterOutput.java

Sat, 07 Nov 2020 10:30:02 +0800

author
aoqi
date
Sat, 07 Nov 2020 10:30:02 +0800
changeset 1921
7166269ef0f1
parent 1620
6df7b161ae4a
permissions
-rw-r--r--

Added tag mips-jdk8u275-b01 for changeset fdbe50121f48

aoqi@0 1 /*
aefimov@1609 2 * Copyright (c) 1997, 2018, 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.bind.v2.runtime.output;
aoqi@0 27
aoqi@0 28 import java.io.IOException;
aefimov@1443 29 import java.io.Writer;
aoqi@0 30 import java.lang.reflect.Constructor;
aoqi@0 31
aoqi@0 32 import javax.xml.stream.XMLStreamException;
aoqi@0 33 import javax.xml.stream.XMLStreamWriter;
aoqi@0 34
aefimov@1443 35 import com.sun.xml.internal.bind.marshaller.CharacterEscapeHandler;
aefimov@1609 36 import com.sun.xml.internal.bind.marshaller.NoEscapeHandler;
aoqi@0 37 import com.sun.xml.internal.bind.v2.runtime.JAXBContextImpl;
aoqi@0 38 import com.sun.xml.internal.bind.v2.runtime.XMLSerializer;
aoqi@0 39
aoqi@0 40 import org.xml.sax.SAXException;
aoqi@0 41
aoqi@0 42 /**
aoqi@0 43 * {@link XmlOutput} that writes to StAX {@link XMLStreamWriter}.
aoqi@0 44 * <p>
aoqi@0 45 * TODO:
aoqi@0 46 * Finding the optimized FI implementations is a bit hacky and not very
aoqi@0 47 * extensible. Can we use the service provider mechanism in general for
aoqi@0 48 * concrete implementations of XmlOutputAbstractImpl.
aoqi@0 49 *
aoqi@0 50 * @author Kohsuke Kawaguchi
aoqi@0 51 */
aoqi@0 52 public class XMLStreamWriterOutput extends XmlOutputAbstractImpl {
aoqi@0 53
aoqi@0 54 /**
aoqi@0 55 * Creates a new {@link XmlOutput} from a {@link XMLStreamWriter}.
aoqi@0 56 * This method recognizes an FI StAX writer.
aoqi@0 57 */
aefimov@1443 58 public static XmlOutput create(XMLStreamWriter out, JAXBContextImpl context, CharacterEscapeHandler escapeHandler) {
aoqi@0 59 // try optimized path
aoqi@0 60 final Class writerClass = out.getClass();
aoqi@0 61 if (writerClass==FI_STAX_WRITER_CLASS) {
aoqi@0 62 try {
aoqi@0 63 return FI_OUTPUT_CTOR.newInstance(out, context);
aoqi@0 64 } catch (Exception e) {
aoqi@0 65 }
aoqi@0 66 }
aoqi@0 67 if (STAXEX_WRITER_CLASS!=null && STAXEX_WRITER_CLASS.isAssignableFrom(writerClass)) {
aoqi@0 68 try {
aoqi@0 69 return STAXEX_OUTPUT_CTOR.newInstance(out);
aoqi@0 70 } catch (Exception e) {
aoqi@0 71 }
aoqi@0 72 }
aoqi@0 73
aefimov@1443 74 CharacterEscapeHandler xmlStreamEscapeHandler = escapeHandler != null ?
aefimov@1609 75 escapeHandler : NoEscapeHandler.theInstance;
aefimov@1443 76
aoqi@0 77 // otherwise the normal writer.
aefimov@1443 78 return new XMLStreamWriterOutput(out, xmlStreamEscapeHandler);
aoqi@0 79 }
aoqi@0 80
aoqi@0 81
aoqi@0 82 private final XMLStreamWriter out;
aoqi@0 83
aefimov@1443 84 private final CharacterEscapeHandler escapeHandler;
aefimov@1443 85
aefimov@1443 86 private final XmlStreamOutWriterAdapter writerWrapper;
aefimov@1443 87
aoqi@0 88 protected final char[] buf = new char[256];
aoqi@0 89
aefimov@1443 90 protected XMLStreamWriterOutput(XMLStreamWriter out, CharacterEscapeHandler escapeHandler) {
aoqi@0 91 this.out = out;
aefimov@1443 92 this.escapeHandler = escapeHandler;
aefimov@1443 93 this.writerWrapper = new XmlStreamOutWriterAdapter(out);
aoqi@0 94 }
aoqi@0 95
aoqi@0 96 // not called if we are generating fragments
aoqi@0 97 @Override
aoqi@0 98 public void startDocument(XMLSerializer serializer, boolean fragment, int[] nsUriIndex2prefixIndex, NamespaceContextImpl nsContext) throws IOException, SAXException, XMLStreamException {
aoqi@0 99 super.startDocument(serializer, fragment,nsUriIndex2prefixIndex,nsContext);
aoqi@0 100 if(!fragment)
aoqi@0 101 out.writeStartDocument();
aoqi@0 102 }
aoqi@0 103
aoqi@0 104 @Override
aoqi@0 105 public void endDocument(boolean fragment) throws IOException, SAXException, XMLStreamException {
aoqi@0 106 if(!fragment) {
aoqi@0 107 out.writeEndDocument();
aoqi@0 108 out.flush();
aoqi@0 109 }
aoqi@0 110 super.endDocument(fragment);
aoqi@0 111 }
aoqi@0 112
aoqi@0 113 public void beginStartTag(int prefix, String localName) throws IOException, XMLStreamException {
aoqi@0 114 out.writeStartElement(
aoqi@0 115 nsContext.getPrefix(prefix),
aoqi@0 116 localName,
aoqi@0 117 nsContext.getNamespaceURI(prefix));
aoqi@0 118
aoqi@0 119 NamespaceContextImpl.Element nse = nsContext.getCurrent();
aoqi@0 120 if(nse.count()>0) {
aoqi@0 121 for( int i=nse.count()-1; i>=0; i-- ) {
aoqi@0 122 String uri = nse.getNsUri(i);
aoqi@0 123 if(uri.length()==0 && nse.getBase()==1)
aoqi@0 124 continue; // no point in definint xmlns='' on the root
aoqi@0 125 out.writeNamespace(nse.getPrefix(i),uri);
aoqi@0 126 }
aoqi@0 127 }
aoqi@0 128 }
aoqi@0 129
aoqi@0 130 public void attribute(int prefix, String localName, String value) throws IOException, XMLStreamException {
aoqi@0 131 if(prefix==-1)
aoqi@0 132 out.writeAttribute(localName,value);
aoqi@0 133 else
aoqi@0 134 out.writeAttribute(
aoqi@0 135 nsContext.getPrefix(prefix),
aoqi@0 136 nsContext.getNamespaceURI(prefix),
aoqi@0 137 localName, value);
aoqi@0 138 }
aoqi@0 139
aoqi@0 140 public void endStartTag() throws IOException, SAXException {
aoqi@0 141 // noop
aoqi@0 142 }
aoqi@0 143
aoqi@0 144 public void endTag(int prefix, String localName) throws IOException, SAXException, XMLStreamException {
aoqi@0 145 out.writeEndElement();
aoqi@0 146 }
aoqi@0 147
aoqi@0 148 public void text(String value, boolean needsSeparatingWhitespace) throws IOException, SAXException, XMLStreamException {
aoqi@0 149 if(needsSeparatingWhitespace)
aoqi@0 150 out.writeCharacters(" ");
aefimov@1443 151 escapeHandler.escape(value.toCharArray(), 0, value.length(), false, writerWrapper);
aoqi@0 152 }
aoqi@0 153
aoqi@0 154 public void text(Pcdata value, boolean needsSeparatingWhitespace) throws IOException, SAXException, XMLStreamException {
aoqi@0 155 if(needsSeparatingWhitespace)
aoqi@0 156 out.writeCharacters(" ");
aoqi@0 157
aoqi@0 158 int len = value.length();
aoqi@0 159 if(len <buf.length) {
aoqi@0 160 value.writeTo(buf,0);
aoqi@0 161 out.writeCharacters(buf,0,len);
aoqi@0 162 } else {
aoqi@0 163 out.writeCharacters(value.toString());
aoqi@0 164 }
aoqi@0 165 }
aoqi@0 166
aoqi@0 167 /**
aoqi@0 168 * Reference to FI's XMLStreamWriter class, if FI can be loaded.
aoqi@0 169 */
aoqi@0 170 private static final Class FI_STAX_WRITER_CLASS = initFIStAXWriterClass();
aoqi@0 171 private static final Constructor<? extends XmlOutput> FI_OUTPUT_CTOR = initFastInfosetOutputClass();
aoqi@0 172
aoqi@0 173 private static Class initFIStAXWriterClass() {
aoqi@0 174 try {
aoqi@0 175 Class<?> llfisw = Class.forName("com.sun.xml.internal.org.jvnet.fastinfoset.stax.LowLevelFastInfosetStreamWriter");
aoqi@0 176 Class<?> sds = Class.forName("com.sun.xml.internal.fastinfoset.stax.StAXDocumentSerializer");
aoqi@0 177 // Check if StAXDocumentSerializer implements LowLevelFastInfosetStreamWriter
aoqi@0 178 if (llfisw.isAssignableFrom(sds))
aoqi@0 179 return sds;
aoqi@0 180 else
aoqi@0 181 return null;
aoqi@0 182 } catch (Throwable e) {
aoqi@0 183 return null;
aoqi@0 184 }
aoqi@0 185 }
aoqi@0 186
aoqi@0 187 private static Constructor<? extends XmlOutput> initFastInfosetOutputClass() {
aoqi@0 188 try {
aoqi@0 189 if (FI_STAX_WRITER_CLASS == null)
aoqi@0 190 return null;
aoqi@0 191 Class c = Class.forName("com.sun.xml.internal.bind.v2.runtime.output.FastInfosetStreamWriterOutput");
aoqi@0 192 return c.getConstructor(FI_STAX_WRITER_CLASS, JAXBContextImpl.class);
aoqi@0 193 } catch (Throwable e) {
aoqi@0 194 return null;
aoqi@0 195 }
aoqi@0 196 }
aoqi@0 197
aoqi@0 198 //
aoqi@0 199 // StAX-ex
aoqi@0 200 //
aoqi@0 201 private static final Class STAXEX_WRITER_CLASS = initStAXExWriterClass();
aoqi@0 202 private static final Constructor<? extends XmlOutput> STAXEX_OUTPUT_CTOR = initStAXExOutputClass();
aoqi@0 203
aoqi@0 204 private static Class initStAXExWriterClass() {
aoqi@0 205 try {
aoqi@0 206 return Class.forName("com.sun.xml.internal.org.jvnet.staxex.XMLStreamWriterEx");
aoqi@0 207 } catch (Throwable e) {
aoqi@0 208 return null;
aoqi@0 209 }
aoqi@0 210 }
aoqi@0 211
aoqi@0 212 private static Constructor<? extends XmlOutput> initStAXExOutputClass() {
aoqi@0 213 try {
aoqi@0 214 Class c = Class.forName("com.sun.xml.internal.bind.v2.runtime.output.StAXExStreamWriterOutput");
aoqi@0 215 return c.getConstructor(STAXEX_WRITER_CLASS);
aoqi@0 216 } catch (Throwable e) {
aoqi@0 217 return null;
aoqi@0 218 }
aoqi@0 219 }
aoqi@0 220
aefimov@1443 221 private static final class XmlStreamOutWriterAdapter extends Writer {
aefimov@1443 222
aefimov@1443 223 private final XMLStreamWriter writer;
aefimov@1443 224
aefimov@1443 225 private XmlStreamOutWriterAdapter(XMLStreamWriter writer) {
aefimov@1443 226 this.writer = writer;
aefimov@1443 227 }
aefimov@1443 228
aefimov@1443 229 @Override
aefimov@1443 230 public void write(char[] cbuf, int off, int len) throws IOException {
aefimov@1443 231 try {
aefimov@1443 232 writer.writeCharacters(cbuf, off, len);
aefimov@1443 233 } catch (XMLStreamException e) {
aefimov@1443 234 throw new IOException("Error writing XML stream", e);
aefimov@1443 235 }
aefimov@1443 236 }
aefimov@1443 237
aefimov@1443 238 public void writeEntityRef(String entityReference) throws XMLStreamException {
aefimov@1443 239 writer.writeEntityRef(entityReference);
aefimov@1443 240 }
aefimov@1443 241
aefimov@1443 242 @Override
aefimov@1443 243 public void flush() throws IOException {
aefimov@1443 244 try {
aefimov@1443 245 writer.flush();
aefimov@1443 246 } catch (XMLStreamException e) {
aefimov@1443 247 throw new IOException("Error flushing XML stream", e);
aefimov@1443 248 }
aefimov@1443 249 }
aefimov@1443 250
aefimov@1443 251 @Override
aefimov@1443 252 public void close() throws IOException {
aefimov@1443 253 try {
aefimov@1443 254 writer.close();
aefimov@1443 255 } catch (XMLStreamException e) {
aefimov@1443 256 throw new IOException("Error closing XML stream", e);
aefimov@1443 257 }
aefimov@1443 258 }
aefimov@1443 259 }
aoqi@0 260 }

mercurial