目录
一、什么是统一功能处理
SpringBoot 统一功能处理:
- 定义:指在SpringBoot应用中,通过一些统一的处理机制,对具有共性的功能进行集中管理,以提高代码的可维护性和开发效率。
- 应用范围:通常包括统一异常处理、统一数据返回格式、统一用户认证与授权等。
二、统一用户登录权限验证
2.1 定义拦截器
在 SpringMVC 中 HandlerInterceptor 接口是一种用于拦截 Http 请求、响应的机制,它允许开发者在访问控制器(Controller)之前加入一些自定义的规范,如拦截器中设置了 session ,当 Controller 请求成功时返回 true 则进行后续相关操作返回 false 则结束。
首先在项目中有以下类:
- ApplicationValue类 自定义的 session
- UserInterceptor类 自定义拦截器
- ApplicationConfig类 自定义拦截规范
- UserController类 自定义请求
ApplicationValue:
/**
* 全局变量
* */
public class ApplicationValue {
//自定义session key
public static final String SESSION_KEY = "SESSION_KEY";
}
UserInterceptor:
/**
* 自定义拦截器
*/
@Component
public class UserInterceptor implements HandlerInterceptor {
/**
* * 返回true代表拦截器验证成功继续执行后续代码
* * 返回false代表拦截器验证失败不执行后续代码
* @param request
* @param response
* @param handler
* @return
* @throws Exception
*/
//前置通知
@Override
public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
System.out.println("拦截器开始执行");
//业务方法
HttpSession session = request.getSession(false);
if(session!=null && session.getAttribute(ApplicationValue.SESSION_KEY) != null) {
//登录成功
return true;
}
return false;
}
}
preHandle 方法是 UserInterceptor 接口底下的一个请求达到控制器之间(Controller)执行的一个方法。类似于 Spring AOP 中的前置通知。当请求满足拦截规则即 preHandle 方法返回 true 时则放行该请求,返回 false 则拦截该请求。
上述获取 session 时,在 getSession 方法中加入一个 false 是为了当获取的 session 不存在时不创建一个新的会话而是返回一个 null 。
2.2 制定拦截规则
WebMvcConfigurer 是 Spring Framework 提供的一个接口,它可以使开发者在 Spring MVC 中不使用 xml 文件来约束规则,而是通过 java 代码来扩展 SpringMVC 相关操作。如制定自定义对某些 Web 请求的拦截规则等。
ApplicationConfig:
@Configuration
public class ApplicationConfig implements WebMvcConfigurer {
//将拦截器注入进来
@Autowired
private UserInterceptor userInterceptor;
@Override
public void addInterceptors(InterceptorRegistry registry) {
//指定拦截规则
registry.addInterceptor(userInterceptor)
.addPathPatterns("/**") //拦截所有请求
.excludePathPatterns("/user/getuser") //排除拦截某个请求
;
}
}
InterceptorRegistry 是 Spring Framework 中的一个制定和管理拦截器的一个类。通过它底下的方法能放行或拦截某些请求。如 addPathPatterns 方法为拦截所有请求,excludePathPatterns 为排除拦截某个请求。
2.3 创建请求
UserController:
@RestController
@RequestMapping("/user")
public class UserController {
@RequestMapping("/getuser")
public String getUser() {
System.out.println("user访问成功");
return "hello user";
}
@RequestMapping("/login")
public String login() {
System.out.println("登录成功");
return "welcome login";
}
}
上述代码中 @RequestMapping 为路由选择,它是 SpringMVC 中将 Http 请求映射到某一个类或方法上的注解。
postman 测试,在浏览器或第三方软件中输入 localhost:8080/user/getuser 就能访问到上面制定的请求。
编译器输出:
此外,当你需要放行所有图片格式的数据时,图片我们知道有 png、jpg、jpeg 等类型的,我们不可能一个个的类型进行放行,此时可以将这些类型的图片放入一个文件夹,只需要放行这一个文件夹即可。
2.4 拦截器底层原理
通过日常观察,在控制台打印信息中所有的 Controller 代码都会执行 DispatcherServlet 中的 doDispatch 调度方法,DispatcherServlet 是前端控制器。
在 SpringBoot 中,当一个 Http 请求到达 Spring MVC 中时,DispatcherServlet 作为前端控制器,负责接收并处理这个请求,在这个过程中会用到 doDispatch 调度方法。
doDispatch 方法主要是负责调度和处理 Http 请求,
拦截器底层主要代码:
//调用预处理
if (!mappedHandler.applyPreHandle(processedRequest, respon
se)) {
return;
}
// 执⾏ Controller 中的业务
mv = ha.handle(processedRequest, response, mappedHandler.g
etHandler());
if (asyncManager.isConcurrentHandlingStarted()) {
return;
}
开始执⾏ Controller 之前,会先调⽤ 预处理⽅法 applyPreHandle:
boolean applyPreHandle(HttpServletRequest request, HttpServletResponse res
ponse) throws Exception {
for(int i = 0; i < this.interceptorList.size(); this.interceptorIndex
= i++) {
// 获取项⽬中使⽤的拦截器 HandlerInterceptor
HandlerInterceptor interceptor = (HandlerInterceptor)this.intercep
torList.get(i);
if (!interceptor.preHandle(request, response, this.handler)) {
this.triggerAfterCompletion(request, response, (Exception)null
);
return false;
}
}
return true;
}
在 applyPreHandle 中会获取所有的拦截器 HandlerInterceptor 并执⾏拦截器中
三、统一异常处理
当访问一个不存在的请求时,浏览器会出现一大片英文表示访问错误,并显示错误状态码为500,如下所示:
@RequestMapping("/user")
public class UserController {
@RequestMapping("/reg")
public String reg(){
Object obj = null;
System.out.println(obj.hashCode());
return "注册成功";
}
}
为了解决上述问题,可以使用 @ControllerAdvice 和 @ExceptionAdvice 这两个注解自定义设置某些异常对应返回的类型格式。如上述的空指针异常浏览器显示一大串英文字母,而通过这两个注解进行自定义设置一个 Ajax 格式,浏览器会显示一个状态码、异常提示、数据,让人看起来比较简洁,当然你也可以设置其他格式。
@ControllerAdvice 注解是 SpringFramework 中的一个注解,它是一个控制器通知类,主要的作用是用于定义全局异常处理器,以及用于全局数据绑定或全局数据格式化的逻辑。它提供了一种集中化的方式来处理跨多个控制器的通用逻辑,比如异常处理、数据预处理和后处理等。
@ExceptionAdvice 注解也是 SpringFramework 中的一个注解用于定义全局异常处理逻辑。它的主要功能是捕获特定类型的异常,并返回相应的响应数据,从而避免在每个控制器中重复编写异常处理代码。
ExceptionAdvice:
@RestControllerAdvice
public class ExceptionAdvice {
@ExceptionHandler(NullPointerException.class)//统一访问对象
public RequestAjax doNullPointerException (NullPointerException e) {
RequestAjax requestAjax = new RequestAjax();
requestAjax.setCode(-1);
requestAjax.setMas("空指针异常:" + e.getMessage());
requestAjax.setData(null);
return requestAjax;
}
}
RequestAjax:
@Data
public class RequestAjax {
private int code; //状态码
private String mas; // 状态码描述
private Object data; // 返回数据
}
结果:
以上是针对单个空指针异常进行的异常处理,异常的种类是非常多的如空指针异常、栈溢出异常等。假设一个程序中有多个异常都按照空指针异常来处理是不行的,因此改为以下代码使得无论是什么异常都会报出如异常:XXX,相比一大面英文字母好很多。
@ExceptionHandler(Exception.class)//统一访问对象
public RequestAjax doException (Exception e) {
RequestAjax requestAjax = new RequestAjax();
requestAjax.setCode(-1);
requestAjax.setMas("异常:" + e.getMessage());
requestAjax.setData(null);
return requestAjax;
}
四、统一数据格式返回
ResponseBodyAdvice用途:
ResponseBodyAdvice 是 Spring 4.1 引入的一个接口,用于在 Controller 方法返回响应体之前,对响应体的内容进行处理。它允许开发者拦截 Controller 方法的返回值,并对返回值进行修改、增强或定制化。具体来说,ResponseBodyAdvice 有以下几个主要用途:
统一处理返回值结构:
- 可以对Controller方法的返回值进行统一的封装,例如,将所有API接口的返回值封装成统一的JSON结构,包含状态码、消息和数据等字段。
- 便于前端统一处理不同接口的响应数据,提高开发效率。
数据加密或压缩:
- 可以在响应体写入之前,对返回值进行加密或压缩处理,提高数据的安全性或传输效率。
添加额外信息:
- 可以在返回值中添加一些额外的信息,例如时间戳、签名等,满足特定的业务需求。
则有以下代码:
@RequestMapping("/getuser")
public String getUser() {
System.out.println("user访问成功");
return "hello user";
}
@ControllerAdvice
public class RespondAdvice implements ResponseBodyAdvice {
@Autowired
private ObjectMapper objectMapper;
/**
* 当supports为true时执行后续操作,否则不执行
* @param returnType
* @param converterType
* @return
*/
@Override
public boolean supports(MethodParameter returnType, Class converterType) {
return true;
}
@Override
public Object beforeBodyWrite(Object body, MethodParameter returnType, MediaType selectedContentType, Class selectedConverterType, ServerHttpRequest request, ServerHttpResponse response) {
if (body instanceof RequestAjax) {
return body;
}
if (body instanceof String) {
RequestAjax requestAjax = RequestAjax.succ(body);
try {
return objectMapper.writeValueAsString(requestAjax);
} catch (JsonProcessingException e) {
e.printStackTrace();
}
}
return RequestAjax.succ(body);
}
}