1

zj

yesterday 5e57de9b12ee136e45ce5754c7fe2e7eb12af05a

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
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
/*
 * Copyright (c) 2018-2999 广州市蓝海创新科技有限公司 All rights reserved.
 *
 * https://www.mall4j.com/
 *
 * 未经允许，不可做商业用途！
 *
 * 版权所有，侵权必究！
 */
 
package com.yami.trading.sys.service.impl;
 
 
import cn.hutool.core.collection.CollectionUtil;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.yami.trading.sys.constant.Constant;
import com.yami.trading.sys.dao.SysMenuMapper;
import com.yami.trading.sys.dao.SysRoleMenuMapper;
import com.yami.trading.sys.model.SysMenu;
import com.yami.trading.sys.service.SysMenuService;
import lombok.AllArgsConstructor;
import org.springframework.stereotype.Service;
 
import java.util.*;
import java.util.stream.Collectors;
 
/**
 * @author lgh
 */
@Service("sysMenuService")
@AllArgsConstructor
public class SysMenuServiceImpl extends ServiceImpl<SysMenuMapper, SysMenu> implements SysMenuService {
 
    private final SysRoleMenuMapper sysRoleMenuMapper;
 
    private final SysMenuMapper sysMenuMapper;
 
    @Override
    public List<SysMenu> listMenuByUserId(Long userId,String appType) {
        List<SysMenu> sysMenus ;
        if(userId == Constant.SUPER_ADMIN_ID||userId == Constant.SUPER_ROOT_ID){
            sysMenus = sysMenuMapper.listMenu(appType);
        }else {
            sysMenus = sysMenuMapper.listMenuByUserId(userId,appType);
 
            Map<Long,SysMenu> mapSysMenu = new HashMap<>();
            for (SysMenu sysMenu : sysMenus) {
                mapSysMenu.put(sysMenu.getMenuId(), sysMenu);
            }
 
            List<Long> bigIntArrays = new Vector<>();
 
            List<SysMenu> sysMenus3 = sysMenuMapper.queryAllPerms2(userId);
            for (SysMenu sysMenu : sysMenus3) {
                if (!mapSysMenu.containsKey(sysMenu.getParentId())
                        && !bigIntArrays.contains(sysMenu.getParentId())
                        && sysMenu.getParentId() != 0) {
                    bigIntArrays.add(sysMenu.getParentId());
                }
            }
 
            if(!bigIntArrays.isEmpty()){
                List<SysMenu> sysMenus2 = sysMenuMapper.listMenuByUserId2(bigIntArrays,appType);
                for (SysMenu sysMenu : sysMenus2) {
                    mapSysMenu.put(sysMenu.getMenuId(), sysMenu);
                }
                sysMenus.addAll(sysMenus2);
            }
 
            appendMissingMenuAncestors(sysMenus, mapSysMenu, appType);
        }
 
        Map<Long, List<SysMenu>> sysMenuLevelMap = sysMenus.stream()
                .sorted(Comparator.comparing(SysMenu::getOrderNum))
                .collect(Collectors.groupingBy(SysMenu::getParentId));
 
        List<SysMenu> rootMenu = sysMenuLevelMap.get(0L);
        if (CollectionUtil.isEmpty(rootMenu)) {
            return Collections.emptyList();
        }
 
        for (SysMenu sysMenu : rootMenu) {
            attachMenuChildren(sysMenu, sysMenuLevelMap);
        }
 
        return rootMenu;
    }
 
    private void appendMissingMenuAncestors(List<SysMenu> sysMenus, Map<Long, SysMenu> mapSysMenu, String appType) {
        while (true) {
            List<Long> missingParents = new ArrayList<>();
            for (SysMenu sysMenu : sysMenus) {
                Long parentId = sysMenu.getParentId();
                if (parentId != null && parentId != 0
                        && !mapSysMenu.containsKey(parentId)
                        && !missingParents.contains(parentId)) {
                    missingParents.add(parentId);
                }
            }
            if (missingParents.isEmpty()) {
                return;
            }
            List<SysMenu> parents = sysMenuMapper.listMenuByUserId2(missingParents, appType);
            if (CollectionUtil.isEmpty(parents)) {
                return;
            }
            for (SysMenu parent : parents) {
                mapSysMenu.put(parent.getMenuId(), parent);
            }
            sysMenus.addAll(parents);
        }
    }
 
    private void attachMenuChildren(SysMenu menu, Map<Long, List<SysMenu>> sysMenuLevelMap) {
        List<SysMenu> children = sysMenuLevelMap.get(menu.getMenuId());
        menu.setList(children);
        if (CollectionUtil.isEmpty(children)) {
            return;
        }
        for (SysMenu child : children) {
            attachMenuChildren(child, sysMenuLevelMap);
        }
    }
 
    @Override
    public void deleteMenuAndRoleMenu(Long menuId){
        this.removeById(menuId);
        sysRoleMenuMapper.deleteByMenuId(menuId);
    }
 
 
    @Override
    public List<Long> listMenuIdByRoleId(Long roleId,String appType) {
        return sysMenuMapper.listMenuIdByRoleId(roleId,appType);
    }
 
 
    @Override
    public List<SysMenu> listSimpleMenuNoButton(String appType) {
        return sysMenuMapper.listSimpleMenuNoButton(appType);
    }
 
    @Override
    public List<SysMenu> listRootMenu() {
        return sysMenuMapper.listRootMenu();
    }
 
    @Override
    public List<SysMenu> listChildrenMenuByParentId(Long parentId) {
        return sysMenuMapper.listChildrenMenuByParentId(parentId);
    }
 
    @Override
    public List<SysMenu> listMenuAndBtn( String appType) {
        return sysMenuMapper.listMenuAndBtn(appType);
    }
 
}