org.fabric3.execution.runtime.PropagatingCallable Maven / Gradle / Ivy

Go to download
package org.fabric3.execution.runtime;

import java.util.List;
import java.util.Map;
import java.util.concurrent.Callable;

import org.fabric3.api.SecuritySubject;
import org.fabric3.spi.invocation.CallbackReference;
import org.fabric3.spi.invocation.WorkContext;
import org.fabric3.spi.invocation.WorkContextCache;

/**
 * Propagates a work context to the current thread executing the delegate callable.
 */
public class PropagatingCallable implements Callable {
    private Callable delegate;
    private List stack;
    private Map headers;
    private SecuritySubject subject;

    public PropagatingCallable(Callable delegate, List stack, Map headers, SecuritySubject subject) {
        this.delegate = delegate;
        this.stack = stack;
        this.headers = headers;
        this.subject = subject;
    }

    public T call() throws Exception {
        WorkContext workContext = WorkContextCache.getAndResetThreadWorkContext();
        try {
            workContext.setSubject(subject);
            workContext.addHeaders(headers);
            workContext.addCallbackReferences(stack);
            return delegate.call();
        } finally {
            workContext.reset();
        }
    }
}