
package org.apache.beam.runners.dataflow.worker.util.common.worker;

import java.io.Closeable;
import javax.annotation.concurrent.NotThreadSafe;
import org.apache.beam.sdk.util.common.Reiterator;
import org.checkerframework.checker.nullness.qual.Nullable;

/**
 * ShuffleEntryReader provides an interface for reading key/value entries from a shuffle dataset.
 */
@NotThreadSafe
public interface ShuffleEntryReader extends Closeable {
  /**
   * Returns an iterator that reads a range of entries from a shuffle dataset.
   *
   * @param startPosition encodes the initial key from where to read. This parameter may be null,
   *     indicating that the read should start with the first key in the dataset.
   * @param endPosition encodes the key "just past" the end of the range to be read; keys up to
   *     endPosition will be returned, but keys equal to or greater than endPosition will not. This
   *     parameter may be null, indicating that the read should end just past the last key in the
   *     dataset (that is, the last key in the dataset will be included in the read, as long as that
   *     key is greater than or equal to startPosition).
   * @return a {@link Reiterator} over the requested range of entries.
   */
  Reiterator<ShuffleEntry> read(
      @Nullable ShufflePosition startPosition, @Nullable ShufflePosition endPosition);
}
