Учитывая Java InputStream, как я могу определить текущее смещение в потоке? - PullRequest
13 голосов
/ 27 октября 2008

Я бы хотел что-то вроде универсального, многократно используемого getPosition() метода, который сообщит мне количество байтов, прочитанных из начальной точки потока. В идеале, я бы предпочел, чтобы это работало со всеми InputStreams, чтобы мне не нужно было оборачивать каждый из них, поскольку я получаю их из разнородных источников.

Существует ли такой зверь? Если нет, то может ли кто-нибудь порекомендовать существующую реализацию подсчета InputStream?

Ответы [ 4 ]

20 голосов
/ 27 октября 2008

Вам нужно будет следовать шаблону Decorator, установленному в java.io, чтобы реализовать это.

Давайте попробуем здесь:

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();
}
12 голосов
/ 27 октября 2008

Взгляните на CountingInputStream в пакете ввода-вывода Commons. У них также есть неплохая коллекция других полезных вариантов InputStream.

2 голосов
/ 17 сентября 2015
2 голосов
/ 27 октября 2008

Нет. InputStream предназначен для обработки потенциально бесконечных объемов данных, поэтому счетчик будет мешать. Помимо того, что вы можете обернуть их все, вы можете сделать что-то с аспектами.

...