package com.gkhy.safePlatform.safeCheck.mq.consumer; import com.gkhy.safePlatform.commons.utils.JsonUtils; import com.gkhy.safePlatform.safeCheck.entity.SafeCheckTask; import com.gkhy.safePlatform.safeCheck.entity.SafeCheckWork; import com.gkhy.safePlatform.safeCheck.enums.TaskStatusEnum; import com.gkhy.safePlatform.safeCheck.mq.msg.SafeCheckTimeOutTaskMsg; import com.gkhy.safePlatform.safeCheck.service.SafeCheckTaskUnitSchedulesService; import com.gkhy.safePlatform.safeCheck.service.baseService.SafeCheckWorkService; import com.gkhy.safePlatform.safeCheck.util.CalcuteDelayLevelUtils; import com.gkhy.safePlatform.safeCheck.util.ConsolePrintIfEnvIsDevUtil; import org.apache.commons.lang3.ObjectUtils; import org.apache.rocketmq.client.consumer.DefaultMQPushConsumer; import org.apache.rocketmq.client.exception.MQClientException; import org.apache.rocketmq.client.producer.SendResult; 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.RocketMQPushConsumerLifecycleListener; 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.safeCheckTimeOutTaskTopic}",consumerGroup = "${rocketmq.consumers.safeCheckTimeOutGroup}") public class SafeCheckTimeOutTaskConsumer implements RocketMQListener, RocketMQPushConsumerLifecycleListener { private final Logger logger = LoggerFactory.getLogger(this.getClass()); @Autowired private RocketMQTemplate rocketMQTemplate; @Value("${rocketmq.topic.safeCheckTimeOutTaskTopic}") private String safeCheckTimeOutTaskTopic; @Autowired private SafeCheckWorkService safeCheckWorkService; @Autowired private SafeCheckTaskUnitSchedulesService safeCheckTaskUnitSchedulesService; @Override public void onMessage(MessageExt messageExt) { ConsolePrintIfEnvIsDevUtil.consolePrintIfEnvIsDev("【##】SafeCheckTimeOutTaskConsumer接收到消息 MSG ID : " + messageExt.getMsgId() + new Date()); // System.out.println("\n【##】SafeCheckTimeOutTaskConsumer接收到消息 \t MSG ID : " + messageExt.getMsgId() + new Date()); SafeCheckTimeOutTaskMsg msg = JsonUtils.parse(new String(messageExt.getBody(), StandardCharsets.UTF_8), SafeCheckTimeOutTaskMsg.class); //logger.info("TimeOutTaskConsumer接收到消息内容" + msg.getTaskId()+msg.getWorkId()+msg.getOutTaskTime()); //TODO 要对过期的消息做处理,可能要使用配置中心 if(msg!=null && msg.getWorkId()>0){ //当前时间早于处理时间时 Long relaySeconds = (msg.getOutTaskTime().getTime() - new Date().getTime())/1000; if(relaySeconds > 0){ //重发消息,实现连续延时 Message message = MessageBuilder.withPayload(msg).build(); SendResult sendResult = rocketMQTemplate.syncSend(safeCheckTimeOutTaskTopic, message, 10000, CalcuteDelayLevelUtils.calcuteDelayLevel(relaySeconds)); //logger.info("重发内容= "+ message + " " + message.getPayload() + " " +relaySeconds+ " " + CalcuteDelayLevelUtils.calcuteDelayLevel(relaySeconds)); if(sendResult.getSendStatus() != SendStatus.SEND_OK){ throw new RuntimeException("【智能巡检】任务到期倒计时通知消息失败" + msg.getOutTaskTime()); } ConsolePrintIfEnvIsDevUtil.consolePrintIfEnvIsDev("【智能巡检】SafeCheckTimeOut重发消息" + sendResult.getMsgId()); }else { SafeCheckTask task = safeCheckTaskUnitSchedulesService.getTaskById(msg.getTaskId()); if (ObjectUtils.isEmpty(task.getResultStatus())){ //设置任务超时 ConsolePrintIfEnvIsDevUtil.consolePrintIfEnvIsDev("【智能巡检】SafeCheckTimeOut设置任务状态为超时..."); if (safeCheckTaskUnitSchedulesService.updateTaskStatus(msg.getTaskId(), TaskStatusEnum.TASK_TIMEOUT_NO_CHECK.getStatus()) <= 0){ throw new RuntimeException("【智能巡检】任务状态设置超时失败" + msg.getOutTaskTime()); } } Date execTaskTime = msg.getExecTaskTime(); SafeCheckWork workById = safeCheckWorkService.getWorkById(msg.getWorkId()); ConsolePrintIfEnvIsDevUtil.consolePrintIfEnvIsDev("【智能巡检】任务开始执行时在调度信息中记录的开始时间"+execTaskTime+"----->>任务结束后调度信息中的时间:"+workById.getNextCheckTime()); //说明这次调度时间没有因为中断 导致调度时间被修改了 if (workById.getNextCheckTime().compareTo(execTaskTime) == 0) { //设置作业为开启 ConsolePrintIfEnvIsDevUtil.consolePrintIfEnvIsDev("【智能巡检】SafeCheckTimeOut重置作业状态开始..."); if (safeCheckTaskUnitSchedulesService.resetWorkStatus(msg.getWorkId()) <= 0) { throw new RuntimeException("【智能巡检】作业状态重置为开启状态失败" + msg.getOutTaskTime()); } } ConsolePrintIfEnvIsDevUtil.consolePrintIfEnvIsDev("【智能巡检】SafeCheckTimeOut重置作业状态完成,等待下一次调度..."); } } } @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(safeCheckTimeOutTaskTopic,"*"); } catch (MQClientException e) { e.printStackTrace(); } } }