KKSU
2024-06-24 6b7a98a866dc46bfddb5663b9a3a15ea2972996e
src/main/java/cc/mrbird/febs/rabbit/consumer/ChainConsumer.java
@@ -1,12 +1,19 @@
package cc.mrbird.febs.rabbit.consumer;
import cc.mrbird.febs.dapp.entity.DbMemberNode;
import cc.mrbird.febs.dapp.mapper.DbMemberNodeMapper;
import cc.mrbird.febs.dapp.service.DappSystemService;
import cc.mrbird.febs.rabbit.QueueConstants;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.util.ObjectUtil;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.rabbit.annotation.RabbitListener;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
import org.springframework.stereotype.Component;
import java.util.List;
/**
 * @author wzy
@@ -19,6 +26,8 @@
    @Autowired
    private DappSystemService dappSystemService;
    @Autowired
    private DbMemberNodeMapper dbMemberNodeMapper;
    @RabbitListener(queues = QueueConstants.ACHIEVE_TREE)
    public void achieveTree(String id) {
@@ -42,4 +51,63 @@
    public void feeDistribute(String data) {
        dappSystemService.feeDistribute(data);
    }
    @RabbitListener(queues = QueueConstants.TFC_INVITE_PERK_QUEEN)
    public void invitePerkMsg(Long id) {
        log.info("消费推荐规则奖励,流水ID:{}", id);
        dappSystemService.invitePerkMsg(id);
    }
    @RabbitListener(queues = QueueConstants.TFC_NODE_PERK_QUEEN)
    public void nodePerkMsg(Long id) {
        log.info("消费节点投资,流水ID:{}", id);
        dappSystemService.nodePerkMsg(id);
    }
    @RabbitListener(queues = QueueConstants.TFC_NODE_AGAIN)
    public void nodeMsg(Long id) {
        log.info("复投:{}", id);
        dappSystemService.achieveTreeV2(id);
    }
    @RabbitListener(queues = QueueConstants.TFC_NODE_AGAIN_LONG)
    public void nodeMsgLong(Long id) {
        log.info("复投:{}", id);
        /**
         * 获取所有左右节点都有值,轮数为 1,未复投work_state = 1 的节点
         *      判断右节点有值
         *          有继续向判断右节点
         *              有,原纪录复投逻辑
         */
        QueryWrapper<DbMemberNode> objectQueryWrapper = new QueryWrapper<>();
        objectQueryWrapper.eq("count_fund", 1);
        objectQueryWrapper.eq("work_state", 1);
        objectQueryWrapper.eq("type", 1);
        objectQueryWrapper.eq("member_id", id);
        List<DbMemberNode> dbMemberNodes = dbMemberNodeMapper.selectList(objectQueryWrapper);
        if(CollUtil.isEmpty(dbMemberNodes)){
            return;
        }
        for(DbMemberNode dbMemberNode : dbMemberNodes){
            if(ObjectUtil.isEmpty(dbMemberNode.getRightNode())){
                continue;
            }
            //子节点
            Long rightNode = dbMemberNode.getRightNode();
            DbMemberNode dbMemberNode1 = dbMemberNodeMapper.selectById(rightNode);
            if(ObjectUtil.isEmpty(dbMemberNode1)){
                continue;
            }
            if(ObjectUtil.isEmpty(dbMemberNode1.getRightNode())){
                continue;
            }
            //有,则上级节点出局复投,轮数加1,复投逻辑
            dappSystemService.memberNodeNext(dbMemberNode);
        }
    }
}