com.emc.mongoose.base.logging.LogContextThreadFactory Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mongoose-base Show documentation

Mongoose is a high-load storage performance testing tool

There is a newer version: 4.3.10

package com.emc.mongoose.base.logging;

import static com.emc.mongoose.base.Exceptions.throwUncheckedIfInterrupted;

import com.github.akurilov.commons.concurrent.ContextAwareThreadFactory;
import java.util.Map;
import java.util.logging.Level;
import java.util.logging.Logger;
import org.apache.logging.log4j.CloseableThreadContext;
import org.apache.logging.log4j.ThreadContext;

/** Created by kurila on 19.07.16. */
public final class LogContextThreadFactory extends ContextAwareThreadFactory {

	private static final Logger LOG = Logger.getLogger(LogContextThreadFactory.class.getName());

	public LogContextThreadFactory(final String threadNamePrefix) {
		super(threadNamePrefix, ThreadContext.getContext());
	}

	public LogContextThreadFactory(final String threadNamePrefix, final boolean daemonFlag) {
		super(threadNamePrefix, daemonFlag, ThreadContext.getContext());
	}

	private static final class LogContextThread extends ContextAwareThread {

		private LogContextThread(
						final Runnable task,
						final String name,
						final boolean daemonFlag,
						final UncaughtExceptionHandler exceptionHandler,
						final Map threadContext) {
			super(task, name, daemonFlag, exceptionHandler, threadContext);
		}

		@Override
		public final void run() {
			try (final var ctx = CloseableThreadContext.putAll(threadContext)) {
				super.run();
			} catch (final Throwable cause) {
				throwUncheckedIfInterrupted(cause);
				LOG.log(Level.SEVERE, "Unhandled thread failure", cause);
			}
		}
	}

	@Override
	public final Thread newThread(final Runnable task) {
		return new LogContextThread(
						task,
						threadNamePrefix + "#" + threadNumber.incrementAndGet(),
						daemonFlag,
						exceptionHandler,
						threadContext);
	}
}