57 lines
1.9 KiB
Java
57 lines
1.9 KiB
Java
package com.newyetai.mq.consumer;
|
||
|
||
import com.google.common.collect.Lists;
|
||
import com.newyetai.common.constant.CommonConstants;
|
||
import com.newyetai.common.util.EnvironmentUtils;
|
||
import com.newyetai.mq.config.ExchangeManager;
|
||
import com.newyetai.mq.config.QueueManager;
|
||
import com.newyetai.mq.config.RabbitCustomProperties;
|
||
import com.newyetai.mq.constant.MqConstant;
|
||
import com.newyetai.mq.consumer.handler.IMqHandler;
|
||
import org.springframework.beans.factory.InitializingBean;
|
||
import org.springframework.beans.factory.annotation.Autowired;
|
||
|
||
import java.util.List;
|
||
|
||
/**
|
||
* 初始化消息消费处理类，初始queue/binding
|
||
*
|
||
* @author sr on 2020-07-24
|
||
*/
|
||
public class InitHandler implements InitializingBean {
|
||
|
||
@Autowired(required = false)
|
||
private List<IMqHandler<?>> handlers = Lists.newArrayList();
|
||
|
||
@Autowired
|
||
private QueueManager queueManager;
|
||
|
||
@Autowired
|
||
private ExchangeManager exchangeManager;
|
||
|
||
@Autowired
|
||
private RabbitCustomProperties rabbitCustomProperties;
|
||
|
||
@Override
|
||
public void afterPropertiesSet() throws Exception {
|
||
String applicationName = EnvironmentUtils.getApplicationName();
|
||
handlers.forEach(handler -> {
|
||
String routingKey = handler.getRoutingKey();
|
||
String queueName = handler.getQueueName();
|
||
// 初始化binding
|
||
if (queueName == null) {
|
||
queueName = String.join(CommonConstants.UNDERLINE, applicationName,
|
||
EnvironmentUtils.getProfile(), routingKey);
|
||
if (handler.isBroadcast()) {
|
||
// queueName = String.join(CommonConstants.UNDERLINE, queueName,rabbitCustomProperties.getBroadcastId());
|
||
String nacosDiscoveryIp = EnvironmentUtils.getProperty(MqConstant.NACOS_DISCOVERY_IP);
|
||
queueName = String.join(CommonConstants.UNDERLINE, queueName, nacosDiscoveryIp);
|
||
}
|
||
}
|
||
queueManager.addQueueAndRoutingKey(
|
||
exchangeManager.getExchangeByType(handler.getExchangeType()), queueName, routingKey);
|
||
});
|
||
}
|
||
|
||
}
|