Spring Boot统一功能处理

107 阅读1分钟

1.同意用户登录权限验证

拦截器的实现分为两步:

1)创建自定义拦截器,实现HandlerInterceptor接口并重写preHandle方法

2)配置拦截器和拦截规则,(将自定义拦截器加入WebMvcConfigurer的addInterceptors方法中)

1.1自定义拦截器

public class LoginInterceptor implements HandlerInterceptor {
    @Override
        public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
        HttpSession session = request.getSession(false);
        if (session != null && session.getAttribute("user") != null) {
            return true;
        }
        response.setStatus(401);
        return false;
    }
}

1.2将自定义拦截器加入到系统配置

@Configuration
public class AppConfig implements WebMvcConfigurer {
    @Override
    public void addInterceptors(InterceptorRegistry registry) {
        registry.addInterceptor(new LoginInterceptor())
                .addPathPatterns("/**") // 拦截所有的接口
                .excludePathPatterns("/**/login.html"); // 排除的接口
    }
}
**addPathPatterns**:表示要拦截的URL

**excludePathPatterns**:表示需要排除的URL

1.3运行结果

image.png

1.4总结

Spring中的拦截器是通过动态代理和环绕通知的思想实现的

QQ截图20220911193204.png

image.png

2.统一异常处理

统一异常处理使用的是@ControllerAdvice + @ExceptionHandler来实现的

    @ControllerAdvice表示控制器通知类
    @ExceptionHandler是异常处理器
    
两个结合表示当出现异常的时候执行某个通知,也就是执行某个方法事件

2.1代码实现

@ControllerAdvice
@ResponseBody
public class ErrorAdvice {

    @ExceptionHandler(NullPointerException.class)
    public Object nullPointerException(NullPointerException e) {
        HashMap<String, Object> map = new HashMap<>();
        map.put("success",200);
        map.put("status",-1);
        map.put("message","空指针异常!");
        return map;
    }
}

2.2运行结果

@RestController
@RequestMapping("/user")
public class UserController {
    @RequestMapping("/hi")
    public String say() {
        Object object = null;
        int i = object.hashCode();
        return "Hello";
    }
}

2.3运行结果:

image.png

3.统一数据返回格式

统一的数据返回格式可以使用@ControllerAdvice + ResponseBodyAdvice的方式实现

1)添加@ControllerAdvice
2)实现ResponseBodyAdvice接口,并重写beforeBodyWrite方法

3.1代码实现

@ControllerAdvice
public class ResponseAdvice implements ResponseBodyAdvice {
    /**
     * 内容是否重写, 返回true表示重写
     */
    @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) {
    // 统一数据格式的封装
        HashMap<String, Object> result = new HashMap<>();
        result.put("success",200);
        result.put("status",1);
        result.put("data",body);
        return result;
    }
}

3.2运行结果

@RestController
@RequestMapping("/user")
public class UserController {
    @RequestMapping("/index")
    public Object index() {
        User user = new User();
        user.setUsername("SpringAOP");
        user.setPassword("123456");
        return user;
    }
}

image.png