reactor.rx.stream.SingleTimerStream Maven / Gradle / Ivy

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

import org.reactivestreams.Subscriber;
import reactor.core.support.Exceptions;
import reactor.fn.Consumer;
import reactor.fn.Pausable;
import reactor.fn.timer.Timer;
import reactor.rx.Stream;
import reactor.rx.subscription.PushSubscription;

import java.util.concurrent.TimeUnit;

/**
 * A Stream that emits {@link 0} after an initial delay and then complete
 * 
 * The SingleTimerStream will manage dedicated timers for new subscriber assigned via
 * {@link this#subscribe(org.reactivestreams.Subscriber)}.
 * 

 * Create such stream with the provided factory, E.g.:
 * 
 * {@code
 * Streams.timer(env, 1).consume(
 *log::info,
 *log::error,
 * (-> log.info("complete"))
 * )
 * }
 * 
 * Will log:
 *  * {@code
 * 0
 * complete
 * }
 * 
 *
 * @author Stephane Maldini
 */
public final class SingleTimerStream extends Stream {

	final private long     delay;
	final private TimeUnit unit;
	final private Timer    timer;

	public SingleTimerStream(long delay, TimeUnit unit, Timer timer) {
		this.delay = delay >= 0l ? delay : 0l;
		this.unit = unit != null ? unit : TimeUnit.SECONDS;
		this.timer = timer;
	}

	@Override
	public void subscribe(final Subscriber subscriber) {
		try {
			subscriber.onSubscribe(new TimerSubscription(this, subscriber));
		}catch (Throwable throwable){
			Exceptions.throwIfFatal(throwable);
			subscriber.onError(throwable);
		}
	}

	private class TimerSubscription extends PushSubscription{

		final Pausable registration = timer.submit(new Consumer() {
			@Override
			public void accept(Long aLong) {
				subscriber.onNext(0l);
				subscriber.onComplete();
			}
		}, delay, unit);

		public TimerSubscription(Stream publisher, Subscriber subscriber) {
			super(publisher, subscriber);
		}

		@Override
		public void cancel() {
			registration.cancel();
			super.cancel();
		}
	}

	@Override
	public String toString() {
		return "delay=" + delay + " " + unit;
	}
}