2.7.18

shikeying

2024-01-11 3b67e947e36133e2a40eb2737b15ea375e157ea0

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
package com.walker.async;
 
import java.util.HashMap;
import java.util.Map;
 
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.BeansException;
import org.springframework.beans.factory.config.BeanPostProcessor;
 
public class AsyncBeanManager implements BeanPostProcessor {
 
    private final transient Logger logger = LoggerFactory.getLogger(getClass());
 
    private static final Map<Class<?>, Consumer> consumerMap = new HashMap<>(4);
    private static final Map<Class<?>, Producer> producerMap = new HashMap<>(2);
 
    @Override
    public Object postProcessAfterInitialization(Object bean, String beanName) throws BeansException {
        if(Consumer.class.isAssignableFrom(bean.getClass())){
            consumerMap.put(bean.getClass(), (Consumer)bean);
            logger.info("........[AsyncMsg] find a Consumer: " + beanName);
        } else if(Producer.class.isAssignableFrom(bean.getClass())){
            producerMap.put(bean.getClass(), (Producer)bean);
            logger.info("........[AsyncMsg] find a Producer: " + beanName);
        }
        return bean;
    }
 
    @Override
    public Object postProcessBeforeInitialization(Object bean, String arg1) throws BeansException {
        return bean;
    }
 
    public static final Map<Class<?>, Consumer> getConsumerMap(){
        return consumerMap;
    }
 
    public static final Map<Class<?>, Producer> getProducerMap(){
        return producerMap;
    }
 
    public static final Consumer getConsumer(Class<?> clazz){
        return consumerMap.get(clazz);
    }
    public static final Producer getProducer(Class<?> clazz){
        return producerMap.get(clazz);
    }
}