java 文件上傳,多次上傳文件名稱、格式相同的數據,需要對文件進行重命名

一.文件重複判斷(文件名和後綴判斷):

 /**
     * 獲取文件夾中相同文件名的文件個數
     *
     * @param filePath
     * @param fileName
     * @return
     */
    private int getFileMax(String filePath, String fileName) {
        File file = new File(filePath);
        File[] files;
        int number = 0;
        if (file.isDirectory()) {
            files = file.listFiles();
            for (int i = 0; i < files.length; i++) {
                File f = files[i];
                if (f.isFile()
                        && (f.getName().substring(0, f.getName().lastIndexOf("."))
                        .contains(fileName.substring(0, fileName.lastIndexOf(".")))
                        && f.getName().substring(f.getName().lastIndexOf("."))
                        .equals(fileName.substring(fileName.lastIndexOf("."))))) {
                    number = number + 1;
                }
            }
        }
        return number;
    }

二、文件上傳

 /**
     * 上傳文件
     *
     * @return
     */
    @Override
    public ServiceResult<FileInfoAO> uploadFile(MultipartFile file) throws Exception {
        // 獲取文件名
        String fileName = file.getOriginalFilename();
        //使用System.getProperty("user.dir")獲取當前程序運行的工作根目錄
        String uploadFilePath = System.getProperty("user.dir") + File.separator + "upload" + File.separator;
        File dest = new File(uploadFilePath);
        // 檢測是否存在目錄
        if (!dest.exists()) {
            dest.mkdirs();
        }
        //文件重命名,防止覆蓋
        int fileMax = getFileMax(uploadFilePath, fileName);
        StringBuilder destFilePath = new StringBuilder();
        if (fileMax > 0) {
            destFilePath.append(uploadFilePath).append(fileName.substring(0, fileName.lastIndexOf("."))).
                    append("(" + fileMax + ")").append(fileName.substring(fileName.lastIndexOf(".")));
        } else {
            destFilePath.append(uploadFilePath).append(fileName);
        }
        File destFile = new File(destFilePath.toString());
        file.transferTo(destFile);
        FileInfoAO fileInfoAO = new FileInfoAO();
        fileInfoAO.setCreateTime(DateTimeUtil.formatDateTime(new Date()));
        fileInfoAO.setFileName(fileName);
        fileInfoAO.setFilePath(destFile.getAbsolutePath());
        fileInfoAO.setFileType(StringUtils.substringAfter(fileName, "."));// 獲取文件的擴展名
        UserAO user = AuthUtil.getCurrentUser();
        fileInfoAO.setPulisher(user != null ? user.getUserName() : null);
        insert(fileInfoAO);
        return ServiceResultHelper.genResultWithSuccess(fileInfoAO);
    }

三、測試

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章