package cc.mrbird.febs.job;

import cc.mrbird.febs.dapp.entity.DappAchieveTreeEntity;
import cc.mrbird.febs.dapp.mapper.DappAchieveTreeDao;
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.stereotype.Component;

import javax.annotation.PostConstruct;
import java.util.List;

/**
 * @author wzy
 * @date 2022-08-26
 **/
@Slf4j
@Component
public class MatrixTreeInit {

    @Autowired
    private DappAchieveTreeDao dappAchieveTreeDao;

    @PostConstruct
    public void init() {
        List<DappAchieveTreeEntity> tree = dappAchieveTreeDao.selectTreeList();

        MatrixTree instance = MatrixTree.getInstance();
        for (DappAchieveTreeEntity treeNode : tree) {

            MemberNode node = new MemberNode();
            node.setAddress(treeNode.getAddress());
            node.setInviteId(treeNode.getInviteId());
            node.setRefererId(treeNode.getRefererId());
            node.setMemberId(treeNode.getMidNode());
            instance.addNode(node);
        }
    }
}