Refactor ThreadPoolInstance Registry to ThreadPoolRegistry.

pull/1339/head
lucca 2 years ago
parent 88165c8e4a
commit 9e89f4d6de

@ -41,7 +41,7 @@ public class SpringThreadPoolRegisterSupport {
private static final Logger LOGGER = LoggerFactory.getLogger(SpringThreadPoolRegisterSupport.class); private static final Logger LOGGER = LoggerFactory.getLogger(SpringThreadPoolRegisterSupport.class);
public static void registerThreadPoolInstances(ApplicationContext context) { public static void registerThreadPoolInstances(ApplicationContext context) {
Map<ThreadPoolExecutor, Class<?>> referencedClassMap = ThreadPoolRegistry.getReferencedClassMap(); Map<ThreadPoolExecutor, Class<?>> referencedClassMap = ThreadPoolRegistry.REFERENCED_CLASS_MAP;
for (Map.Entry<ThreadPoolExecutor, Class<?>> entry : referencedClassMap.entrySet()) { for (Map.Entry<ThreadPoolExecutor, Class<?>> entry : referencedClassMap.entrySet()) {
ThreadPoolExecutor enhancedInstance = entry.getKey(); ThreadPoolExecutor enhancedInstance = entry.getKey();
Class<?> declaredClass = entry.getValue(); Class<?> declaredClass = entry.getValue();

@ -52,7 +52,7 @@ public class ThreadPoolExecutorConstructorMethodInterceptor implements InstanceC
StackTraceElement declaredClassStackTraceElement = stackTraceElements.get(0); StackTraceElement declaredClassStackTraceElement = stackTraceElements.get(0);
String declaredClassName = declaredClassStackTraceElement.getClassName(); String declaredClassName = declaredClassStackTraceElement.getClassName();
Class<?> declaredClass = Thread.currentThread().getContextClassLoader().loadClass(declaredClassName); Class<?> declaredClass = Thread.currentThread().getContextClassLoader().loadClass(declaredClassName);
ThreadPoolRegistry.putReferencedClass((ThreadPoolExecutor) objInst, declaredClass); ThreadPoolRegistry.REFERENCED_CLASS_MAP.put((ThreadPoolExecutor) objInst, declaredClass);
} }
private List<StackTraceElement> getStackTraceElements() { private List<StackTraceElement> getStackTraceElements() {

@ -49,12 +49,4 @@ public class ThreadPoolRegistry {
public static Map<ThreadPoolExecutor, Class<?>> getReferencedClassMap() { public static Map<ThreadPoolExecutor, Class<?>> getReferencedClassMap() {
return REFERENCED_CLASS_MAP; return REFERENCED_CLASS_MAP;
} }
public static void putReferencedClass(ThreadPoolExecutor executor, Class<?> referencedClass) {
REFERENCED_CLASS_MAP.put(executor, referencedClass);
}
public static Class<?> getReferencedClass(ThreadPoolExecutor executor) {
return REFERENCED_CLASS_MAP.get(executor);
}
} }

Loading…
Cancel
Save