代码优化

稚屿

2022-02-09 c99eb98001e22f671ce514268231edcd27fc73ef

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

@@ -2,6 +2,8 @@


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.config.RuoYiConfig;

@@ -100,7 +102,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 +114,7 @@


        File desc = getAbsoluteFile(baseDir, fileName);

        file.transferTo(desc);

        String pathFileName = getPathFileName(baseDir, fileName);

        return pathFileName;

        return getPathFileName(baseDir, fileName);

    }



    /**

@@ -145,8 +146,7 @@
    {

        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;

    }



    /**

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

        }

@@ -228,7 +228,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;

    }