Administrator
2023-06-19 49588f5a462ae7425e7eb030438a35fd80c246fa
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
package com.gk.firework.Service.ServiceImpl;
 
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.gk.firework.Domain.PermissionInfo;
import com.gk.firework.Domain.UserInfo;
import com.gk.firework.Domain.Vo.Menu;
import com.gk.firework.Domain.Vo.SortData;
import com.gk.firework.Mapper.PermissionInfoMapper;
import com.gk.firework.Service.PermissionService;
import com.gk.firework.Service.UserService;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
 
import java.util.ArrayList;
import java.util.Comparator;
import java.util.List;
import java.util.stream.Collectors;
 
@Service("PermissionService")
public class PermissionServiceImpl extends ServiceImpl<PermissionInfoMapper, PermissionInfo> implements PermissionService {
    @Autowired
    private PermissionInfoMapper permissionInfoMapper;
    @Autowired
    private UserService userService;
 
    @Override
    public List<Menu> selectByUserId(String userId) {
        List<Menu> menus = new ArrayList<>();
 
        UserInfo userInfo = userService.getById(Integer.parseInt(userId));
        if (null != userInfo && null != userInfo.getType()){
            List<PermissionInfo> permissionInfoList = permissionInfoMapper.selectAll();
            if (userInfo.getType() == 1){
                //返回所有菜单
                for (PermissionInfo permissionInfo : permissionInfoList){
                    //一级菜单
                    if (null == permissionInfo.getParentid()){
                        Menu pmenu = new Menu();
                        BeanUtils.copyProperties(permissionInfo,pmenu,Menu.class);
                        List<Menu> iterateMenuList = iterateMenus(permissionInfoList,permissionInfo.getId());
                        pmenu.setChildren(iterateMenuList);
                        menus.add(pmenu);
                    }
                }
            }else if (userInfo.getType() == 2){
                //排除系统管理
                for (PermissionInfo permissionInfo : permissionInfoList){
                    //一级菜单
                    if (null == permissionInfo.getParentid() && !permissionInfo.getTitle().equals("权限管理")){
                        Menu pmenu = new Menu();
                        BeanUtils.copyProperties(permissionInfo,pmenu,Menu.class);
                        List<Menu> iterateMenuList = iterateMenus(permissionInfoList,permissionInfo.getId());
                        pmenu.setChildren(iterateMenuList);
                        menus.add(pmenu);
                    }
                }
            }else {
                permissionInfoList = permissionInfoMapper.selectByUserId(userId);
                menus = addMenuList(permissionInfoList);
//                for (PermissionInfo permissionInfo : permissionInfoList){
//                    Menu pmenu = new Menu();
//                    BeanUtils.copyProperties(permissionInfo,pmenu,Menu.class);
//                    List<Menu> iterateMenuList = iterateMenus(permissionInfoList,permissionInfo.getId());
//                    if (iterateMenuList.size() > 0) {
//                        pmenu.setChildren(iterateMenuList);
//                        menus.add(pmenu);
//                    }
//                }
            }
        }
        return menus;
    }
 
    private List<Menu> iterateMenus(List<PermissionInfo> permissionInfoList, Integer pid) {
        List<Menu> result = new ArrayList<Menu>();
        for (PermissionInfo resource : permissionInfoList) {
            Integer menuid = resource.getId();//获取菜单的id
            Integer parentid = resource.getParentid();//获取菜单的父id
            if(null != parentid && (parentid.equals(pid))){
                Menu pmenu = new Menu();
                BeanUtils.copyProperties(resource,pmenu,Menu.class);
                List<Menu> iterateMenu = iterateMenus(permissionInfoList,menuid);
                pmenu.setChildren(iterateMenu);
                result.add(pmenu);
            }
        }
        return result;
    }
 
    private List<Menu> addMenuList(List<PermissionInfo> permissionInfoList) {
        List<Menu> menuList = new ArrayList<Menu>();
        //一级菜单
        for (PermissionInfo permissionInfo : permissionInfoList) {
            if (permissionInfo.getType() == 1 && permissionInfo.getLevel() == 1 && permissionInfo.getStatus() == 1){
                Menu menu = new Menu();
                BeanUtils.copyProperties(permissionInfo,menu,Menu.class);
                List<Menu> childList = new ArrayList<Menu>();
                menu.setChildren(childList);
                menuList.add(menu);
            }
        }
        //二级菜单
        for (PermissionInfo permissionInfo : permissionInfoList) {
            if (permissionInfo.getType() == 1 && permissionInfo.getLevel() == 2 && permissionInfo.getStatus() == 1){
                boolean isfound = false;
                for (Menu menu : menuList){
                    if (menu.getId().equals(permissionInfo.getParentid())){
                        Menu pmenu = new Menu();
                        BeanUtils.copyProperties(permissionInfo,pmenu,Menu.class);
                        List<Menu> secList = iterateMenus(permissionInfoList,permissionInfo.getId());
                        pmenu.setChildren(secList);
                        menu.getChildren().add(pmenu);
                        isfound = true;
                        break;
                    }
                }
                if (!isfound){
                    PermissionInfo onePermission = this.getById(permissionInfo.getParentid());
                    if (onePermission != null) {
                        Menu childmenu = new Menu();
                        BeanUtils.copyProperties(permissionInfo,childmenu,Menu.class);
                        List<Menu> secList = new ArrayList<Menu>();
                        childmenu.setChildren(secList);
 
                        Menu onemenu = new Menu();
                        BeanUtils.copyProperties(onePermission,onemenu,Menu.class);
                        List<Menu> childList = new ArrayList<Menu>();
                        childList.add(childmenu);
                        onemenu.setChildren(childList);
                        menuList.add(onemenu);
                    }
                }
            }
        }
 
 
        return menuList.stream().sorted(Comparator.comparing(Menu::getSortorder)).collect(Collectors.toList());
    }
 
 
}