RabbitAdmin spring的mq底层封装,由于各种declare操作比如declareQueue,declareExchange
SimpleMessageListenerContainer,spring的监听封装
创建mq动态添加类
@Slf4j
public class MqListener {
public MqListener() {
param.put("x-queue-type", "classic");
}
private String mqServiceName;
private SimpleMessageListenerContainer container = null;
private Map<String, Object> param = new HashMap<>();
private RabbitAdmin rabbitAdmin = null;
private IMqHandler mqHandler;
public MqListener(ConnectionFactory connectionFactory, String mqServiceName,IMqHandler mqHandler) {
this.mqServiceName = mqServiceName;
this.mqHandler = mqHandler;
this.rabbitAdmin = new RabbitAdmin(connectionFactory);
rabbitAdmin.setAutoStartup(true);
rabbitAdmin.afterPropertiesSet();
container = new SimpleMessageListenerContainer();
container.setAmqpAdmin(rabbitAdmin);
container.setConcurrentConsumers(1);
// 最大的并发消费者
container.setMaxConcurrentConsumers(5);
// 设置是否重回队列
container.setDefaultRequeueRejected(false);
// 设置签收模式
container.setAcknowledgeMode(AcknowledgeMode.MANUAL);
//设置消费者的Arguments
Map<String, Object> args = new HashMap<>();
args.put("type", "my-mq-listener");
container.setConsumerArguments(args);
container.setConsumerTagStrategy((queue) -> mqServiceName + "_queue_" + queue + "_" + UUID.randomUUID());
// 设置非独占模式
container.setExclusive(false);
// 设置consumer未被 ack 的消息个数
container.setPrefetchCount(1);
container.setMessageListener((ChannelAwareMessageListener) (message, channel) -> {
try {
this.mqHandler.todo(this.mqServiceName, message);
channel.basicAck(message.getMessageProperties().getDeliveryTag(), false);
} catch (Exception e) {
channel.basicNack(message.getMessageProperties().getDeliveryTag(), false, true);
}
});
container.setConnectionFactory(connectionFactory);
container.afterPropertiesSet();
}
/**
* 需要监听的queueNames
*
* @param queueNames
*/
public SimpleMessageListenerContainer addListenerQueue(String... queueNames) {
this.container.addQueueNames(queueNames);
return this.container;
}
/**
* 再服务端声明 关系
*
* @param queueName
* @param routeKey
* @param exchange
* @param exchangeType
*/
public void declareQueue(String queueName, String routeKey, String exchange, String exchangeType) {
if (StringUtils.isEmpty(routeKey)) {
log.warn("add routeKey is null");
return;
}
if (topic.equals(exchangeType)) {
TopicExchange ex = new TopicExchange(exchange, true, false);
Queue queue = new Queue(queueName, true, false, false, param);
this.rabbitAdmin.declareExchange(ex);
this.rabbitAdmin.declareQueue(queue);
Binding binding = BindingBuilder.bind(queue).to(ex).with(routeKey);
this.rabbitAdmin.declareBinding(binding);
} else {
throw new RuntimeException("not support " + exchangeType);
}
}
public int removeQueueNames(String... queueNames) {
this.container.removeQueueNames(queueNames);
return this.container.getQueueNames().length;
}
public void removeBind(String queueName, String routeKey, String exchange, String exchangeType) {
if (StringUtils.isEmpty(routeKey)) {
log.warn("add routeKey is null");
return;
}
if ("topic".equals(exchangeType)) {
TopicExchange ex = new TopicExchange(exchange, true, false);
Queue queue = new Queue(queueName, true, false, false, param);
Binding binding = BindingBuilder.bind(queue).to(ex).with(routeKey);
this.rabbitAdmin.removeBinding(binding);
} else {
//其他类型自己添加
throw new RuntimeException("not support " + exchangeType);
}
}
public void start() {
if (container.isRunning()) {
return;
}
this.container.start();
}
public String printInfo() {
StringBuilder sb = new StringBuilder(this.mqServiceName);
sb.append(" is ").append(this.container.isRunning() == true ? "Running" : "Stopped")
.append(" queues is ").append(JsonUtil.to(this.container.getQueueNames()));
return sb.toString();
}
public void stop() {
this.container.stop();
}
}
使用上面的类如下
1、先根据自己的环境创建connection
@Component
@Data
@ConfigurationProperties(prefix = "my.mq")
public class MqProperties {
private List<MqYml> mqYmlList;
@Data
public static class MqYml {
private String addresses;
private int port;
private String username;
private String password;
private String virtualHost;
private String mqServiceName;
}
}
private Map<String, ConnectionFactory> getMqConnection() {
Map<String, ConnectionFactory> returns = Maps.newConcurrentMap();
for (MqProperties.MqYml mqYml : mqProperties.getMqYmlList()) {
CachingConnectionFactory connectionFactory = new CachingConnectionFactory();
connectionFactory.setHost(mqYml.getAddresses());
connectionFactory.setPort(mqYml.getPort());
connectionFactory.setUsername(mqYml.getUsername());
connectionFactory.setPassword(mqYml.getPassword());
connectionFactory.setVirtualHost(mqYml.getVirtualHost());
connectionFactory.afterPropertiesSet();
returns.put(mqYml.getMqServiceName(), connectionFactory);
}
return returns;
}
2、根据自己需求创建自己服务监听
这里mqServiceName对应的是一个mqserver 可以是不通ip或者不用vhost
//mqHisHandler 自己实现了todo的监听接口
MqListener mqListener = new MqListener(connectionFactory, "serviceAname", mqHisHandler);
//开启声明,routingkey和queue和exchange的关系
mqListener.declareQueue("QueueName", "RouteKey"
, "ExchangeName", "ExchangeType");
//监听列队
addListenerQueue("QueueName1","QueueName2").start()
//动态删除列队
mqListener.stop();
//forListenerQueueNames 返回的是该监听器里面还有多少被监听的queue
int forListenerQueueNames = mqListener.removeQueueNames(lmRecordHis.getQueueName());
//如果此监听器没有监听的列队了,就不需要start了
if (forListenerQueueNames != 0) {
mqListener.start();
}
handler的实现
@Override
public void todo(String mqServiceName, Message message) {
try {
String routeKey = message.getMessageProperties().getReceivedRoutingKey();
String exchange = message.getMessageProperties().getReceivedExchange();
String queueName = message.getMessageProperties().getConsumerQueue();
//这里需要判断routeKey 是否是自己需要的
//为什么需要判断请看我上一个文章
} catch (Exception e) {
log.error("save message error ", e);
}
}