角色啊搞定

zen-api/src/main/kotlin/com/gxzc/zen/api/sys/service/ISysMenuRoleService.kt

@@ -13,4 +13,11 @@ import com.gxzc.zen.common.base.BaseService
  */
 interface ISysMenuRoleService : BaseService<SysMenuRole> {
     fun removeByMenuId(menuId: Long)
+
+    /**
+     * 通过roleid来获取
+     */
+    fun getMenuIdListByRoleId(roleId: Long): List<Long>
+
+    fun updateMenuRole(roleId: Long, menuIdList: List<Long>)
 }

zen-api/src/main/kotlin/com/gxzc/zen/api/sys/service/impl/SysMenuRoleServiceImpl.kt

@@ -5,6 +5,9 @@ import com.baomidou.mybatisplus.service.impl.ServiceImpl
 import com.gxzc.zen.api.sys.mapper.SysMenuRoleMapper
 import com.gxzc.zen.api.sys.model.SysMenuRole
 import com.gxzc.zen.api.sys.service.ISysMenuRoleService
+import com.gxzc.zen.orm.annotation.ZenTransactional
+import com.gxzc.zen.umps.constant.ZenHttpSession
+import com.gxzc.zen.umps.util.ShiroRedisUtil
 import org.springframework.stereotype.Service
 
 /**
@@ -17,7 +20,6 @@ import org.springframework.stereotype.Service
  */
 @Service
 class SysMenuRoleServiceImpl : ServiceImpl<SysMenuRoleMapper, SysMenuRole>(), ISysMenuRoleService {
-
     override fun removeByMenuId(menuId: Long) {
         val condition = SysMenuRole().apply {
             this.menuId = menuId
@@ -25,4 +27,44 @@ class SysMenuRoleServiceImpl : ServiceImpl<SysMenuRoleMapper, SysMenuRole>(), IS
         baseMapper.physicalDelete(EntityWrapper(condition))
     }
 
+    override fun getMenuIdListByRoleId(roleId: Long): List<Long> {
+        val condition = SysMenuRole().apply {
+            this.roleId = roleId
+        }
+        return baseMapper.selectWOLogic(EntityWrapper(condition)).map {
+            it.menuId!!
+        }
+    }
+
+    @ZenTransactional
+    override fun updateMenuRole(roleId: Long, menuIdList: List<Long>) {
+        // 先获取roleid的所有menuid
+        val oldMenuIdList = getMenuIdListByRoleId(roleId)
+        val deleteIds = oldMenuIdList.subtract(menuIdList)
+        val insertIds = menuIdList.subtract(oldMenuIdList)
+
+        var eff = false
+        if (deleteIds.isNotEmpty()) {
+            baseMapper.physicalDelete(EntityWrapper<SysMenuRole>().eq("role_id", roleId).`in`("menu_id", deleteIds))
+            eff = true
+        }
+
+        if (insertIds.isNotEmpty()) {
+            val entityList = mutableListOf<SysMenuRole>()
+            insertIds.forEach {
+                entityList.add(SysMenuRole().apply {
+                    this.roleId = roleId
+                    this.menuId = it
+                })
+            }
+            this.insertBatch(entityList)
+            eff = true
+        }
+
+        if (eff) {
+            // 更新一下缓存SESSION_KEY_USER_MENU
+            ShiroRedisUtil.removeAllSessionsAttributeKey(ZenHttpSession.SESSION_KEY_USER_MENU)
+        }
+    }
+
 }

zen-api/src/main/kotlin/com/gxzc/zen/api/sys/service/impl/SysMenuServiceImpl.kt

@@ -99,6 +99,10 @@ class SysMenuServiceImpl : ServiceImpl<SysMenuMapper, SysMenu>(), ISysMenuServic
                 }
             }
         }
+        // 排序
+        menu.forEach {
+            it.sortChildren(null)
+        }
         return menu
     }
 

zen-web/src/main/kotlin/com/gxzc/zen/web/sys/controller/MenuRoleController.kt

@@ -0,0 +1,42 @@
+package com.gxzc.zen.web.sys.controller
+
+import com.gxzc.zen.api.sys.service.ISysMenuRoleService
+import com.gxzc.zen.common.config.request.annotation.KVType
+import com.gxzc.zen.common.config.request.annotation.KVTypeEnum
+import com.gxzc.zen.common.config.request.annotation.ZenRequestTypes
+import com.gxzc.zen.common.dto.RequestDto
+import com.gxzc.zen.common.dto.ResponseDto
+import io.swagger.annotations.ApiOperation
+import org.springframework.beans.factory.annotation.Autowired
+import org.springframework.http.ResponseEntity
+import org.springframework.web.bind.annotation.*
+
+/**
+ *
+ * @author NorthLan
+ * @date 2018/5/5
+ * @url https://noahlan.com
+ */
+@RestController
+class MenuRoleController {
+
+    @Autowired
+    private lateinit var menuRoleSerivce: ISysMenuRoleService
+
+    @ApiOperation("获取对应角色拥有的菜单id列表")
+    @GetMapping("/role/{id}/menu/idlist")
+    fun getMenuIdListByRoleId(@PathVariable id: Long): ResponseEntity<*> {
+        return ResponseEntity.ok(ResponseDto().apply {
+            this.data = menuRoleSerivce.getMenuIdListByRoleId(id)
+        })
+    }
+
+    @Suppress("UNCHECKED_CAST")
+    @ApiOperation("更新列表")
+    @PutMapping("/role/{id}/menu")
+    @ZenRequestTypes(KVType(key = "menuIdList", value = Long::class, type = KVTypeEnum.LIST))
+    fun updateMenuRole(@PathVariable id: Long, @RequestBody requestDto: RequestDto): ResponseEntity<*> {
+        menuRoleSerivce.updateMenuRole(id, requestDto["menuIdList"] as List<Long>)
+        return ResponseEntity.ok(ResponseDto())
+    }
+}