package com.consum.base.support;

import com.consum.model.po.FinSysTenant;
import com.walker.infrastructure.tree.AbstractTreeGenerator;
import com.walker.infrastructure.tree.TreeNode;

import java.util.List;

public class FinSysTenantGenerator extends AbstractTreeGenerator<FinSysTenant> {

    public FinSysTenantGenerator(String dummyRootName) {
        super(dummyRootName);
        this.setMultiRoot(true);
    }

    protected TreeNode toTreeNode(FinSysTenant entity) {
        TreeNode treeNode = new TreeNode(entity.getId(), entity.getName(), (List) null, entity.getParentId(), entity.getCode());
        return treeNode;
    }
}