reactor.rx.subscription.FanOutSubscription Maven / Gradle / Ivy

Go to download
/*
 * Copyright (c) 2011-2014 Pivotal Software, Inc.
 *
 *  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 reactor.rx.subscription;

import reactor.fn.Consumer;
import reactor.rx.Stream;

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Queue;
import java.util.concurrent.LinkedTransferQueue;

/**
 * A composite subscription used to achieve pub/sub pattern. When more than 1 subscriber is attached to a Stream,
 * in particular an Action, the previous subscription is replaced by a composite fanOutSubscription delegating to
 * both the previous and the new subscriptions.
 *
 * @author Stephane Maldini
 * @since 2.0
 */
public class FanOutSubscription extends PushSubscription {

	private final List>  subscriptions = new ArrayList>(2);
	private final Queue> deleteQueue   = new LinkedTransferQueue<>();

	public FanOutSubscription(Stream publisher, PushSubscription reactiveSubscriptionA,
	                          PushSubscription reactiveSubscriptionB) {
		super(publisher, null);
		subscriptions.add(reactiveSubscriptionA);
		subscriptions.add(reactiveSubscriptionB);
	}

	@Override
	public void onComplete() {
		forEach(new Consumer>() {
			@Override
			public void accept(PushSubscription subscription) {
				try {
					subscription.onComplete();
				} catch (Throwable throwable) {
					subscription.onError(throwable);
				}
			}
		});
	}

	@Override
	public void onNext(final O ev) {
		forEach(new Consumer>() {
			@Override
			public void accept(PushSubscription subscription) {
				try {
					subscription.onNext(ev);
				} catch (Throwable throwable) {
					subscription.onError(throwable);
				}
			}
		});

		drainErrors();
	}

	private void drainErrors(){
		PushSubscription deleteSubscription;
		synchronized (deleteQueue) {
			synchronized (subscriptions) {
				while ((deleteSubscription = deleteQueue.poll()) != null) {
					subscriptions.remove(deleteSubscription);
				}
			}
		}
	}

	@Override
	public void start() {
		forEach(new Consumer>() {
			@Override
			public void accept(PushSubscription oPushSubscription) {
				oPushSubscription.start();
			}
		});
		super.start();
	}

	@Override
	public void cancel() {
		forEach(new Consumer>() {
			@Override
			public void accept(PushSubscription oPushSubscription) {
				oPushSubscription.cancel();
			}
		});
		super.cancel();
	}

	@Override
	public void onError(final Throwable ev) {
		forEach(new Consumer>() {
			@Override
			public void accept(PushSubscription oPushSubscription) {
				oPushSubscription.onError(ev);
			}
		});
	}

	@Override
	public boolean isComplete() {
		boolean isComplete = false;
		synchronized (subscriptions) {
			for (PushSubscription subscription : subscriptions) {
				isComplete = subscription.isComplete();
				if (!isComplete) break;
			}
			return isComplete;
		}
	}

	public void forEach(Consumer> consumer) {
		synchronized (subscriptions) {
			if(subscriptions.isEmpty()) return;
			for (PushSubscription subscription : subscriptions) {
				if (subscription != null) {
					consumer.accept(subscription);
				}
			}
		}
	}

	public List> getSubscriptions() {
		return Collections.unmodifiableList(subscriptions);
	}

	public boolean isEmpty() {
		synchronized (subscriptions) {
			return subscriptions.isEmpty();
		}
	}

	public boolean remove(PushSubscription subscription) {
		synchronized (deleteQueue) {
			return deleteQueue.add(subscription);
		}
	}


	public boolean add(PushSubscription subscription) {
		synchronized (subscriptions) {
		return subscriptions.add(subscription);
		}
	}

	public boolean contains(PushSubscription subscription) {
		synchronized (subscriptions) {
			return subscriptions.contains(subscription);
		}
	}
}