org.neo4j.driver.internal.shaded.reactor.core.publisher.FluxName Maven / Gradle / Ivy

/*
 * 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.Collections;
import java.util.HashSet;
import java.util.Objects;
import java.util.Set;

import reactor.core.CoreSubscriber;
import reactor.core.Fuseable;
import reactor.util.annotation.Nullable;
import reactor.util.function.Tuple2;
import reactor.util.function.Tuples;

import static reactor.core.Scannable.Attr.RUN_STYLE;
import static reactor.core.Scannable.Attr.RunStyle.SYNC;

/**
 * An operator that just bears a name or a set of tags, which can be retrieved via the
 * {@link reactor.core.Scannable.Attr#TAGS TAGS}
 * attribute.
 *
 * @author Simon Baslé
 * @author Stephane Maldini
 */
final class FluxName extends InternalFluxOperator {

	final String name;

	final Set> tags;

	@SuppressWarnings("unchecked")
	static  Flux createOrAppend(Flux source, String name) {
		Objects.requireNonNull(name, "name");

		if (source instanceof FluxName) {
			FluxName s = (FluxName) source;
			return new FluxName<>(s.source, name, s.tags);
		}
		if (source instanceof FluxNameFuseable) {
			FluxNameFuseable s = (FluxNameFuseable) source;
			return new FluxNameFuseable<>(s.source, name, s.tags);
		}
		if (source instanceof Fuseable) {
			return new FluxNameFuseable<>(source, name, null);
		}
		return new FluxName<>(source, name, null);
	}

	@SuppressWarnings("unchecked")
	static  Flux createOrAppend(Flux source, String tagName, String tagValue) {
		Objects.requireNonNull(tagName, "tagName");
		Objects.requireNonNull(tagValue, "tagValue");

		Set> tags = Collections.singleton(Tuples.of(tagName, tagValue));

		if (source instanceof FluxName) {
			FluxName s = (FluxName) source;
			if(s.tags != null) {
				tags = new HashSet<>(tags);
				tags.addAll(s.tags);
			}
			return new FluxName<>(s.source, s.name, tags);
		}
		if (source instanceof FluxNameFuseable) {
			FluxNameFuseable s = (FluxNameFuseable) source;
			if (s.tags != null) {
				tags = new HashSet<>(tags);
				tags.addAll(s.tags);
			}
			return new FluxNameFuseable<>(s.source, s.name, tags);
		}
		if (source instanceof Fuseable) {
			return new FluxNameFuseable<>(source, null, tags);
		}
		return new FluxName<>(source, null, tags);
	}

	FluxName(Flux source,
			@Nullable String name,
			@Nullable Set> tags) {
		super(source);
		this.name = name;
		this.tags = tags;
	}

	@Override
	public CoreSubscriber subscribeOrReturn(CoreSubscriber actual) {
		return actual;
	}

	@Nullable
	@Override
	public Object scanUnsafe(Attr key) {
		if (key == Attr.NAME) {
			return name;
		}

		if (key == Attr.TAGS && tags != null) {
			return tags.stream();
		}

		if (key == RUN_STYLE) {
		    return SYNC;
		}

		return super.scanUnsafe(key);
	}


}