mutiny.zero.internal.TubePublisher Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mutiny-zero Show documentation

Mutiny Zero is a minimal API for creating reactive-streams compliant publishers

There is a newer version: 1.1.1

package mutiny.zero.internal;

import static java.util.Objects.requireNonNull;

import java.util.concurrent.Flow;
import java.util.concurrent.Flow.Publisher;
import java.util.function.Consumer;

import mutiny.zero.BackpressureStrategy;
import mutiny.zero.Tube;
import mutiny.zero.TubeConfiguration;

public class TubePublisher implements Publisher {

    private final BackpressureStrategy backpressureStrategy;
    private final int bufferSize;
    private final Consumer> tubeConsumer;

    public TubePublisher(TubeConfiguration configuration, Consumer> tubeConsumer) {
        this.backpressureStrategy = configuration.getBackpressureStrategy();
        this.bufferSize = configuration.getBufferSize();
        this.tubeConsumer = tubeConsumer;
    }

    @Override
    public void subscribe(Flow.Subscriber subscriber) {
        requireNonNull(subscriber, "The subscriber cannot be null");
        TubeBase tube = null;
        switch (backpressureStrategy) {
            case BUFFER:
                tube = new BufferingTube<>(subscriber, bufferSize);
                break;
            case UNBOUNDED_BUFFER:
                tube = new UnbounbedBufferingTube<>(subscriber);
                break;
            case DROP:
                tube = new DroppingTube<>(subscriber);
                break;
            case ERROR:
                tube = new ErroringTube<>(subscriber);
                break;
            case IGNORE:
                tube = new IgnoringTube<>(subscriber);
                break;
            case LATEST:
                tube = new LatestTube<>(subscriber, bufferSize);
                break;
        }
        subscriber.onSubscribe(tube);
        tubeConsumer.accept(tube);
    }
}