io.descoped.rawdata.api.RawdataSubscriber Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of rawdata-client-api Show documentation
Rawdata Client API
The newest version!
package io.descoped.rawdata.api;

import java.util.Objects;
import java.util.concurrent.Flow;
import java.util.concurrent.atomic.AtomicLong;
import java.util.concurrent.atomic.AtomicReference;
import java.util.function.BiConsumer;
import java.util.function.Consumer;
import java.util.function.Supplier;

class RawdataSubscriber implements Flow.Subscriber {

    final Supplier producerSupplier;
    final AtomicReference producerRef = new AtomicReference<>();
    final AtomicReference subscriptionRef = new AtomicReference<>();
    final AtomicLong received = new AtomicLong();
    final BiConsumer onError;
    final Consumer onComplete;

    RawdataSubscriber(Supplier producerSupplier, Consumer onComplete, BiConsumer onError) {
        this.producerSupplier = producerSupplier;
        this.onError = onError;
        this.onComplete = onComplete;
    }

    RawdataSubscriber(Supplier producerSupplier, Consumer onComplete) {
        this.producerSupplier = producerSupplier;
        this.onError = null;
        this.onComplete = onComplete;
    }

    RawdataSubscriber(Supplier producerSupplier, BiConsumer onError) {
        this.producerSupplier = producerSupplier;
        this.onError = onError;
        this.onComplete = null;
    }

    RawdataSubscriber(Supplier producerSupplier) {
        this.producerSupplier = producerSupplier;
        this.onError = null;
        this.onComplete = null;
    }

    @Override
    public void onSubscribe(Flow.Subscription subscription) {
        if (!subscriptionRef.compareAndSet(null, subscription)) {
            subscription.cancel();
            return;
        }
        producerRef.set(producerSupplier.get());
        subscription.request(10);
    }

    @Override
    public void onNext(RawdataMessage item) {
        RawdataProducer producer = producerRef.get();
        producer.publish(item);
        if (received.incrementAndGet() % 5L == 0L) {
            subscriptionRef.get().request(5);
        }
    }

    @Override
    public void onError(Throwable throwable) {
        try {
            Objects.requireNonNull(throwable);
            if (onError != null) {
                onError.accept(throwable, producerRef.get());
            }
        } finally {
            close();
        }
    }

    @Override
    public void onComplete() {
        try {
            if (onComplete != null) {
                onComplete.accept(producerRef.get());
            }
        } finally {
            close();
        }
    }

    private void close() {
        subscriptionRef.set(null);
        RawdataProducer producer = producerRef.getAndSet(null);
        if (producer != null) {
            try {
                producer.close();
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
    }
}