package cc.mrbird.febs.dapp.service.impl;
|
|
import cc.mrbird.febs.common.tree.MatrixTree;
|
import cc.mrbird.febs.common.tree.MemberNode;
|
import cc.mrbird.febs.dapp.entity.DappMemberEntity;
|
import cc.mrbird.febs.dapp.entity.MatrixTreeNode;
|
import cc.mrbird.febs.dapp.mapper.DappMemberDao;
|
import cc.mrbird.febs.dapp.mapper.MatrixTreeNodeMapper;
|
import cc.mrbird.febs.dapp.service.IMatrixTreeNodeService;
|
import cn.hutool.core.util.ObjectUtil;
|
import cn.hutool.core.util.StrUtil;
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
import lombok.RequiredArgsConstructor;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.stereotype.Service;
|
|
@Slf4j
|
@Service
|
@RequiredArgsConstructor
|
public class MatrixTreeNodeServiceImpl extends ServiceImpl<MatrixTreeNodeMapper, MatrixTreeNode> implements IMatrixTreeNodeService {
|
|
private final DappMemberDao mallMemberMapper;
|
private final MatrixTreeNodeMapper matrixTreeNodeMapper;
|
|
@Override
|
public MemberNode addTreeNode(Long memberId) {
|
DappMemberEntity mallMember = mallMemberMapper.selectById(memberId);
|
if (mallMember == null) {
|
return null;
|
}
|
|
MatrixTree instance = MatrixTree.getInstance();
|
MemberNode node = instance.getNode(memberId);
|
//节点存在则直接返回
|
if (node != null) {
|
return null;
|
}
|
//未找到节点
|
if (StrUtil.isNotBlank(mallMember.getRefererId())) {
|
//返回直推上级所在节点
|
MemberNode parentNode = instance.getNode(mallMember.getRefererId());
|
|
MemberNode notWholeNode = null;
|
if (parentNode != null) {
|
Long parentNodeMemberId = parentNode.getMemberId();
|
MatrixTreeNode matrixTreeNode = matrixTreeNodeMapper.selectByParentNodeAndType(parentNodeMemberId,mallMember.getNodeType());
|
if(ObjectUtil.isNotEmpty(matrixTreeNode)){
|
//直推上级的第一代子节点
|
MemberNode nodeParentChild = instance.getNode(matrixTreeNode.getTreeNode());
|
notWholeNode = instance.findNotWholeNode(nodeParentChild);
|
}else{
|
notWholeNode = instance.findNotWholeNode(parentNode);
|
}
|
} else {
|
notWholeNode = instance.findNotWholeNode();
|
}
|
|
MatrixTreeNode treeNode = new MatrixTreeNode();
|
treeNode.setTreeNode(mallMember.getId());
|
treeNode.setParentNode(notWholeNode.getMemberId());
|
treeNode.setType(mallMember.getNodeType());
|
matrixTreeNodeMapper.insert(treeNode);
|
} else {
|
MatrixTreeNode treeNode = new MatrixTreeNode();
|
treeNode.setTreeNode(mallMember.getId());
|
treeNode.setType(0);
|
matrixTreeNodeMapper.insert(treeNode);
|
}
|
MatrixTreeNode matrixTreeNode = matrixTreeNodeMapper.selectByTreeNode(memberId);
|
DappMemberEntity dappMemberEntity = mallMemberMapper.selectById(matrixTreeNode.getParentNode());
|
node = new MemberNode();
|
node.setMemberId(memberId);
|
node.setInviteId(mallMember.getInviteId());
|
node.setRefererId(dappMemberEntity.getInviteId());
|
node.setType(mallMember.getNodeType());
|
return instance.addNode(node);
|
}
|
}
|
