org.broadleafcommerce.common.classloader.release.ThreadLocalManager Maven / Gradle / Ivy
/*
* #%L
* BroadleafCommerce Common Libraries
* %%
* Copyright (C) 2009 - 2013 Broadleaf Commerce
* %%
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
* #L%
*/
package org.broadleafcommerce.common.classloader.release;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import java.util.LinkedHashMap;
import java.util.Map;
/**
* @author Jeff Fischer
*/
public class ThreadLocalManager {
private static final Log LOG = LogFactory.getLog(ThreadLocalManager.class);
private static final ThreadLocal THREAD_LOCAL_MANAGER = new ThreadLocal() {
@Override
protected ThreadLocalManager initialValue() {
ThreadLocalManager manager = new ThreadLocalManager();
String checkOrphans = System.getProperty("ThreadLocalManager.notify.orphans");
if ("true".equals(checkOrphans)) {
manager.marker = new RuntimeException("Thread Local Manager is not empty - the following is the culprit call that setup the thread local but did not clear it.");
}
return manager;
}
};
protected Map threadLocals = new LinkedHashMap();
protected RuntimeException marker = null;
public static void addThreadLocal(ThreadLocal threadLocal) {
Long position;
synchronized (threadLock) {
count++;
position = count;
}
THREAD_LOCAL_MANAGER.get().threadLocals.put(position, threadLocal);
}
public static ThreadLocal createThreadLocal(final Class type) {
return createThreadLocal(type, true);
}
public static ThreadLocal createThreadLocal(final Class type, final boolean createInitialValue) {
ThreadLocal response = new ThreadLocal() {
@Override
protected T initialValue() {
addThreadLocal(this);
if (!createInitialValue) {
return null;
}
try {
return type.newInstance();
} catch (InstantiationException e) {
throw new RuntimeException(e);
} catch (IllegalAccessException e) {
throw new RuntimeException(e);
}
}
@Override
public void set(T value) {
super.get();
super.set(value);
}
};
return response;
}
public static void remove() {
for (Map.Entry entry : THREAD_LOCAL_MANAGER.get().threadLocals.entrySet()) {
if (LOG.isDebugEnabled()) {
LOG.debug("Removing ThreadLocal #" + entry.getKey() + " from request thread.");
}
entry.getValue().remove();
}
THREAD_LOCAL_MANAGER.get().threadLocals.clear();
THREAD_LOCAL_MANAGER.remove();
}
public static void remove(ThreadLocal threadLocal) {
Long removePosition = null;
for (Map.Entry entry : THREAD_LOCAL_MANAGER.get().threadLocals.entrySet()) {
if (entry.getValue().equals(threadLocal)) {
if (LOG.isDebugEnabled()) {
LOG.debug("Removing ThreadLocal #" + entry.getKey() + " from request thread.");
}
entry.getValue().remove();
removePosition = entry.getKey();
}
}
THREAD_LOCAL_MANAGER.get().threadLocals.remove(removePosition);
}
private static Long count = 0L;
private static final Object threadLock = new Object();
@Override
public String toString() {
if (!threadLocals.isEmpty() && marker != null) {
marker.printStackTrace();
}
return super.toString();
}
}