bobol
6 months ago
12 changed files with 440 additions and 78 deletions
@ -0,0 +1,172 @@ |
|||||
|
package com.lzbi.common.config; |
||||
|
|
||||
|
import org.springframework.amqp.core.AcknowledgeMode; |
||||
|
import org.springframework.amqp.rabbit.config.RetryInterceptorBuilder; |
||||
|
import org.springframework.amqp.rabbit.config.SimpleRabbitListenerContainerFactory; |
||||
|
import org.springframework.amqp.rabbit.connection.ConnectionFactory; |
||||
|
import org.springframework.amqp.rabbit.core.RabbitTemplate; |
||||
|
import org.springframework.amqp.rabbit.retry.RejectAndDontRequeueRecoverer; |
||||
|
import org.springframework.amqp.support.converter.Jackson2JsonMessageConverter; |
||||
|
import org.springframework.beans.factory.annotation.Autowired; |
||||
|
import org.springframework.boot.autoconfigure.amqp.RabbitProperties; |
||||
|
import org.springframework.context.annotation.Bean; |
||||
|
import org.springframework.context.annotation.Configuration; |
||||
|
import org.springframework.retry.RetryCallback; |
||||
|
import org.springframework.retry.RetryContext; |
||||
|
import org.springframework.retry.RetryListener; |
||||
|
import org.springframework.retry.backoff.ExponentialBackOffPolicy; |
||||
|
import org.springframework.retry.policy.SimpleRetryPolicy; |
||||
|
import org.springframework.retry.support.RetryTemplate; |
||||
|
|
||||
|
/** |
||||
|
* 常用的三个配置如下 |
||||
|
* 1---设置手动应答（acknowledge-mode: manual） |
||||
|
* 2---设置生产者消息发送的确认回调机制 ( #这个配置是保证提供者确保消息推送到交换机中，不管成不成功，都会回调 |
||||
|
* publisher-confirm-type: correlated |
||||
|
* # 保证交换机能把消息推送到队列中 |
||||
|
* publisher-returns: true |
||||
|
* template: |
||||
|
* # 以下是rabbitmqTemplate配置 |
||||
|
* mandatory: true) |
||||
|
* 3---设置重试 |
||||
|
*/ |
||||
|
@Configuration |
||||
|
public class RabbitConfig { |
||||
|
|
||||
|
@Autowired |
||||
|
private ConnectionFactory rabbitConnectionFactory; |
||||
|
|
||||
|
//@Bean 缓存连接池
|
||||
|
//public CachingConnectionFactory rabbitConnectionFactory
|
||||
|
|
||||
|
@Autowired |
||||
|
private RabbitProperties properties; |
||||
|
|
||||
|
//这里因为使用自动配置的connectionFactory，所以把自定义的connectionFactory注解掉
|
||||
|
// 存在此名字的bean 自带的连接工厂会不加载（也就是说yml中rabbitmq下一级不生效），如果想自定义来区分开 需要改变bean 的名称
|
||||
|
// @Bean
|
||||
|
// public ConnectionFactory connectionFactory() throws Exception {
|
||||
|
// //创建工厂类
|
||||
|
// CachingConnectionFactory cachingConnectionFactory=new CachingConnectionFactory();
|
||||
|
// //用户名
|
||||
|
// cachingConnectionFactory.setUsername("gust");
|
||||
|
// //密码
|
||||
|
// cachingConnectionFactory.setPassword("gust");
|
||||
|
// //rabbitMQ地址
|
||||
|
// cachingConnectionFactory.setHost("127.0.0.1");
|
||||
|
// //rabbitMQ端口
|
||||
|
// cachingConnectionFactory.setPort(Integer.parseInt("5672"));
|
||||
|
//
|
||||
|
// //设置发布消息后回调
|
||||
|
// cachingConnectionFactory.setPublisherReturns(true);
|
||||
|
// //设置发布后确认类型，此处确认类型为交互
|
||||
|
// cachingConnectionFactory.setPublisherConfirmType(CachingConnectionFactory.ConfirmType.CORRELATED);
|
||||
|
//
|
||||
|
// cachingConnectionFactory.setCacheMode(CachingConnectionFactory.CacheMode.CHANNEL);
|
||||
|
// return cachingConnectionFactory;
|
||||
|
// }
|
||||
|
|
||||
|
|
||||
|
// 存在此名字的bean 自带的容器工厂会不加载（yml下rabbitmq下的listener下的simple配置），如果想自定义来区分开 需要改变bean 的名称
|
||||
|
@Bean |
||||
|
public SimpleRabbitListenerContainerFactory rabbitListenerContainerFactory() { |
||||
|
SimpleRabbitListenerContainerFactory containerFactory = new SimpleRabbitListenerContainerFactory(); |
||||
|
containerFactory.setConnectionFactory(rabbitConnectionFactory); |
||||
|
|
||||
|
// 并发消费者数量
|
||||
|
containerFactory.setConcurrentConsumers(1); |
||||
|
containerFactory.setMaxConcurrentConsumers(20); |
||||
|
// 预加载消息数量 -- QOS
|
||||
|
containerFactory.setPrefetchCount(1); |
||||
|
// 应答模式（此处设置为手动）
|
||||
|
containerFactory.setAcknowledgeMode(AcknowledgeMode.MANUAL); |
||||
|
//消息序列化方式
|
||||
|
containerFactory.setMessageConverter(new Jackson2JsonMessageConverter()); |
||||
|
// 设置通知调用链 （这里设置的是重试机制的调用链）
|
||||
|
containerFactory.setAdviceChain(RetryInterceptorBuilder.stateless().recoverer(new RejectAndDontRequeueRecoverer()).retryOperations(rabbitRetryTemplate()).build()); |
||||
|
return containerFactory; |
||||
|
} |
||||
|
|
||||
|
// 存在此名字的bean 自带的容器工厂会不加载（yml下rabbitmq下的template的配置），如果想自定义来区分开 需要改变bean 的名称
|
||||
|
@Bean |
||||
|
public RabbitTemplate rabbitTemplate() { |
||||
|
RabbitTemplate rabbitTemplate = new RabbitTemplate(rabbitConnectionFactory); |
||||
|
//默认是用jdk序列化
|
||||
|
//数据转换为json存入消息队列，方便可视化界面查看消息数据
|
||||
|
rabbitTemplate.setMessageConverter(new Jackson2JsonMessageConverter()); |
||||
|
//设置开启Mandatory,才能触发回调函数,无论消息推送结果怎么样都强制调用回调函数
|
||||
|
rabbitTemplate.setMandatory(true); |
||||
|
//此处设置重试template后，会再生产者发送消息的时候，调用该template中的调用链
|
||||
|
rabbitTemplate.setRetryTemplate(rabbitRetryTemplate()); |
||||
|
//CorrelationData correlationData, boolean b, String s
|
||||
|
rabbitTemplate.setConfirmCallback((correlationData, b, s) -> { |
||||
|
System.out.println("ConfirmCallback " + "相关数据：" + correlationData); |
||||
|
System.out.println("ConfirmCallback " + "确认情况：" + b); |
||||
|
System.out.println("ConfirmCallback " + "原因：" + s); |
||||
|
}); |
||||
|
//Message message, int i, String s, String s1, String s2
|
||||
|
rabbitTemplate.setReturnCallback((message, i, s, s1, s2) -> { |
||||
|
System.out.println("ReturnCallback： " + "消息：" + message); |
||||
|
System.out.println("ReturnCallback： " + "回应码：" + i); |
||||
|
System.out.println("ReturnCallback： " + "回应消息：" + s); |
||||
|
System.out.println("ReturnCallback： " + "交换机：" + s1); |
||||
|
System.out.println("ReturnCallback： " + "路由键：" + s2); |
||||
|
}); |
||||
|
|
||||
|
return rabbitTemplate; |
||||
|
} |
||||
|
|
||||
|
// 重试的Template
|
||||
|
@Bean |
||||
|
public RetryTemplate rabbitRetryTemplate() { |
||||
|
RetryTemplate retryTemplate = new RetryTemplate(); |
||||
|
// 设置监听 调用重试处理过程
|
||||
|
retryTemplate.registerListener(new RetryListener() { |
||||
|
@Override |
||||
|
public <T, E extends Throwable> boolean open(RetryContext retryContext, RetryCallback<T, E> retryCallback) { |
||||
|
// 执行之前调用 （返回false时会终止执行）
|
||||
|
return true; |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public <T, E extends Throwable> void close(RetryContext retryContext, RetryCallback<T, E> retryCallback, Throwable throwable) { |
||||
|
// 重试结束的时候调用 （最后一次重试 ）
|
||||
|
System.out.println("---------------最后一次调用"); |
||||
|
|
||||
|
return; |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public <T, E extends Throwable> void onError(RetryContext retryContext, RetryCallback<T, E> retryCallback, Throwable throwable) { |
||||
|
// 异常 都会调用
|
||||
|
System.err.println("-----第{}次调用" + retryContext.getRetryCount()); |
||||
|
} |
||||
|
}); |
||||
|
retryTemplate.setBackOffPolicy(backOffPolicyByProperties()); |
||||
|
retryTemplate.setRetryPolicy(retryPolicyByProperties()); |
||||
|
return retryTemplate; |
||||
|
} |
||||
|
|
||||
|
@Bean |
||||
|
public ExponentialBackOffPolicy backOffPolicyByProperties() { |
||||
|
ExponentialBackOffPolicy backOffPolicy = new ExponentialBackOffPolicy(); |
||||
|
long maxInterval = properties.getListener().getSimple().getRetry().getMaxInterval().getSeconds(); |
||||
|
long initialInterval = properties.getListener().getSimple().getRetry().getInitialInterval().getSeconds(); |
||||
|
double multiplier = properties.getListener().getSimple().getRetry().getMultiplier(); |
||||
|
// 重试间隔
|
||||
|
backOffPolicy.setInitialInterval(initialInterval * 1000); |
||||
|
// 重试最大间隔
|
||||
|
backOffPolicy.setMaxInterval(maxInterval * 1000); |
||||
|
// 重试间隔乘法策略
|
||||
|
backOffPolicy.setMultiplier(multiplier); |
||||
|
return backOffPolicy; |
||||
|
} |
||||
|
|
||||
|
@Bean |
||||
|
public SimpleRetryPolicy retryPolicyByProperties() { |
||||
|
SimpleRetryPolicy retryPolicy = new SimpleRetryPolicy(); |
||||
|
int maxAttempts = properties.getListener().getSimple().getRetry().getMaxAttempts(); |
||||
|
retryPolicy.setMaxAttempts(maxAttempts); |
||||
|
return retryPolicy; |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,37 @@ |
|||||
|
package com.lzbi.common.config; |
||||
|
|
||||
|
import org.springframework.amqp.core.Binding; |
||||
|
import org.springframework.amqp.core.BindingBuilder; |
||||
|
import org.springframework.amqp.core.DirectExchange; |
||||
|
import org.springframework.amqp.core.Queue; |
||||
|
import org.springframework.context.annotation.Bean; |
||||
|
import org.springframework.context.annotation.Configuration; |
||||
|
|
||||
|
@Configuration |
||||
|
public class ReportExportRabbitConfig { |
||||
|
//创建一个名为TestDirectQueue的队列
|
||||
|
@Bean |
||||
|
public Queue ReportExportDirectQueue() { |
||||
|
// durable:是否持久化,默认是false,持久化队列：会被存储在磁盘上，当消息代理重启时仍然存在，暂存队列：当前连接有效
|
||||
|
// exclusive:默认也是false，只能被当前创建的连接使用，而且当连接关闭后队列即被删除。此参考优先级高于durable
|
||||
|
// autoDelete:是否自动删除，有消息者订阅本队列，然后所有消费者都解除订阅此队列，会自动删除。
|
||||
|
// arguments：队列携带的参数，比如设置队列的死信队列，消息的过期时间等等。
|
||||
|
return new Queue("ReportExportDirectQueue", true); |
||||
|
} |
||||
|
|
||||
|
//创建一个名为TestDirectExchange的Direct类型的交换机
|
||||
|
@Bean |
||||
|
public DirectExchange ReportExportDirectExchange() { |
||||
|
// durable:是否持久化,默认是false,持久化交换机。
|
||||
|
// autoDelete:是否自动删除，交换机先有队列或者其他交换机绑定的时候，然后当该交换机没有队列或其他交换机绑定的时候，会自动删除。
|
||||
|
// arguments：交换机设置的参数，比如设置交换机的备用交换机（Alternate Exchange），当消息不能被路由到该交换机绑定的队列上时，会自动路由到备用交换机
|
||||
|
return new DirectExchange("ReportExportDirectExchange", true, false); |
||||
|
} |
||||
|
|
||||
|
//绑定交换机和队列
|
||||
|
@Bean |
||||
|
public Binding bindingDirect() { |
||||
|
//bind队列to交换机中with路由key（routing key）
|
||||
|
return BindingBuilder.bind(ReportExportDirectQueue()).to(ReportExportDirectExchange()).with("123"); |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,26 @@ |
|||||
|
package com.lzbi.rabbit; |
||||
|
|
||||
|
import com.lzbi.common.core.domain.AjaxResult; |
||||
|
import com.lzbi.rabbit.producer.ReportExportProducer; |
||||
|
import org.springframework.beans.factory.annotation.Autowired; |
||||
|
import org.springframework.web.bind.annotation.GetMapping; |
||||
|
import org.springframework.web.bind.annotation.RequestMapping; |
||||
|
import org.springframework.web.bind.annotation.RestController; |
||||
|
|
||||
|
import java.util.concurrent.atomic.AtomicLong; |
||||
|
|
||||
|
@RestController |
||||
|
@RequestMapping("/test/rabbit") |
||||
|
public class TestRabbitController { |
||||
|
|
||||
|
@Autowired |
||||
|
private ReportExportProducer reportExportProducer; |
||||
|
|
||||
|
private static final AtomicLong id = new AtomicLong(0L); |
||||
|
|
||||
|
@GetMapping("/sendMessage") |
||||
|
public AjaxResult sendDirectMessage(){ |
||||
|
reportExportProducer.sendMsg(id.addAndGet(1L)); |
||||
|
return AjaxResult.success(); |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,11 @@ |
|||||
|
package com.lzbi.rabbit.consumer; |
||||
|
|
||||
|
import com.rabbitmq.client.Channel; |
||||
|
import org.springframework.amqp.core.Message; |
||||
|
|
||||
|
public interface IConsumerService { |
||||
|
|
||||
|
void process(Object msg, Channel channel, Message message); |
||||
|
|
||||
|
void business(Object msg); |
||||
|
} |
||||
@ -0,0 +1,81 @@ |
|||||
|
package com.lzbi.rabbit.consumer.impl; |
||||
|
|
||||
|
import cn.hutool.core.io.IoUtil; |
||||
|
import cn.hutool.poi.excel.BigExcelWriter; |
||||
|
import cn.hutool.poi.excel.ExcelUtil; |
||||
|
import com.lzbi.common.constant.BizConstants; |
||||
|
import com.lzbi.rabbit.consumer.IConsumerService; |
||||
|
import com.lzbi.report.domain.DcBusiReportRecord; |
||||
|
import com.lzbi.report.service.DcBusiReportRecordService; |
||||
|
import com.rabbitmq.client.Channel; |
||||
|
import lombok.extern.slf4j.Slf4j; |
||||
|
import org.springframework.amqp.core.Message; |
||||
|
import org.springframework.amqp.rabbit.annotation.RabbitHandler; |
||||
|
import org.springframework.amqp.rabbit.annotation.RabbitListener; |
||||
|
import org.springframework.beans.factory.annotation.Autowired; |
||||
|
import org.springframework.stereotype.Component; |
||||
|
|
||||
|
import java.io.IOException; |
||||
|
import java.io.OutputStream; |
||||
|
import java.nio.file.Files; |
||||
|
import java.nio.file.Path; |
||||
|
import java.nio.file.Paths; |
||||
|
import java.util.List; |
||||
|
import java.util.Map; |
||||
|
|
||||
|
@Slf4j |
||||
|
@RabbitListener(queues = "ReportExportDirectQueue") |
||||
|
@Component |
||||
|
public class ReportExportConsumer implements IConsumerService { |
||||
|
|
||||
|
@Autowired |
||||
|
private DcBusiReportRecordService dcBusiReportRecordService; |
||||
|
|
||||
|
@RabbitHandler |
||||
|
@Override |
||||
|
public void process(Object msg, Channel channel, Message message) { |
||||
|
log.info("收到消息：{}", msg); |
||||
|
try { |
||||
|
this.business(msg); |
||||
|
//由于配置设置了手动应答，所以这里要进行一个手动应答。注意：如果设置了自动应答，这里又进行手动应答，会出现double ack，那么程序会报错。
|
||||
|
channel.basicAck(message.getMessageProperties().getDeliveryTag(), false); |
||||
|
} catch (IOException e) { |
||||
|
throw new RuntimeException(e); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void business(Object msg) { |
||||
|
Long reportRecordId = (Long) msg; |
||||
|
DcBusiReportRecord dcBusiReportRecord = dcBusiReportRecordService.selectDcBusiReportRecordById(reportRecordId); |
||||
|
// try {
|
||||
|
//
|
||||
|
// // 更新报表记录
|
||||
|
// dcBusiReportRecord.setStatus(BizConstants.ReportRecordStatus.NOT_DOWNLOAD);
|
||||
|
// dcBusiReportRecordService.updateDcBusiReportRecord(dcBusiReportRecord);
|
||||
|
// } catch (IOException e) {
|
||||
|
// log.error("导出数据异常", e);
|
||||
|
// // 更新报表记录
|
||||
|
// dcBusiReportRecord.setStatus(BizConstants.ReportRecordStatus.ERROR);
|
||||
|
// dcBusiReportRecordService.updateDcBusiReportRecord(dcBusiReportRecord);
|
||||
|
// }
|
||||
|
} |
||||
|
|
||||
|
|
||||
|
|
||||
|
// private void toExcel(DcBusiReportRecord dcBusiReportRecord, List<Map<String, Object>> mapList) throws IOException {
|
||||
|
// Path folder = Paths.get(reportFilePath);
|
||||
|
// if (!Files.exists(folder)) {
|
||||
|
// Files.createDirectories(folder);
|
||||
|
// }
|
||||
|
// OutputStream out = Files.newOutputStream(Paths.get(dcBusiReportRecord.getUrl()));
|
||||
|
// BigExcelWriter writer = ExcelUtil.getBigWriter();
|
||||
|
// writer.write(mapList, true);
|
||||
|
// writer.flush(out, true);
|
||||
|
// // 关闭writer，释放内存
|
||||
|
// writer.close();
|
||||
|
// //此处记得关闭输出流
|
||||
|
// IoUtil.close(out);
|
||||
|
// }
|
||||
|
|
||||
|
} |
||||
@ -0,0 +1,25 @@ |
|||||
|
package com.lzbi.rabbit.producer; |
||||
|
|
||||
|
import lombok.extern.slf4j.Slf4j; |
||||
|
import org.springframework.amqp.rabbit.connection.CorrelationData; |
||||
|
import org.springframework.amqp.rabbit.core.RabbitTemplate; |
||||
|
import org.springframework.beans.factory.annotation.Autowired; |
||||
|
import org.springframework.stereotype.Component; |
||||
|
|
||||
|
import java.util.UUID; |
||||
|
|
||||
|
/** |
||||
|
* 报表生产者 |
||||
|
*/ |
||||
|
@Slf4j |
||||
|
@Component |
||||
|
public class ReportExportProducer { |
||||
|
|
||||
|
@Autowired |
||||
|
private RabbitTemplate rabbitTemplate; |
||||
|
|
||||
|
public void sendMsg(Long reportRecordId) { |
||||
|
log.info("发送消息，导出报表，报表记录id：{}", reportRecordId); |
||||
|
rabbitTemplate.convertAndSend("ReportExportDirectExchange", "123", reportRecordId, new CorrelationData(UUID.randomUUID().toString())); |
||||
|
} |
||||
|
} |
||||
Loading…
Reference in new issue