reactor.core.publisher.FluxMaterialize 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-2021 VMware Inc. or its affiliates, All Rights Reserved.
*
* 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
*
* https://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 reactor.core.publisher;
import java.util.AbstractQueue;
import java.util.Iterator;
import java.util.concurrent.atomic.AtomicLongFieldUpdater;
import java.util.function.BooleanSupplier;
import org.reactivestreams.Subscription;
import reactor.core.CoreSubscriber;
import reactor.util.annotation.Nullable;
import reactor.util.context.Context;
/**
* @author Stephane Maldini
*/
final class FluxMaterialize extends InternalFluxOperator> {
FluxMaterialize(Flux source) {
super(source);
}
@Override
public CoreSubscriber super T> subscribeOrReturn(CoreSubscriber super Signal> actual) {
return new MaterializeSubscriber<>(actual);
}
@Override
public Object scanUnsafe(Attr key) {
if (key == Attr.RUN_STYLE) return Attr.RunStyle.SYNC;
return super.scanUnsafe(key);
}
final static class MaterializeSubscriber
extends AbstractQueue>
implements InnerOperator>, BooleanSupplier {
final CoreSubscriber super Signal> actual;
final Context cachedContext;
Signal terminalSignal;
volatile boolean cancelled;
volatile long requested;
@SuppressWarnings("rawtypes")
static final AtomicLongFieldUpdater REQUESTED =
AtomicLongFieldUpdater.newUpdater(MaterializeSubscriber.class, "requested");
long produced;
Subscription s;
MaterializeSubscriber(CoreSubscriber super Signal> subscriber) {
this.actual = subscriber;
this.cachedContext = actual.currentContext();
}
@Override
public Context currentContext() {
return cachedContext;
}
@Override
@Nullable
public Object scanUnsafe(Attr key) {
if (key == Attr.PARENT) return s;
if (key == Attr.TERMINATED) return terminalSignal != null;
if (key == Attr.ERROR) return terminalSignal != null ? terminalSignal.getThrowable() : null;
if (key == Attr.CANCELLED) return getAsBoolean();
if (key == Attr.REQUESTED_FROM_DOWNSTREAM) return requested;
if (key == Attr.BUFFERED) return size();
if (key == Attr.RUN_STYLE) return Attr.RunStyle.SYNC;
return InnerOperator.super.scanUnsafe(key);
}
@Override
public CoreSubscriber super Signal> actual() {
return actual;
}
@Override
public void onSubscribe(Subscription s) {
if (Operators.validate(this.s, s)) {
this.s = s;
actual.onSubscribe(this);
}
}
@Override
public void onNext(T ev) {
if(terminalSignal != null){
Operators.onNextDropped(ev, this.cachedContext);
return;
}
produced++;
actual.onNext(Signal.next(ev, this.cachedContext));
}
@Override
public void onError(Throwable ev) {
if(terminalSignal != null){
Operators.onErrorDropped(ev, this.cachedContext);
return;
}
terminalSignal = Signal.error(ev, this.cachedContext);
long p = produced;
if (p != 0L) {
Operators.addCap(REQUESTED, this, -p);
}
DrainUtils.postComplete(actual, this, REQUESTED, this, this);
}
@Override
public void onComplete() {
if(terminalSignal != null){
return;
}
terminalSignal = Signal.complete(this.cachedContext);
long p = produced;
if (p != 0L) {
Operators.addCap(REQUESTED, this, -p);
}
DrainUtils.postComplete(actual, this, REQUESTED, this, this);
}
@Override
public void request(long n) {
if (Operators.validate(n)) {
if (!DrainUtils.postCompleteRequest(n, actual, this, REQUESTED, this, this)) {
s.request(n);
}
}
}
@Override
public void cancel() {
if(cancelled){
return;
}
cancelled = true;
s.cancel();
}
@Override
public boolean getAsBoolean() {
return cancelled;
}
@Override
public boolean offer(Signal e) {
throw new UnsupportedOperationException();
}
@Override
@Nullable
@SuppressWarnings("unchecked")
public Signal poll() {
Signal v = terminalSignal;
if (v != null && v != empty) {
terminalSignal = (Signal)empty;
return v;
}
return null;
}
@Override
@Nullable
public Signal peek() {
return empty == terminalSignal ? null : terminalSignal;
}
@Override
public Iterator> iterator() {
throw new UnsupportedOperationException();
}
@Override
public int size() {
return terminalSignal == null || terminalSignal == empty ? 0 : 1;
}
@Override
public String toString() {
return "MaterializeSubscriber";
}
static final Signal empty = new ImmutableSignal<>(Context.empty(), SignalType.ON_NEXT, null, null, null);
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy