org.wildfly.clustering.context.ContextualThreadFactory Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of wildfly-clustering-context Show documentation

There is a newer version: 5.0.7.Final

Show newest version

/*
 * Copyright The WildFly Authors
 * SPDX-License-Identifier: Apache-2.0
 */
package org.wildfly.clustering.context;

import java.security.AccessController;
import java.security.PrivilegedAction;
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 = AccessController.doPrivileged(new PrivilegedAction<>() {
			@Override
			public Thread run() {
				return ContextualThreadFactory.this.factory.newThread(ContextualThreadFactory.this.contextualizer.contextualize(task));
			}
		});
		this.reference.accept(thread, this.targetContext);
		return thread;
	}
}