package reactor.core.publisher;
import java.util.Objects;
import java.util.stream.Stream;
import org.reactivestreams.Publisher;
import org.reactivestreams.Subscriber;
import org.reactivestreams.Subscription;
import reactor.core.CoreSubscriber;
import reactor.core.Exceptions;
import reactor.core.Scannable;
import reactor.util.annotation.Nullable;
import reactor.util.context.Context;
final class DelegateProcessor<IN, OUT> extends FluxProcessor<IN, OUT> {
final Publisher<OUT> downstream;
final Subscriber<IN> upstream;
DelegateProcessor(Publisher<OUT> downstream,
Subscriber<IN> upstream) {
this.downstream = Objects.requireNonNull(downstream, "Downstream must not be null");
this.upstream = Objects.requireNonNull(upstream, "Upstream must not be null");
}
@Override
public Context currentContext() {
if(upstream instanceof CoreSubscriber){
return ((CoreSubscriber)upstream).currentContext();
}
return Context.empty();
}
@Override
public void onComplete() {
upstream.onComplete();
}
@Override
public void onError(Throwable t) {
upstream.onError(t);
}
@Override
public void onNext(IN in) {
upstream.onNext(in);
}
@Override
public void onSubscribe(Subscription s) {
upstream.onSubscribe(s);
}
@Override
public void subscribe(CoreSubscriber<? super OUT> actual) {
Objects.requireNonNull(actual, "subscribe");
downstream.subscribe(actual);
}
@Override
@SuppressWarnings("unchecked")
public boolean isSerialized() {
return upstream instanceof SerializedSubscriber ||
(upstream instanceof FluxProcessor &&
((FluxProcessor<?, ?>)upstream).isSerialized());
}
@Override
public Stream<? extends Scannable> inners() {
return Scannable.from(upstream)
.inners();
}
@Override
public int getBufferSize() {
return Scannable.from(upstream)
.scanOrDefault(Attr.CAPACITY, super.getBufferSize());
}
@Override
@Nullable
public Throwable getError() {
return Scannable.from(upstream)
.scanOrDefault(Attr.ERROR, super.getError());
}
@Override
public boolean isTerminated() {
return Scannable.from(upstream)
.scanOrDefault(Attr.TERMINATED, super.isTerminated());
}
@Override
@Nullable
public Object scanUnsafe(Attr key) {
if (key == Attr.PARENT) {
return downstream;
}
return Scannable.from(upstream)
.scanUnsafe(key);
}
}