消息队列 MessageListener 中无法自动注入bean的解决方法

骑猪看日落 2023-09-30 08:31 10阅读 0赞

参考 [https://www.cnblogs.com/loong-hon/p/11414164.html][https_www.cnblogs.com_loong-hon_p_11414164.html]

背景：使用监听器处理业务，需要使用自己的service方法；

错误：使用@Autowired 注入service对象，最终得到的为null；

原因：listener、fitter 都不是Spring容器管理的，无法在这些类中直接使用Spring注解的方式来注入我们需要的对象。  
解决：写一个bean工厂，从spring的上下文WebApplicationContext 中获取。

import org.springframework.context.ApplicationContext;
    import org.springframework.context.ApplicationContextAware;
     
    @Component 
    public class SpringContextHolder implements ApplicationContextAware {
        
        private static ApplicationContext applicationContext;
     
        /**
         * 实现ApplicationContextAware接口的context注入函数, 将其存入静态变量.
         */
        @Override
        public void setApplicationContext(ApplicationContext applicationContext) {
        
            SpringContextHolder.applicationContext = applicationContext; // NOSONAR
        }
     
        /**
         * 取得存储在静态变量中的ApplicationContext.
         */
        public static ApplicationContext getApplicationContext() {
        
            checkApplicationContext();
            return applicationContext;
        }
     
        /**
         * 从静态变量ApplicationContext中取得Bean, 自动转型为所赋值对象的类型.
         */
        @SuppressWarnings("unchecked")
        public static <T> T getBean(String name) {
        
            checkApplicationContext();
            return (T) applicationContext.getBean(name);
        }
     
        /**
         * 从静态变量ApplicationContext中取得Bean, 自动转型为所赋值对象的类型.
         */
        @SuppressWarnings("unchecked")
        public static <T> T getBean(Class<T> clazz) {
        
            checkApplicationContext();
            return (T) applicationContext.getBeansOfType(clazz);
        }
     
        /**
         * 清除applicationContext静态变量.
         */
        public static void cleanApplicationContext() {
        
            applicationContext = null;
        }
     
        private static void checkApplicationContext() {
        
            if (applicationContext == null) {
        
            	// 使用注解注入或者 xml 注入，我使用的 @Component 
                throw new IllegalStateException("applicaitonContext未注入");
            }
        }
    }

监听器里获取service，使用 getBean("")，根据 bean 名字获取的是一个对象，如果根据类型获取到的是 Map 集合，需要再自行处理

public class ConsumerListener implements MessageListener {
        
    
    	private GaussdbService gaussdbService;
        public ConsumeStatus consume(Message message) {
        
           
            try {
        
              	// 名字获取
                gaussdbService = SpringContextHolder.getBean("gaussdbService");
                // 类型获取
                // Map<String, GaussdbService> map = (Map<String, GaussdbService>) SpringContextHolder.getBean(GaussdbService.class);
                // GaussdbService gaussdbService = map.get("gaussdbService");
            } catch (Exception e) {
        
                log.error("SpringContextHolder 获取 bean 错误" + e.getMessage(),e);
                return ConsumeStatus.SUSPEND_CURRENT_QUEUE_A_MOMENT;
            }
            // 业务实现
    	}
    }

[https_www.cnblogs.com_loong-hon_p_11414164.html]: https://www.cnblogs.com/loong-hon/p/11414164.html