io.smallrye.mutiny.helpers.spies.MultiOnSubscribeSpy Maven / Gradle / Ivy

Go to download

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

Intuitive Event-Driven Reactive Programming Library for Java

There is a newer version: 2.8.0

package io.smallrye.mutiny.helpers.spies;

import java.util.concurrent.Flow.Subscription;

import io.smallrye.mutiny.Multi;
import io.smallrye.mutiny.subscription.MultiSubscriber;

public class MultiOnSubscribeSpy extends MultiSpyBase {

    private volatile Subscription lastSubscription;

    public Subscription lastSubscription() {
        return lastSubscription;
    }

    MultiOnSubscribeSpy(Multi upstream) {
        super(upstream);
    }

    @Override
    public void reset() {
        super.reset();
        lastSubscription = null;
    }

    @Override
    public void subscribe(MultiSubscriber downstream) {
        upstream.onSubscription().invoke(subscription -> {
            incrementInvocationCount();
            lastSubscription = subscription;
        }).subscribe().withSubscriber(downstream);
    }

    @Override
    public String toString() {
        return "MultiOnSubscribeSpy{" +
                "lastSubscription=" + lastSubscription +
                "} " + super.toString();
    }
}