JAVA通过文件后缀名进行校验,校验文件是否符合后缀
定义上传的文件类型白名单,可放在conf配置文件中。
# yml文件
# 校验上传文件类型白名单(配置多个类型逗号分隔)
suffix:
whiteList: XLSX,XLS,TXT
@Value("${suffix.whiteList}")
public String[] suffixWhiteList;
public String upload(@RequestParam("file") MultipartFile file) {
String fileType = "public";
String suffix =
file.getOriginalFilename().substring(file.getOriginalFilename().lastIndexOf(".") + 1);
boolean anyMatch = Arrays.stream(suffixWhiteList).anyMatch(x -> x.equalsIgnoreCase(suffix));
if(anyMatch){
String fileUrl = fileAdminService.fileUpload(file.getBytes(), fileType, suffix.toLowerCase());
return "fileUrl"
}else{
return new RestResultResponse().data("不支持此类型文件!");
}
}
请先 后发表评论~