对用户上传文件类型限制的工具类

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("文件上传类型错误,请重新选择!");
        }

  

posted @ 2020-11-29 11:56  江遇晚风  阅读(200)  评论(0)    收藏  举报