Supwisdom Source - institute/sw-backend.git/blob

   1 package com.supwisdom.institute.backend.system.domain.service;
   2
   3 import java.util.ArrayList;
   4 import java.util.HashMap;
   5 import java.util.LinkedHashMap;
   6 import java.util.List;
   7 import java.util.Map;
   8
   9 import lombok.extern.slf4j.Slf4j;
  10
  11 import org.springframework.beans.factory.annotation.Autowired;
  12 import org.springframework.stereotype.Service;
  13
  14 import com.supwisdom.institute.backend.common.framework.entity.EntityUtils;
  15 import com.supwisdom.institute.backend.common.framework.service.ABaseService;
  16 import com.supwisdom.institute.backend.system.domain.entity.Permission;
  17 import com.supwisdom.institute.backend.system.domain.model.PermissionTreeNode;
  18 import com.supwisdom.institute.backend.system.domain.repo.PermissionRepository;
  19
  20 @Slf4j
  21 @Service
  22 public class PermissionService extends ABaseService<Permission, PermissionRepository> {
  23
  24   @Override
  25   public PermissionRepository getRepo() {
  26     return permissionRepository;
  27   }
  28
  29   @Autowired
  30   private PermissionRepository permissionRepository;
  31
  32   public PermissionTreeNode selectPermissionTree(Map<String, Object> mapBean) {
  33
  34     List<Permission> permissions = permissionRepository.selectList(true, -1, -1, mapBean, null);
  35
  36     Map<String, PermissionTreeNode> parentTreeNode = new LinkedHashMap<String, PermissionTreeNode>();
  37
  38     PermissionTreeNode rootTreeNode = new PermissionTreeNode();
  39     rootTreeNode.setId(Permission.ROOT_PARENT_ID);
  40     rootTreeNode.setCode("ROOT");
  41     rootTreeNode.setName("顶级菜单");
  42     rootTreeNode.setChildren(new ArrayList<PermissionTreeNode>());
  43
  44     parentTreeNode.put(rootTreeNode.getId(), rootTreeNode);
  45
  46     for (Permission permission : permissions) {
  47       PermissionTreeNode treeNode = EntityUtils.copy(permission, new PermissionTreeNode());
  48       treeNode.setChildren(new ArrayList<PermissionTreeNode>());
  49
  50       parentTreeNode.put(treeNode.getId(), treeNode);
  51     }
  52
  53     for (PermissionTreeNode treeNode : parentTreeNode.values()) {
  54       if (!parentTreeNode.containsKey(treeNode.getParentId())) {
  55         continue;
  56       }
  57
  58       parentTreeNode.get(treeNode.getParentId()).getChildren().add(treeNode);
  59     }
  60
  61     return rootTreeNode;
  62   }
  63
  64 }