package reactor.core.publisher;
import org.reactivestreams.Subscription;
import reactor.core.CoreSubscriber;
import reactor.util.annotation.Nullable;
import reactor.util.context.Context;
final class FluxSkip<T> extends InternalFluxOperator<T, T> {
final long n;
FluxSkip(Flux<? extends T> source, long n) {
super(source);
if (n < 0) {
throw new IllegalArgumentException("n >= 0 required but it was " + n);
}
this.n = n;
}
@Override
public CoreSubscriber<? super T> subscribeOrReturn(CoreSubscriber<? super T> actual) {
return new SkipSubscriber<>(actual, n);
}
@Override
public Object scanUnsafe(Attr key) {
if (key == Attr.RUN_STYLE) return Attr.RunStyle.SYNC;
return super.scanUnsafe(key);
}
static final class SkipSubscriber<T>
implements InnerOperator<T, T> {
final CoreSubscriber<? super T> actual;
final Context ctx;
long remaining;
Subscription s;
SkipSubscriber(CoreSubscriber<? super T> actual, long n) {
this.actual = actual;
this.ctx = actual.currentContext();
this.remaining = n;
}
@Override
public void onSubscribe(Subscription s) {
if (Operators.validate(this.s, s)) {
this.s = s;
long n = remaining;
actual.onSubscribe(this);
s.request(n);
}
}
@Override
public void onNext(T t) {
long r = remaining;
if (r == 0L) {
actual.onNext(t);
}
else {
Operators.onDiscard(t, ctx);
remaining = r - 1;
}
}
@Override
public void onError(Throwable t) {
actual.onError(t);
}
@Override
public void onComplete() {
actual.onComplete();
}
@Override
@Nullable
public Object scanUnsafe(Attr key) {
if (key == Attr.PARENT) return s;
if (key == Attr.RUN_STYLE) return Attr.RunStyle.SYNC;
return InnerOperator.super.scanUnsafe(key);
}
@Override
public CoreSubscriber<? super T> actual() {
return actual;
}
@Override
public void request(long n) {
s.request(n);
}
@Override
public void cancel() {
s.cancel();
}
}
}