menuIds;
}
diff --git a/src/main/java/com/mini/mybigscreen/biz/controller/HomeRoleController.java b/src/main/java/com/mini/mybigscreen/biz/controller/HomeRoleController.java
index ff49ca2..36304a0 100644
--- a/src/main/java/com/mini/mybigscreen/biz/controller/HomeRoleController.java
+++ b/src/main/java/com/mini/mybigscreen/biz/controller/HomeRoleController.java
@@ -1,12 +1,20 @@
package com.mini.mybigscreen.biz.controller;
+import cn.hutool.core.util.StrUtil;
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.mini.mybigscreen.Model.Message;
import com.mini.mybigscreen.Model.Result;
+import com.mini.mybigscreen.Model.RoleAssignPerm;
+import com.mini.mybigscreen.biz.domain.ErpCategory;
+import com.mini.mybigscreen.biz.domain.HomeRole;
+import com.mini.mybigscreen.biz.domain.HomeRoleMenu;
+import com.mini.mybigscreen.biz.service.HomeRoleMenuService;
import com.mini.mybigscreen.biz.service.HomeRoleService;
import jakarta.annotation.Resource;
-import org.springframework.web.bind.annotation.GetMapping;
-import org.springframework.web.bind.annotation.RequestMapping;
-import org.springframework.web.bind.annotation.RestController;
+import org.springframework.web.bind.annotation.*;
+
+import java.util.List;
+import java.util.stream.Collectors;
/**
*
@@ -20,16 +28,54 @@ import org.springframework.web.bind.annotation.RestController;
@RequestMapping("/biz/homeRole")
public class HomeRoleController {
+ @Resource
+ private HomeRoleMenuService roleMenuService;
@Resource
private HomeRoleService roleService;
@GetMapping("list")
- public Result> getList(){
+ public Result> getList() {
return Result.success(roleService.list());
}
+ /**
+ * 角色授权
+ */
+ @PostMapping("assign")
+ public Result getRoleAssign(@RequestBody RoleAssignPerm assignPerm) {
+ HomeRole role = roleService.getById(assignPerm.getRoleId());
+ role.setRoleName(assignPerm.getRoleName());
+ roleService.updateById(role);
+ roleMenuService.remove(new LambdaQueryWrapper()
+ .eq(HomeRoleMenu::getRoleId, assignPerm.getRoleId()));
+ List newRoleMenus = assignPerm.getMenuIds().stream()
+ .map(menuId -> {
+ HomeRoleMenu roleMenu = new HomeRoleMenu();
+ roleMenu.setRoleId(assignPerm.getRoleId());
+ roleMenu.setMenuId(menuId);
+ roleMenu.setUstatus("1");
+ return roleMenu;
+ })
+ .collect(Collectors.toList());
+ roleMenuService.saveBatch(newRoleMenus);
+ return Result.success(new Message("角色菜单分配成功", 200));
+ }
+
+ @GetMapping("userMenus")
+ public Result getUserMenus() {
+ return Result.success();
+ }
+
+
+ @GetMapping("roleMenus")
+ public Result> getRoleMenus(String roleId) {
+ LambdaQueryWrapper query = new LambdaQueryWrapper()
+ .eq(StrUtil.isNotBlank(roleId), HomeRoleMenu::getRoleId, roleId)
+ .orderByDesc(HomeRoleMenu::getCreateTime);
+ return Result.success(roleMenuService.list(query));
+ }
public Result save() {
diff --git a/src/main/java/com/mini/mybigscreen/biz/domain/HomeRoleMenu.java b/src/main/java/com/mini/mybigscreen/biz/domain/HomeRoleMenu.java
index 768507e..ba0e6f2 100644
--- a/src/main/java/com/mini/mybigscreen/biz/domain/HomeRoleMenu.java
+++ b/src/main/java/com/mini/mybigscreen/biz/domain/HomeRoleMenu.java
@@ -48,5 +48,5 @@ public class HomeRoleMenu extends BaseEntity implements Serializable {
private String menuId;
@TableField("ustatus")
- private Integer ustatus;
+ private String ustatus;
}