Examples with TaskQueue com.dtp.core.support.TaskQueue used on opensource projects

Example 1 with TaskQueue

use of com.dtp.core.support.TaskQueue in project dynamic-tp by dromara.

the class DtpPostProcessor method postProcessAfterInitialization.

@Override
public Object postProcessAfterInitialization(Object bean, String beanName) throws BeansException {
    if (!(bean instanceof ThreadPoolExecutor)) {
        return bean;
    }
    if (bean instanceof DtpExecutor) {
        DtpExecutor dtpExecutor = (DtpExecutor) bean;
        if (bean instanceof EagerDtpExecutor) {
            ((TaskQueue) dtpExecutor.getQueue()).setExecutor((EagerDtpExecutor) dtpExecutor);
        }
        registerDtp(dtpExecutor);
        return dtpExecutor;
    }
    ApplicationContext applicationContext = ApplicationContextHolder.getInstance();
    DynamicTp dynamicTp;
    try {
        dynamicTp = applicationContext.findAnnotationOnBean(beanName, DynamicTp.class);
        if (dynamicTp == null) {
            return bean;
        }
    } catch (NoSuchBeanDefinitionException e) {
        log.error("There is no bean with the given name {}", beanName, e);
        return bean;
    }
    String poolName = StringUtils.isNotBlank(dynamicTp.value()) ? dynamicTp.value() : beanName;
    registerCommon(poolName, (ThreadPoolExecutor) bean);
    return bean;
}