简单写了个文件类型的校验(图片、动图、视频)

134 阅读1分钟

整体设计:

  1. 对传过来的文件名截取获取文件后缀;
  2. 和本地定义好的类型进行比对;

直接上代码


/**
 * 文件类型校验
 */
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;
        }
    }

}