package com.gkhy.safePlatform.doublePrevention.mq.consumer; import com.gkhy.safePlatform.commons.utils.CalcuteDelayLevelUtils; import com.gkhy.safePlatform.doublePrevention.mq.msg.PreventCreateTaskMsg; import com.gkhy.safePlatform.doublePrevention.service.DangerService; import com.gkhy.safePlatform.commons.utils.JsonUtils; import org.apache.rocketmq.client.consumer.DefaultMQPushConsumer; import org.apache.rocketmq.client.exception.MQClientException; import org.apache.rocketmq.client.producer.SendStatus; import org.apache.rocketmq.common.message.MessageExt; import org.apache.rocketmq.spring.annotation.RocketMQMessageListener; import org.apache.rocketmq.spring.core.RocketMQListener; import org.apache.rocketmq.spring.core.RocketMQTemplate; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.annotation.Value; import org.springframework.messaging.Message; import org.springframework.messaging.support.MessageBuilder; import org.springframework.stereotype.Service; import java.nio.charset.StandardCharsets; import java.util.Date; @Service @RocketMQMessageListener(topic = "${rocketmq.topic.preventCreateTaskTopic}",consumerGroup = "${rocketmq.consumers.CreateGroup}",consumeThreadNumber = 1) public class CreateTaskConsumer implements RocketMQListener { private final Logger logger = LoggerFactory.getLogger(this.getClass()); @Value("${rocketmq.topic.preventCreateTaskTopic}") private String preventCreateTaskTopic; @Autowired private RocketMQTemplate rocketMQTemplate; @Autowired private DangerService dangerService; @Override public void onMessage(MessageExt messageExt) { System.out.println("\n【双重预防】preventCreateTaskConsumer接收到消息 MSG ID : " + messageExt.getMsgId() + "---->> 【##】接收时间" + new Date()); PreventCreateTaskMsg msg = JsonUtils.parse(new String(messageExt.getBody(), StandardCharsets.UTF_8), PreventCreateTaskMsg.class); //Message message = MessageBuilder.withPayload(msgStr).build(); if(msg!=null && msg.getWorkId()>0){ //当前时间早于目标处理时间前0秒 Long relaySeconds = (msg.getCreateTaskTime().getTime() - new Date().getTime())/1000; if(relaySeconds > 0){ //重发消息,实现连续延时 Message message = MessageBuilder.withPayload(msg).build(); logger.info("【双重预防】创建任务的消息延时"); //, 3000, CalcuteDelayLevelUtils.calcuteDelayLevel(relaySeconds) if(rocketMQTemplate.syncSend(preventCreateTaskTopic, message,10000, CalcuteDelayLevelUtils.calcuteDelayLevel(relaySeconds)).getSendStatus() != SendStatus.SEND_OK){ throw new RuntimeException("重发创建任务消息失败" + msg.getWorkId()); } }else if(relaySeconds < -4 * 60 * 60L){ logger.info("【双重预防】任务消息过时,不执行任务创建"); }else { //创建task logger.info("【双重预防】开始创建任务" + new Date()); System.out.println(msg); if(dangerService.createAutoTask(msg.getWorkId()) != true){ throw new RuntimeException("mq消息创建任务失败"); } } } } // @Override public void prepareStart(DefaultMQPushConsumer consumer) { // consumer.setConsumeFromWhere(ConsumeFromWhere.CONSUME_FROM_LAST_OFFSET); consumer.setConsumeThreadMin(5); // consumer.setConsumeTimestamp(UtilAll.timeMillisToHumanString3(System.currentTimeMillis())); try { consumer.subscribe(preventCreateTaskTopic,"*"); } catch (MQClientException e) { e.printStackTrace(); } } }