io.reactiverse.reactivecontexts.propagators.rxjava2.ContextPropagatorOnObservableAssemblyAction Maven / Gradle / Ivy

Go to download

package io.reactiverse.reactivecontexts.propagators.rxjava2;

import io.reactiverse.reactivecontexts.core.Context;
import io.reactiverse.reactivecontexts.core.ContextState;
import io.reactivex.Observable;
import io.reactivex.Observer;
import io.reactivex.functions.Function;

public class ContextPropagatorOnObservableAssemblyAction implements Function {

	@Override
	public Observable apply(Observable t) throws Exception {
		return new ContextPropagatorObservable(t);
	}

	public class ContextPropagatorObservable extends Observable {

		private Observable source;
		private ContextState context;

		public ContextPropagatorObservable(Observable t) {
			this.source = t;
			this.context = Context.capture();
		}

		@Override
		protected void subscribeActual(Observer observer) {
			ContextState previousContext = context.install();
			try {
				source.subscribe(observer);
			}finally {
				previousContext.restore();
			}
		}

	}

}