KKSU
2024-06-12 8986d6d479bf56432ebb8e18bfa0f344d028eee6
src/main/java/cc/mrbird/febs/job/MatrixTreeInit.java
@@ -1,11 +1,14 @@
package cc.mrbird.febs.job;
import cc.mrbird.febs.dapp.entity.DappAchieveTreeEntity;
import cc.mrbird.febs.dapp.entity.DappMemberEntity;
import cc.mrbird.febs.dapp.mapper.DappAchieveTreeDao;
import cc.mrbird.febs.dapp.mapper.DappMemberDao;
import cc.mrbird.febs.tree.MatrixTree;
import cc.mrbird.febs.tree.MemberNode;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
import org.springframework.stereotype.Component;
import javax.annotation.PostConstruct;
@@ -17,25 +20,33 @@
 **/
@Slf4j
@Component
@ConditionalOnProperty(prefix = "system", name = "chain-listener", havingValue = "true")
public class MatrixTreeInit {
    @Autowired
    private DappAchieveTreeDao dappAchieveTreeDao;
    @Autowired
    private DappMemberDao dappMemberDao;
    @PostConstruct
    public void init() {
        List<DappAchieveTreeEntity> tree = dappAchieveTreeDao.selectTreeList();
//        List<DappAchieveTreeEntity> tree = dappAchieveTreeDao.selectTreeList();
        List<DappAchieveTreeEntity> tree = dappAchieveTreeDao.selectTreeListV2();
        MatrixTree instance = MatrixTree.getInstance();
        for (DappAchieveTreeEntity treeNode : tree) {
            Long memberId = treeNode.getMemberId();
            DappMemberEntity dappMemberEntity = dappMemberDao.selectById(memberId);
            MemberNode node = new MemberNode();
            node.setAddress(treeNode.getAddress());
            node.setInviteId(treeNode.getInviteId());
            node.setRefererId(treeNode.getRefererId());
            node.setMemberId(treeNode.getMidNode());
            node.setAddress(dappMemberEntity.getAddress());
            node.setInviteId(dappMemberEntity.getInviteId());
            node.setRefererId(dappMemberEntity.getRefererId());
            node.setMemberId(treeNode.getMemberId());
            node.setFundId(treeNode.getFundId());
            instance.addNode(node);
        }
        System.out.println(11);
    }
}