io.opentracing.threadcontext.ThreadContextSpan Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of opentracing-thread-context Show documentation

opentracing-thread-context

There is a newer version: 0.8

Show newest version

package io.opentracing.threadcontext;

import com.github.threadcontext.Saver;
import com.github.threadcontext.ThreadContext;
import com.github.threadcontext.ThreadLocalSaver;
import io.opentracing.NoopSpan;
import io.opentracing.Span;
import java.util.concurrent.Callable;
import java.util.function.Supplier;

public final class ThreadContextSpan {

    private static final ThreadLocal span = new ThreadLocal() {
        protected Span initialValue() {
            return defaultSpan.get();
        }
    };
    private static final Saver saver = new ThreadLocalSaver<>(span);
    static {
        ThreadContext.savers.add(saver);
    }

    public static Supplier defaultSpan = () -> NoopSpan.INSTANCE;

    private ThreadContextSpan() {
    }

    public static Span get() {
        return span.get();
    }

    public static void withSpan(Span span, Runnable runnable) {
        saver.runAndRestore(() -> {
            ThreadContextSpan.span.set(span);
            runnable.run();
        });
    }

    public static  T withSpan(Span span, Callable callable) throws Exception {
        return saver.runAndRestore(() -> {
            ThreadContextSpan.span.set(span);
            return callable.call();
        });
    }

    public static  T withSpan(Span span, Supplier supplier) throws Exception {
        return saver.runAndRestore(() -> {
            ThreadContextSpan.span.set(span);
            return supplier.get();
        });
    }

}