角色赋予添加验证

This commit is contained in:
176yunxuan 2024-01-19 08:46:33 +08:00
parent f564b64622
commit 90af016e16
3 changed files with 13 additions and 13 deletions

View File

@ -9,6 +9,7 @@ import org.springframework.lang.Nullable;
import org.springframework.web.bind.annotation.*;
import javax.servlet.http.HttpServletRequest;
import javax.validation.constraints.NotNull;
/**
* <h1>角色控制器</h1>
@ -43,12 +44,12 @@ public class RoleController {
* @return
*/
@PostMapping("role/user/add")
public BaseResponse roleAddUser(@RequestParam Long uid, @RequestParam Long rid) {
public BaseResponse roleAddUser(HttpServletRequest request, @RequestParam Long uid, @RequestParam Long rid) {
// 判断是否有参数错误
if (uid == null || rid == null) {
return ResultUtil.error(ErrorCode.PARAMETER_ERROR);
}
return roleService.roleAddUser(uid, rid);
return roleService.roleAddUser(request,uid, rid);
}
/**
@ -57,11 +58,11 @@ public class RoleController {
* @return
*/
@DeleteMapping("role/user/remove")
public BaseResponse roleRemoveUser(@RequestParam Long uid) {
public BaseResponse roleRemoveUser(HttpServletRequest request,@RequestParam Long uid) {
// 判断是否有参数错误
if (uid == null) {
return ResultUtil.error(ErrorCode.PARAMETER_ERROR);
}
return roleService.roleRemoveUser(uid);
return roleService.roleRemoveUser(request,uid);
}
}

View File

@ -13,9 +13,9 @@ import javax.servlet.http.HttpServletRequest;
* @since v1.1.0
*/
public interface RoleService {
BaseResponse roleAddUser(Long uid, Long rid);
BaseResponse roleAddUser(HttpServletRequest request,Long uid, Long rid);
BaseResponse roleRemoveUser(Long uid);
BaseResponse roleRemoveUser(HttpServletRequest request,Long uid);
BaseResponse roleGet(HttpServletRequest request, String id);
}

View File

@ -20,22 +20,21 @@ import java.util.regex.Pattern;
public class RoleServiceImpl implements RoleService {
private final RoleDAO roleDAO;
private final UserDAO userDAO;
@Override
public BaseResponse roleAddUser(Long uid, Long rid) {
if (userDAO.isExistUser(uid)) {
public BaseResponse roleAddUser(HttpServletRequest request,Long uid, Long rid) {
if (Processing.checkUserIsAdmin(request,roleDAO.roleMapper)) {
roleDAO.roleAddUser(uid, rid);
return ResultUtil.success();
} else return ResultUtil.error(ErrorCode.USER_NOT_EXIST);
} else return ResultUtil.error(ErrorCode.NOT_ADMIN);
}
@Override
public BaseResponse roleRemoveUser(Long uid) {
if (userDAO.isExistUser(uid)) {
public BaseResponse roleRemoveUser(HttpServletRequest request,Long uid) {
if (Processing.checkUserIsAdmin(request,roleDAO.roleMapper)) {
roleDAO.roleRemoveUser(uid);
return ResultUtil.success();
} else return ResultUtil.error(ErrorCode.USER_NOT_EXIST);
} else return ResultUtil.error(ErrorCode.NOT_ADMIN);
}
@Override