Mercurial > jdk8-mips64-public > jaxws / file revision

 /*

  * Copyright (c) 2005, 2010, 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.txw2.output;

 import com.sun.xml.internal.txw2.TxwException;

 import javax.xml.transform.stream.StreamResult;

 import java.io.BufferedWriter;

 import java.io.FileOutputStream;

 import java.io.IOException;

 import java.io.OutputStream;

 import java.io.OutputStreamWriter;

 import java.io.Writer;

 import java.io.UnsupportedEncodingException;

 /**

  * {@link XmlSerializer} for {@link javax.xml.transform.stream.StreamResult}.

  *

  * @author Ryan.Shoemaker@Sun.COM

  */

 public class StreamSerializer implements XmlSerializer {

     // delegate to SaxSerializer

     private final SaxSerializer serializer;

     private final XMLWriter writer;

     public StreamSerializer(OutputStream out) {

         this(createWriter(out));

     }

     public StreamSerializer(OutputStream out,String encoding) throws UnsupportedEncodingException {

         this(createWriter(out,encoding));

     }

     public StreamSerializer(Writer out) {

         this(new StreamResult(out));

     }

     public StreamSerializer(StreamResult streamResult) {

         // if this method opened a stream, let it close it

         final OutputStream[] autoClose = new OutputStream[1];

         if (streamResult.getWriter() != null)

             writer = createWriter(streamResult.getWriter());

         else if (streamResult.getOutputStream() != null)

             writer = createWriter(streamResult.getOutputStream());

         else if (streamResult.getSystemId() != null) {

             String fileURL = streamResult.getSystemId();

             fileURL = convertURL(fileURL);

             try {

                 FileOutputStream fos = new FileOutputStream(fileURL);

                 autoClose[0] = fos;

                 writer = createWriter(fos);

             } catch (IOException e) {

                 throw new TxwException(e);

             }

         } else

             throw new IllegalArgumentException();

         // now delegate to the SaxSerializer

         serializer = new SaxSerializer(writer,writer,false) {

             public void endDocument() {

                 super.endDocument();

                 if(autoClose[0]!=null) {

                     try {

                         autoClose[0].close();

                     } catch (IOException e) {

                         throw new TxwException(e);

                     }

                     autoClose[0] = null;

                 }

             }

         };

     }

     private StreamSerializer(XMLWriter writer) {

         this.writer = writer;

         // now delegate to the SaxSerializer

         serializer = new SaxSerializer(writer,writer,false);

     }

     private String convertURL(String url) {

         url = url.replace('\\', '/');

         url = url.replaceAll("//","/");

         url = url.replaceAll("//","/");

         if (url.startsWith("file:/")) {

             if (url.substring(6).indexOf(":") > 0)

                 url = url.substring(6);

             else

                 url = url.substring(5);

         } // otherwise assume that it's a file name

         return url;

     }

     // XmlSerializer api's - delegate to SaxSerializer

     public void startDocument() {

         serializer.startDocument();

     }

     public void beginStartTag(String uri, String localName, String prefix) {

         serializer.beginStartTag(uri, localName, prefix);

     }

     public void writeAttribute(String uri, String localName, String prefix, StringBuilder value) {

         serializer.writeAttribute(uri, localName, prefix, value);

     }

     public void writeXmlns(String prefix, String uri) {

         serializer.writeXmlns(prefix, uri);

     }

     public void endStartTag(String uri, String localName, String prefix) {

         serializer.endStartTag(uri, localName, prefix);

     }

     public void endTag() {

         serializer.endTag();

     }

     public void text(StringBuilder text) {

         serializer.text(text);

     }

     public void cdata(StringBuilder text) {

         serializer.cdata(text);

     }

     public void comment(StringBuilder comment) {

         serializer.comment(comment);

     }

     public void endDocument() {

         serializer.endDocument();

     }

     public void flush() {

         serializer.flush();

         try {

             writer.flush();

         } catch (IOException e) {

             throw new TxwException(e);

         }

     }

     // other supporting code

     private static XMLWriter createWriter(Writer w) {

         // buffering improves the performance

         DataWriter dw = new DataWriter(new BufferedWriter(w));

         dw.setIndentStep("  ");

         return dw;

     }

     private static XMLWriter createWriter(OutputStream os, String encoding) throws UnsupportedEncodingException {

         XMLWriter writer = createWriter(new OutputStreamWriter(os,encoding));

         writer.setEncoding(encoding);

         return writer;

     }

     private static XMLWriter createWriter(OutputStream os) {

         try {

             return createWriter(os,"UTF-8");

         } catch (UnsupportedEncodingException e) {

             // UTF-8 is supported on all platforms.

             throw new Error(e);

         }

     }

 }