整体设计:
- 对传过来的文件名截取获取文件后缀;
- 和本地定义好的类型进行比对;
直接上代码
/**
* 文件类型校验
*/
public class FileTypeCheckUtil {
//附件类型检查
final static String image = ".xbm.tif.pjp.svgz.jpg.jpeg.ico.tiff.svg.jfif.webp.png.bmp.pjpeg.avif";
final static String gif = ".gif";
final static String vedio = ".rm.rmvb.3gp.avi.mpeg.mpg.mkv.dat.asf.wmv.flv,mov.mp4.ogg.ogm";
// 代码 0:未知,1:图片 2:动图 3:视频
public static Integer checkFileType(String path) {
String fileType = path.substring(path.lastIndexOf(".")).toLowerCase();
if(image.contains(fileType)){
return MaterialTypeEnum.PICTURE.getCode();//1
}else if(gif.contains(fileType)){
return MaterialTypeEnum.GIF.getCode();//3
}else if (vedio.contains(fileType)){
return MaterialTypeEnum.VIDEO.getCode();//2
}else {
return 0;
}
}
}