代码初始化

k94314517

2024-06-06 5519e6a190a714923eabc3156539a48b572f3fa5

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
package com.doumee.service.system.impl;
 
import com.doumee.dao.system.SystemMenuMapper;
import com.doumee.dao.system.model.SystemMenu;
import com.doumee.dao.system.vo.SystemMenuListVO;
import com.doumee.service.system.SystemMenuService;
import com.baomidou.mybatisplus.core.conditions.Wrapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.util.CollectionUtils;
 
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.stream.Collectors;
 
/**
 * 系统菜单Service实现
 * @author Eva.Caesar Liu
 * @date 2023/03/21 14:49
 */
@Service
public class SystemMenuServiceImpl implements SystemMenuService {
 
    @Autowired
    private SystemMenuMapper systemMenuMapper;
 
    @Override
    public Integer create(SystemMenu systemMenu) {
        systemMenuMapper.insert(systemMenu);
        return systemMenu.getId();
    }
 
    @Override
    public void deleteById(Integer id) {
        SystemMenu systemMenu = new SystemMenu();
        systemMenu.setId(id);
        systemMenu.setDeleted(Boolean.TRUE);
        this.updateById(systemMenu);
    }
 
    @Override
    @Transactional
    public void deleteByIdInBatch(List<Integer> ids) {
        if (CollectionUtils.isEmpty(ids)) return;
        for (Integer id : ids) {
            this.deleteById(id);
        }
    }
 
    @Override
    public void updateById(SystemMenu systemMenu) {
        systemMenuMapper.updateById(systemMenu);
    }
 
    @Override
    @Transactional
    public void updateByIdInBatch(List<SystemMenu> systemMenus) {
        if (CollectionUtils.isEmpty(systemMenus)) return;
        for (SystemMenu systemMenu: systemMenus) {
            this.updateById(systemMenu);
        }
    }
 
    @Override
    public SystemMenu findById(Integer id) {
        return systemMenuMapper.selectById(id);
    }
 
    @Override
    public SystemMenu findOne(SystemMenu systemMenu) {
        Wrapper<SystemMenu> wrapper = new QueryWrapper<>(systemMenu);
        return systemMenuMapper.selectOne(wrapper);
    }
 
    @Override
    public List<SystemMenu> findList(SystemMenu systemMenu) {
        QueryWrapper<SystemMenu> wrapper = new QueryWrapper<>(systemMenu);
        wrapper.lambda().orderByAsc(SystemMenu::getSort);
        return systemMenuMapper.selectList(wrapper);
    }
 
    @Override
    public List<SystemMenuListVO> findList() {
        return systemMenuMapper.selectManageList();
    }
 
    @Override
    public List<SystemMenu> findRootList() {
        QueryWrapper<SystemMenu> queryWrapper = new QueryWrapper<>();
        queryWrapper.lambda()
                .isNull(SystemMenu::getParentId)
                .eq(SystemMenu::getDeleted, Boolean.FALSE)
                .orderByAsc(SystemMenu::getSort);
        return systemMenuMapper.selectList(queryWrapper);
    }
 
    @Override
    public List<SystemMenu> findByUserId(Integer userId) {
        return systemMenuMapper.selectByUserId(userId);
    }
 
    @Override
    public List<SystemMenu> findByRoleId(Integer roleId) {
        return systemMenuMapper.selectByRoleId(roleId);
    }
 
    @Override
    public long count(SystemMenu systemMenu) {
        Wrapper<SystemMenu> wrapper = new QueryWrapper<>(systemMenu);
        return systemMenuMapper.selectCount(wrapper);
    }
 
    @Override
    public List<Integer> findChildren(Integer menuId) {
        List<Integer> pool = new ArrayList<>();
        this.fillChildren(pool, Arrays.asList(menuId));
        return pool;
    }
 
    /**
     * 获取子部门ID
     * @author Eva.Caesar Liu
     * @date 2023/03/21 14:49
     */
    private void fillChildren(List<Integer> pool, List<Integer> parentIds) {
        QueryWrapper<SystemMenu> queryWrapper = new QueryWrapper<>();
        queryWrapper.lambda()
                .eq(SystemMenu::getDeleted, Boolean.FALSE)
                .in(SystemMenu::getParentId, parentIds);
        List<SystemMenu> departments = systemMenuMapper.selectList(queryWrapper);
        List<Integer> ids = departments.stream().map(SystemMenu::getId).collect(Collectors.toList());
        if (ids.size() > 0) {
            pool.addAll(ids);
            this.fillChildren(pool, ids);
        }
    }
}