io.reactivex.rxjava3.internal.operators.flowable.FlowableBuffer Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of redisson-all Show documentation
Show all versions of redisson-all Show documentation
Easy Redis Java client and Real-Time Data Platform. Valkey compatible. Sync/Async/RxJava3/Reactive API. Client side caching. Over 50 Redis based Java objects and services: JCache API, Apache Tomcat, Hibernate, Spring, Set, Multimap, SortedSet, Map, List, Queue, Deque, Semaphore, Lock, AtomicLong, Map Reduce, Bloom filter, Scheduler, RPC
/*
* 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.atomic.*;
import org.reactivestreams.*;
import io.reactivex.rxjava3.core.*;
import io.reactivex.rxjava3.exceptions.Exceptions;
import io.reactivex.rxjava3.functions.*;
import io.reactivex.rxjava3.internal.subscriptions.SubscriptionHelper;
import io.reactivex.rxjava3.internal.util.*;
import io.reactivex.rxjava3.plugins.RxJavaPlugins;
public final class FlowableBuffer> extends AbstractFlowableWithUpstream {
final int size;
final int skip;
final Supplier bufferSupplier;
public FlowableBuffer(Flowable source, int size, int skip, Supplier bufferSupplier) {
super(source);
this.size = size;
this.skip = skip;
this.bufferSupplier = bufferSupplier;
}
@Override
public void subscribeActual(Subscriber super C> s) {
if (size == skip) {
source.subscribe(new PublisherBufferExactSubscriber<>(s, size, bufferSupplier));
} else if (skip > size) {
source.subscribe(new PublisherBufferSkipSubscriber<>(s, size, skip, bufferSupplier));
} else {
source.subscribe(new PublisherBufferOverlappingSubscriber<>(s, size, skip, bufferSupplier));
}
}
static final class PublisherBufferExactSubscriber>
implements FlowableSubscriber, Subscription {
final Subscriber super C> downstream;
final Supplier bufferSupplier;
final int size;
C buffer;
Subscription upstream;
boolean done;
int index;
PublisherBufferExactSubscriber(Subscriber super C> actual, int size, Supplier bufferSupplier) {
this.downstream = actual;
this.size = size;
this.bufferSupplier = bufferSupplier;
}
@Override
public void request(long n) {
if (SubscriptionHelper.validate(n)) {
upstream.request(BackpressureHelper.multiplyCap(n, size));
}
}
@Override
public void cancel() {
upstream.cancel();
}
@Override
public void onSubscribe(Subscription s) {
if (SubscriptionHelper.validate(this.upstream, s)) {
this.upstream = s;
downstream.onSubscribe(this);
}
}
@Override
public void onNext(T t) {
if (done) {
return;
}
C b = buffer;
if (b == null) {
try {
b = Objects.requireNonNull(bufferSupplier.get(), "The bufferSupplier returned a null buffer");
} catch (Throwable e) {
Exceptions.throwIfFatal(e);
cancel();
onError(e);
return;
}
buffer = b;
}
b.add(t);
int i = index + 1;
if (i == size) {
index = 0;
buffer = null;
downstream.onNext(b);
} else {
index = i;
}
}
@Override
public void onError(Throwable t) {
if (done) {
RxJavaPlugins.onError(t);
return;
}
buffer = null;
done = true;
downstream.onError(t);
}
@Override
public void onComplete() {
if (done) {
return;
}
done = true;
C b = buffer;
buffer = null;
if (b != null) {
downstream.onNext(b);
}
downstream.onComplete();
}
}
static final class PublisherBufferSkipSubscriber>
extends AtomicInteger
implements FlowableSubscriber, Subscription {
private static final long serialVersionUID = -5616169793639412593L;
final Subscriber super C> downstream;
final Supplier bufferSupplier;
final int size;
final int skip;
C buffer;
Subscription upstream;
boolean done;
int index;
PublisherBufferSkipSubscriber(Subscriber super C> actual, int size, int skip,
Supplier bufferSupplier) {
this.downstream = actual;
this.size = size;
this.skip = skip;
this.bufferSupplier = bufferSupplier;
}
@Override
public void request(long n) {
if (SubscriptionHelper.validate(n)) {
if (get() == 0 && compareAndSet(0, 1)) {
// n full buffers
long u = BackpressureHelper.multiplyCap(n, size);
// + (n - 1) gaps
long v = BackpressureHelper.multiplyCap(skip - size, n - 1);
upstream.request(BackpressureHelper.addCap(u, v));
} else {
// n full buffer + gap
upstream.request(BackpressureHelper.multiplyCap(skip, n));
}
}
}
@Override
public void cancel() {
upstream.cancel();
}
@Override
public void onSubscribe(Subscription s) {
if (SubscriptionHelper.validate(this.upstream, s)) {
this.upstream = s;
downstream.onSubscribe(this);
}
}
@Override
public void onNext(T t) {
if (done) {
return;
}
C b = buffer;
int i = index;
if (i++ == 0) {
try {
b = Objects.requireNonNull(bufferSupplier.get(), "The bufferSupplier returned a null buffer");
} catch (Throwable e) {
Exceptions.throwIfFatal(e);
cancel();
onError(e);
return;
}
buffer = b;
}
if (b != null) {
b.add(t);
if (b.size() == size) {
buffer = null;
downstream.onNext(b);
}
}
if (i == skip) {
i = 0;
}
index = i;
}
@Override
public void onError(Throwable t) {
if (done) {
RxJavaPlugins.onError(t);
return;
}
done = true;
buffer = null;
downstream.onError(t);
}
@Override
public void onComplete() {
if (done) {
return;
}
done = true;
C b = buffer;
buffer = null;
if (b != null) {
downstream.onNext(b);
}
downstream.onComplete();
}
}
static final class PublisherBufferOverlappingSubscriber>
extends AtomicLong
implements FlowableSubscriber, Subscription, BooleanSupplier {
private static final long serialVersionUID = -7370244972039324525L;
final Subscriber super C> downstream;
final Supplier bufferSupplier;
final int size;
final int skip;
final ArrayDeque buffers;
final AtomicBoolean once;
Subscription upstream;
boolean done;
int index;
volatile boolean cancelled;
long produced;
PublisherBufferOverlappingSubscriber(Subscriber super C> actual, int size, int skip,
Supplier bufferSupplier) {
this.downstream = actual;
this.size = size;
this.skip = skip;
this.bufferSupplier = bufferSupplier;
this.once = new AtomicBoolean();
this.buffers = new ArrayDeque<>();
}
@Override
public boolean getAsBoolean() {
return cancelled;
}
@Override
public void request(long n) {
if (SubscriptionHelper.validate(n)) {
if (QueueDrainHelper.postCompleteRequest(n, downstream, buffers, this, this)) {
return;
}
if (!once.get() && once.compareAndSet(false, true)) {
// (n - 1) skips
long u = BackpressureHelper.multiplyCap(skip, n - 1);
// + 1 full buffer
long r = BackpressureHelper.addCap(size, u);
upstream.request(r);
} else {
// n skips
long r = BackpressureHelper.multiplyCap(skip, n);
upstream.request(r);
}
}
}
@Override
public void cancel() {
cancelled = true;
upstream.cancel();
}
@Override
public void onSubscribe(Subscription s) {
if (SubscriptionHelper.validate(this.upstream, s)) {
this.upstream = s;
downstream.onSubscribe(this);
}
}
@Override
public void onNext(T t) {
if (done) {
return;
}
ArrayDeque bs = buffers;
int i = index;
if (i++ == 0) {
C b;
try {
b = Objects.requireNonNull(bufferSupplier.get(), "The bufferSupplier returned a null buffer");
} catch (Throwable e) {
Exceptions.throwIfFatal(e);
cancel();
onError(e);
return;
}
bs.offer(b);
}
C b = bs.peek();
if (b.size() + 1 == size) {
bs.poll();
b.add(t);
produced++;
downstream.onNext(b);
}
for (C b0 : bs) {
b0.add(t);
}
if (i == skip) {
i = 0;
}
index = i;
}
@Override
public void onError(Throwable t) {
if (done) {
RxJavaPlugins.onError(t);
return;
}
done = true;
buffers.clear();
downstream.onError(t);
}
@Override
public void onComplete() {
if (done) {
return;
}
done = true;
long p = produced;
if (p != 0L) {
BackpressureHelper.produced(this, p);
}
QueueDrainHelper.postComplete(downstream, buffers, this, this);
}
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy