Given a Java InputStream, how can I determine the current offset in the stream?

后端 未结 4 2112
轻奢々
轻奢々 2020-12-01 18:30

I\'d like something like a generic, re-usable getPosition() method that will tell me the number of bytes read from the starting point of the stream. Ideally, I

相关标签:
4条回答
  • 2020-12-01 19:02

    There is also CountingInputStream in Guava.

    Apidocs: https://google.github.io/guava/releases/19.0/api/docs/com/google/common/io/CountingInputStream.html

    Source: https://github.com/google/guava/blob/master/guava/src/com/google/common/io/CountingInputStream.java

    0 讨论(0)
  • 2020-12-01 19:06

    No. InputStream is intended to handle potentially infinite amounts of data, so a counter would get in the way. In addition to wrapping them all, you might be able to do something with aspects.

    0 讨论(0)
  • 2020-12-01 19:08

    You'll need to follow the Decorator pattern established in java.io to implement this.

    Let's give it a try here:

    import java.io.FilterInputStream;
    import java.io.IOException;
    import java.io.InputStream;
    
    public final class PositionInputStream
      extends FilterInputStream
    {
    
      private long pos = 0;
    
      private long mark = 0;
    
      public PositionInputStream(InputStream in)
      {
        super(in);
      }
    
      /**
       * <p>Get the stream position.</p>
       *
       * <p>Eventually, the position will roll over to a negative number.
       * Reading 1 Tb per second, this would occur after approximately three 
       * months. Applications should account for this possibility in their 
       * design.</p>
       *
       * @return the current stream position.
       */
      public synchronized long getPosition()
      {
        return pos;
      }
    
      @Override
      public synchronized int read()
        throws IOException
      {
        int b = super.read();
        if (b >= 0)
          pos += 1;
        return b;
      }
    
      @Override
      public synchronized int read(byte[] b, int off, int len)
        throws IOException
      {
        int n = super.read(b, off, len);
        if (n > 0)
          pos += n;
        return n;
      }
    
      @Override
      public synchronized long skip(long skip)
        throws IOException
      {
        long n = super.skip(skip);
        if (n > 0)
          pos += n;
        return n;
      }
    
      @Override
      public synchronized void mark(int readlimit)
      {
        super.mark(readlimit);
        mark = pos;
      }
    
      @Override
      public synchronized void reset()
        throws IOException
      {
        /* A call to reset can still succeed if mark is not supported, but the 
         * resulting stream position is undefined, so it's not allowed here. */
        if (!markSupported())
          throw new IOException("Mark not supported.");
        super.reset();
        pos = mark;
      }
    
    }
    

    The InputStreams are intended to be thread safe, so that accounts for the liberal use of synchronization. I played around with volatile and AtomicLong position variables, but synchronization is probably best because it allows one thread to operate on the stream and query its position without relinquishing the lock.

    PositionInputStream is = …
    synchronized (is) {
      is.read(buf);
      pos = is.getPosition();
    }
    
    0 讨论(0)
  • 2020-12-01 19:13

    Take a look at CountingInputStream in the Commons IO package. They have a pretty good collection of other useful InputStream variants as well.

    0 讨论(0)
提交回复
热议问题