forked from kidgrow-microservices-platform

zhaoxiaohao
2020-11-04 f5f444c2bbb1e1e90a6ea6d0fbbfa06c07deed2d
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
package com.kidgrow.usercenter.service.impl;
 
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.kidgrow.common.constant.CommonConstant;
import com.kidgrow.common.context.ClientContextHolder;
import com.kidgrow.common.model.ResultBody;
import com.kidgrow.common.model.SysMenu;
import com.kidgrow.common.service.impl.SuperServiceImpl;
import com.kidgrow.usercenter.mapper.SysMenuMapper;
import com.kidgrow.usercenter.model.SysRoleMenu;
import com.kidgrow.usercenter.service.ISysMenuService;
import com.kidgrow.usercenter.service.ISysRoleMenuService;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.collections4.MapUtils;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.util.CollectionUtils;
 
import javax.annotation.Resource;
import java.util.*;
 
/**
 * 石家庄喜高科技有限责任公司 版权所有 © Copyright 2020<br>
 *
 * @Description: <br>
 * @Project: <br>
 * @CreateDate: Created in 2020/2/13 16:33 <br>
 * @Author: <a href="4345453@kidgrow.com">liuke</a>
 */
@Slf4j
@Service
public class SysMenuServiceImpl extends SuperServiceImpl<SysMenuMapper, SysMenu> implements ISysMenuService {
     @Resource
    private ISysRoleMenuService roleMenuService;
 
    @Transactional(rollbackFor = Exception.class)
    @Override
    public void setMenuToRole(Long roleId, Set<Long> menuIds) {
        roleMenuService.delete(roleId, null);
 
        if (!CollectionUtils.isEmpty(menuIds)) {
            List<SysRoleMenu> roleMenus = new ArrayList<>(menuIds.size());
            menuIds.forEach(menuId -> roleMenus.add(new SysRoleMenu(roleId, menuId)));
            roleMenuService.saveBatch(roleMenus);
        }
    }
 
    /**
     * 角色菜单列表
     * @param roleIds
     * @return
     */
    @Override
    public List<SysMenu> findByRoles(Set<Long> roleIds) {
        return roleMenuService.findMenusByRoleIds(roleIds, null);
    }
 
    /**
     * 角色菜单列表
     * @param roleIds 角色ids
     * @param roleIds 是否菜单
     * @return
     */
    @Override
    public List<SysMenu> findByRoles(Set<Long> roleIds, Integer type) {
        return roleMenuService.findMenusByRoleIds(roleIds, type);
    }
 
    @Override
    public List<SysMenu> findByRoleCodes(Set<String> roleCodes, Integer type) {
        String tentid=ClientContextHolder.getClient();
        if (roleCodes.contains(CommonConstant.SYSTEM_ADMIN_ROLE_CODE)) {
            tentid="";
        }
        return roleMenuService.findMenusByRoleCodes(roleCodes, type,tentid);
    }
 
    /**
     * 查询所有菜单
     */
    @Override
    public List<SysMenu> findAll(Map<String, Object> params) {
//        return baseMapper.selectList(
//                new QueryWrapper<SysMenu>().orderByAsc("sort")
//        );
 
        return baseMapper.findAll(params);
    }
 
    /**
     * 查询所有一级菜单
     */
    @Override
    public List<SysMenu> findOnes(String tenantId) {
        return baseMapper.selectList(
                new QueryWrapper<SysMenu>()
                        .eq("type", 1).eq("tenant_id",tenantId)
                        .orderByAsc("sort")
        );
    }
 
    @Override
    public ResultBody updateEnabled(Map<String, Object> params) {
        Long id = MapUtils.getLong(params, "id");
        Boolean enabled = MapUtils.getBoolean(params, "enabled");
 
        SysMenu sysMenu = baseMapper.selectById(id);
        if (sysMenu == null) {
            return ResultBody.failed("菜单不存在");
        }
        sysMenu.setEnabled(enabled);
        sysMenu.setUpdateTime(new Date());
 
        int i = baseMapper.updateById(sysMenu);
        log.info("修改菜单:{}", sysMenu);
 
        return i > 0 ? ResultBody.ok().data(sysMenu).msg("更新成功") : ResultBody.failed("更新失败");
    }
}