MrShi
2025-01-12 426718fb2310abff70f54962f118f4300ead2408
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.baomidou.mybatisplus.core.conditions.Wrapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.doumee.core.model.LoginUserInfo;
import com.doumee.core.model.PageData;
import com.doumee.core.model.PageWrap;
import com.doumee.core.utils.Constants;
import com.doumee.core.utils.Utils;
import com.doumee.dao.system.SystemDataPermissionMapper;
import com.doumee.dao.system.SystemRoleMapper;
import com.doumee.dao.system.dto.QuerySystemRoleDTO;
import com.doumee.dao.system.model.SystemDataPermission;
import com.doumee.dao.system.model.SystemRole;
import com.doumee.dao.system.vo.SystemRoleListVO;
import com.doumee.service.system.SystemMenuService;
import com.doumee.service.system.SystemPermissionService;
import com.doumee.service.system.SystemRoleService;
import com.github.pagehelper.PageHelper;
import com.github.pagehelper.PageInfo;
import org.apache.shiro.SecurityUtils;
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.List;
import java.util.Objects;
 
/**
 * 系统角色Service实现
 * @author Eva.Caesar Liu
 * @date 2023/03/21 14:49
 */
@Service
public class SystemRoleServiceImpl implements SystemRoleService {
 
    @Autowired
    private SystemRoleMapper systemRoleMapper;
 
    @Autowired
    private SystemMenuService systemMenuService;
 
    @Autowired
    private SystemPermissionService systemPermissionService;
 
    @Autowired
    private SystemDataPermissionMapper systemDataPermissionMapper;
 
    @Override
    public Integer create(SystemRole systemRole) {
        LoginUserInfo principal = (LoginUserInfo) SecurityUtils.getSubject().getPrincipal();
        systemRole.setCompanyId(principal.getCompanyId());
        systemRole.setType(principal.getType());
        systemRoleMapper.insert(systemRole);
        return systemRole.getId();
    }
 
    @Override
    public void deleteById(Integer id) {
        SystemRole systemRole = new SystemRole();
        systemRole.setId(id);
        systemRole.setDeleted(Boolean.TRUE);
        this.updateById(systemRole);
    }
 
    @Override
    @Transactional
    public void deleteByIdInBatch(List<Integer> ids) {
        if (CollectionUtils.isEmpty(ids)) return;
        for (Integer id : ids) {
            this.deleteById(id);
        }
    }
 
    @Override
    public void updateById(SystemRole systemRole) {
        systemRoleMapper.updateById(systemRole);
    }
 
    @Override
    @Transactional
    public void updateByIdInBatch(List<SystemRole> systemRoles) {
        if (CollectionUtils.isEmpty(systemRoles)) return;
        for (SystemRole systemRole: systemRoles) {
            this.updateById(systemRole);
        }
    }
 
    @Override
    public SystemRole findById(Integer id) {
        return systemRoleMapper.selectById(id);
    }
 
    @Override
    public List<SystemRole> findByUserId(Integer userId) {
        return systemRoleMapper.selectByUserId(userId);
    }
 
    @Override
    public SystemRole findOne(SystemRole systemRole) {
        Wrapper<SystemRole> wrapper = new QueryWrapper<>(systemRole);
        return systemRoleMapper.selectOne(wrapper);
    }
 
    @Override
    public List<SystemRole> findList(SystemRole systemRole) {
        LoginUserInfo user = (LoginUserInfo) SecurityUtils.getSubject().getPrincipal();
        if(Objects.nonNull(user)&&(user.getType().equals(Constants.ONE)||user.getType().equals(Constants.TWO))){
            systemRole.setCompanyId(user.getCompanyId());
        }
        Wrapper<SystemRole> wrapper = new QueryWrapper<>(systemRole);
        return systemRoleMapper.selectList(wrapper);
    }
 
    @Override
    public PageData<SystemRoleListVO> findPage(PageWrap<QuerySystemRoleDTO> pageWrap) {
        LoginUserInfo principal = (LoginUserInfo) SecurityUtils.getSubject().getPrincipal();
        Utils.MP.blankToNull(pageWrap.getModel());
        QuerySystemRoleDTO model = pageWrap.getModel();
        model.setType(Constants.formatIntegerNum(principal.getType()));
        model.setCompanyId(principal.getCompanyId());
        PageHelper.startPage(pageWrap.getPage(), pageWrap.getCapacity());
        List<SystemRoleListVO> roleList = systemRoleMapper.selectManageList(model, pageWrap.getOrderByClause());
        for (SystemRoleListVO role : roleList) {
            role.setMenus(systemMenuService.findByRoleId(role.getId()));
            role.setPermissions(systemPermissionService.findByRoleId(role.getId()));
            role.setSystemDataPermission(systemDataPermissionMapper.selectOne(new QueryWrapper<SystemDataPermission>().lambda()
                    .eq(SystemDataPermission::getDeleted,Constants.ZERO).eq(SystemDataPermission::getRoleId,role.getId()).last("limit 1")
            ));
        }
        return PageData.from(new PageInfo<>(roleList));
    }
 
    @Override
    public long count(SystemRole systemRole) {
        Wrapper<SystemRole> wrapper = new QueryWrapper<>(systemRole);
        return systemRoleMapper.selectCount(wrapper);
    }
}