给定Java InputStream，如何确定流中的当前偏移量？

StackOverflow https://stackoverflow.com/questions/240294

  •  04-07-2019
  •  | 
  •  

题

我喜欢通用的，可重用的 getPosition（）方法，它会告诉我从流的起始点读取的字节数。理想情况下，我更喜欢使用它来处理所有InputStream，因此当我从不同的源获取它们时，我不必将它们包装在每一个中。

这样的野兽存在吗？如果没有，任何人都可以推荐计数 InputStream 的现有实现吗？

有帮助吗？

解决方案

查看 CountingInputStream 在Commons IO包中。他们也有很多其他有用的InputStream变种。

其他提示

您需要遵循 java.io 中建立的Decorator模式来实现此目的。

我们来试试吧：

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;
  }

}

InputStreams旨在是线程安全的，因此可以自由地使用同步。我使用 volatile 和 AtomicLong 位置变量，但同步可能是最好的，因为它允许一个线程在流上操作并查询其位置而不放弃锁定。 

PositionInputStream is = …
synchronized (is) {
  is.read(buf);
  pos = is.getPosition();
}

没有。 InputStream 旨在处理可能无限量的数据，因此计数器会妨碍。除了将它们全部包装起来之外，您还可以对方面做些什么。

许可以下： CC-BY-SA 和 归因
不隶属于 StackOverflow
scroll top