智慧保安后台管理-外网项目备份
zhongrj
2023-09-17 8853292babb2ad94de4a3207966f1e83b767cd2d
src/main/java/org/springblade/modules/system/service/impl/DictBizServiceImpl.java
@@ -32,6 +32,7 @@
import org.springblade.core.tool.utils.StringPool;
import org.springblade.modules.system.entity.DictBiz;
import org.springblade.modules.system.mapper.DictBizMapper;
import org.springblade.modules.system.node.TreeNode;
import org.springblade.modules.system.service.IDictBizService;
import org.springblade.modules.system.vo.DictBizVO;
import org.springblade.modules.system.wrapper.DictBizWrapper;
@@ -116,4 +117,22 @@
      List<DictBiz> list = this.list(Condition.getQueryWrapper(dict, DictBiz.class).lambda().ne(DictBiz::getId, parentId).eq(DictBiz::getCode, parentDict.getCode()).orderByAsc(DictBiz::getSort));
      return DictBizWrapper.build().listNodeVO(list);
   }
   /**
    * 获取字典树
    */
   @Override
   public List<TreeNode> getTree() {
//      List<TreeNode> dicTreeList = baseMapper.getDicTreeList();
      Map<Long,TreeNode> map = baseMapper.getDicTreeMap();
      List<TreeNode> tree = new ArrayList<>();
      map.forEach((id,treeNode) ->{
         if (map.containsKey(treeNode.getParentId())){
            map.get(treeNode.getParentId()).getChildren().add(treeNode);
         }else {
            tree.add(treeNode);
         }
      });
      return tree;
   }
}