@Test
public void forEachXWithEvents() {
List<Integer> list = new ArrayList<>();
LazyFutureStream<Integer> stream =
LazyFutureStream.of(
() -> 1,
() -> 2,
() -> 3,
(Supplier<Integer>)
() -> {
throw new RuntimeException();
})
.map(Supplier::get);
Subscription s =
stream.forEachXEvents(2, i -> list.add(i), e -> error = e, () -> complete = true);
assertThat(list, hasItems(1, 2));
assertThat(list.size(), equalTo(2));
System.out.println("first batch");
s.request(1);
assertFalse(complete);
assertThat(list, hasItems(1, 2, 3));
assertThat(list.size(), equalTo(3));
assertThat(error, nullValue());
s.request(2);
assertThat(error, instanceOf(RuntimeException.class));
assertTrue(complete);
}
@Override
public void onNext(Long n) {
BackpressureUtils.getAndAdd(COUNTED, AdaptiveConsumerAction.this, n);
Subscription upstreamSubscription = AdaptiveConsumerAction.this.upstreamSubscription;
if (upstreamSubscription != null) {
upstreamSubscription.request(n);
}
Subscription s = this.s;
if (s != null) {
s.request(1l);
}
}
/**
* Atomically sets the single subscription and requests the missed amount from it.
*
* @param s the Subscription to set.
* @return false if this arbiter is cancelled or there was a subscription already set
*/
protected final boolean set(Subscription s) {
Objects.requireNonNull(s, "s");
Subscription a = this.s;
if (a == Operators.cancelledSubscription()) {
s.cancel();
return false;
}
if (a != null) {
s.cancel();
Operators.reportSubscriptionSet();
return false;
}
if (S.compareAndSet(this, null, s)) {
long r = REQUESTED.getAndSet(this, 0L);
if (r != 0L) {
s.request(r);
}
return true;
}
a = this.s;
if (a != Operators.cancelledSubscription()) {
s.cancel();
return false;
}
Operators.reportSubscriptionSet();
return false;
}
/** Requests the deferred amount if not zero. */
protected final void requestDeferred() {
long r = REQUESTED.getAndSet(this, 0L);
if (r != 0L) {
s.request(r);
}
}
@Override
protected void requestMore(long n) {
Subscription subscription = SUBSCRIPTION.get(this);
if (subscription != UNSUBSCRIBED) {
subscription.request(n);
}
}
@Override
protected void doOnSubscribe(Subscription subscription) {
long p = pendingRequests;
if (p > 0) {
subscription.request(p);
}
}
@Test
public void forEachX() {
Subscription s = LazyFutureStream.of(1, 2, 3).forEachX(2, System.out::println);
System.out.println("first batch");
s.request(1);
}
@Override
public void onSubscribe(Subscription s) {
if (SubscriptionHelper.validateSubscription(this.s, s)) {
return;
}
this.s = s;
Subscriber<? super U> actual = this.actual;
U b;
try {
b = bufferSupplier.get();
} catch (Throwable e) {
cancelled = true;
s.cancel();
EmptySubscription.error(e, actual);
return;
}
if (b == null) {
cancelled = true;
s.cancel();
EmptySubscription.error(new NullPointerException("The buffer supplied is null"), actual);
return;
}
buffer = b;
Publisher<B> boundary;
try {
boundary = boundarySupplier.get();
} catch (Throwable ex) {
cancelled = true;
s.cancel();
EmptySubscription.error(ex, actual);
return;
}
if (boundary == null) {
cancelled = true;
s.cancel();
EmptySubscription.error(
new NullPointerException("The boundary publisher supplied is null"), actual);
return;
}
BufferBoundarySubscriber<T, U, B> bs = new BufferBoundarySubscriber<T, U, B>(this);
other = bs;
actual.onSubscribe(this);
if (!cancelled) {
s.request(Long.MAX_VALUE);
boundary.subscribe(bs);
}
}
protected final void normalRequest(long n) {
Subscription a = s;
if (a != null) {
a.request(n);
} else {
Operators.addAndGet(REQUESTED, this, n);
a = s;
if (a != null) {
long r = REQUESTED.getAndSet(this, 0L);
if (r != 0L) {
a.request(r);
}
}
}
}
@Override
public void onSubscribe(Subscription s) {
super.onSubscribe(s);
subscription = s;
long r = pendingRequests;
if (r > 0) {
subscription.request(r);
}
}
@Override
public void onSubscribe(Subscription s) {
if (Operators.validate(this.s, s)) {
this.s = s;
actual.onSubscribe(this);
s.request(Long.MAX_VALUE);
}
}
@Override
public void onSubscribe(final Subscription s) {
if (Operators.validate(subscription, s)) {
subscription = s;
ctx.channel().closeFuture().addListener(this);
s.request(1L);
}
}
@Test
public void forEachXTest() {
List<Integer> list = new ArrayList<>();
Subscription s = LazyFutureStream.of(1, 2, 3).forEachX(2, i -> list.add(i));
assertThat(list, hasItems(1, 2));
assertThat(list.size(), equalTo(2));
s.request(1);
assertThat(list, hasItems(1, 2, 3));
assertThat(list.size(), equalTo(3));
}
@Override
protected void requestUpstream(long capacity, boolean terminated, long elements) {
FallbackSubscriber fallbackSubscriber = this.fallbackSubscriber;
if (fallbackSubscriber == null) {
BackpressureUtils.getAndAdd(REQUESTED, this, elements);
super.requestUpstream(capacity, terminated, elements);
} else {
Subscription subscription = fallbackSubscriber.subscription;
if (subscription != null) {
subscription.request(elements);
}
}
}
@Override
public void operationComplete(ChannelFuture future) throws Exception {
if (!future.isSuccess()) {
promise.tryFailure(future.cause());
if (log.isDebugEnabled()) {
log.debug("Write error", future.cause());
}
return;
}
if (subscription != null) {
subscription.request(1L);
}
}
@Override
public void request(long n) {
if (subscription != null) {
subscription.request(n);
}
}
@Override
public void onSubscribe(Subscription s) {
this.s = s;
s.request(1l);
}
@Override
public void onSubscribe(Subscription s) {
s.request(Long.MAX_VALUE);
}
final void requestMore(long n) {
Subscription s = this.subscription;
if (s != null) {
s.request(n);
}
}