io.reactiverse.reactivecontexts.propagators.rxjava2.ContextPropagatorOnMaybeAssemblyAction 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.Maybe;
import io.reactivex.MaybeObserver;
import io.reactivex.functions.Function;

public class ContextPropagatorOnMaybeAssemblyAction implements Function {

	@Override
	public Maybe apply(Maybe t) throws Exception {
		return new ContextPropagatorMaybe(t);
	}

	public class ContextPropagatorMaybe extends Maybe {

		private Maybe source;
		private ContextState context;

		public ContextPropagatorMaybe(Maybe t) {
			this.source = t;
			this.context = Context.capture();
		}

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

	}

}