com.aliyun.openservices.ons.api.exactlyonce.ExactlyOnceCheckSendbackHook Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of ons-client-ext Show documentation

There is a newer version: 1.9.3.Final

package com.aliyun.openservices.ons.api.exactlyonce;

import com.aliyun.openservices.shade.com.alibaba.rocketmq.client.consumer.listener.ConsumeConcurrentlyContext;
import com.aliyun.openservices.shade.com.alibaba.rocketmq.client.hook.CheckSendBackHook;
import com.aliyun.openservices.shade.com.alibaba.rocketmq.common.message.MessageExt;

import com.aliyun.openservices.ons.api.exactlyonce.aop.model.MQTxContext;
import com.aliyun.openservices.ons.api.exactlyonce.datasource.DataSourceConfig;
import com.aliyun.openservices.ons.api.exactlyonce.manager.util.DBAccessUtil;
import com.aliyun.openservices.ons.api.exactlyonce.manager.util.TxContextUtil;

/**
 * @author gongshi
 */
public class ExactlyOnceCheckSendbackHook implements CheckSendBackHook {
    private String consumerGroup;
    private DataSourceConfig config;

    public ExactlyOnceCheckSendbackHook(String consumerGroup, DataSourceConfig config) {
        this.consumerGroup = consumerGroup;
        this.config = config;
    }

    public DataSourceConfig getConfig() {
        return config;
    }

    public void setConfig(DataSourceConfig config) {
        this.config = config;
    }

    @Override
    public String hookName() {
        return "ExactlyOnceCheckSendbackHook";
    }

    @Override
    public boolean needSendBack(MessageExt msg, ConsumeConcurrentlyContext context) {
        if (consumerGroup == null || config == null) {
            return true;
        }
        MQTxContext txContext = new MQTxContext();
        txContext.setMessageId(TxContextUtil.buildInternalMsgId(msg, consumerGroup));
        txContext.setDataSourceConfig(config);
        return !DBAccessUtil.isRecordExist(txContext);
    }
}