package cc.mrbird.febs;
|
|
import cc.mrbird.febs.dapp.entity.DappAchieveMemberTreeEntity;
|
import cc.mrbird.febs.dapp.entity.DappFundFlowEntity;
|
import cc.mrbird.febs.dapp.entity.DappMemberEntity;
|
import cc.mrbird.febs.dapp.mapper.DappAchieveMemberTreeDao;
|
import cc.mrbird.febs.dapp.mapper.DappFundFlowDao;
|
import cc.mrbird.febs.dapp.mapper.DappMemberDao;
|
import cc.mrbird.febs.dapp.service.DappSystemService;
|
import cc.mrbird.febs.dapp.vo.TeamListVo;
|
import cc.mrbird.febs.tree.MatrixTree;
|
import cc.mrbird.febs.tree.MemberNode;
|
import cn.hutool.core.collection.CollUtil;
|
import cn.hutool.core.util.ObjectUtil;
|
import cn.hutool.core.util.RandomUtil;
|
import cn.hutool.json.JSONUtil;
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
import org.junit.jupiter.api.Test;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.boot.test.context.SpringBootTest;
|
|
import java.math.BigDecimal;
|
import java.util.ArrayList;
|
import java.util.List;
|
|
/**
|
* @author wzy
|
* @date 2022-08-24
|
**/
|
@SpringBootTest
|
public class TreeTest {
|
|
@Test
|
public void matrixTest() {
|
MatrixTree matrixTree = MatrixTree.getInstance();
|
|
|
String refererId = null;
|
for (int i = 0; i < 11; i++) {
|
String inviteId = RandomUtil.randomString(6);
|
if (i == 2) {
|
refererId = inviteId;
|
}
|
MemberNode memberNode = new MemberNode();
|
memberNode.setMemberId(Long.parseLong(i + 1 + ""));
|
memberNode.setInviteId(inviteId);
|
memberNode.setAddress(RandomUtil.randomString(14));
|
memberNode.setRefererId(refererId);
|
matrixTree.addNode(memberNode);
|
}
|
System.out.println(System.currentTimeMillis());
|
}
|
|
@Autowired
|
private DappSystemService dappSystemService;
|
|
@Autowired
|
private DappAchieveMemberTreeDao dappAchieveMemberTreeDao;
|
|
@Autowired
|
private DappMemberDao dappMemberDao;
|
|
@Autowired
|
private DappFundFlowDao dappFundFlowDao;
|
|
@Test
|
public void resetMatrixV3() {
|
Long memberId = 136L;
|
QueryWrapper<DappFundFlowEntity> objectQueryWrappers = new QueryWrapper<>();
|
objectQueryWrappers.eq("member_id", memberId);
|
objectQueryWrappers.orderByDesc("create_time");
|
List<DappFundFlowEntity> dappFundFlowEntitiess = dappFundFlowDao.selectList(objectQueryWrappers);
|
if(ObjectUtil.isNotEmpty(dappFundFlowEntitiess)){
|
DappAchieveMemberTreeEntity topNode = dappAchieveMemberTreeDao.selectNodeByDeep(dappFundFlowEntitiess.get(0).getId(), 1);
|
if (ObjectUtil.isNotEmpty(topNode)) {
|
TeamListVo teamListVo = buildTeamMatrix(topNode);
|
System.out.println(JSONUtil.parseObj(teamListVo));
|
}
|
}
|
}
|
|
@Test
|
public void resetMatrixV2() {
|
dappSystemService.achieveTreeV2(136L);
|
}
|
|
@Test
|
public void resetMatrix() {
|
Long withoutId = 1L;
|
QueryWrapper<DappMemberEntity> objectQueryWrapper = new QueryWrapper<>();
|
objectQueryWrapper.orderByAsc("create_time");
|
List<DappMemberEntity> dappMemberEntities = dappMemberDao.selectList(objectQueryWrapper);
|
for(DappMemberEntity member : dappMemberEntities){
|
System.out.println(member.getId());
|
if(withoutId != member.getId()){
|
DappFundFlowEntity fundFlow = new DappFundFlowEntity(member.getId(), new BigDecimal("1"), 1, 2,
|
BigDecimal.ZERO, "测试");
|
dappFundFlowDao.insert(fundFlow);
|
dappSystemService.achieveTreeV2(fundFlow.getId());
|
}
|
}
|
}
|
|
public TeamListVo buildTeamMatrix(DappAchieveMemberTreeEntity node) {
|
List<DappAchieveMemberTreeEntity> childNodes = dappAchieveMemberTreeDao.selectMatrixChildNode(node.getTopNode(), node.getTreeNode());
|
|
TeamListVo teamListVo = new TeamListVo();
|
teamListVo.setName(node.getInviteId());
|
teamListVo.setHasMoney(node.getHasMoney());
|
|
if (CollUtil.isEmpty(childNodes)) {
|
return teamListVo;
|
}
|
|
List<TeamListVo> list = new ArrayList<>();
|
for (DappAchieveMemberTreeEntity childNode : childNodes) {
|
list.add(buildTeamMatrix(childNode));
|
}
|
teamListVo.setChildren(list);
|
return teamListVo;
|
}
|
|
@Test
|
public void limitResetProfitTest() {
|
dappSystemService.putIntoProfit(59L, 1);
|
}
|
}
|
