io.reactivex.internal.operators.flowable.FlowableFilter Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of rxjava Show documentation
Show all versions of rxjava Show documentation
Reactive Extensions for Java
/**
* 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.internal.operators.flowable;
import org.reactivestreams.Subscriber;
import io.reactivex.Flowable;
import io.reactivex.annotations.Nullable;
import io.reactivex.functions.Predicate;
import io.reactivex.internal.fuseable.*;
import io.reactivex.internal.subscribers.*;
public final class FlowableFilter extends AbstractFlowableWithUpstream {
final Predicate predicate;
public FlowableFilter(Flowable source, Predicate predicate) {
super(source);
this.predicate = predicate;
}
@Override
protected void subscribeActual(Subscriber s) {
if (s instanceof ConditionalSubscriber) {
source.subscribe(new FilterConditionalSubscriber(
(ConditionalSubscriber)s, predicate));
} else {
source.subscribe(new FilterSubscriber(s, predicate));
}
}
static final class FilterSubscriber extends BasicFuseableSubscriber
implements ConditionalSubscriber {
final Predicate filter;
FilterSubscriber(Subscriber actual, Predicate filter) {
super(actual);
this.filter = filter;
}
@Override
public void onNext(T t) {
if (!tryOnNext(t)) {
upstream.request(1);
}
}
@Override
public boolean tryOnNext(T t) {
if (done) {
return false;
}
if (sourceMode != NONE) {
downstream.onNext(null);
return true;
}
boolean b;
try {
b = filter.test(t);
} catch (Throwable e) {
fail(e);
return true;
}
if (b) {
downstream.onNext(t);
}
return b;
}
@Override
public int requestFusion(int mode) {
return transitiveBoundaryFusion(mode);
}
@Nullable
@Override
public T poll() throws Exception {
QueueSubscription qs = this.qs;
Predicate f = filter;
for (;;) {
T t = qs.poll();
if (t == null) {
return null;
}
if (f.test(t)) {
return t;
}
if (sourceMode == ASYNC) {
qs.request(1);
}
}
}
}
static final class FilterConditionalSubscriber extends BasicFuseableConditionalSubscriber {
final Predicate filter;
FilterConditionalSubscriber(ConditionalSubscriber actual, Predicate filter) {
super(actual);
this.filter = filter;
}
@Override
public void onNext(T t) {
if (!tryOnNext(t)) {
upstream.request(1);
}
}
@Override
public boolean tryOnNext(T t) {
if (done) {
return false;
}
if (sourceMode != NONE) {
return downstream.tryOnNext(null);
}
boolean b;
try {
b = filter.test(t);
} catch (Throwable e) {
fail(e);
return true;
}
return b && downstream.tryOnNext(t);
}
@Override
public int requestFusion(int mode) {
return transitiveBoundaryFusion(mode);
}
@Nullable
@Override
public T poll() throws Exception {
QueueSubscription qs = this.qs;
Predicate f = filter;
for (;;) {
T t = qs.poll();
if (t == null) {
return null;
}
if (f.test(t)) {
return t;
}
if (sourceMode == ASYNC) {
qs.request(1);
}
}
}
}
}