package cc.mrbird.febs; 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 * @date 2023-04-18 **/ @Slf4j @SpringBootTest public class TreeTest { @Test public void treeTest() { MatrixTree instance = MatrixTree.getInstance(); for (int i = 0; i < 20; i++) { MemberNode node = new MemberNode(); node.setMemberId((long) i); node.setInviteId(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 orderQuery = new LambdaQueryWrapper<>(); orderQuery.eq(MallOrderInfo::getStatus, 2) .eq(MallOrderInfo::getOrderType, 1) .orderByAsc(MallOrderInfo::getId); List mallOrderInfos = mallOrderInfoMapper.selectList(orderQuery); for (MallOrderInfo mallOrderInfo : mallOrderInfos) { agentService.autoUpAgentLevel(mallOrderInfo.getMemberId()); } MatrixTree instance = MatrixTree.getInstance(); System.out.println(111); } }