Open Source Repository

Home /spring/spring-core-3.0.5 | Repository Home



org/springframework/util/xml/StaxUtils.java
/*
 * Copyright 2002-2010 the original author or authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.springframework.util.xml;

import javax.xml.stream.XMLEventFactory;
import javax.xml.stream.XMLEventReader;
import javax.xml.stream.XMLEventWriter;
import javax.xml.stream.XMLStreamException;
import javax.xml.stream.XMLStreamReader;
import javax.xml.stream.XMLStreamWriter;
import javax.xml.transform.Result;
import javax.xml.transform.Source;
import javax.xml.transform.stax.StAXResult;
import javax.xml.transform.stax.StAXSource;

import org.xml.sax.ContentHandler;
import org.xml.sax.XMLReader;

import org.springframework.util.Assert;
import org.springframework.util.ClassUtils;

/**
 * Convenience methods for working with the StAX API.
 *
 <p>In particular, methods for using StAX ({@link javax.xml.stream}) in combination with the TrAX API
 * ({@link javax.xml.transform}), and converting StAX readers/writers into SAX readers/handlers and vice-versa.
 *
 @author Arjen Poutsma
 @author Juergen Hoeller
 @since 3.0
 */
public abstract class StaxUtils {

  private static boolean jaxp14Available =
      ClassUtils.isPresent("javax.xml.transform.stax.StAXSource", StaxUtils.class.getClassLoader());

  // Stax Source

  /**
   * Create a custom, non-JAXP 1.4 StAX {@link Source} for the given {@link XMLStreamReader}.
   *
   @param streamReader the StAX stream reader
   @return a source wrapping the {@code streamReader}
   */
  public static Source createCustomStaxSource(XMLStreamReader streamReader) {
    return new StaxSource(streamReader);
  }

  /**
   * Create a StAX {@link Source} for the given {@link XMLStreamReader}.
   *
   <p>If JAXP 1.4 is available, this method returns a {@link StAXSource}; otherwise it returns a
   * custom StAX Source.
   @param streamReader the StAX stream reader
   @return a source wrapping the {@code streamReader}
   @see #createCustomStaxSource(XMLStreamReader)
   */
  public static Source createStaxSource(XMLStreamReader streamReader) {
    if (jaxp14Available) {
      return Jaxp14StaxHandler.createStaxSource(streamReader);
    }
    else {
      return createCustomStaxSource(streamReader);
    }
  }

  /**
   * Create a custom, non-JAXP 1.4 StAX {@link Source} for the given {@link XMLEventReader}.
   *
   @param eventReader the StAX event reader
   @return a source wrapping the {@code eventReader}
   */
  public static Source createCustomStaxSource(XMLEventReader eventReader) {
    return new StaxSource(eventReader);
  }

