WarpPI/core/src/main/java/it/cavallium/warppi/flow/ObservableZipped.java
2018-09-22 11:17:30 +02:00

77 lines
1.9 KiB
Java

package it.cavallium.warppi.flow;
import org.apache.commons.lang3.tuple.Pair;
public class ObservableZipped<T, U> extends Observable<Pair<T, U>> {
private volatile boolean initialized = false;
private final Observable<T> a;
private final Observable<U> b;
private Disposable disposableA;
private Disposable disposableB;
private volatile T lastA;
private volatile U lastB;
private volatile boolean didA;
private volatile boolean didB;
public ObservableZipped(final Observable<T> a, final Observable<U> b) {
super();
this.a = a;
this.b = b;
}
private void initialize() {
this.disposableA = a.subscribe((t) -> {
lastA = t;
didA = true;
receivedNext();
}, (e) -> {
for (final Subscriber<? super Pair<T, U>> sub : subscribers)
sub.onError(e);;
}, () -> {
for (final Subscriber<? super Pair<T, U>> sub : subscribers)
sub.onComplete();;
});
this.disposableB = b.subscribe((t) -> {
lastB = t;
didB = true;
receivedNext();
}, (e) -> {
for (final Subscriber<? super Pair<T, U>> sub : subscribers)
sub.onError(e);;
}, () -> {
for (final Subscriber<? super Pair<T, U>> sub : subscribers)
sub.onComplete();;
});
}
private void receivedNext() {
if (didA && didB) {
didA = false;
didB = false;
for (final Subscriber<? super Pair<T, U>> sub : subscribers)
sub.onNext(Pair.of(lastA, lastB));;
}
}
private void chechInitialized() {
if (!initialized) {
initialized = true;
initialize();
}
}
@Override
public Disposable subscribe(final Subscriber<? super Pair<T, U>> sub) {
final Disposable disp = super.subscribe(sub);
chechInitialized();
return disp;
}
@Override
public void onDisposed(final Subscriber<? super Pair<T, U>> sub) {
super.onDisposed(sub);
this.disposableA.dispose();
this.disposableB.dispose();
}
}