补充提交

MercuryZ

2022-11-14 2479dc444595f8771b44f7d720abd366b8e4aa64

 sandu-common/src/main/java/com/sandu/common/file/impl/LocalFileUploadImpl.java

@@ -5,14 +5,15 @@
import cn.hutool.core.io.FileTypeUtil;
import cn.hutool.core.io.FileUtil;
import cn.hutool.core.lang.UUID;
import com.alibaba.fastjson.JSON;
import com.sandu.common.enums.FileStorageEnums;
import com.sandu.common.execption.BusinessException;
import com.sandu.common.file.FileUploadDto;
import com.sandu.common.file.IFileUpload;
import com.sandu.common.file.config.FileProperties;
import com.sandu.common.file.*;
import com.sandu.common.file.config.NovaFileConfig;
import com.sandu.common.util.SpringContextHolder;
import lombok.AllArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang.StringEscapeUtils;
import org.springframework.stereotype.Service;
import org.springframework.web.multipart.MultipartFile;

@@ -20,8 +21,13 @@
import java.awt.image.BufferedImage;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.math.BigInteger;
import java.security.MessageDigest;
import java.time.LocalDate;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Optional;

/**
@@ -32,21 +38,23 @@
@AllArgsConstructor
public class LocalFileUploadImpl implements IFileUpload {

    private static List<String> videoSuffixList = CollectionUtil.newArrayList("mp4", "mov", "avi", "mkv", "m4v", "wmv",
            "asf", "asx", "rm", "rmvb", "3gp", "dat", "flv", "vob");
    private static List<String> videoSuffixList = CollectionUtil.newArrayList("mp4", "mov", "avi", "mkv", "m4v", "wmv", "asf", "asx", "rm", "rmvb", "3gp", "dat", "flv", "vob");

    private final FileToThumbnail fileToThumbnail;

    @Override
    public FileUploadDto uploadFile(MultipartFile multipartFile) {
        FileProperties fileProperties = SpringContextHolder.getBean(FileProperties.class);
        NovaFileConfig fileProperties = SpringContextHolder.getBean(NovaFileConfig.class);
        String originalFilename = multipartFile.getOriginalFilename();
        long size = multipartFile.getSize();
        String suffix = FileUtil.getSuffix(originalFilename);
        String date = LocalDateTimeUtil.formatNormal(LocalDate.now());
        String fileName = UUID.fastUUID().toString() + "." + suffix;
        String uuidStr = UUID.fastUUID().toString();
        String fileName = uuidStr + "." + suffix;
        File dest;
        try {
            String type = Optional.ofNullable(FileTypeUtil.getType(multipartFile.getInputStream(), originalFilename)).orElse("");
            dest = new File(fileProperties.getUploadRootPath() + type + File.separator + date + File.separator + fileName).getCanonicalFile();
            dest = new File(fileProperties.getUploadRootPath() + File.separator + type + File.separator + date + File.separator + fileName).getCanonicalFile();

            if (!dest.getParentFile().exists()) {
                dest.getParentFile().mkdirs();
@@ -61,10 +69,13 @@
        FileUploadDto uploadDto = new FileUploadDto();
        uploadDto.setOriginName(originalFilename);
        uploadDto.setFileName(fileName);
        uploadDto.setFileUrl(dest.getPath().replace(fileProperties.getUploadRootPath(), ""));
        uploadDto.setRealUrl(dest.getPath().replace(File.separator,"/"));
        uploadDto.setFileUrl(dest.getPath().replace(fileProperties.getUploadRootPath(), "").replace(File.separator,"/"));
        uploadDto.setFileType(FileUtil.getType(dest));
        uploadDto.setStorageType(FileStorageEnums.LOCAL.getCode());
        uploadDto.setFileSize(size);

        uploadDto.setMd5(getMD5(multipartFile) + "." + uploadDto.getFileType());

        if (IMAGE_SUFFIX_LIST.contains(uploadDto.getFileType())) {
            BufferedImage img = null;
@@ -77,39 +88,61 @@
                uploadDto.setWidth(img.getWidth());
                uploadDto.setHeight(img.getHeight());
            }
        } else if (FileService.videoSuffixList.contains(uploadDto.getFileType())) {
            try {
                long duration = VideoUtil.getDuration(uploadDto.getRealUrl());
                uploadDto.setDuration((int) (duration * 1000));
            } catch (IOException e) {
                throw new RuntimeException(e);
            }
            try {
                InputStream inputStream = multipartFile.getInputStream();
                String screenShot = fileProperties.getUploadRootPath() + uploadDto.getFileType() + File.separator + date + File.separator + "screenShot" + uuidStr + ".jpg";
                String thumbnailForVideo = fileToThumbnail.getThumbnailForVideo(inputStream, screenShot);
                uploadDto.setScreenShot(thumbnailForVideo);
                inputStream.close();
            } catch (IOException e) {
                throw new RuntimeException(e);
            }
        }

        return uploadDto;
    }

    public FileUploadDto uploadCert(MultipartFile multipartFile) {
        String originalFilename = multipartFile.getOriginalFilename();
        long size = multipartFile.getSize();
        String suffix = FileUtil.getSuffix(originalFilename);
        String date = LocalDateTimeUtil.formatNormal(LocalDate.now());
        String fileName = UUID.fastUUID().toString() + "." + suffix;
        File dest;

    /**
     * 获取上传文件的MD5值
     *
     * @param file
     * @return
     */
    public String getMD5(MultipartFile file) {

        try {
            String type = Optional.ofNullable(FileTypeUtil.getType(multipartFile.getInputStream(), originalFilename)).orElse("");
//            dest = new File(fileProperties.getUploadRootPath() + type + File.separator + date + File.separator + fileName).getCanonicalFile();

//            if (!dest.getParentFile().exists()) {
//                dest.getParentFile().mkdirs();
//            }
//            multipartFile.transferTo(dest);

            byte[] uploadBytes = file.getBytes();
            MessageDigest md5 = MessageDigest.getInstance("MD5");
            byte[] digest = md5.digest(uploadBytes);
            String hashString = new BigInteger(1, digest).toString(16);
            if (hashString.length() % 2 != 0) {
                hashString = "0" + hashString;
            }
            return hashString;
        } catch (Exception e) {
            e.printStackTrace();
            throw new BusinessException(e.getMessage());
        }
        return null;
    }

        FileUploadDto uploadDto = new FileUploadDto();
        uploadDto.setOriginName(originalFilename);
        uploadDto.setFileName(fileName);
//        uploadDto.setFileUrl(dest.getPath());
//        uploadDto.setFileType(FileUtil.getType(dest));
        uploadDto.setStorageType(FileStorageEnums.LOCAL.getCode());
        uploadDto.setFileSize(size);
        return uploadDto;
    public static void main(String[] args) {
//        String a = "E:\\\\file\\\\png\\\\2022-11-14\\\\bc8cbe61-2c81-4dac-911a-91b711eebc24\\\\png";
//        System.out.println(a);
//        System.out.println(a.replaceAll("\\\\", "/"));
        File f = null;
        try {
            f = new File("E:\\file\\png\\2022-11-14\\cc4ce72e-2304-4f45-81e5-3b57519d57b6.png").getCanonicalFile();
        } catch (IOException e) {
            e.printStackTrace();
        }
        System.out.println(File.separator);
        System.out.println(f.getPath());
    }
}