Add closeStreamOnPersist option when creating GridFSInputFile

Author: greenlaw110

src/main/com/mongodb/gridfs/GridFS.java

@@ -232,7 +232,7 @@ public void remove( DBObject query ){
      * @return
      */
     public GridFSInputFile createFile( byte[] data ){
-        return createFile( new ByteArrayInputStream( data ) );
+        return createFile( new ByteArrayInputStream( data ), true );
     }
 
 
@@ -245,7 +245,7 @@ public GridFSInputFile createFile( byte[] data ){
      */
     public GridFSInputFile createFile( File f )
         throws IOException {
-        return createFile( new FileInputStream( f ) , f.getName() );
+        return createFile( new FileInputStream( f ) , f.getName(), true );
     }
 
     /**
@@ -258,6 +258,18 @@ public GridFSInputFile createFile( InputStream in ){
         return createFile( in , null );
     }
 
+    /**
+     * creates a file entry.
+     * after calling this method, you have to call save() on the GridFSInputFile file
+     * @param in an inputstream containing the file's data
+     * @param closeStreamOnPersist indicate the passed in input stream should be closed 
+     *        once the data chunk persisted
+     * @return
+     */
+    public GridFSInputFile createFile( InputStream in, boolean closeStreamOnPersist ){
+        return createFile( in , null, closeStreamOnPersist );
+    }
+
     /**
      * creates a file entry.
      * After calling this method, you have to call save() on the GridFSInputFile file
@@ -269,6 +281,19 @@ public GridFSInputFile createFile( InputStream in , String filename ){
         return new GridFSInputFile( this , in , filename );
     }
 
+    /**
+     * creates a file entry.
+     * After calling this method, you have to call save() on the GridFSInputFile file
+     * @param in an inputstream containing the file's data
+     * @param filename the file name as stored in the db
+     * @param closeStreamOnPersist indicate the passed in input stream should be closed 
+     *        once the data chunk persisted
+     * @return
+     */
+    public GridFSInputFile createFile( InputStream in , String filename, boolean closeStreamOnPersist ){
+        return new GridFSInputFile( this , in , filename, closeStreamOnPersist );
+    }
+
     /**
      * @see {@link GridFS#createFile()} on how to use this method
      * @param filename the file name as stored in the db

src/main/com/mongodb/gridfs/GridFSInputFile.java

@@ -52,11 +52,14 @@ public class GridFSInputFile extends GridFSFile {
      *            Stream used for reading data from.
      * @param filename
      *            Name of the file to be created.
+     * @param closeStreamOnPersist 
+                  indicate the passed in input stream should be closed once the data chunk persisted
      */
-    GridFSInputFile( GridFS fs , InputStream in , String filename ) {
+    GridFSInputFile( GridFS fs , InputStream in , String filename, boolean closeStreamOnPersist ) {
         _fs = fs;
         _in = in;
         _filename = filename;
+        _closeStreamOnPersist = closeStreamOnPersist;
 
         _id = new ObjectId();
         _chunkSize = GridFS.DEFAULT_CHUNKSIZE;
@@ -66,6 +69,21 @@ public class GridFSInputFile extends GridFSFile {
         _buffer = new byte[(int) _chunkSize];
     }
 
+    /**
+     * Default constructor setting the GridFS file name and providing an input
+     * stream containing data to be written to the file.
+     * 
+     * @param fs
+     *            The GridFS connection handle.
+     * @param in
+     *            Stream used for reading data from.
+     * @param filename
+     *            Name of the file to be created.
+     */
+    GridFSInputFile( GridFS fs , InputStream in , String filename ) {
+        this( fs, in, filename, false);
+    }
+    
     /**
      * Constructor that only provides a file name, but does not rely on the
      * presence of an {@link java.io.InputStream}. An
@@ -290,14 +308,17 @@ private void _finishData() {
             _length = _totalBytes;
             _savedChunks = true;
             try {
-                if (null != _in) _in.close();
+                if (null != _in && _closeStreamOnPersist) {
+                  _in.close();
+                }
             } catch (IOException e) {
                 //ignore
             }
         }
     }
 
     private final InputStream _in;
+    private boolean _closeStreamOnPersist;
     private boolean _savedChunks = false;
     private byte[] _buffer = null;
     private int _currentChunkNumber = 0;