git-svn-id: https://svn.apache.org/repos/asf/maven/shared/trunk@560022 13f79535-47bb...
authorJohn Dennis Casey <jdcasey@apache.org>
Thu, 26 Jul 2007 22:42:00 +0000 (22:42 +0000)
committerJohn Dennis Casey <jdcasey@apache.org>
Thu, 26 Jul 2007 22:42:00 +0000 (22:42 +0000)
src/main/java/org/apache/maven/shared/io/location/FileLocation.java
src/main/java/org/apache/maven/shared/io/location/Location.java
src/test/java/org/apache/maven/shared/io/location/FileLocationTest.java

index 0aee171..698c20b 100644 (file)
@@ -3,6 +3,7 @@ package org.apache.maven.shared.io.location;
 import java.io.File;
 import java.io.FileInputStream;
 import java.io.IOException;
+import java.io.InputStream;
 import java.nio.ByteBuffer;
 import java.nio.channels.FileChannel;
 
@@ -14,13 +15,14 @@ public class FileLocation
     private File file;
     private FileChannel channel;
     private final String specification;
-    
+    private FileInputStream stream;
+
     public FileLocation( File file, String specification )
     {
         this.file = file;
         this.specification = specification;
     }
-    
+
     protected FileLocation( String specification )
     {
         this.specification = specification;
@@ -28,7 +30,7 @@ public class FileLocation
 
     public void close()
     {
-        if ( channel != null && channel.isOpen() )
+        if ( ( channel != null ) && channel.isOpen() )
         {
             try
             {
@@ -38,17 +40,29 @@ public class FileLocation
             {
                 //swallow it.
             }
-        }        
+        }
+
+        if ( stream != null )
+        {
+            try
+            {
+                stream.close();
+            }
+            catch( IOException e )
+            {
+                // swallow it.
+            }
+        }
     }
 
     public File getFile()
         throws IOException
     {
         initFile();
-        
+
         return unsafeGetFile();
     }
-    
+
     protected File unsafeGetFile()
     {
         return file;
@@ -63,14 +77,14 @@ public class FileLocation
             file = new File( specification );
         }
     }
-    
+
     protected void setFile( File file )
     {
         if ( channel != null )
         {
-            throw new IllegalStateException( "Location is already open; cannot setFile(..)." ); 
+            throw new IllegalStateException( "Location is already open; cannot setFile(..)." );
         }
-        
+
         this.file = file;
     }
 
@@ -82,21 +96,34 @@ public class FileLocation
     public void open()
         throws IOException
     {
-        initFile();
-        
-        channel = new FileInputStream( file ).getChannel();
+        if ( stream == null )
+        {
+            initFile();
+
+            stream = new FileInputStream( file );
+            channel = stream.getChannel();
+        }
     }
 
     public int read( ByteBuffer buffer )
         throws IOException
     {
+        open();
         return channel.read( buffer );
     }
 
     public int read( byte[] buffer )
         throws IOException
     {
+        open();
         return channel.read( ByteBuffer.wrap( buffer ) );
     }
 
+    public InputStream getInputStream()
+        throws IOException
+    {
+        open();
+        return stream;
+    }
+
 }
index c618a50..0701772 100644 (file)
@@ -2,21 +2,24 @@ package org.apache.maven.shared.io.location;
 
 import java.io.File;
 import java.io.IOException;
+import java.io.InputStream;
 import java.nio.ByteBuffer;
 
 public interface Location
 {
-    
+
     File getFile() throws IOException;
-    
+
     void open() throws IOException;
-    
+
     void close();
-    
+
     int read( ByteBuffer buffer ) throws IOException;
-    
+
     int read( byte[] buffer ) throws IOException;
-    
+
+    InputStream getInputStream() throws IOException;
+
     String getSpecification();
 
 }
index 3cb5afa..7904c85 100644 (file)
 package org.apache.maven.shared.io.location;
 
+import org.apache.maven.shared.io.TestUtils;
+import org.codehaus.plexus.util.IOUtil;
+
 import java.io.File;
 import java.io.IOException;
+import java.io.InputStream;
+import java.io.StringWriter;
 import java.nio.ByteBuffer;
 
