Changeset 8b8d32e

Timestamp:

Dec 18, 2017 10:23:42 PM (3 years ago)

Author:

zzz <zzz@…>

Branches:

master

Children:

73796458

Parents:

be004cd

Message:

Util: LookaheadInputStream? speedups and cleanups

File:

• core/java/src/net/i2p/util/LookaheadInputStream.java (modified) (9 diffs)

core/java/src/net/i2p/util/LookaheadInputStream.java

@@ -7 +7 @@
 import java.util.Arrays;
 
+import net.i2p.data.DataHelper;
+
 /**
  * Simple lookahead buffer to keep the last K bytes in reserve,
  * configured to easily be reused.  Currently only used by the
- * ResettableGZIPInputStream
- *
+ * ResettableGZIPInputStream.
  */
 public class LookaheadInputStream extends FilterInputStream {
     private boolean _eofReached;
     private final byte[] _footerLookahead;
+    private final int size;
+    // Next byte to read.
+    private int index;
     private static final InputStream _fakeInputStream = new ByteArrayInputStream(new byte[0]);
     
@@ -30 +34 @@
         super(_fakeInputStream);
         _footerLookahead = new byte[lookaheadSize];
+        size = lookaheadSize;
     }
     
@@ -38 +43 @@
      *  Resets everything if the LookaheadInputStream was previously used.
      *  WARNING - blocking until lookaheadSize bytes are read!
+     *
+     *  @throws IOException if less than lookaheadSize bytes could be read.
      */
     public void initialize(InputStream src) throws IOException {
         in = src;
         _eofReached = false;
-        Arrays.fill(_footerLookahead, (byte)0x00);
-        int footerRead = 0;
-        while (footerRead < _footerLookahead.length) {
-            int read = in.read(_footerLookahead, footerRead, _footerLookahead.length - footerRead);
-            if (read == -1) throw new IOException("EOF reading the footer lookahead");
-            footerRead += read;
-        }
+        index = 0;
+        DataHelper.read(in, _footerLookahead);
     }
     
@@ -54 +56 @@
     public int read() throws IOException {
         if (_eofReached) 
-            return -1; //throw new IOException("Already past the EOF");
+            return -1;
         int c = in.read();
         if (c == -1) {
@@ -60 +62 @@
             return -1;
         }
-        int rv = _footerLookahead[0];
-        // FIXME use an index!!!!!!!!!!!!
-        System.arraycopy(_footerLookahead, 1, _footerLookahead, 0, _footerLookahead.length-1);
-        _footerLookahead[_footerLookahead.length-1] = (byte)c;
-        if (rv < 0) rv += 256;
+        int rv = _footerLookahead[index] & 0xff;
+        _footerLookahead[index] = (byte)c;
+        index++;
+        if (index >= size)
+            index = 0;
         return rv;
-    }
-
-    @Override
-    public int read(byte buf[]) throws IOException {
-        return read(buf, 0, buf.length);
     }
 
@@ -91 +88 @@
     }
     
-    /** grab the lookahead footer */
-    public byte[] getFooter() { return _footerLookahead; }
+    /**
+     * Grab the lookahead footer.
+     * This will be of size lookaheadsize given in constructor.
+     * The last byte received will be in the last byte of the array.
+     */
+    public byte[] getFooter() {
+        if (index == 0)
+            return _footerLookahead;
+        byte[] rv = new byte[size];
+        System.arraycopy(_footerLookahead, index, rv, 0, size - index);
+        System.arraycopy(_footerLookahead, 0, rv, size - index, index);
+        return rv;
+    }
+    
+    /**
+     *  @since 0.9.33
+     */
+    @Override
+    public long skip(long n) throws IOException {
+        long rv = 0;
+        int c;
+        while (rv < n && (c = read()) >= 0) {
+            rv++;
+        }
+        return rv;
+    }
     
 /*******
@@ -112 +133 @@
                 if (lis.getFooter()[i] != (byte)(i+24))
                     throw new RuntimeException("Error at footer " + i + " [" + lis.getFooter()[i] + "]");
-            System.out.println("Everything is fine in general");
         } catch (Exception e) {
             e.printStackTrace();
@@ -118 +138 @@
         
         for (int i = 9; i < 32*1024; i++) {
-            if (!test(i)) break;
+            if (!test(i)) {
+                System.out.println("Everything is NOT fine at size=" + i);
+                break;
+            }
         }
+        System.out.println("Everything is fine in general");
     }
     
@@ -138 +162 @@
                 if (lis.getFooter()[i] != buf[i+(size-8)])
                     throw new RuntimeException("Error at footer " + i + " [" + lis.getFooter()[i] + "]");
-            System.out.println("Everything is fine at size=" + size);
+            //System.out.println("Everything is fine at size=" + size);
             return true;
         } catch (Exception e) {