View Javadoc

1   /*
2    * $Id: XMLStreamCloser.java 19191 2010-08-25 21:05:23Z tcarlson $
3    * --------------------------------------------------------------------------------------
4    * Copyright (c) MuleSoft, Inc.  All rights reserved.  http://www.mulesoft.com
5    *
6    * The software in this package is published under the terms of the CPAL v1.0
7    * license, a copy of which has been included with this distribution in the
8    * LICENSE.txt file.
9    */
10  
11  package org.mule.module.xml.util;
12  
13  import org.mule.api.util.StreamCloser;
14  import org.mule.module.xml.stax.StaxSource;
15  
16  import javax.xml.stream.XMLStreamReader;
17  
18  public class XMLStreamCloser implements StreamCloser
19  {
20  
21      public boolean canClose(Class streamType)
22      {
23          return StaxSource.class.isAssignableFrom(streamType)
24                 || XMLStreamReader.class.isAssignableFrom(streamType);
25      }
26  
27      public void close(Object stream) throws Exception
28      {
29          if (stream instanceof XMLStreamReader)
30          {
31              ((XMLStreamReader) stream).close();
32          }
33          else if (stream instanceof StaxSource)
34          {
35              ((StaxSource) stream).getXMLStreamReader().close();
36          }
37      }
38  
39  }