增加权限管理

admin

2022-05-19 56364722d3ed70d48ec41f567a4e59e5ccbbb868

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
package com.ks.app.service.impl.admin;
 
import com.ks.app.dao.admin.AdminUserRolesDao;
import com.ks.app.entity.admin.AdminRole;
import com.ks.app.entity.admin.AdminUserRoles;
import com.ks.app.service.inter.admin.AdminUserRolesService;
import com.ks.app.service.query.admin.AdminUserRolesQuery;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.stereotype.Service;
import org.yeshi.utils.bean.BeanUtil;
 
import javax.annotation.Resource;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.stream.Collectors;
 
@Service
public class AdminUserRolesServiceImpl implements AdminUserRolesService {
 
    @Resource
    private AdminUserRolesDao adminUserRolesDao;
 
    @Override
    public List<AdminUserRoles> list(AdminUserRolesQuery adminUserRolesQuery, int page, int pageSize) {
        AdminUserRolesDao.DaoQuery daoQuery = new AdminUserRolesDao.DaoQuery();
        try {
            BeanUtil.copyProperties(adminUserRolesQuery, daoQuery);
        } catch (IllegalAccessException e) {
            e.printStackTrace();
        }
        daoQuery.start = (page - 1) * pageSize;
        daoQuery.count = pageSize;
        return adminUserRolesDao.list(daoQuery);
    }
 
    @Override
    public long count(AdminUserRolesQuery adminUserRolesQuery) {
        AdminUserRolesDao.DaoQuery daoQuery = new AdminUserRolesDao.DaoQuery();
        try {
            BeanUtil.copyProperties(adminUserRolesQuery, daoQuery);
        } catch (IllegalAccessException e) {
            e.printStackTrace();
        }
        return adminUserRolesDao.count(daoQuery);
    }
 
    @Override
    public AdminUserRoles get(String id) {
        Query query = new Query();
        query.addCriteria(Criteria.where("_id").is(id));
        return adminUserRolesDao.findOne(query);
    }
 
    @Override
    public void add(AdminUserRoles adminUserRoles) throws Exception {
        //查询主键ID是否存在
        if (adminUserRolesDao.get(adminUserRoles.getId()) != null) {
            throw new Exception("已存在");
        }
 
        if (adminUserRoles.getCreateTime() == null) {
            adminUserRoles.setCreateTime(new Date());
        }
        //保存
        adminUserRolesDao.save(adminUserRoles);
    }
 
    @Override
    public void update(AdminUserRoles adminUserRoles) {
        if (adminUserRoles.getUpdateTime() == null) {
            adminUserRoles.setUpdateTime(new Date());
        }
        //更新
        adminUserRolesDao.updateSelective(adminUserRoles);
    }
 
    @Override
    public void delete(List<String> idList) {
        for (String id : idList) {
            adminUserRolesDao.delete(id);
        }
    }
 
    @Override
    public List<AdminRole> listRoleByAdminId(String adminId) {
        AdminUserRolesDao.DaoQuery daoQuery = new AdminUserRolesDao.DaoQuery();
        daoQuery.adminId = adminId;
        daoQuery.count = Integer.MAX_VALUE;
        List<AdminUserRoles> adminUserRoles = adminUserRolesDao.list(daoQuery);
        if (adminUserRoles == null) {
            adminUserRoles = new ArrayList<>();
        }
        return adminUserRoles.stream().map(e -> e.getRole()
        ).collect(Collectors.toList());
    }
 
    @Override
    public void setRoles(String adminId, List<AdminRole> roles) {
        if(roles==null){
            return;
        }
        //删除之前的权限
        Query query = new Query();
        query.addCriteria(Criteria.where("adminId").is(adminId));
        adminUserRolesDao.delete(query);
 
        //添加新的权限
        for (AdminRole role : roles) {
            AdminUserRoles adminUserRoles = new AdminUserRoles();
            adminUserRoles.setAdminId(adminId);
            adminUserRoles.setCreateTime(new Date());
            adminUserRoles.setRole(role);
            adminUserRoles.setId(adminUserRoles.toId());
            adminUserRolesDao.save(adminUserRoles);
        }
    }
 
 
}