package com.gk.hotwork.doublePrevention.mq.consumer;
|
|
import com.gk.hotwork.doublePrevention.utils.CalcuteDelayLevelUtils;
|
import com.gk.hotwork.doublePrevention.mq.msg.PreventCreateTaskMsg;
|
import com.gk.hotwork.doublePrevention.service.DangerService;
|
import com.gk.hotwork.doublePrevention.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<MessageExt> {
|
|
|
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【##】CreateTaskConsumer接收到消息 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)
|
System.out.println("【##】创建任务的消息延时");
|
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();
|
}
|
}
|
|
|
}
|