Java @PreAuthorize 自定义code
时间: 2023-10-09 13:07:12 浏览: 128
在 Spring Security 中,@PreAuthorize 注解本身并不直接支持自定义返回 code 值。它的返回值类型固定为 boolean,并用于判断是否允许访问该方法。
如果你想在权限验证失败时返回自定义的 code 值,可以通过结合使用 @PreAuthorize 注解和自定义的异常处理来实现。
首先,你可以创建一个自定义的异常类,用于表示权限验证失败的情况。例如:
```java
public class AccessDeniedException extends RuntimeException {
private int code;
public AccessDeniedException(int code, String message) {
super(message);
this.code = code;
}
public int getCode() {
return code;
}
}
```
然后,在需要进行权限验证的方法上使用 @PreAuthorize 注解,并在权限验证失败时抛出自定义的异常。例如:
```java
PreAuthorize("hasRole('ROLE_ADMIN')")
public void doSomething() {
// 检查权限...
if (!hasPermission()) {
throw new AccessDeniedException(403, "Access denied");
}
// 方法逻辑
}
```
最后,在你的应用程序中,可以定义一个异常处理器来处理权限验证失败时抛出的自定义异常。在异常处理器中,你可以根据自定义异常的 code 值自定义返回结果。
```java
@ControllerAdvice
public class ExceptionHandlerController {
@ExceptionHandler(AccessDeniedException.class)
public ResponseEntity<ErrorResponse> handleAccessDeniedException(AccessDeniedException ex) {
ErrorResponse errorResponse = new ErrorResponse(ex.getCode(), ex.getMessage());
return new ResponseEntity<>(errorResponse, HttpStatus.FORBIDDEN);
}
// 其他异常处理方法...
}
```
在上面的示例中,我们定义了一个异常处理器,当捕获到 AccessDeniedException 异常时,将返回一个包含自定义 code 值和错误信息的 ErrorResponse 对象。
这样,当权限验证失败时,就可以返回自定义的 code 值来表示权限错误。当然,具体的处理方式可以根据你的需求进行调整和扩展。
阅读全文