  /**
   * Create a StAX {@link Source} for the given {@link XMLEventReader}.
   *
   <p>If JAXP 1.4 is available, this method returns a {@link StAXSource}; otherwise it returns a
   * custom StAX Source.
   @param eventReader the StAX event reader
   @return a source wrapping the {@code eventReader}
   @throws XMLStreamException in case of StAX errors
   @see #createCustomStaxSource(XMLEventReader)
   */
  public static Source createStaxSource(XMLEventReader eventReaderthrows XMLStreamException {
    if (jaxp14Available) {
      return Jaxp14StaxHandler.createStaxSource(eventReader);
    }
    else {
      return createCustomStaxSource(eventReader);
    }
  }

  /**
   * Indicate whether the given {@link Source} is a StAX Source.
   @return {@code true} if {@code source} is a custom StAX source or JAXP
   * 1.4 {@link StAXSource}; {@code false} otherwise.
   */
  public static boolean isStaxSource(Source source) {
    return (source instanceof StaxSource || (jaxp14Available && Jaxp14StaxHandler.isStaxSource(source)));
  }

  // Stax Result

  /**
   * Create a custom, non-JAXP 1.4 StAX {@link Result} for the given {@link XMLStreamWriter}.
   *
   @param streamWriter the StAX stream writer
   @return a source wrapping the {@code streamWriter}
   */
  public static Result createCustomStaxResult(XMLStreamWriter streamWriter) {
    return new StaxResult(streamWriter);
  }

  /**
   * Create a StAX {@link Result} for the given {@link XMLStreamWriter}.
   *
   <p>If JAXP 1.4 is available, this method returns a {@link StAXResult}; otherwise it returns a
   * custom StAX Result.
   @param streamWriter the StAX stream writer
   @return a result wrapping the {@code streamWriter}
   @see #createCustomStaxResult(XMLStreamWriter)
   */
  public static Result createStaxResult(XMLStreamWriter streamWriter) {
    if (jaxp14Available) {
      return Jaxp14StaxHandler.createStaxResult(streamWriter);
    }
    else {
      return createCustomStaxResult(streamWriter);
    }
  }

  /**
   * Create a custom, non-JAXP 1.4 StAX {@link Result} for the given {@link XMLEventWriter}.
   *
   @param eventWriter the StAX event writer
   @return a source wrapping the {@code eventWriter}
   */
  public static Result createCustomStaxResult(XMLEventWriter eventWriter) {
    return new StaxResult(eventWriter);
  }

  /**
   * Create a StAX {@link Result} for the given {@link XMLEventWriter}.
   *
   <p>If JAXP 1.4 is available, this method returns a {@link StAXResult}; otherwise it returns a
   * custom StAX Result.
   @param eventWriter the StAX event writer
   @return a result wrapping {@code streamReader}
   @throws XMLStreamException in case of StAX errors
   @see #createCustomStaxResult(XMLEventWriter)
   */
  public static Result createStaxResult(XMLEventWriter eventWriterthrows XMLStreamException {
    if (jaxp14Available) {
      return Jaxp14StaxHandler.createStaxResult(eventWriter);
    }
    else {
      return createCustomStaxResult(eventWriter);
    }
  }

  /**
   * Indicate whether the given {@link javax.xml.transform.Result} is a StAX Result.
   @return {@code true} if {@code result} is a custom Stax Result or JAXP
   * 1.4 {@link StAXResult}; {@code false} otherwise.
   */
  public static boolean isStaxResult(Result result) {
    return (result instanceof StaxResult || (jaxp14Available && Jaxp14StaxHandler.isStaxResult(result)));
  }

  /**
   * Return the {@link XMLStreamReader} for the given StAX Source.
   @param source a {@linkplain #createCustomStaxSource(XMLStreamReader) custom StAX Source} or
   * JAXP 1.4 {@link StAXSource}
   @return the {@link XMLStreamReader}
   @throws IllegalArgumentException if {@code source} is neither a custom StAX Source
   * nor JAXP 1.4 {@link StAXSource}
   */
  public static XMLStreamReader getXMLStreamReader(Source source) {
    if (source instanceof StaxSource) {
      return ((StaxSourcesource).getXMLStreamReader();
    }
    else if (jaxp14Available) {
      return Jaxp14StaxHandler.getXMLStreamReader(source);
    }
    else {
      throw new IllegalArgumentException("Source '" + source + "' is neither StaxSource nor StAXSource");
    }
  }

  /**
   * Return the {@link XMLEventReader} for the given StAX Source.
   @param source a {@linkplain #createCustomStaxSource(XMLEventReader) custom StAX Source} or
   * JAXP 1.4 {@link StAXSource}
   @return the {@link XMLEventReader}
   @throws IllegalArgumentException if {@code source} is neither a custom StAX Source
   * nor a JAXP 1.4 {@link StAXSource}
   */
  public static XMLEventReader getXMLEventReader(Source source) {
    if (source instanceof StaxSource) {
      return ((StaxSourcesource).getXMLEventReader();
    }
    else if (jaxp14Available) {
      return Jaxp14StaxHandler.getXMLEventReader(source);
    }
    else {
      throw new IllegalArgumentException("Source '" + source + "' is neither StaxSource nor StAXSource");
    }
  }

  /**
   * Return the {@link XMLStreamWriter} for the given StAX Result.
   @param result a {@linkplain #createCustomStaxResult(XMLStreamWriter) custom StAX Result} or
   * JAXP 1.4 {@link StAXResult}
   @return the {@link XMLStreamReader}
   @throws IllegalArgumentException if {@code source} is neither a custom StAX Result
   * nor a JAXP 1.4 {@link StAXResult}
   */
  public static XMLStreamWriter getXMLStreamWriter(Result result) {
    if (result instanceof StaxResult) {
      return ((StaxResultresult).getXMLStreamWriter();
    }
    else if (jaxp14Available) {
      return Jaxp14StaxHandler.getXMLStreamWriter(result);
    }
    else {
      throw new IllegalArgumentException("Result '" + result + "' is neither StaxResult nor StAXResult");
    }
  }

  /**
   * Return the {@link XMLEventWriter} for the given StAX Result.
   @param result a {@linkplain #createCustomStaxResult(XMLEventWriter)  custom StAX Result} or
   * JAXP 1.4 {@link StAXResult}
   @return the {@link XMLStreamReader}
   @throws IllegalArgumentException if {@code source} is neither a custom StAX Result
   * nor a JAXP 1.4 {@link StAXResult}
   */
  public static XMLEventWriter getXMLEventWriter(Result result) {
    if (result instanceof StaxResult) {
      return ((StaxResultresult).getXMLEventWriter();
    }
    else if (jaxp14Available) {
      return Jaxp14StaxHandler.getXMLEventWriter(result);
    }
    else {
      throw new IllegalArgumentException("Result '" + result + "' is neither StaxResult nor StAXResult");
    }
  }

  /**
   * Create a SAX {@link ContentHandler} that writes to the given StAX {@link XMLStreamWriter}.
   @param streamWriter the StAX stream writer
   @return a content handler writing to the <code>streamWriter</code>
   */
  public static ContentHandler createContentHandler(XMLStreamWriter streamWriter) {
    return new StaxStreamContentHandler(streamWriter);
  }

  /**
   * Create a SAX {@link ContentHandler} that writes events to the given StAX {@link XMLEventWriter}.
   @param eventWriter the StAX event writer
   @return a content handler writing to the <code>eventWriter</code>
   */
  public static ContentHandler createContentHandler(XMLEventWriter eventWriter) {
    return new StaxEventContentHandler(eventWriter);
  }

  /**
   * Create a SAX {@link XMLReader} that reads from the given StAX {@link XMLStreamReader}.
   @param streamReader the StAX stream reader
   @return a XMLReader reading from the <code>streamWriter</code>
   */
  public static XMLReader createXMLReader(XMLStreamReader streamReader) {
    return new StaxStreamXMLReader(streamReader);
  }

  /**
   * Create a SAX {@link XMLReader} that reads from the given StAX {@link XMLEventReader}.
   @param eventReader the StAX event reader
   @return a XMLReader reading from the <code>eventWriter</code>
   */
  public static XMLReader createXMLReader(XMLEventReader eventReader) {
    return new StaxEventXMLReader(eventReader);
  }

  /**
   * Return a {@link XMLStreamReader} that reads from a {@link XMLEventReader}. Useful, because the StAX
   <code>XMLInputFactory</code> allows one to create a event reader from a stream reader, but not vice-versa.
   @return a stream reader that reads from an event reader
   */
  public static XMLStreamReader createEventStreamReader(XMLEventReader eventReaderthrows XMLStreamException {
    return new XMLEventStreamReader(eventReader);
  }

  /**
   * Return a {@link XMLStreamWriter} that writes to a {@link XMLEventWriter}.
   @return a stream writer that writes to an event writer
   @since 3.0.5
   */
  public static XMLStreamWriter createEventStreamWriter(XMLEventWriter eventWriter, XMLEventFactory eventFactory) {
    return new XMLEventStreamWriter(eventWriter, eventFactory);
  }

  /**
   * Inner class to avoid a static JAXP 1.4 dependency.
   */
  private static class Jaxp14StaxHandler {

    private static Source createStaxSource(XMLStreamReader streamReader) {
      return new StAXSource(streamReader);
    }

    private static Source createStaxSource(XMLEventReader eventReaderthrows XMLStreamException {
      return new StAXSource(eventReader);
    }

    private static Result createStaxResult(XMLStreamWriter streamWriter) {
      return new StAXResult(streamWriter);
    }

    private static Result createStaxResult(XMLEventWriter eventWriter) {
      return new StAXResult(eventWriter);
    }

    private static boolean isStaxSource(Source source) {
      return source instanceof StAXSource;
    }

    private static boolean isStaxResult(Result result) {
      return result instanceof StAXResult;
    }

    private static XMLStreamReader getXMLStreamReader(Source source) {
      Assert.isInstanceOf(StAXSource.class, source);
      return ((StAXSourcesource).getXMLStreamReader();
    }

    private static XMLEventReader getXMLEventReader(Source source) {
      Assert.isInstanceOf(StAXSource.class, source);
      return ((StAXSourcesource).getXMLEventReader();
    }

    private static XMLStreamWriter getXMLStreamWriter(Result result) {
      Assert.isInstanceOf(StAXResult.class, result);
      return ((StAXResultresult).getXMLStreamWriter();
    }

    private static XMLEventWriter getXMLEventWriter(Result result) {
      Assert.isInstanceOf(StAXResult.class, result);
      return ((StAXResultresult).getXMLEventWriter();
    }
  }

}