异步迭代器

我有以下代码：

while(slowIterator.hasNext()) {
  performLengthTask(slowIterator.next());
}

由于迭代器和任务都很慢，因此将它们放入单独的线程中是有意义的。这是对迭代器包装器的快速而肮脏的尝试：

class AsyncIterator<T> implements Iterator<T> {
    private final BlockingQueue<T> queue = new ArrayBlockingQueue<T>(100);

    private AsyncIterator(final Iterator<T> delegate) {
      new Thread() {
        @Override
        public void run() {
          while(delegate.hasNext()) {
            queue.put(delegate.next()); // try/catch removed for brevity
          }
        }
      }.start();
    }

    @Override
    public boolean hasNext() {
      return true;
    }

    @Override
    public T next() {
        return queue.take(); // try/catch removed for brevity
    }
    // ... remove() throws UnsupportedOperationException
  }

然而这个实现缺乏对“hasNext()”的支持。当然，hasNext() 方法可以阻塞，直到它知道是否返回 true。我可以在 AsyncIterator 中有一个 peek 对象，并且可以更改 hasNext() 以从队列中获取一个对象，并让 next() 返回此 peek。但是，如果已到达委托迭代器的末尾，这将导致 hasNext() 无限期地阻塞。

我当然可以自己进行线程通信，而不是使用 ArrayBlockingQueue：

private static class AsyncIterator<T> implements Iterator<T> {

  private final Queue<T> queue = new LinkedList<T>();
  private boolean delegateDone = false;

  private AsyncIterator(final Iterator<T> delegate) {
    new Thread() {
      @Override
      public void run() {
        while (delegate.hasNext()) {
          final T next = delegate.next();
          synchronized (AsyncIterator.this) {
            queue.add(next);
            AsyncIterator.this.notify();
          }
        }
        synchronized (AsyncIterator.this) {
          delegateDone = true;
          AsyncIterator.this.notify();
        }
      }
    }.start();
  }

  @Override
  public boolean hasNext() {
    synchronized (this) {
      while (queue.size() == 0 && !delegateDone) {
        try {
          wait();
        } catch (InterruptedException e) {
          throw new Error(e);
        }
      }
    }
    return queue.size() > 0;
  }

  @Override
  public T next() {
    return queue.remove();
  }

  @Override
  public void remove() {
    throw new UnsupportedOperationException();
  }
}

然而，所有额外的同步、等待和通知并没有真正使代码更具可读性，并且很容易在某处隐藏竞争条件。

还有更好的想法吗？

Update

是的，我确实了解常见的观察者/可观察模式。然而，通常的实现并没有预见到数据流的结束，并且它们不是迭代器。

我在这里特别想要一个迭代器，因为实际上上面提到的循环存在于外部库中并且它需要一个迭代器。

这是一个棘手的问题，但我想这次我得到了正确的答案。 （我删除了我的第一个答案。）

答案是使用哨兵。我还没有测试过这段代码，为了清楚起见，我删除了 try/catch：

public class AsyncIterator<T> implements Iterator<T> {

    private BlockingQueue<T> queue = new ArrayBlockingQueue<T>(100);
    private T sentinel = (T) new Object();
    private T next;

    private AsyncIterator(final Iterator<T> delegate) {
        new Thread() {
            @Override
            public void run() {
                while (delegate.hasNext()) {
                    queue.put(delegate.next());
                }
                queue.put(sentinel);
            }
        }.start();
    }

    @Override
    public boolean hasNext() {
        if (next != null) {
            return true;
        }
        next = queue.take(); // blocks if necessary
        if (next == sentinel) {
            return false;
        }
        return true;
    }

    @Override
    public T next() {
        T tmp = next;
        next = null;
        return tmp;
    }

}

这里的见解是 hasNext() 需要阻塞，直到下一个项目准备好。它还需要某种退出条件，并且由于线程问题，它不能使用空队列或布尔标志。哨兵无需任何锁定或同步即可解决问题。

编辑：缓存“下一个”，因此可以多次调用 hasNext() 。