net.pincette.rs.Separator Maven / Gradle / Ivy

Go to download
package net.pincette.rs;

import java.util.concurrent.Flow.Processor;
import java.util.function.Supplier;

/**
 * A processor which emits a given value between the incoming value stream.
 *
 * @param  the value type.
 * @author Werner Donn\u00e9
 * @since 1.0
 */
public class Separator extends Mapper {
  private final Supplier value;
  private boolean extra;
  private boolean first = true;

  public Separator(final T value) {
    this(() -> value);
  }

  public Separator(final Supplier value) {
    super(v -> v);
    this.value = value;
  }

  public static  Processor separator(final T value) {
    return new Separator<>(value);
  }

  public static  Processor separator(final Supplier value) {
    return new Separator<>(value);
  }

  @Override
  protected boolean canRequestMore(long n) {
    if (extra) {
      extra = false;

      return false;
    }

    return true;
  }

  @Override
  public void onNext(final T value) {
    if (first) {
      first = false;
    } else {
      extra = true;
      super.onNext(this.value.get());
    }

    super.onNext(value);
  }
}