From 079499c3f90e813e59ebeba62a22ca930e0df370 Mon Sep 17 00:00:00 2001
From: huanghongfa <18228131219@163.com>
Date: 星期五, 25 十二月 2020 16:28:10 +0800
Subject: [PATCH] 运营后台系统管理完成

---
 springcloud_k8s_panzhihuazhihuishequ/service_user/src/main/java/com/panzhihua/service_user/service/impl/RoleServiceImpl.java |   56 +++++++++++++++++++++++++++++++++++++++++++++++++++++++-
 1 files changed, 55 insertions(+), 1 deletions(-)

diff --git a/springcloud_k8s_panzhihuazhihuishequ/service_user/src/main/java/com/panzhihua/service_user/service/impl/RoleServiceImpl.java b/springcloud_k8s_panzhihuazhihuishequ/service_user/src/main/java/com/panzhihua/service_user/service/impl/RoleServiceImpl.java
index 8282b08..df2117e 100644
--- a/springcloud_k8s_panzhihuazhihuishequ/service_user/src/main/java/com/panzhihua/service_user/service/impl/RoleServiceImpl.java
+++ b/springcloud_k8s_panzhihuazhihuishequ/service_user/src/main/java/com/panzhihua/service_user/service/impl/RoleServiceImpl.java
@@ -1,25 +1,32 @@
 package com.panzhihua.service_user.service.impl;
 
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
+import com.panzhihua.common.constants.SecurityConstants;
 import com.panzhihua.common.model.vos.MenuVO;
 import com.panzhihua.common.model.vos.R;
+import com.panzhihua.common.model.vos.user.MenuRoleVO;
 import com.panzhihua.common.model.vos.user.RoleVO;
 import com.panzhihua.common.model.vos.user.SysMenuVO;
 import com.panzhihua.service_user.dao.RoleDAO;
 import com.panzhihua.service_user.dao.SysMenuDAO;
+import com.panzhihua.service_user.dao.SysRoleMenuDAO;
 import com.panzhihua.service_user.dao.SysUserRoleDAO;
 import com.panzhihua.service_user.model.dos.SysMenuDO;
 import com.panzhihua.service_user.model.dos.SysRoleDO;
+import com.panzhihua.service_user.model.dos.SysRoleMenuDO;
 import com.panzhihua.service_user.model.dos.SysUserRoleDO;
 import com.panzhihua.service_user.service.RoleService;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.BeanUtils;
+import org.springframework.data.redis.core.StringRedisTemplate;
 import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
 import org.springframework.util.ObjectUtils;
 
 import javax.annotation.Resource;
 import java.util.ArrayList;
 import java.util.List;
+import java.util.stream.Collectors;
 
 /**
  * @program: springcloud_k8s_panzhihuazhihuishequ
@@ -36,6 +43,10 @@
     private SysUserRoleDAO sysUserRoleDAO;
     @Resource
     private SysMenuDAO sysMenuDAO;
+    @Resource
+    private SysRoleMenuDAO sysRoleMenuDAO;
+    @Resource
+    private StringRedisTemplate stringRedisTemplate;
     /**
      * 获取平台所有的菜单按钮权限
      *
@@ -159,8 +170,51 @@
             sysMenuDOS.forEach(sysMenuDO -> {
                 SysMenuVO sysMenuVO=new SysMenuVO();
                 BeanUtils.copyProperties(sysMenuDO,sysMenuVO);
+                Long menuId = sysMenuDO.getMenuId();
+                List<SysMenuDO> sysMenuDOS1 = sysMenuDAO.selectList(new QueryWrapper<SysMenuDO>().lambda().eq(SysMenuDO::getParentId, menuId).orderByAsc(SysMenuDO::getOrderNum));
+                List<SysMenuVO> sysMenuVOS1=new ArrayList<>();
+                if (!ObjectUtils.isEmpty(sysMenuDOS1)) {
+                    sysMenuDOS1.forEach(sysMenuDO1 -> {
+                        SysMenuVO sysMenuVO1=new SysMenuVO();
+                        BeanUtils.copyProperties(sysMenuDO1,sysMenuVO1);
+                        sysMenuVOS1.add(sysMenuVO1);
+                    });
+                    sysMenuVO.setSysMenuVOList(sysMenuVOS1);
+                }
+                sysMenuVOS.add(sysMenuVO);
             });
         }
-        return null;
+        return R.ok(sysMenuVOS);
+    }
+
+    /**
+     * 设置角色的权限
+     *
+     * @param menuRoleVO 用户设置的权限id
+     * @return 设置结果
+     */
+    @Override
+    @Transactional(rollbackFor = Exception.class)
+    public R putMenuRole(MenuRoleVO menuRoleVO) {
+        List<Long>menuIds=new ArrayList<>();
+        Long roleId = menuRoleVO.getRoleId();
+//        删除旧的
+        sysRoleMenuDAO.delete(new QueryWrapper<SysRoleMenuDO>().lambda().eq(SysRoleMenuDO::getRoleId, roleId));
+//        创建新的
+        Integer isAll = menuRoleVO.getIsAll();
+        if (isAll.intValue()==1) {
+            menuIds = sysMenuDAO.selectList(null).stream().map(sysMenuDO -> sysMenuDO.getMenuId()).collect(Collectors.toList());
+        }else{
+            menuIds=menuRoleVO.getMenuIds();
+        }
+        for(int i=0;i<menuIds.size();i++){
+            SysRoleMenuDO sysRoleMenuDO=new SysRoleMenuDO();
+            sysRoleMenuDO.setMenuId(menuIds.get(i));
+            sysRoleMenuDO.setRoleId(roleId);
+            sysRoleMenuDAO.insert(sysRoleMenuDO);
+        }
+//        清除缓存
+        stringRedisTemplate.delete(SecurityConstants.ROLE_ALL);
+        return R.ok();
     }
 }

--
Gitblit v1.7.1