Examples with ContextMapHolder io.servicetalk.context.api.ContextMapHolder used on opensource projects

Example 16 with ContextMapHolder

use of io.servicetalk.context.api.ContextMapHolder in project servicetalk by apple.

the class ContextPreservingCompletableSubscriber method onComplete.

@Override
public final void onComplete() {
    final Thread currentThread = Thread.currentThread();
    if (currentThread instanceof ContextMapHolder) {
        final ContextMapHolder asyncContextMapHolder = (ContextMapHolder) currentThread;
        ContextMap prev = asyncContextMapHolder.context();
        try {
            asyncContextMapHolder.context(saved);
            subscriber.onComplete();
        } finally {
            asyncContextMapHolder.context(prev);
        }
    } else {
        onCompleteSlowPath();
    }
}

Example 17 with ContextMapHolder

use of io.servicetalk.context.api.ContextMapHolder in project servicetalk by apple.

the class ContextPreservingCompletableSubscriber method onError.

@Override
public final void onError(Throwable t) {
    final Thread currentThread = Thread.currentThread();
    if (currentThread instanceof ContextMapHolder) {
        final ContextMapHolder asyncContextMapHolder = (ContextMapHolder) currentThread;
        ContextMap prev = asyncContextMapHolder.context();
        try {
            asyncContextMapHolder.context(saved);
            subscriber.onError(t);
        } finally {
            asyncContextMapHolder.context(prev);
        }
    } else {
        onErrorSlowPath(t);
    }
}

Example 18 with ContextMapHolder

use of io.servicetalk.context.api.ContextMapHolder in project servicetalk by apple.

the class ContextPreservingConsumer method accept.

@Override
public void accept(T t) {
    final Thread currentThread = Thread.currentThread();
    if (currentThread instanceof ContextMapHolder) {
        final ContextMapHolder asyncContextMapHolder = (ContextMapHolder) currentThread;
        ContextMap prev = asyncContextMapHolder.context();
        try {
            asyncContextMapHolder.context(saved);
            delegate.accept(t);
        } finally {
            asyncContextMapHolder.context(prev);
        }
    } else {
        slowPath(t);
    }
}

Example 19 with ContextMapHolder

use of io.servicetalk.context.api.ContextMapHolder in project servicetalk by apple.

the class ContextPreservingRunnable method run.

@Override
public void run() {
    final Thread currentThread = Thread.currentThread();
    if (currentThread instanceof ContextMapHolder) {
        final ContextMapHolder asyncContextMapHolder = (ContextMapHolder) currentThread;
        ContextMap prev = asyncContextMapHolder.context();
        try {
            asyncContextMapHolder.context(saved);
            delegate.run();
        } finally {
            asyncContextMapHolder.context(prev);
        }
    } else {
        slowPath();
    }
}

Example 20 with ContextMapHolder

use of io.servicetalk.context.api.ContextMapHolder in project servicetalk by apple.

the class ContextPreservingSubscriber method onNext.

@Override
public final void onNext(T t) {
    final Thread currentThread = Thread.currentThread();
    if (currentThread instanceof ContextMapHolder) {
        final ContextMapHolder asyncContextMapHolder = (ContextMapHolder) currentThread;
        ContextMap prev = asyncContextMapHolder.context();
        try {
            asyncContextMapHolder.context(saved);
            subscriber.onNext(t);
        } finally {
            asyncContextMapHolder.context(prev);
        }
    } else {
        onNextSlowPath(t);
    }
}