郑永安
2023-06-19 7a6abd05683528032687c75e80e0bd2030a3e46c
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
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<MessageExt>, 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();
        }
    }
}