最近在做项目的时候遇到一个问题,就是在spring加载完所有bean的时候要获取所有AMQ消息队列的容器对象,从而可以实现方便地对其进行控制,比如运行某一个消息队列容器对象,使其开始消费AMQ消息。
问了下度娘,总体解决方案如下:
1、写一个类继承spring的ApplicationListener<ContextRefreshedEvent>监听,并监控ContextRefreshedEvent事件(容易初始化完成事件)
2、在XML配置一下上面的bean,或者直接使用@Component或者@Service等注解
代码如下:
import java.util.HashMap;
import java.util.Map;
import org.springframework.context.ApplicationListener;
import org.springframework.context.event.ContextRefreshedEvent;
import org.springframework.jms.listener.DefaultMessageListenerContainer;
import org.springframework.stereotype.Service;
@Service
public class AMQListener implements ApplicationListener<ContextRefreshedEvent> {
/**
* key:队列名称 value:messageListener
*/
private final Map<String, DefaultMessageListenerContainer> queueContainerMap = new HashMap<String, DefaultMessageListenerContainer>();
@Override
public void onApplicationEvent(ContextRefreshedEvent event) {
// 获取所有的MQ消息监听者,用于后续对消息消费的控制
Map<String, DefaultMessageListenerContainer> containMap = event.getApplicationContext().getBeansOfType(DefaultMessageListenerContainer.class);
//......对containMap进行处理
}
public Map<String, DefaultMessageListenerContainer> getQueueContainerMap() {
return queueContainerMap;
}
}
但是这时候又遇到一个问题,就是发现ContextRefreshedEvent被触发了两次。查了下原因,原来项目中存在两个上下文环境,一个是root上下文环境,还有一个是非root上下文环境,为了防止重复初始化,所以在ContextRefreshedEvent事件代码中得加上以下的判断
if(context.getParent == null){
// 初始化代码....
}
这样的话,只有是root上下文环境才会触发该事件。