package com.genersoft.iot.vmp.vmanager.user; import com.genersoft.iot.vmp.conf.exception.ControllerException; import com.genersoft.iot.vmp.conf.security.SecurityUtils; import com.genersoft.iot.vmp.service.IRoleService; import com.genersoft.iot.vmp.storager.dao.dto.Role; import com.genersoft.iot.vmp.utils.DateUtil; import com.genersoft.iot.vmp.vmanager.bean.ErrorCode; import com.genersoft.iot.vmp.vmanager.bean.WVPResult; import io.swagger.v3.oas.annotations.Operation; import io.swagger.v3.oas.annotations.Parameter; import io.swagger.v3.oas.annotations.tags.Tag; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.http.HttpStatus; import org.springframework.http.ResponseEntity; import org.springframework.web.bind.annotation.*; import java.util.List; @Tag(name = "角色管理") @CrossOrigin @RestController @RequestMapping("/api/role") public class RoleController { @Autowired private IRoleService roleService; @PostMapping("/add") @Operation(summary = "添加角色") @Parameter(name = "name", description = "角色名", required = true) @Parameter(name = "authority", description = "权限(自行定义内容,目前未使用)", required = true) public void add(@RequestParam String name, @RequestParam(required = false) String authority){ // 获取当前登录用户id int currenRoleId = SecurityUtils.getUserInfo().getRole().getId(); if (currenRoleId != 1) { // 只用角色id为1才可以删除和添加用户 throw new ControllerException(ErrorCode.ERROR403); } Role role = new Role(); role.setName(name); role.setAuthority(authority); role.setCreateTime(DateUtil.getNow()); role.setUpdateTime(DateUtil.getNow()); int addResult = roleService.add(role); if (addResult <= 0) { throw new ControllerException(ErrorCode.ERROR100); } } @DeleteMapping("/delete") @Operation(summary = "删除角色") @Parameter(name = "id", description = "用户Id", required = true) public void delete(@RequestParam Integer id){ // 获取当前登录用户id int currenRoleId = SecurityUtils.getUserInfo().getRole().getId(); if (currenRoleId != 1) { // 只用角色id为0才可以删除和添加用户 throw new ControllerException(ErrorCode.ERROR403); } int deleteResult = roleService.delete(id); if (deleteResult <= 0) { throw new ControllerException(ErrorCode.ERROR100); } } @GetMapping("/all") @Operation(summary = "查询角色") public List all(){ // 获取当前登录用户id List allRoles = roleService.getAll(); return roleService.getAll(); } }