com.github.life.lab.leisure.common.rocketmq.consumer.RocketMqConsumerHolder Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of common-rocketmq Show documentation
There is a newer version: 0.0.9-RELEASE
package com.github.life.lab.leisure.common.rocketmq.consumer;


import com.aliyun.openservices.ons.api.Action;
import com.aliyun.openservices.ons.api.Consumer;
import com.aliyun.openservices.ons.api.ONSFactory;
import com.aliyun.openservices.ons.api.PropertyKeyConst;
import com.github.life.lab.leisure.common.rocketmq.consumer.enhance.ConsumeFailEnhance;
import com.github.life.lab.leisure.common.rocketmq.properties.RocketMqProperties;
import com.github.life.lab.leisure.common.utils.ByteArrayUtils;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.collections.CollectionUtils;
import org.apache.commons.lang3.StringUtils;
import org.springframework.context.ApplicationListener;
import org.springframework.context.event.ContextRefreshedEvent;

import java.text.MessageFormat;
import java.util.ArrayList;
import java.util.List;
import java.util.Properties;

/**
 * 阿里云 ONS 消费者管理器
 *
 * @author weichao.li ([email protected])
 * @date 2018/8/6
 */
@Slf4j
public class RocketMqConsumerHolder implements ApplicationListener {

    private List enhances;

    private List consumerHandlers;

    private List consumers = new ArrayList<>();

    public RocketMqConsumerHolder(RocketMqProperties aliyunOnsProperties,
                                  List consumerHandlers,
                                  List enhances) {
        log.info("rocket mq consumer is loading！");
        this.enhances = enhances;
        this.consumerHandlers = consumerHandlers;
        if (CollectionUtils.isNotEmpty(consumerHandlers)) {
            Properties properties = new Properties();
            properties.put(PropertyKeyConst.AccessKey, aliyunOnsProperties.getAccessKey());
            properties.put(PropertyKeyConst.SecretKey, aliyunOnsProperties.getSecretKey());
            if (StringUtils.isNotBlank(aliyunOnsProperties.getNameServerAddress())) {
                properties.put(PropertyKeyConst.NAMESRV_ADDR, aliyunOnsProperties.getNameServerAddress());
            }
            for (RocketMqConsumer consumerHandler : consumerHandlers) {
                properties.put(PropertyKeyConst.ConsumerId, consumerHandler.getConsumerId());
                properties.put(PropertyKeyConst.ConsumeThreadNums, consumerHandler.getThreadNums());
                Consumer consumer = ONSFactory.createConsumer(properties);
                consumer.subscribe(consumerHandler.getTopic(), consumerHandler.getTag(), (message, context) -> {
                    try {
                        consumerHandler.handle(message, context);
                        return Action.CommitMessage;
                    } catch (Exception e) {
                        if (message.getReconsumeTimes() >= consumerHandler.getTriggerNotificationTimes()) {
                            for (ConsumeFailEnhance enhance : enhances) {
                                //// 消息 消费 失败，进行增强处理。
                                String consumerId = consumerHandler.getConsumerId();
                                if (enhance.support(consumerId, message)) {
                                    enhance.enhance(consumerId, message, e);
                                }
                            }
                        }
                        log.error(MessageFormat.format("consume ons message fail - [message_id:{} -> message_body:{}]", message.getMsgID(), ByteArrayUtils.convertByteArray2JsonString(message.getBody())), e);
                        return Action.ReconsumeLater;
                    }
                });
                consumers.add(consumer);
            }
        }

    }

    public List getEnhances() {
        return enhances;
    }

    public RocketMqConsumerHolder setEnhances(List enhances) {
        this.enhances = enhances;
        return this;
    }

    public List getConsumerHandlers() {
        return consumerHandlers;
    }

    public RocketMqConsumerHolder setConsumerHandlers(List consumerHandlers) {
        this.consumerHandlers = consumerHandlers;
        return this;
    }

    @Override
    public void onApplicationEvent(ContextRefreshedEvent event) {
        log.info("RocketMqConsumerHolder onApplicationEvent : {}", event.getApplicationContext().getId());
        long startTime = System.currentTimeMillis();
        if (CollectionUtils.isNotEmpty(consumers)) {
            log.info("aliyun ons consumer 正在装载！");
            consumers.forEach(Consumer::start);
            log.info("aliyun ons consumer 启动耗时[{}]", System.currentTimeMillis() - startTime);
        }

    }
}