io.scalecube.services.HeadAndTail Maven / Gradle / Ivy

Go to download
package io.scalecube.services;

import org.reactivestreams.Publisher;

import java.util.concurrent.atomic.AtomicBoolean;

import reactor.core.publisher.Flux;
import reactor.core.publisher.UnicastProcessor;

public final class HeadAndTail {

  private final T head;
  private final Publisher tail;

  private HeadAndTail(T head, Publisher tail) {
    this.head = head;
    this.tail = tail;
  }

  public static  Publisher> createFrom(Publisher publisher) {
    AtomicBoolean first = new AtomicBoolean(true);
    UnicastProcessor tail = UnicastProcessor.create();
    UnicastProcessor> firstResult = UnicastProcessor.create();

    return Flux.from(publisher)
        .doOnComplete(tail::onComplete)
        .doOnError(tail::onError)
        .flatMap(message -> {
          if (first.compareAndSet(true, false)) {
            firstResult.onNext(new HeadAndTail<>(message, tail));
            firstResult.onComplete();
            return firstResult;
          } else {
            tail.onNext(message);
            return Flux.empty();
          }
        });
  }

  public T head() {
    return head;
  }

  public Publisher tail() {
    return tail;
  }
}