package com.eloancn.back.schedule.core.config.mq;
import com.rabbitmq.client.Channel;
import org.springframework.amqp.core.AcknowledgeMode;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.config.SimpleRabbitListenerContainerFactory;
import org.springframework.amqp.rabbit.connection.ConnectionFactory;
import org.springframework.amqp.rabbit.core.ChannelAwareMessageListener;
import org.springframework.amqp.rabbit.listener.SimpleMessageListenerContainer;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.beans.factory.config.ConfigurableBeanFactory;
import org.springframework.boot.autoconfigure.amqp.SimpleRabbitListenerContainerFactoryConfigurer;
import org.springframework.context.annotation.*;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.amqp.rabbit.connection.CachingConnectionFactory;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.amqp.core.Queue;
/**
* Created by famous on 2017/8/8.
*/
@Configuration
public class RabbitConfig {
@Bean(name = "payAdvanceConnectionFactory")
@Primary
public ConnectionFactory firstConnectionFactory(
@Value("${spring.rabbitmq.first.host}") String host,
@Value("${spring.rabbitmq.first.port}") int port,
@Value("${spring.rabbitmq.first.username}") String username,
@Value("${spring.rabbitmq.first.password}") String password,
@Value("${spring.rabbitmq.first.vhost}") String vhost) {
CachingConnectionFactory connectionFactory = new CachingConnectionFactory();
connectionFactory.setHost(host);
connectionFactory.setPort(port);
connectionFactory.setUsername(username);
connectionFactory.setPassword(password);
connectionFactory.setVirtualHost(vhost);
connectionFactory.setPublisherConfirms(true);必须要设置
return connectionFactory;
}
@Bean(name = "advanceRabbitTemplate")
@Scope(ConfigurableBeanFactory.SCOPE_PROTOTYPE)
@Primary
public RabbitTemplate advanceRabbitTemplate(@Qualifier("payAdvanceConnectionFactory") ConnectionFactory connectionFactory) {
RabbitTemplate advanceRabbitTemplate = new RabbitTemplate(connectionFactory);
return advanceRabbitTemplate;
}
@Bean(name = "payRabbitTemplate")
@Scope(ConfigurableBeanFactory.SCOPE_PROTOTYPE)
public RabbitTemplate payRabbitTemplate(@Qualifier("payAdvanceConnectionFactory") ConnectionFactory connectionFactory) {
RabbitTemplate payRabbitTemplate = new RabbitTemplate(connectionFactory);
return payRabbitTemplate;
}
@Bean
public SimpleMessageListenerContainer messageContainer(@Qualifier("payAdvanceConnectionFactory") ConnectionFactory connectionFactory) {
SimpleMessageListenerContainer container = new SimpleMessageListenerContainer(connectionFactory);
container.setQueueNames("queue_backadvance_for_app");
container.setExposeListenerChannel(true);
container.setMaxConcurrentConsumers(1);//??
container.setConcurrentConsumers(1);//??
container.setAcknowledgeMode(AcknowledgeMode.MANUAL); //设置确认模式手工确认
container.setMessageListener(new ChannelAwareMessageListener() {
@Override
public void onMessage(Message message, Channel channel) throws Exception {
byte[] body = message.getBody();
System.out.println("###########RRRreceive msg : " + new String(body));
channel.basicAck(message.getMessageProperties().getDeliveryTag(), false); //确认消息成功消费
}
});
return container;
}
}
package com.eloancn.back.schedule.core.custom.confirm.repayment.mq;
import com.eloancn.back.schedule.api.definition.BizTypeEnum;
import com.eloancn.back.schedule.core.config.biz.DTCRoleEnum;
import com.eloancn.back.schedule.core.config.biz.InteractionResult;
import com.eloancn.back.schedule.core.config.biz.InteractiveStatusEnum;
import com.eloancn.back.schedule.core.config.biz.ScheduleStatusEnum;
import com.eloancn.back.schedule.core.config.schedule.DTCConfig;
import com.eloancn.back.schedule.core.model.ScheduleLog;
import com.eloancn.back.schedule.core.model.ScheduleParam;
import com.eloancn.back.schedule.core.model.ScheduleRecord;
import com.eloancn.back.schedule.core.schedule.template.BizConfirm;
import com.eloancn.back.schedule.core.service.RecordQueryService;
import com.eloancn.back.schedule.core.service.ScheduleBizService;
import com.eloancn.back.schedule.core.service.ScheduleParamService;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.amqp.rabbit.support.CorrelationData;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.context.annotation.Profile;
import org.springframework.stereotype.Service;
/**
* Created by famous on 2017/8/8.
*/
@Service("overdueAdvSendMqConfirm")
@DTCConfig(bizType = {BizTypeEnum.OVERDUE_ADVANCE_NOTIFY}, tccRole = DTCRoleEnum.REMOTE_CONFIRM)
public class OverdueAdvSendMqConfirm implements RabbitTemplate.ConfirmCallback, BizConfirm {
private static Logger logger = LoggerFactory.getLogger(OverdueAdvSendMqConfirm.class);
private RabbitTemplate rabbitTemplate;
private RecordQueryService recordQueryService;
private ScheduleBizService scheduleBizService;
private ScheduleParamService paramService;
@Autowired
public OverdueAdvSendMqConfirm(@Qualifier("payRabbitTemplate") RabbitTemplate rabbitTemplate,
RecordQueryService recordQueryService,
ScheduleBizService scheduleBizService,
ScheduleParamService paramService) {
this.rabbitTemplate = rabbitTemplate;
this.recordQueryService = recordQueryService;
this.scheduleBizService = scheduleBizService;
this.paramService = paramService;
rabbitTemplate.setConfirmCallback(this);
}
@Override
public InteractionResult confirm(ScheduleRecord scheduleRecord, ScheduleLog recordLog) {
ScheduleParam scheduleRecordParam = paramService.queryByBizId(scheduleRecord.getBizId());
CorrelationData correlationId = new CorrelationData(scheduleRecord.getBizId());
logger.info("逾期垫付事件的广播scheduleRecord********{} 推送还款的消息是:{}", scheduleRecord.getId(), scheduleRecordParam.getConfirmParam());
rabbitTemplate.convertAndSend("exchange_topic_schedule", "key.backrepayment.advance.overdue", scheduleRecordParam.getConfirmParam(), correlationId);
InteractionResult interactionResult = new InteractionResult();
// 默认是UNKNOW 再这个阶段 是不知道MQ是否发送成功的
interactionResult.setInteractiveStatusEnum(InteractiveStatusEnum.UNKNOW);
return interactionResult;
}
@Override
public void confirm(CorrelationData correlationData, boolean ack, String cause) {
String bizId = correlationData.getId();
ScheduleRecord scheduleRecord = recordQueryService.queryByBizId(bizId);
logger.info("逾期垫付事件的广播scheduleRecord********{} 收到MQ发送成功回调的消息是:correlationData:{} ack:{} cause:{}",
scheduleRecord.getId(), correlationData.getId(), ack, cause);
if (ack) {
scheduleBizService.updateStatus(scheduleRecord, ScheduleStatusEnum.SUCCESS);
} else {
scheduleBizService.updateStatus(scheduleRecord, ScheduleStatusEnum.INIT);
}
}
}
# RabbitMQ
spring.rabbitmq.first.host=****
spring.rabbitmq.first.port=****
spring.rabbitmq.first.username=****
spring.rabbitmq.first.password=****
spring.rabbitmq.first.vhost=/***