io.reactivex.rxjava3.internal.operators.flowable.BlockingFlowableNext Maven / Gradle / Ivy
Show all versions of redisson-all Show documentation
/*
* Copyright (c) 2016-present, RxJava Contributors.
*
* Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in
* compliance with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software distributed under the License is
* distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See
* the License for the specific language governing permissions and limitations under the License.
*/
package io.reactivex.rxjava3.internal.operators.flowable;
import java.util.*;
import java.util.concurrent.*;
import java.util.concurrent.atomic.AtomicInteger;
import org.reactivestreams.Publisher;
import io.reactivex.rxjava3.core.*;
import io.reactivex.rxjava3.internal.util.*;
import io.reactivex.rxjava3.plugins.RxJavaPlugins;
import io.reactivex.rxjava3.subscribers.DisposableSubscriber;
/**
* Returns an Iterable that blocks until the Observable emits another item, then returns that item.
*
* 
*
* @param the value type
*/
public final class BlockingFlowableNext implements Iterable {
final Publisher source;
public BlockingFlowableNext(Publisher source) {
this.source = source;
}
@Override
public Iterator iterator() {
NextSubscriber nextSubscriber = new NextSubscriber<>();
return new NextIterator<>(source, nextSubscriber);
}
// test needs to access the observer.waiting flag
static final class NextIterator implements Iterator {
private final NextSubscriber subscriber;
private final Publisher items;
private T next;
private boolean hasNext = true;
private boolean isNextConsumed = true;
private Throwable error;
private boolean started;
NextIterator(Publisher items, NextSubscriber subscriber) {
this.items = items;
this.subscriber = subscriber;
}
@Override
public boolean hasNext() {
if (error != null) {
// If any error has already been thrown, throw it again.
throw ExceptionHelper.wrapOrThrow(error);
}
// Since an iterator should not be used in different thread,
// so we do not need any synchronization.
if (!hasNext) {
// the iterator has reached the end.
return false;
}
// next has not been used yet.
return !isNextConsumed || moveToNext();
}
private boolean moveToNext() {
try {
if (!started) {
started = true;
// if not started, start now
subscriber.setWaiting();
Flowable.fromPublisher(items)
.materialize().subscribe(subscriber);
}
Notification nextNotification = subscriber.takeNext();
if (nextNotification.isOnNext()) {
isNextConsumed = false;
next = nextNotification.getValue();
return true;
}
// If an observable is completed or fails,
// hasNext() always return false.
hasNext = false;
if (nextNotification.isOnComplete()) {
return false;
}
error = nextNotification.getError();
throw ExceptionHelper.wrapOrThrow(error);
} catch (InterruptedException e) {
subscriber.dispose();
error = e;
throw ExceptionHelper.wrapOrThrow(e);
}
}
@Override
public T next() {
if (error != null) {
// If any error has already been thrown, throw it again.
throw ExceptionHelper.wrapOrThrow(error);
}
if (hasNext()) {
isNextConsumed = true;
return next;
}
else {
throw new NoSuchElementException("No more elements");
}
}
@Override
public void remove() {
throw new UnsupportedOperationException("Read only iterator");
}
}
static final class NextSubscriber extends DisposableSubscriber> {
private final BlockingQueue> buf = new ArrayBlockingQueue<>(1);
final AtomicInteger waiting = new AtomicInteger();
@Override
public void onComplete() {
// ignore
}
@Override
public void onError(Throwable e) {
RxJavaPlugins.onError(e);
}
@Override
public void onNext(Notification args) {
if (waiting.getAndSet(0) == 1 || !args.isOnNext()) {
Notification toOffer = args;
while (!buf.offer(toOffer)) {
Notification concurrentItem = buf.poll();
// in case if we won race condition with onComplete/onError method
if (concurrentItem != null && !concurrentItem.isOnNext()) {
toOffer = concurrentItem;
}
}
}
}
public Notification takeNext() throws InterruptedException {
setWaiting();
BlockingHelper.verifyNonBlocking();
return buf.take();
}
void setWaiting() {
waiting.set(1);
}
}
}