ModifyAwareIterator.java

package io.ebeaninternal.server.type;

import java.util.Iterator;

/**
 * Wraps an iterator for the purposes of detecting modifications.
 */
public class ModifyAwareIterator<E> implements Iterator<E> {

  private final ModifyAwareOwner owner;

  private final Iterator<E> it;

  /**
   * Create with an Owner and the underlying Iterator this wraps.
   * <p>
   * The owner is notified of the removals.
   * </p>
   */
  public ModifyAwareIterator(ModifyAwareOwner owner, Iterator<E> it) {
    this.owner = owner;
    this.it = it;
  }

  @Override
  public boolean hasNext() {
    return it.hasNext();
  }

  @Override
  public E next() {
    return it.next();
  }

  @Override
  public void remove() {
    owner.markAsModified();
    it.remove();
  }

}