#9 - Added URI based stream factories to WarcRecordIteratorFactory.

Author: laxika

java-warc/src/main/java/com/github/bottomlessarchive/warc/service/WarcRecordIteratorFactory.java

@@ -6,6 +6,8 @@
 import java.io.BufferedInputStream;
 import java.io.IOException;
 import java.io.InputStream;
+import java.net.MalformedURLException;
+import java.net.URI;
 import java.net.URL;
 import java.nio.charset.Charset;
 import java.util.Iterator;
@@ -15,6 +17,14 @@ public class WarcRecordIteratorFactory {
 
     private static final int ONE_MEGABYTE_IN_BYTE = 1048576;
 
+    public static <T extends WarcContentBlock> Iterator<WarcRecord<T>> iteratorOf(final URI uri) {
+        try {
+            return iteratorOf(uri.toURL());
+        } catch (MalformedURLException e) {
+            throw new WarcNetworkException("Unable to parse WARC location: " + uri + "!", e);
+        }
+    }
+
     public static <T extends WarcContentBlock> Iterator<WarcRecord<T>> iteratorOf(final URL url) {
         try {
             return iteratorOf(new AvailableInputStream(new BufferedInputStream(url.openStream(), ONE_MEGABYTE_IN_BYTE)),
@@ -33,7 +43,7 @@ public static <T extends WarcContentBlock> Iterator<WarcRecord<T>> iteratorOf(fi
     }
 
     public static <T extends WarcContentBlock> Iterator<WarcRecord<T>> iteratorOf(final InputStream inputStream, final Charset charset,
-            final boolean compressed) {
+                                                                                  final boolean compressed) {
         final WarcReader warcReader = new WarcReader(inputStream, charset, compressed);
 
         return (Iterator<WarcRecord<T>>) new WarcRecordIterator(warcReader);