若依 3.5.0

RuoYi

2022-04-11 d14f27e5eb3a804fa1c1f6cb5d471b63ed80f57a

 ruoyi-modules/ruoyi-file/src/main/java/com/ruoyi/file/utils/FileUploadUtils.java

@@ -2,16 +2,17 @@


import java.io.File;

import java.io.IOException;

import java.nio.file.Paths;

import java.util.Objects;

import org.apache.commons.io.FilenameUtils;

import org.springframework.beans.factory.annotation.Value;

import org.springframework.web.multipart.MultipartFile;

import com.ruoyi.common.core.exception.file.FileNameLengthLimitExceededException;

import com.ruoyi.common.core.exception.file.FileSizeLimitExceededException;

import com.ruoyi.common.core.exception.file.InvalidExtensionException;

import com.ruoyi.common.core.utils.DateUtils;

import com.ruoyi.common.core.utils.IdUtils;

import com.ruoyi.common.core.utils.StringUtils;

import com.ruoyi.common.core.utils.file.MimeTypeUtils;

import com.ruoyi.common.core.utils.uuid.Seq;



/**

 * 文件上传工具类

@@ -29,12 +30,6 @@
     * 默认的文件名最大长度 100

     */

    public static final int DEFAULT_FILE_NAME_LENGTH = 100;



    /**

     * 资源映射路径 前缀

     */

    @Value("${file.prefix}")

    public String localFilePrefix;



    /**

     * 根据文件路径上传

@@ -61,7 +56,7 @@
     *

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

     * @param file 上传的文件

     * @param extension 上传文件类型

     * @param allowedExtension 上传文件类型

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

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

     * @throws FileNameLengthLimitExceededException 文件名太长

@@ -72,7 +67,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);

@@ -82,10 +77,9 @@


        String fileName = extractFilename(file);



        File desc = getAbsoluteFile(baseDir, fileName);

        file.transferTo(desc);

        String pathFileName = getPathFileName(fileName);

        return pathFileName;

        String absPath = getAbsoluteFile(baseDir, fileName).getAbsolutePath();

        file.transferTo(Paths.get(absPath));

        return getPathFileName(fileName);

    }



    /**

@@ -93,10 +87,8 @@
     */

    public static final String extractFilename(MultipartFile file)

    {

        String fileName = file.getOriginalFilename();

        String extension = getExtension(file);

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

        return fileName;

        return StringUtils.format("{}/{}_{}.{}", DateUtils.datePath(),

                FilenameUtils.getBaseName(file.getOriginalFilename()), Seq.getId(Seq.uploadSeqType), getExtension(file));

    }



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

@@ -110,7 +102,7 @@
                desc.getParentFile().mkdirs();

            }

        }

        return desc;

        return desc.isAbsolute() ? desc : desc.getAbsoluteFile();

    }



    private static final String getPathFileName(String fileName) throws IOException

@@ -123,15 +115,14 @@
     * 文件大小校验

     *

     * @param file 上传的文件

     * @return

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

     * @throws InvalidExtensionException

     * @throws InvalidExtensionException 文件校验异常

     */

    public static final void assertAllowed(MultipartFile file, String[] allowedExtension)

            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);

        }

@@ -170,9 +161,9 @@
    /**

     * 判断MIME类型是否是允许的MIME类型

     *

     * @param extension

     * @param allowedExtension

     * @return

     * @param extension 上传文件类型

     * @param allowedExtension 允许上传文件类型

     * @return true/false

     */

    public static final boolean isAllowedExtension(String extension, String[] allowedExtension)

    {

@@ -197,7 +188,7 @@
        String extension = FilenameUtils.getExtension(file.getOriginalFilename());

        if (StringUtils.isEmpty(extension))

        {

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

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

        }

        return extension;

    }