优化代码

RuoYi

2022-02-13 34d87784df7535c30ad9dfb7065568820e13ea24

 src/main/java/com/ruoyi/common/utils/file/FileUploadUtils.java

@@ -2,6 +2,7 @@


import java.io.File;

import java.io.IOException;

import java.util.Objects;

import org.apache.commons.io.FilenameUtils;

import org.springframework.web.multipart.MultipartFile;

import com.ruoyi.common.constant.Constants;

@@ -15,7 +16,7 @@


/**

 * 文件上传工具类

 * 
 *

 * @author ruoyi

 */

public class FileUploadUtils

@@ -89,7 +90,7 @@
     *

     * @param baseDir 相对应用的基目录

     * @param file 上传的文件

     * @param extension 上传文件类型

     * @param allowedExtension 上传文件类型

     * @return 返回上传成功的文件名

     * @throws FileSizeLimitExceededException 如果超出最大大小

     * @throws FileNameLengthLimitExceededException 文件名太长

@@ -100,7 +101,7 @@
            throws FileSizeLimitExceededException, IOException, FileNameLengthLimitExceededException,

            InvalidExtensionException

    {

        int fileNamelength = file.getOriginalFilename().length();

        int fileNamelength = Objects.requireNonNull(file.getOriginalFilename()).length();

        if (fileNamelength > FileUploadUtils.DEFAULT_FILE_NAME_LENGTH)

        {

            throw new FileNameLengthLimitExceededException(FileUploadUtils.DEFAULT_FILE_NAME_LENGTH);

@@ -112,8 +113,7 @@


        File desc = getAbsoluteFile(baseDir, fileName);

        file.transferTo(desc);

        String pathFileName = getPathFileName(baseDir, fileName);

        return pathFileName;

        return getPathFileName(baseDir, fileName);

    }



    /**

@@ -121,33 +121,28 @@
     */

    public static final String extractFilename(MultipartFile file)

    {

        String fileName = file.getOriginalFilename();

        String extension = getExtension(file);

        fileName = DateUtils.datePath() + "/" + IdUtils.fastUUID() + "." + extension;

        return fileName;

        return DateUtils.datePath() + "/" + IdUtils.fastUUID() + "." + getExtension(file);

    }



    private static final File getAbsoluteFile(String uploadDir, String fileName) throws IOException

    public static final File getAbsoluteFile(String uploadDir, String fileName) throws IOException

    {

        File desc = new File(uploadDir + File.separator + fileName);



        if (!desc.getParentFile().exists())

        {

            desc.getParentFile().mkdirs();

        }

        if (!desc.exists())

        {

            desc.createNewFile();

            if (!desc.getParentFile().exists())

            {

                desc.getParentFile().mkdirs();

            }

        }

        return desc;

    }



    private static final String getPathFileName(String uploadDir, String fileName) throws IOException

    public static final String getPathFileName(String uploadDir, String fileName) throws IOException

    {

        int dirLastIndex = RuoYiConfig.getProfile().length() + 1;

        String currentDir = StringUtils.substring(uploadDir, dirLastIndex);

        String pathFileName = Constants.RESOURCE_PREFIX + "/" + currentDir + "/" + fileName;

        return pathFileName;

        return Constants.RESOURCE_PREFIX + "/" + currentDir + "/" + fileName;

    }



    /**

@@ -162,7 +157,7 @@
            throws FileSizeLimitExceededException, InvalidExtensionException

    {

        long size = file.getSize();

        if (DEFAULT_MAX_SIZE != -1 && size > DEFAULT_MAX_SIZE)

        if (size > DEFAULT_MAX_SIZE)

        {

            throw new FileSizeLimitExceededException(DEFAULT_MAX_SIZE / 1024 / 1024);

        }

@@ -184,6 +179,11 @@
            else if (allowedExtension == MimeTypeUtils.MEDIA_EXTENSION)

            {

                throw new InvalidExtensionException.InvalidMediaExtensionException(allowedExtension, extension,

                        fileName);

            }

            else if (allowedExtension == MimeTypeUtils.VIDEO_EXTENSION)

            {

                throw new InvalidExtensionException.InvalidVideoExtensionException(allowedExtension, extension,

                        fileName);

            }

            else

@@ -215,7 +215,7 @@


    /**

     * 获取文件名的后缀

     * 
     *

     * @param file 表单文件

     * @return 后缀名

     */

@@ -224,8 +224,8 @@
        String extension = FilenameUtils.getExtension(file.getOriginalFilename());

        if (StringUtils.isEmpty(extension))

        {

            extension = MimeTypeUtils.getExtension(file.getContentType());

            extension = MimeTypeUtils.getExtension(Objects.requireNonNull(file.getContentType()));

        }

        return extension;

    }

}
}