对用户上传文件类型限制的工具类
1.编写工具类
/**
* 校验上传的文件类型是否符合要求
* @param mf 需要校验的文件
* @param types 允许的文件类型。多个文件类型中间用“,”隔开。
* @return
*/
public static boolean checkFileType(MultipartFile mf,String types){
String[] typeArr = types.split(",");
// 文件不能为空
if(mf.isEmpty()){
return false;
}
// 获取文件扩展名
String extension = FilenameUtils.getExtension(mf.getOriginalFilename());
for(int i=0;i<typeArr.length;i++){
if(typeArr[i].equalsIgnoreCase(extension)){
return true;
}
}
return false;
}
2.对应的控制层引入
MultipartFile file = request.getFileMap().get("file");
//调用工具类 对上传的文件类型做校验
if (FileUtils.checkFileType(file, "PNG,JPG,GIF")){
return RetMessage.error("文件上传类型错误,请重新选择!");
}
浙公网安备 33010602011771号