com.carrotsearch.ant.tasks.junit4.slave.StdInLineIterator Maven / Gradle / Ivy

Go to download
package com.carrotsearch.ant.tasks.junit4.slave;

import java.io.*;
import java.nio.charset.Charset;

import com.carrotsearch.ant.tasks.junit4.events.IdleEvent;
import com.carrotsearch.ant.tasks.junit4.events.Serializer;
import com.google.common.collect.AbstractIterator;

/**
 * Iterates over lines from standard input. 
 */
class StdInLineIterator extends AbstractIterator {
  private BufferedReader reader;
  private Serializer serializer;

  public StdInLineIterator(Serializer serializer) {
    this.serializer = serializer;
    this.reader = new BufferedReader(
      new InputStreamReader(
          System.in,
          Charset.defaultCharset()));
  }

  @Override
  protected String computeNext() {
    try {
      serializer.serialize(new IdleEvent());
      serializer.flush();

      String line = reader.readLine();
      return line != null ? line : endOfData();
    } catch (IOException e) {
      throw new RuntimeException(e);
    }
  }
}