首次提交

duxinglangzi

2022-04-22 de8c2b2a4654893dc2c80f1fe095c165485bee5f

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
package com.duxinglangzi.canal.starter.configuration; 
 
 
import com.duxinglangzi.canal.starter.annotation.CanalListener; 
import com.duxinglangzi.canal.starter.factory.TransponderContainerFactory; 
import org.slf4j.Logger; 
import org.slf4j.LoggerFactory; 
import org.springframework.aop.support.AopUtils; 
import org.springframework.beans.BeansException; 
import org.springframework.beans.factory.SmartInitializingSingleton; 
import org.springframework.beans.factory.config.BeanFactoryPostProcessor; 
import org.springframework.beans.factory.config.BeanPostProcessor; 
import org.springframework.beans.factory.config.ConfigurableListableBeanFactory; 
import org.springframework.core.MethodIntrospector; 
import org.springframework.core.annotation.AnnotatedElementUtils; 
 
import java.lang.reflect.Method; 
import java.util.Collections; 
import java.util.Map; 
import java.util.Set; 
import java.util.concurrent.ConcurrentHashMap; 
import java.util.stream.Collectors; 
 
/** 
 * @author wuqiong 2022/4/11 
 * @description 
 */ 
public class CanalListenerAnnotationBeanPostProcessor implements 
        BeanPostProcessor, SmartInitializingSingleton, BeanFactoryPostProcessor { 
 
    private static final Logger logger = LoggerFactory.getLogger(CanalListenerAnnotationBeanPostProcessor.class); 
 
    private final Set<Class<?>> notAnnotatedClasses = Collections.newSetFromMap(new ConcurrentHashMap<>(256)); 
    private Set<CanalListenerEndpointRegistrar> registrars = Collections.newSetFromMap(new ConcurrentHashMap<>()); 
    private ConfigurableListableBeanFactory configurableListableBeanFactory; 
    private CanalAutoConfigurationProperties canalAutoConfigurationProperties; 
 
    @Override 
    public Object postProcessAfterInitialization(final Object bean, final String beanName) throws BeansException { 
        if (notAnnotatedClasses.contains(bean.getClass())) return bean; 
        Class<?> targetClass = AopUtils.getTargetClass(bean); 
        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(), 
                    annotatedMethods.keySet().stream().map(e -> e.getName()).collect(Collectors.toSet())); 
        } 
        return bean; 
    } 
 
 
    private CanalListener findListenerAnnotations(Method method) { 
        return AnnotatedElementUtils.findMergedAnnotation(method, CanalListener.class); 
    } 
 
 
    @Override 
    public void afterSingletonsInstantiated() { 
        canalAutoConfigurationProperties = configurableListableBeanFactory.getBean(CanalAutoConfigurationProperties.class); 
        TransponderContainerFactory.registerListenerContainer(configurableListableBeanFactory, canalAutoConfigurationProperties, registrars); 
    } 
 
    @Override 
    public void postProcessBeanFactory(ConfigurableListableBeanFactory configurableListableBeanFactory) throws BeansException { 
        this.configurableListableBeanFactory = configurableListableBeanFactory; 
    } 
 
 
}