1.配置文件
application-rabbit.yml
####################################################
####### Rabbit MQ Exchange Queue Config ##########
####################################################
rabbit:
# 交换机
exchanges:
# # 自定义-延迟
# - name: delay.mode
# type: CUSTOM
# custom-type: x-delayed-message
# arguments:
# x-delayed-type: direct
# # 通配符订阅
# - name: topic.mode
# type: TOPIC
# # 广播
# - name: fanout.mode
# type: FANOUT
# # 消息头
# - name: headers.mode
# type: HEADERS
# 直连交换机
- name: centerDeliverExchange
type: DIRECT
# 队列
queues:
# 直连队列
- name: queue-PLUS2-9002
routing-key: route-PLUS2-9002
exchange-name: centerDeliverExchange
- name: queue-PLUS2-9003
routing-key: route-PLUS2-9003
exchange-name: centerDeliverExchange
# # 队列2
# - name: queue.2
# routing-key: queue.*
# exchange-name: fanout.mode,topic.mode
# # 延迟队列
# - name: delay.queue
# routing-key: delay.queue
# exchange-name: delay.mode
将以上配置文件引入application.yml
spring:
profiles:
include: rabbit
注入配置文件 并定义交换机 队列
SpringBeanUtils.java
package com.mybatis.plus.utils;
import org.springframework.beans.BeansException;
import org.springframework.beans.factory.NoSuchBeanDefinitionException;
import org.springframework.beans.factory.config.BeanFactoryPostProcessor;
import org.springframework.beans.factory.config.ConfigurableListableBeanFactory;
import org.springframework.stereotype.Component;
/**
* spring工具类 方便在非spring管理环境中获取bean
*
* @author gch
*/
@Component
public final class SpringBeanUtils implements BeanFactoryPostProcessor {
/**
* Spring应用上下文环境
*/
private static ConfigurableListableBeanFactory beanFactory;
@Override
public void postProcessBeanFactory(ConfigurableListableBeanFactory beanFactory) throws BeansException {
SpringBeanUtils.beanFactory = beanFactory;
}
/**
* 获取对象
*
* @param name
* @return Object 一个以所给名字注册的bean的实例
* @throws BeansException
*/
@SuppressWarnings("unchecked")
public static <T> T getBean(String name) throws BeansException {
return (T) beanFactory.getBean(name);
}
/**
* 获取类型为requiredType的对象
*
* @param clz
* @return
* @throws BeansException
*/
public static <T> T getBean(Class<T> clz) throws BeansException {
T result = (T) beanFactory.getBean(clz);
return result;
}
public static <T> T getBean(String name, Class<T> clz) throws BeansException {
T result = (T) beanFactory.getBean(name, clz);
return result;
}
/**
* 如果BeanFactory包含一个与所给名称匹配的bean定义,则返回true
*
* @param name
* @return boolean
*/
public static boolean containsBean(String name) {
return beanFactory.containsBean(name);
}
/**
* 判断以给定名字注册的bean定义是一个singleton还是一个prototype。 如果与给定名字相应的bean定义没有被找到,将会抛出一个异常(NoSuchBeanDefinitionException)
*
* @param name
* @return boolean
* @throws NoSuchBeanDefinitionException
*/
public static boolean isSingleton(String name) throws NoSuchBeanDefinitionException {
return beanFactory.isSingleton(name);
}
/**
* @param name
* @return Class 注册对象的类型
* @throws NoSuchBeanDefinitionException
*/
public static Class<?> getType(String name) throws NoSuchBeanDefinitionException {
return beanFactory.getType(name);
}
/**
* 如果给定的bean名字在bean定义中有别名,则返回这些别名
*
* @param name
* @return
* @throws NoSuchBeanDefinitionException
*/
public static String[] getAliases(String name) throws NoSuchBeanDefinitionException {
return beanFactory.getAliases(name);
}
/**
* 将bean对象注册到bean工厂
*
* @param beanName
* @param bean
* @param <T>
* @return
*/
public static <T> boolean registerBean(String beanName, T bean) {
// 将bean对象注册到bean工厂
beanFactory.registerSingleton(beanName, bean);
return true;
}
}
RabbitMqProperties.java
package com.mybatis.plus.config.mq;
import com.baomidou.mybatisplus.core.toolkit.StringPool;
import com.mybatis.plus.utils.SpringBeanUtils;
import lombok.Data;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.collections.CollectionUtils;
import org.springframework.amqp.core.*;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.context.annotation.Bean;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;
/**
* rabbitmq 消息队列和交换机 配置文件
*
* @author gch
*/
@Slf4j
@Data
@ConfigurationProperties(
prefix = "rabbit"
)
public class RabbitMqProperties {
/**
* 装载自定义配置交换机
*/
private List<ExchangeConfig> exchanges = new ArrayList<>();
/**
* 装载自定义配置队列
*/
private List<QueueConfig> queues = new ArrayList<>();
@Data
public static class QueueConfig {
/**
* 队列名(每个队列的名称应该唯一)
* 必须*
*/
private String name;
/**
* 指定绑定交互机,可绑定多个(逗号分隔)
* 必须*
*/
private String exchangeName;
/**
* 队列路由键(队列绑定交换机的匹配键,例如:“user” 将会匹配到指定路由器下路由键为:【*.user、#.user】的队列)
*/
private String routingKey;
/**
* 是否为持久队列(该队列将在服务器重启后保留下来)
*/
private Boolean durable = Boolean.TRUE;
/**
* 是否为排它队列
*/
private Boolean exclusive = Boolean.FALSE;
/**
* 如果队列为空是否删除(如果服务器在不使用队列时是否删除队列)
*/
private Boolean autoDelete = Boolean.FALSE;
/**
* 头队列是否全部匹配
* 默认:是
*/
private Boolean whereAll = Boolean.TRUE;
/**
* 参数
*/
private Map<String, Object> args;
/**
* 消息头
*/
private Map<String, Object> headers;
}
@Data
public static class ExchangeConfig {
/**
* 交换机名
*/
private String name;
/**
* 交换机类型
*/
private ExchangeType type;
/**
* 自定义交换机类型
*/
private String customType;
/**
* 交换机参数(自定义交换机)
*/
private Map<String, Object> arguments;
}
public enum ExchangeType {
/**
* 自定义交换机
*/
CUSTOM,
/**
* 直连交换机(全文匹配)
*/
DIRECT,
/**
* 通配符交换机(两种通配符:*只能匹配一个单词,#可以匹配零个或多个)
*/
TOPIC,
/**
* 头交换机(自定义键值对匹配,根据发送消息内容中的headers属性进行匹配)
*/
HEADERS,
/**
* 扇形(广播)交换机 (将消息转发到所有与该交互机绑定的队列上)
*/
FANOUT;
}
public ExchangeConfig getExchangeConfig(String name) {
Map<String, ExchangeConfig> collect = exchanges.stream().collect(Collectors.toMap(e -> e.getName(), e -> e));
return collect.get(name);
}
/**
* 动态创建交换机
*
* @return
*/
@Bean
public Object createExchange() {
List<ExchangeConfig> exchanges = getExchanges();
if (!CollectionUtils.isEmpty(exchanges)) {
exchanges.forEach(e -> {
// 声明交换机
Exchange exchange = null;
switch (e.getType()) {
case DIRECT:
exchange = new DirectExchange(e.getName());
break;
case TOPIC:
exchange = new TopicExchange(e.getName());
break;
case HEADERS:
exchange = new HeadersExchange(e.getName());
break;
case FANOUT:
exchange = new FanoutExchange(e.getName());
break;
case CUSTOM:
exchange = new CustomExchange(e.getName(), e.getCustomType(), true, false, e.getArguments());
break;
default:
break;
}
// 将交换机注册到spring bean工厂 让spring实现交换机的管理
if (exchange != null) {
SpringBeanUtils.registerBean(e.getName(), exchange);
}
});
}
return null;
}
/**
* 动态绑定队列和交换机
*
* @return
*/
@Bean
public Object bindingQueueToExchange() {
List<QueueConfig> queues = getQueues();
if (!CollectionUtils.isEmpty(queues)) {
queues.forEach(q -> {
// 创建队列
Queue queue = new Queue(q.getName(), q.getDurable(),
q.getExclusive(), q.getAutoDelete(), q.getArgs());
// 注入队列bean
SpringBeanUtils.registerBean(q.getName(), queue);
// 获取队列绑定交换机名
List<String> exchangeNameList;
if (q.getExchangeName().indexOf(StringPool.COMMA) != -1) {
String[] split = q.getExchangeName().split(StringPool.COMMA);
exchangeNameList = Arrays.asList(split);
} else {
exchangeNameList = Arrays.asList(q.getExchangeName());
}
exchangeNameList.forEach(name -> {
// 获取交换机配置参数
ExchangeConfig exchangeConfig = getExchangeConfig(name);
Binding binding = bindingBuilder(queue, q, exchangeConfig);
// 将绑定关系注册到spring bean工厂 让spring实现绑定关系的管理
if (binding != null) {
log.debug("queue [{}] binding exchange [{}] success!", q.getName(), exchangeConfig.getName());
SpringBeanUtils.registerBean(q.getName() + StringPool.DASH + name, binding);
}
});
});
}
return null;
}
public Binding bindingBuilder(Queue queue, QueueConfig q, ExchangeConfig exchangeConfig) {
// 声明绑定关系
Binding binding = null;
// 根据不同的交换机模式 获取不同的交换机对象(注意:刚才注册时使用的是父类Exchange,这里获取的时候将类型获取成相应的子类)生成不同的绑定规则
switch (exchangeConfig.getType()) {
case TOPIC:
binding = BindingBuilder.bind(queue)
.to(SpringBeanUtils.getBean(exchangeConfig.getName(), TopicExchange.class))
.with(q.getRoutingKey());
break;
case DIRECT:
binding = BindingBuilder.bind(queue)
.to(SpringBeanUtils.getBean(exchangeConfig.getName(), DirectExchange.class))
.with(q.getRoutingKey());
break;
case HEADERS:
if (q.getWhereAll()) {
binding = BindingBuilder.bind(queue)
.to(SpringBeanUtils.getBean(exchangeConfig.getName(), HeadersExchange.class))
.whereAll(q.getHeaders()).match();
} else {
binding = BindingBuilder.bind(queue)
.to(SpringBeanUtils.getBean(exchangeConfig.getName(), HeadersExchange.class))
.whereAny(q.getHeaders()).match();
}
break;
case FANOUT:
binding = BindingBuilder.bind(queue)
.to(SpringBeanUtils.getBean(exchangeConfig.getName(), FanoutExchange.class));
break;
case CUSTOM:
binding = BindingBuilder.bind(queue)
.to(SpringBeanUtils.getBean(exchangeConfig.getName(), CustomExchange.class))
.with(q.getRoutingKey()).noargs();
break;
default:
log.warn("queue [{}] config unspecified exchange!", q.getName());
break;
}
return binding;
}
}
添加rabbitMQ配置
package com.mybatis.plus.config.mq;
import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.connection.CachingConnectionFactory;
import org.springframework.amqp.rabbit.connection.ConnectionFactory;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.boot.context.properties.EnableConfigurationProperties;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.PropertySource;
/**
* RabbitMQ配置
*
* @author gch
*/
@Slf4j
@Configuration
@EnableConfigurationProperties(RabbitMqProperties.class)
//@PropertySource(value = "classpath:application-rabbit.yml")
public class RabbitConfiguration {
@Bean
public RabbitTemplate createRabbitTemplate(ConnectionFactory connectionFactory){
RabbitTemplate rabbitTemplate = new RabbitTemplate();
rabbitTemplate.setConnectionFactory(connectionFactory);
//设置开启Mandatory,才能触发回调函数,无论消息推送结果怎么样都强制调用回调函数
rabbitTemplate.setMandatory(true);
rabbitTemplate.setConfirmCallback(new RabbitTemplate.ConfirmCallback() {
@Override
public void confirm(CorrelationData correlationData, boolean ack, String cause) {
System.out.println("ConfirmCallback: "+"相关数据:"+correlationData);
System.out.println("ConfirmCallback: "+"确认情况:"+ack);
System.out.println("ConfirmCallback: "+"原因:"+cause);
}
});
rabbitTemplate.setReturnCallback(new RabbitTemplate.ReturnCallback() {
@Override
public void returnedMessage(Message message, int replyCode, String replyText, String exchange, String routingKey) {
System.out.println("ReturnCallback: "+"消息:"+message);
System.out.println("ReturnCallback: "+"回应码:"+replyCode);
System.out.println("ReturnCallback: "+"回应信息:"+replyText);
System.out.println("ReturnCallback: "+"交换机:"+exchange);
System.out.println("ReturnCallback: "+"路由键:"+routingKey);
}
});
return rabbitTemplate;
}
}
配置文件注意两点
1.
@EnableConfigurationProperties(RabbitMqProperties.class)
2.就是监听需要配置文件配置:
spring:
#配置rabbitMq 服务器
rabbitmq:
host: 127.0.0.1
port: 5672
username: root
password: root
#确认消息已发送到交换机(Exchange)
#publisher-confirms: true
publisher-confirm-type: correlated
#确认消息已发送到队列(Queue)
publisher-returns: true
经调试:springBoot 版本 2.1.6.RELEASE 使用 publisher-confirm-type: correlated ; 2.1.6.RELEASE 版本使用 publisher-confirms: true
接下来就可以愉快的生产消息了
rabbitTemplate.convertAndSend("testDirectExchange", "testDirectRouting", JSONObject.toJSONString(logEntity));
贴个消费的方式
MessageListenerConfig
package com.mybatis.plus.config.mq;
import org.springframework.amqp.core.AcknowledgeMode;
import org.springframework.amqp.rabbit.connection.CachingConnectionFactory;
import org.springframework.amqp.rabbit.listener.SimpleMessageListenerContainer;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
/**
* @Author : JCccc
* @CreateTime : 2019/9/4
* @Description :
**/
@Configuration
public class MessageListenerConfig {
@Autowired
private CachingConnectionFactory connectionFactory;
@Autowired
private EurekaReceiver eurekaReceiver;//消息接收处理类
@Bean
public SimpleMessageListenerContainer simpleMessageListenerContainer() {
SimpleMessageListenerContainer container = new SimpleMessageListenerContainer(connectionFactory);
container.setConcurrentConsumers(1);
container.setMaxConcurrentConsumers(1);
container.setAcknowledgeMode(AcknowledgeMode.MANUAL); // RabbitMQ默认是自动确认,这里改为手动确认消息
//设置一个队列
container.setQueueNames("testDirectQueue");
//如果同时设置多个如下: 前提是队列都是必须已经创建存在的
// container.setQueueNames("TestDirectQueue","TestDirectQueue2","TestDirectQueue3");
//另一种设置队列的方法,如果使用这种情况,那么要设置多个,就使用addQueues
//container.setQueues(new Queue("TestDirectQueue",true));
//container.addQueues(new Queue("TestDirectQueue2",true));
//container.addQueues(new Queue("TestDirectQueue3",true));
container.setMessageListener(eurekaReceiver);
return container;
}
}
package com.mybatis.plus.config.mq;
import cn.hutool.core.bean.BeanUtil;
import cn.hutool.core.bean.copier.CopyOptions;
import com.alibaba.fastjson.JSONObject;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.mybatis.plus.entity.Log;
import com.mybatis.plus.service.ILogService;
import com.mybatis.plus.utils.EurekaUtils;
import com.mybatis.plus.utils.hash.ConsistentHash;
import com.mybatis.plus.utils.hash.pojo.ConsistentHashNode;
import com.rabbitmq.client.Channel;
import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.amqp.rabbit.listener.api.ChannelAwareMessageListener;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;
import org.springframework.transaction.PlatformTransactionManager;
import org.springframework.transaction.TransactionDefinition;
import javax.annotation.Resource;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
@Component
@Slf4j
public class EurekaReceiver implements ChannelAwareMessageListener {
@Value("${server.port}")
private String port;
@Autowired
RabbitTemplate rabbitTemplate;
@Autowired
ILogService logService;
@Override
public void onMessage(Message message, Channel channel) throws Exception {
long deliveryTag = message.getMessageProperties().getDeliveryTag();
try {
if ("testDirectQueue".equals(message.getMessageProperties().getConsumerQueue())) {
String msg = new String(message.getBody(), "UTF-8");
Log parseObject = JSONObject.parseObject(msg, Log.class);
EurekaReceiver.log.info("消费的消息来自的队列名为:" + message.getMessageProperties().getConsumerQueue());
log.info("消息成功消费到 messageId:" + parseObject.getLogUuid() + " messageData:" + parseObject.getLogTitle() + " createTime:" + parseObject.getCreateTime());
log.info("================================");
// 收到来自主机的消息 进行一致性hash分配 发往不同的服务
// 获取服务节点 创建一致hash环
ConsistentHash consistentHash = InitConfig.consistentHash;
List<Map<String, String>> allServiceAddr = EurekaUtils.getAllServiceInfo("127.0.0.1", port, "PLUS2");
if (!allServiceAddr.isEmpty()) {
for (Map<String, String> stringMap : allServiceAddr) {
String instanceId = stringMap.get("routeKey");
// 新增1个物理节点和150个对应的虚拟节点
// String instanceId = stringMap.get("queueKey");
// 如果hash环中没有该节点 才新增
ConsistentHashNode node = consistentHash.getAccurateNode(instanceId);
if (null == node) {
consistentHash.putNode(new ConsistentHashNode(consistentHash.getPoint(instanceId),instanceId),150);
}
}
} else {
//没有服务提供者 将消息返回队列
channel.basicReject(deliveryTag, true);
return;
}
// 提取消息中的某个代表来源主机的标识 然后在hash环上分配目标节点
String logUuid = parseObject.getLogUuid();
ConsistentHashNode node = consistentHash.getNode(logUuid);
log.info("主机标识:{},分配节点:{}", logUuid, node.getTarget());
//向指定路由发送消息
// todo 问题 这里怎么保证队列预先创建初始化好 解决方案 先从配置文件获取队列名称 新增服务时 需要重启服务
rabbitTemplate.convertAndSend("centerDeliverExchange", node.getTarget().toString(), msg);
// planTwo(parseObject);
channel.basicAck(deliveryTag, false); //第二个参数,手动确认可以被批处理,当该参数为 true 时,则可以一次性确认 delivery_tag 小于等于传入值的所有消息
log.info(">>>>>>>>>>>>消费消息成功!");
}
} catch (Exception e) {
log.info(">>>>>>>>>>>>消费消息失败!失败消息ID:{}, 失败原因:{}", deliveryTag, e.getMessage());
channel.basicReject(deliveryTag, true);
}
}
private void planTwo(Log parseObject) {
//先查询出数据库存在的数据(如果有)
String uuid = parseObject.getLogUuid();
//结合数据库索引 防止并发插入多条重复数据
Log byId = logService.getOne((new QueryWrapper<Log>()).eq("log_uuid", uuid));
if (byId == null) {
logService.save(parseObject);
} else {
String logOrder = byId.getLogOrder();
String logOrder1 = parseObject.getLogOrder();
//编号大于数据库编号才更新
if (Integer.valueOf(logOrder1) > Integer.valueOf(logOrder)) {
log.info("============= 新序号:{},旧序号:{}; 执行更新操作", logOrder1, logOrder);
BeanUtil.copyProperties(parseObject, byId, CopyOptions.create().ignoreNullValue());
logService.updateById(byId);
}
}
}
}
EurekaReceiver 这个类有些包我就不贴了 业务那一块替换成自己的