com.gs.api.accelrx.context.rxjava3.ContextPropagatorOnSingleAssemblyAction Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of application Show documentation

The newest version!

package com.gs.api.accelrx.context.rxjava3;

import io.reactivex.rxjava3.annotations.NonNull;
import io.reactivex.rxjava3.core.Single;
import io.reactivex.rxjava3.core.SingleObserver;
import io.reactivex.rxjava3.functions.Function;

import java.util.concurrent.Executor;

@SuppressWarnings("rawtypes")
public class ContextPropagatorOnSingleAssemblyAction implements Function {

    private final ThreadContext threadContext;

    public ContextPropagatorOnSingleAssemblyAction(ThreadContext threadContext) {
        this.threadContext = threadContext;
    }

    @Override
    @SuppressWarnings({"rawtypes", "unchecked"})
    public Single apply(Single single) throws Throwable {
        return new ContextPropagatorSingle<>(single, threadContext.currentContextExecutor());
    }

    private static final class ContextPropagatorSingle extends Single {

        private final Single source;
        private final Executor contextExecutor;

        public ContextPropagatorSingle(Single single, Executor contextExecutor) {
            this.source = single;
            this.contextExecutor = contextExecutor;
        }

        @Override
        protected void subscribeActual(@NonNull SingleObserver observer) {
            contextExecutor.execute(() -> source.subscribe(observer));
        }
    }
}