-import org.apache.maven.shared.io.TestUtils;
-
 import junit.framework.TestCase;
 
 public class FileLocationTest
     extends TestCase
 {
-    
+
     public void testShouldConstructWithFileThenRetrieveSameFile() throws IOException
     {
         File file = File.createTempFile( "test.", ".file-location" );
         file.deleteOnExit();
-        
+
         FileLocation location = new FileLocation( file, file.getAbsolutePath() );
-        
+
         assertSame( file, location.getFile() );
         assertEquals( file.getAbsolutePath(), location.getSpecification() );
     }
-    
+
     public void testShouldReadFileContentsUsingByteBuffer() throws IOException
     {
         File file = File.createTempFile( "test.", ".file-location" );
         file.deleteOnExit();
-        
+
         String testStr = "This is a test";
-        
+
         TestUtils.writeToFile( file, testStr );
-        
+
         FileLocation location = new FileLocation( file, file.getAbsolutePath() );
-        
+
         location.open();
-        
+
         ByteBuffer buffer = ByteBuffer.allocate( testStr.length() );
         location.read( buffer );
-        
+
         assertEquals( testStr, new String( buffer.array() ) );
     }
-    
+
+    public void testShouldReadFileContentsUsingStream() throws IOException
+    {
+        File file = File.createTempFile( "test.", ".file-location" );
+        file.deleteOnExit();
+
+        String testStr = "This is a test";
+
+        TestUtils.writeToFile( file, testStr );
+
+        FileLocation location = new FileLocation( file, file.getAbsolutePath() );
+
+        location.open();
+
+        InputStream stream = location.getInputStream();
+        StringWriter writer = new StringWriter();
+        IOUtil.copy( stream, writer );
+
+        assertEquals( testStr, writer.toString() );
+    }
+
     public void testShouldReadFileContentsUsingByteArray() throws IOException
     {
         File file = File.createTempFile( "test.", ".file-location" );
         file.deleteOnExit();
-        
+
         String testStr = "This is a test";
-        
+
         TestUtils.writeToFile( file, testStr );
-        
+
         FileLocation location = new FileLocation( file, file.getAbsolutePath() );
-        
+
         location.open();
-        
+
         byte[] buffer = new byte[ testStr.length() ];
         location.read( buffer );
-        
+
         assertEquals( testStr, new String( buffer ) );
     }
-    
+
     public void testShouldReadThenClose() throws IOException
     {
         File file = File.createTempFile( "test.", ".file-location" );
         file.deleteOnExit();
-        
+
         String testStr = "This is a test";
-        
+
         TestUtils.writeToFile( file, testStr );
-        
+
         FileLocation location = new FileLocation( file, file.getAbsolutePath() );
-        
+
         location.open();
-        
+
         byte[] buffer = new byte[ testStr.length() ];
         location.read( buffer );
-        
+
         assertEquals( testStr, new String( buffer ) );
-        
+
         location.close();
     }
-    
+
     public void testShouldOpenThenFailToSetFile() throws IOException
     {
         File file = File.createTempFile( "test.", ".file-location" );
         file.deleteOnExit();
-        
+
         TestFileLocation location = new TestFileLocation( file.getAbsolutePath() );
-        
+
         location.open();
-        
+
         try
         {
             location.setFile( file );
-            
+
             fail( "should not succeed." );
         }
         catch( IllegalStateException e )
         {
         }
     }
-    
+
     public void testShouldConstructWithoutFileThenSetFileThenOpen() throws IOException
     {
         File file = File.createTempFile( "test.", ".file-location" );
         file.deleteOnExit();
-        
+
         TestFileLocation location = new TestFileLocation( file.getAbsolutePath() );
-        
+
         location.setFile( file );
         location.open();
     }
-    
+
     public void testShouldConstructWithLocationThenRetrieveEquivalentFile() throws IOException
     {
         File file = File.createTempFile( "test.", ".file-location" );
         file.deleteOnExit();
-        
+
         Location location = new TestFileLocation( file.getAbsolutePath() );
-        
+
         assertEquals( file, location.getFile() );
         assertEquals( file.getAbsolutePath(), location.getSpecification() );
     }
-    
+
     private static final class TestFileLocation extends FileLocation
     {
 
@@ -131,7 +154,7 @@ public class FileLocationTest
         {
             super( specification );
         }
-        
+
     }
 
 }