SpringBoot统一功能处理

一、统一用户登录权限验证

1.1Spring拦截器

实现拦截器需要以下两步：
1.创建自定义拦截器，实现 HandlerInterceptor 接⼝的 preHandle（执行具体方法之前的预处理）方法。
2.将⾃定义拦截器加⼊ WebMvcConfigurer 的 addInterceptors 方法中

1.1.1自定义拦截器

package com.example.demo.interceptor;import org.springframework.web.servlet.HandlerInterceptor;import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpSession;/*** 创建一个登录的拦截器*/
public class LoginInterceptor implements HandlerInterceptor {//返回true表示验证通过，可以执行后面的方法；// 但是返回false表示验证失败，后面的代码就不能执行了@Overridepublic boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {HttpSession session = request.getSession(false);if (session != null && session.getAttribute("userinfo") != null){//表明用户已登录return true;}//执行到此行，表明验证未通过，自动跳转到登录页面response.sendRedirect("login.html");return false;}
}

1.1.2将拦截器配置给当前项目

package com.example.demo.config;import com.example.demo.interceptor.LoginInterceptor;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.servlet.config.annotation.InterceptorRegistry;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;@Configuration
public class AppConfig implements WebMvcConfigurer {@Autowiredprivate LoginInterceptor loginInterceptor;/*** 给当前项目添加拦截器* @param registry*/@Overridepublic void addInterceptors(InterceptorRegistry registry) {registry.addInterceptor(loginInterceptor).addPathPatterns("/**")//拦截使用的url.excludePathPatterns("/user/reg")//不拦截/user/reg.excludePathPatterns("/**/*.html");//拦截/**/*.html}
}

1.2拦截器实现原理

二、统⼀异常处理

使用 @ControllerAdvice + @ExceptionHandler来实现
@ControllerAdvice：控制器通知类
@ExceptionHandler：异常处理器
结合表示当出现异常的时候执行某个通知。

2.1创建异常类

添加@ControllerAdvice 注解

2.2实现异常的封装方法

添加@ExceptionHandler注解

package com.example.demo.config;import org.springframework.web.bind.annotation.ControllerAdvice;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.ResponseBody;import java.util.HashMap;@ControllerAdvice//对控制器进行功能增强（当前类为统一封装类）
public class MyExceptionResult {@ResponseBody@ExceptionHandler(Exception.class)public HashMap<String,Object> myException(Exception e){HashMap<String,Object> result = new HashMap<String,Object>();result.put("state",-1);result.put("msg","默认异常"+e.getMessage());result.put("data",null);return result;}
}

三、统⼀数据格式返回

以使⽤ @ControllerAdvice + ResponseBodyAdvice实现

package com.example.demo.config;import org.springframework.core.MethodParameter;
import org.springframework.http.MediaType;
import org.springframework.http.server.ServerHttpRequest;
import org.springframework.http.server.ServerHttpResponse;
import org.springframework.web.bind.annotation.ControllerAdvice;
import org.springframework.web.servlet.mvc.method.annotation.ResponseBodyAdvice;import java.util.HashMap;@ControllerAdvice
public class MyResponseBodyAdvice implements ResponseBodyAdvice {//是否要重写的方法改为true,true表示在返回数据之前，进行统一的格式封装@Overridepublic boolean supports(MethodParameter returnType, Class converterType) {return true;}@Overridepublic Object beforeBodyWrite(Object body, MethodParameter returnType, MediaType selectedContentType, Class selectedConverterType, ServerHttpRequest request, ServerHttpResponse response) {HashMap<String,Object> result = new HashMap<>();result.put("state",1);result.put("data",body);result.put("msg","");return result;}
}