ecel

Hentua

2023-06-10 72e7d01b5b249ae1a079d19cdc62ecd2782c325c

 src/test/java/cc/mrbird/febs/TreeTest.java

@@ -2,9 +2,20 @@

import cc.mrbird.febs.common.tree.MatrixTree;
import cc.mrbird.febs.common.tree.MemberNode;
import cc.mrbird.febs.mall.entity.MallMember;
import cc.mrbird.febs.mall.entity.MallOrderInfo;
import cc.mrbird.febs.mall.mapper.MallMemberMapper;
import cc.mrbird.febs.mall.mapper.MallOrderInfoMapper;
import cc.mrbird.febs.mall.service.IAgentService;
import cc.mrbird.febs.mall.service.IMatrixTreeNodeService;
import cn.hutool.core.collection.CollUtil;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import lombok.extern.slf4j.Slf4j;
import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;

import java.util.List;

/**
 * @author wzy
@@ -21,10 +32,39 @@
            MemberNode node = new MemberNode();
            node.setMemberId((long) i);
            node.setInviteId(String.valueOf(i));
            node.setAddress(String.valueOf(i));
            node.setPhone(String.valueOf(i));
            instance.addNode(node);
        }

        System.out.println(111);
    }

    @Autowired
    private MallMemberMapper mallMemberMapper;

    @Autowired
    private IMatrixTreeNodeService matrixTreeNodeService;

    @Autowired
    private MallOrderInfoMapper mallOrderInfoMapper;

    @Autowired
    private IAgentService agentService;

    @Test
    public void initMemberTreeTest() {
        LambdaQueryWrapper<MallOrderInfo> orderQuery = new LambdaQueryWrapper<>();
        orderQuery.eq(MallOrderInfo::getStatus, 2)
                .eq(MallOrderInfo::getOrderType, 1)
                .orderByAsc(MallOrderInfo::getId);
        List<MallOrderInfo> mallOrderInfos = mallOrderInfoMapper.selectList(orderQuery);

        for (MallOrderInfo mallOrderInfo : mallOrderInfos) {
            agentService.autoUpAgentLevel(mallOrderInfo.getMemberId());
        }

        MatrixTree instance = MatrixTree.getInstance();
        System.out.println(111);
    }

}