com.github.liaomengge.base_common.mq.rabbitmq.listener.RetryMQMessageListener Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of base-common-middleware Show documentation
base common tool for developer
The newest version!
package com.github.liaomengge.base_common.mq.rabbitmq.listener;

import com.fasterxml.jackson.databind.JsonMappingException;
import com.github.liaomengge.base_common.mq.consts.MetricsConst;
import com.github.liaomengge.base_common.mq.domain.MQMessage;
import com.github.liaomengge.base_common.mq.domain.MessageHeader;
import com.github.liaomengge.base_common.mq.rabbitmq.domain.QueueConfig;
import com.github.liaomengge.base_common.mq.rabbitmq.enums.DeadLetterReasonEnum;
import com.github.liaomengge.base_common.mq.rabbitmq.monitor.DefaultMQMonitor;
import com.github.liaomengge.base_common.utils.date.LyJdk8DateUtil;
import com.github.liaomengge.base_common.utils.json.LyJsonUtil;
import com.github.liaomengge.base_common.utils.trace.LyTraceLogUtil;
import com.rabbitmq.client.Channel;
import com.rabbitmq.client.ShutdownSignalException;
import lombok.Setter;
import org.apache.commons.collections4.CollectionUtils;
import org.apache.commons.collections4.MapUtils;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.core.MessageProperties;
import org.springframework.amqp.rabbit.support.ConsumerCancelledException;

import java.io.IOException;
import java.util.List;
import java.util.Map;

/**
 * Created by liaomengge on 16/12/22.
 */
public abstract class RetryMQMessageListener extends BaseMQMessageListener {

    @Setter
    private int retryPublish = 3;

    public RetryMQMessageListener(QueueConfig queueConfig, DefaultMQMonitor mqMonitor) {
        super(queueConfig, mqMonitor);
    }

    @Override
    public void onMessage(Message message, Channel channel) throws Exception {
        long startTime = LyJdk8DateUtil.getMilliSecondsTime();
        long endTime;
        T t = null;
        try {
            t = parseMessage(message);
            if (t == null) {
                return;
            }
            MessageHeader messageHeader = resolveMessageHeader(message);
            mqMonitor.monitorTime(MetricsConst.SEND_2_RECEIVE_EXEC_TIME + "." + queueConfig.getExchangeName(),
                    LyJdk8DateUtil.getMilliSecondsTime() - messageHeader.getSendTime());

            LyTraceLogUtil.put(messageHeader.getMqTraceId());
            startTime = LyJdk8DateUtil.getMilliSecondsTime();
            //业务逻辑
            processListener(t);

            mqMonitor.monitorCount(MetricsConst.DEQUEUE_COUNT + "." + queueConfig.getExchangeName());
            channel.basicAck(message.getMessageProperties().getDeliveryTag(), false);
        } catch (ClassCastException | JsonMappingException e) {
            log.error("Receive Message[" + LyJsonUtil.toJson4Log(t) + "] Format Error ===> ", e);
            channel.basicAck(message.getMessageProperties().getDeliveryTag(), false);
        } catch (ShutdownSignalException | ConsumerCancelledException e) {
            log.error("Handle Message[" + LyJsonUtil.toJson4Log(t) + "] Failed ===> ", e);
        } catch (Exception e) {
            mqMonitor.monitorCount(MetricsConst.EXEC_EXCEPTION + "." + queueConfig.getExchangeName());
            log.error("Handle Message[" + LyJsonUtil.toJson4Log(t) + "] Failed ===> ", e);
            try {
                MessageProperties messageProperties = message.getMessageProperties();
                Map headerMap = messageProperties.getHeaders();
                if (MapUtils.isNotEmpty(headerMap)) {
                    List> xDeathList = (List>) headerMap.get("x-death");
                    if (CollectionUtils.isNotEmpty(xDeathList)) {
                        for (Map xDeath : xDeathList) {
                            if (DeadLetterReasonEnum.EXPIRED.getCode().equals(xDeath.get("reason")) &&
                                    MapUtils.getIntValue(xDeath, "count") >= retryPublish) {
                                channel.basicAck(messageProperties.getDeliveryTag(), false);
                                return;
                            }
                        }
                    }
                }

                channel.basicNack(message.getMessageProperties().getDeliveryTag(), false, false);
            } catch (IOException ioe) {
                log.error("Enq Message[" + message.toString() + "], Reject/Nack Exception ===> ", ioe);
                channel.basicAck(message.getMessageProperties().getDeliveryTag(), false);
            }
        } finally {
            endTime = LyJdk8DateUtil.getMilliSecondsTime();
            mqMonitor.monitorTime(MetricsConst.RECEIVE_2_HANDLE_EXEC_TIME + "." + queueConfig.getExchangeName(),
                    endTime - startTime);
            LyTraceLogUtil.clear();
        }
    }
}