add LICENSE.

duxinglangzi

2022-09-06 ed3a1614384279b7d3a97e7411b649476a934ddb

 src/main/java/com/duxinglangzi/canal/starter/configuration/CanalListenerAnnotationBeanPostProcessor.java

@@ -11,6 +11,8 @@
import org.springframework.beans.factory.config.BeanFactoryPostProcessor;

import org.springframework.beans.factory.config.BeanPostProcessor;

import org.springframework.beans.factory.config.ConfigurableListableBeanFactory;

import org.springframework.context.ApplicationContext;

import org.springframework.context.ApplicationContextAware;

import org.springframework.core.MethodIntrospector;

import org.springframework.core.annotation.AnnotatedElementUtils;



@@ -23,10 +25,9 @@


/**

 * @author wuqiong 2022/4/11

 * @description

 */

public class CanalListenerAnnotationBeanPostProcessor implements

        BeanPostProcessor, SmartInitializingSingleton, BeanFactoryPostProcessor {

        BeanPostProcessor, SmartInitializingSingleton, BeanFactoryPostProcessor, ApplicationContextAware {



    private static final Logger logger = LoggerFactory.getLogger(CanalListenerAnnotationBeanPostProcessor.class);



@@ -34,20 +35,23 @@
    private Set<CanalListenerEndpointRegistrar> registrars = Collections.newSetFromMap(new ConcurrentHashMap<>());

    private ConfigurableListableBeanFactory configurableListableBeanFactory;

    private CanalAutoConfigurationProperties canalAutoConfigurationProperties;

    private ApplicationContext applicationContext;



    @Override

    public Object postProcessAfterInitialization(final Object bean, final String beanName) throws BeansException {

        if (notAnnotatedClasses.contains(bean.getClass())) return bean;

        Class<?> targetClass = AopUtils.getTargetClass(bean);

        // 只扫描类的方法，目前 CanalListener 只支持在方法上

        Map<Method, CanalListener> annotatedMethods = MethodIntrospector.selectMethods(targetClass,

                (MethodIntrospector.MetadataLookup<CanalListener>) method -> findListenerAnnotations(method));

        if (annotatedMethods.isEmpty()) {

            this.notAnnotatedClasses.add(bean.getClass());

        } else {

            // 先加入到待注册里面

            annotatedMethods.entrySet().stream()

                    .filter(e -> e != null)

                    .forEach(ele -> registrars.add(new CanalListenerEndpointRegistrar(bean, ele)));

            logger.info("Registered @CanalListener methods processed on bean:{} , Methods :{} ", bean.getClass().getName(),

                    .forEach(ele -> registrars.add(createRegistrar(bean, ele)));

            logger.info("Registered @CanalListener methods processed on bean:{} , Methods :{} ", beanName,

                    annotatedMethods.keySet().stream().map(e -> e.getName()).collect(Collectors.toSet()));

        }

        return bean;

@@ -70,5 +74,15 @@
        this.configurableListableBeanFactory = configurableListableBeanFactory;

    }



    @Override

    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {

        this.applicationContext = applicationContext;

    }



    private CanalListenerEndpointRegistrar createRegistrar(Object bean, Map.Entry<Method, CanalListener> entry) {

        return new CanalListenerEndpointRegistrar(bean, entry.getKey(),

                applicationContext.getEnvironment().resolvePlaceholders(entry.getValue().destination()),

                applicationContext.getEnvironment().resolvePlaceholders(entry.getValue().database()),

                entry.getValue().table(), entry.getValue().eventType());

    }

}