fix #243, offer a way to close the input source for XMLStreamReaderHandle

Author: sammefford

src/main/java/com/marklogic/client/io/XMLStreamReaderHandle.java

@@ -48,7 +48,7 @@
  * An XML Stream Reader Handle represents XML content as an XML stream reader
  * for reading as a StAX pull stream.
  * 
- * When finished with the stream reader, close the stream reader to release
+ * When finished with the stream reader, call {@link #close} to release
  * the response.
  */
 public class XMLStreamReaderHandle
@@ -61,6 +61,7 @@ public class XMLStreamReaderHandle
 
 	private XMLResolver     resolver;
 	private XMLStreamReader content;
+	private InputStream contentSource;
 	private XMLInputFactory factory;
 
 	/**
@@ -125,7 +126,7 @@ public void setResolver(XMLResolver resolver) {
 	 * Returns an XML Stream Reader for for reading a resource from the database
 	 * as a StAX pull stream.
 	 * 
-     * When finished with the stream reader, close the stream reader to release
+	 * When finished with the stream reader, call {@link #close} to release
      * the response.
 	 * 
 	 * @return	the XML stream reader
@@ -195,6 +196,16 @@ public byte[] toBuffer() {
 			throw new MarkLogicIOException(e);
 		}
 	}
+	/**
+	 * Closes the XMLStreamReader and the InputStream, if any, used to populate
+	 * the XMLStreamReader.  This method should always be called when finished
+	 * with the stream reader.
+	 */
+	public void close() throws XMLStreamException, IOException {
+		if ( content != null ) content.close();
+		if ( contentSource != null ) contentSource.close();
+	}
+
 	/**
 	 * Buffers the StAX stream and returns the buffer as an XML string.
 	 */
@@ -255,6 +266,7 @@ protected void receiveContent(InputStream content) {
 				factory.setXMLResolver(resolver);
 
 			this.content = factory.createXMLStreamReader(content, "UTF-8");
+			this.contentSource = content;
 		} catch (XMLStreamException e) {
 			logger.error("Failed to parse StAX stream from input stream",e);
 			throw new MarkLogicInternalException(e);

src/test/java/com/marklogic/client/test/XMLDocumentTest.java

@@ -137,7 +137,8 @@ public void startElement(String uri, String localName, String qName, Attributes
 		assertTrue("Failed to process XML document with SAX",
 				counter.get("elementCount") == 2 && counter.get("attributeCount") == 2);
 
-		XMLStreamReader streamReader = docMgr.read(docId, new XMLStreamReaderHandle()).get();
+		XMLStreamReaderHandle streamReaderHandle = docMgr.read(docId, new XMLStreamReaderHandle());
+		XMLStreamReader streamReader = streamReaderHandle.get();
 		int elementCount = 0;
 		int attributeCount = 0;
 		while (streamReader.hasNext()) {
@@ -148,7 +149,7 @@ public void startElement(String uri, String localName, String qName, Attributes
 			if (elementAttributeCount > 0)
 				attributeCount += elementAttributeCount;
 		}
-		streamReader.close();
+		streamReaderHandle.close();
 		assertTrue("Failed to process XML document with StAX stream reader",
 				elementCount == 2 && attributeCount == 2);