package com.matrix.component.rabbitmq;
|
|
import com.matrix.core.tools.LogUtil;
|
import com.matrix.core.tools.StringUtils;
|
import com.rabbitmq.client.Channel;
|
import com.rabbitmq.client.Connection;
|
|
import java.io.IOException;
|
import java.util.HashMap;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.concurrent.TimeoutException;
|
|
/**
|
* mq消息模板
|
*
|
* @Author jyy
|
*/
|
public class RabiitMqTemplate {
|
|
private Connection connection;
|
|
private Map<String, MqTask> taskMap = new HashMap<>();
|
|
public RabiitMqTemplate(Connection connection) {
|
this.connection = connection;
|
}
|
|
/**
|
* 申明一个交换机
|
*
|
* @param exchangeName
|
* @param direct
|
* @throws IOException
|
*/
|
public void exchangeDeclare(String exchangeName, String direct) throws IOException {
|
Channel channel = connection.createChannel();
|
channel.exchangeDeclare(exchangeName, direct);
|
}
|
|
/**
|
* 绑定任务队列以及交换机的关系
|
*
|
* @param taskList
|
* @throws IOException
|
*/
|
public void binding(List<MqTask> taskList) throws IOException {
|
for (MqTask task : taskList) {
|
LogUtil.info("绑定任务{}", task.getRoutingKey());
|
Channel channel = connection.createChannel();
|
//申明队列
|
channel.queueDeclare(task.getQueue(), true, false, false, null);
|
//绑定队列
|
channel.queueBind(task.getQueue(), task.getExchange(), task.getRoutingKey());
|
|
if (task.getHander() != null) {
|
LogUtil.info("消费者不为空{}", task.getHander());
|
channel.basicConsume(task.getQueue(), task.getAutoAck(), task.getHander(), consumerTag -> {
|
LogUtil.debug("客户端取消");
|
});
|
|
} else {
|
LogUtil.info("消费者未定义，跳过绑定");
|
}
|
taskMap.put(task.getRoutingKey(), task);
|
|
}
|
|
|
}
|
|
/**
|
* 根据路由key发送消息到交换机
|
*
|
* @param routingKey
|
* @param content
|
*/
|
public void sendMsg(String routingKey, String content) {
|
Channel channel = null;
|
try {
|
channel = connection.createChannel();
|
if (channel != null) {
|
MqTask task = taskMap.get(routingKey);
|
if (task != null) {
|
// 消息内容
|
if (StringUtils.isNotBlank(content)) {
|
channel.basicPublish(task.getExchange(), routingKey, false, null, content.getBytes());
|
} else {
|
LogUtil.info("本次发送空消息");
|
channel.basicPublish(task.getExchange(), routingKey, false, null, null);
|
}
|
//关闭通道和连接
|
channel.close();
|
} else {
|
throw new IllegalArgumentException("根据【" + routingKey + "】未获取到绑定的MqTask,请检查路由key是否正确");
|
}
|
} else {
|
LogUtil.error("消息发送失败，通道获取失败 chnnel={},routingKey={},content={}", channel, routingKey, content);
|
}
|
|
} catch (IOException | TimeoutException e) {
|
LogUtil.error("通道IO异常", e);
|
}
|
}
|
|
/**
|
* 发送消息到指定的交换机
|
* 可以试用与自定义消息和topic广播消息
|
*
|
* @param routingKey
|
* @param content
|
*/
|
public void sendTopicMsg(String exchangeName, String routingKey, String content) {
|
Channel channel = null;
|
try {
|
channel = connection.createChannel();
|
if (channel != null) {
|
|
// 消息内容
|
if (StringUtils.isNotBlank(content)) {
|
channel.basicPublish(exchangeName, routingKey, false, null, content.getBytes());
|
} else {
|
LogUtil.info("本次发送空消息");
|
channel.basicPublish(exchangeName, routingKey, false, null, null);
|
}
|
//关闭通道和连接
|
channel.close();
|
|
} else {
|
LogUtil.error("消息发送失败，通道获取失败 chnnel={},routingKey={},content={}", channel, routingKey, content);
|
}
|
|
} catch (IOException | TimeoutException e) {
|
LogUtil.error("通道IO异常", e);
|
}
|
}
|
|
|
|
/**
|
* 发送消息到指定的交换机
|
* 可以适用与自定义消息和topic广播消息
|
*
|
* @param routingKey
|
* @param content
|
*/
|
public void sendMsg(String exchangeName, String routingKey, String content) {
|
Channel channel = null;
|
try {
|
channel = connection.createChannel();
|
if (channel != null) {
|
|
// 消息内容
|
if (StringUtils.isNotBlank(content)) {
|
channel.basicPublish(exchangeName, routingKey, false, null, content.getBytes());
|
} else {
|
LogUtil.info("本次发送空消息");
|
channel.basicPublish(exchangeName, routingKey, false, null, null);
|
}
|
//关闭通道和连接
|
channel.close();
|
|
} else {
|
LogUtil.error("消息发送失败，通道获取失败 chnnel={},routingKey={},content={}", channel, routingKey, content);
|
}
|
|
} catch (IOException | TimeoutException e) {
|
LogUtil.error("通道IO异常", e);
|
}
|
}
|
|
public Connection getConnection() {
|
return connection;
|
}
|
|
public void setConnection(Connection connection) {
|
this.connection = connection;
|
}
|
}
|
