org.wildfly.clustering.context.ContextualThreadFactory Maven / Gradle / Ivy
/*
* Copyright The WildFly Authors
* SPDX-License-Identifier: Apache-2.0
*/
package org.wildfly.clustering.context;
import java.util.concurrent.ThreadFactory;
/**
* {@link ThreadFactory} decorator that contextualizes its threads.
* @param the context type
* @author Paul Ferraro
*/
public class ContextualThreadFactory implements ThreadFactory {
private final ThreadFactory factory;
private final C targetContext;
private final ThreadContextReference reference;
private final Contextualizer contextualizer;
public ContextualThreadFactory(ThreadFactory factory, C targetContext, ThreadContextReference reference) {
this(factory, targetContext, reference, Contextualizer.withContextProvider(reference.provide(targetContext)));
}
ContextualThreadFactory(ThreadFactory factory, C targetContext, ThreadContextReference reference, Contextualizer contextualizer) {
this.factory = factory;
this.targetContext = targetContext;
this.reference = reference;
this.contextualizer = contextualizer;
}
@Override
public Thread newThread(Runnable task) {
Thread thread = this.factory.newThread(this.contextualizer.contextualize(task));
this.reference.accept(thread, this.targetContext);
return thread;
}
}