Java上传文件工具类

第一个类getUploadDir获取上传文件的目录,我这里获取的jar运行时的所在的目录,开发时就是项目目录,这样就保证了无论在那个系统下运行该jar包都可以适用。如果没有该目录就创建。

 @SneakyThrows
    public static String getUploadDir() {
        String path = ProjectApplication.class.getProtectionDomain().getCodeSource().getLocation().getPath();//这个class是启动类,这里自行修改
        path = URLDecoder.decode(path, StandardCharsets.UTF_8.toString());
        if (System.getProperty("os.name").contains("Windows")) {
            int firstIndex = path.indexOf("/");
            path = path.substring(firstIndex+1);
        }
        if (path.contains("jar")) {
            path = path.substring(0, path.lastIndexOf("."));
        }
        path = path.replace("target/classes/", "");
        int lastIndex = path.lastIndexOf("/");
        path = path.substring(0,lastIndex);
        String url = path + "/upload";
        File uploadDirectory = new File(url);
        if (!uploadDirectory.exists()) {
            uploadDirectory.mkdir(); // 创建目录
        }
        return url;
    }

  

第二个类上传文件

    public static final String UPLOAD_DIR = getUploadDir();

    @SneakyThrows
    public static String uploadFile(MultipartFile file){
        String filePah = UPLOAD_DIR;
        String originalFilename = file.getOriginalFilename();
        String fileExtension = getFileExtension(originalFilename);
//在扩展名限制文件,如果不想限制直接去除即可
        if (isValidFileExtension(fileExtension)) {
            File destFile = new File(filePah + File.separator + originalFilename);
            file.transferTo(destFile);
            String fileUrl = filePah + '\\' + originalFilename;
            return fileUrl;
        } else {
            throw new ProjectException(MessageEnum.PAK_ERR.getCode(),MessageEnum.PAK_ERR.getMessage());
        }
    }

//提取扩展名
    private static String getFileExtension(String filename) {
        int dotIndex = filename.lastIndexOf(".");
        if (dotIndex > 0 && dotIndex < filename.length() - 1) {
            return filename.substring(dotIndex + 1).toLowerCase();
        }
        return "";
    }
//限制文件
   private static boolean isValidFileExtension(String fileExtensions) {
    List<String> allowedExtensions = Arrays.asList("","");//文件类别
    return Arrays.stream(fileExtensions)
            .allMatch(allowedExtensions::contains);
}
    }

  • 9
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值