com.github.kristofa.brave.BraveCallable Maven / Gradle / Ivy

Go to download
package com.github.kristofa.brave;

import java.util.concurrent.Callable;

import org.apache.commons.lang3.builder.EqualsBuilder;
import org.apache.commons.lang3.builder.HashCodeBuilder;

/**
 * Callable implementation that wraps another Callable and makes sure the wrapped Callable will be executed in the same
 * Span/Trace context as the thread from which the Callable was executed.
 * 
 * Is used by {@link BraveExecutorService}.
 * 
 * @author kristof
 * @param  Return type.
 * @see BraveExecutorService
 */
public class BraveCallable implements Callable {

    private final Callable wrappedCallable;
    private final ServerSpanThreadBinder serverTracer;
    private final ServerSpan currentServerSpan;

    /**
     * Creates a new instance.
     * 
     * @param wrappedCallable The wrapped Callable.
     * @param serverSpanThreadBinder ServerSpan thread binder.
     */
    public BraveCallable(final Callable wrappedCallable, final ServerSpanThreadBinder serverSpanThreadBinder) {
        this.wrappedCallable = wrappedCallable;
        this.serverTracer = serverSpanThreadBinder;
        this.currentServerSpan = serverSpanThreadBinder.getCurrentServerSpan();
    }

    /**
     * {@inheritDoc}
     */
    @Override
    public T call() throws Exception {
        serverTracer.setCurrentSpan(currentServerSpan);
        final long start = System.currentTimeMillis();
        try {
            return wrappedCallable.call();
        } finally {
            final long duration = System.currentTimeMillis() - start;
            currentServerSpan.incThreadDuration(duration);
        }
    }

    /**
     * {@inheritDoc}
     */
    @Override
    public int hashCode() {
        return HashCodeBuilder.reflectionHashCode(this, false);
    }

    /**
     * {@inheritDoc}
     */
    @Override
    public boolean equals(final Object arg0) {
        return EqualsBuilder.reflectionEquals(this, arg0, false);
    }

}