SpringBoot拦截器 addResourceHandlers()用法和小坑

addResourceHandlers()用法

  1. 先上代码 然后再说
@Configuration
public class MyWebMVCConfig implements WebMvcConfigurer {
    @Value("${file.location}") //         D:/test/ 
    String filelocation;  // 这两个是路径 
    @Value("${file.path}") //         /file/**
    String filepath;

    @Override
    public void addResourceHandlers(ResourceHandlerRegistry registry) {
        //匹配到resourceHandler,将URL映射至location,也就是本地文件夹
        registry.addResourceHandler(filepath).addResourceLocations("file:///" + filelocation);//这里最后一个/不能不写
    }
}
  1. 这段代码意思就配置一个拦截器, 如果访问路径是addResourceHandler中的filepath 这个路径 那么就 映射到访问本地的addResourceLocations 的参数的这个路径上,这样就可以让别人访问服务器的本地文件了,比如本地图片或者本地音乐视频什么的。
在 Spring Boot 中,你可以通过使用拦截器来统一限制上传文件的类型和大小。拦截器可以拦截所有的上传请求,然后在上传之前进行校验。 首先,你需要创建一个拦截器类,并实现 HandlerInterceptor 接口。在拦截器中,你可以重写 preHandle 方法来实现文件上传的校验。以下是一个简单的例子: ``` public class FileUploadInterceptor implements HandlerInterceptor { private static final long MAX_FILE_SIZE = 10 * 1024 * 1024; // 10MB private static final List<String> ALLOWED_CONTENT_TYPES = Arrays.asList("image/jpeg", "image/png"); @Override public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception { if (handler instanceof HandlerMethod) { HandlerMethod handlerMethod = (HandlerMethod) handler; // 检查方法是否有 @PostMapping 注解 PostMapping postMapping = handlerMethod.getMethodAnnotation(PostMapping.class); if (postMapping == null) { return true; } // 检查方法参数是否有 MultipartFile 类型 MethodParameter[] methodParameters = handlerMethod.getMethodParameters(); for (MethodParameter methodParameter : methodParameters) { if (methodParameter.getParameterType().equals(MultipartFile.class)) { MultipartFile file = (MultipartFile) request.getAttribute(methodParameter.getParameterName()); // 检查文件大小 if (file.getSize() > MAX_FILE_SIZE) { response.setStatus(HttpStatus.BAD_REQUEST.value()); response.getWriter().write("File size too large"); return false; } // 检查文件类型 String contentType = file.getContentType(); if (!ALLOWED_CONTENT_TYPES.contains(contentType)) { response.setStatus(HttpStatus.BAD_REQUEST.value()); response.getWriter().write("File type not allowed"); return false; } } } } return true; } } ``` 在上面的代码中,我们首先检查请求处理的方法是否有 @PostMapping 注解,并且是否有 MultipartFile 类型的参数。如果都符合条件,我们就可以从请求中获取到上传的文件,并进行校验。如果上传的文件不符合要求,我们就返回一个错误响应。否则,就放行请求,让请求继续被处理。 接下来,你需要在 Spring Boot 应用程序中注册这个拦截器。你可以创建一个配置类,并实现 WebMvcConfigurer 接口。在这个配置类中,你可以重写 addInterceptors 方法来注册你的拦截器。以下是一个简单的例子: ``` @Configuration public class WebMvcConfig implements WebMvcConfigurer { @Autowired private FileUploadInterceptor fileUploadInterceptor; @Override public void addInterceptors(InterceptorRegistry registry) { registry.addInterceptor(fileUploadInterceptor) .addPathPatterns("/upload"); } } ``` 在上面的代码中,我们首先创建了一个 WebMvcConfigurer 类,并注入了我们之前创建的 FileUploadInterceptor。然后,我们重写了 addInterceptors 方法,并将我们的拦截器注册到了 /upload 路径下。这样,当用户上传文件时,我们的拦截器就会拦截请求,并对上传的文件进行校验。
评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值