From ddfbc40f9ca8546a2c34865abef51630f054d5e9 Mon Sep 17 00:00:00 2001
From: liuhaonan <31457034@qq.com>
Date: 星期一, 07 十一月 2022 14:00:45 +0800
Subject: [PATCH] changes

---
 sandu-common/src/main/java/com/sandu/common/file/impl/AliOssFileServiceImpl.java |   47 ++++++++++++++++++++++++++++++++++++++++-------
 1 files changed, 40 insertions(+), 7 deletions(-)

diff --git a/sandu-common/src/main/java/com/sandu/common/file/impl/AliOssFileServiceImpl.java b/sandu-common/src/main/java/com/sandu/common/file/impl/AliOssFileServiceImpl.java
index ac43eab..97df32c 100644
--- a/sandu-common/src/main/java/com/sandu/common/file/impl/AliOssFileServiceImpl.java
+++ b/sandu-common/src/main/java/com/sandu/common/file/impl/AliOssFileServiceImpl.java
@@ -13,8 +13,8 @@
 import com.sandu.common.execption.BusinessException;
 import com.sandu.common.file.FileUploadDto;
 import com.sandu.common.file.IFileUpload;
-import lombok.AllArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
+import org.springframework.beans.factory.annotation.Value;
 import org.springframework.stereotype.Component;
 import org.springframework.web.multipart.MultipartFile;
 
@@ -23,6 +23,8 @@
 import java.io.ByteArrayInputStream;
 import java.io.File;
 import java.io.IOException;
+import java.math.BigInteger;
+import java.security.MessageDigest;
 import java.time.LocalDateTime;
 import java.time.format.DateTimeFormatter;
 
@@ -34,10 +36,15 @@
 @Component
 public class AliOssFileServiceImpl implements IFileUpload {
 
-    private final String endPoint="oss-cn-shanghai.aliyuncs.com";
-    private final String keyId="LTAI5tPdpt5wvJyLipRijFSP";
-    private final String keySecret="1ahYfCKd0yTddsUnuDLQzI23MLh4VQ";
-    private final String bucketName= "ximonsmart";
+    @Value("${oss-conf.end-point}")
+    private String endPoint;
+    @Value("${oss-conf.key-id}")
+    private String keyId;
+    @Value("${oss-conf.key-secret}")
+    private String keySecret;
+    @Value("${oss-conf.bucket-name}")
+    private String bucketName;
+
     private String host;
 
     @Override
@@ -46,8 +53,9 @@
         long size = multipartFile.getSize();
         String suffix = FileUtil.getSuffix(originalFilename);
         String dayStr = LocalDateTime.now().format(DateTimeFormatter.ofPattern("yyyyMMddHHmm"));
-        String fileName = dayStr + UUID.fastUUID()+ "." + suffix;
+        String fileName = dayStr + UUID.fastUUID() + "." + suffix;
 
+        String md5 = getMD5(multipartFile);
         FileUploadDto uploadDto = new FileUploadDto();
         OSS ossClient = new OSSClientBuilder().build(endPoint, keyId, keySecret);
         try {
@@ -75,6 +83,7 @@
                         url = host + fileName;
                     }
                 }
+
                 uploadDto.setFileUrl(url);
             }
         } catch (Exception e) {
@@ -90,6 +99,7 @@
         uploadDto.setFileType(FileUtil.extName(originalFilename));
         uploadDto.setStorageType(FileStorageEnums.ALIBABA.getCode());
         uploadDto.setFileSize(size);
+        uploadDto.setMd5(md5);
 
         if (IMAGE_SUFFIX_LIST.contains(uploadDto.getFileType())) {
             BufferedImage img = null;
@@ -112,7 +122,7 @@
 //        long size = multipartFile.get();
         String suffix = FileUtil.getSuffix(originalFilename);
         String dayStr = LocalDateTime.now().format(DateTimeFormatter.ofPattern("yyyyMMddHHmm"));
-        String fileName = dayStr + UUID.fastUUID()+ "." + suffix;
+        String fileName = dayStr + UUID.fastUUID() + "." + suffix;
 
         FileUploadDto uploadDto = new FileUploadDto();
         OSS ossClient = new OSSClientBuilder().build(endPoint, keyId, keySecret);
@@ -158,4 +168,27 @@
 
         return uploadDto;
     }
+
+    /**
+     * 鑾峰彇涓婁紶鏂囦欢鐨凪D5鍊�
+     *
+     * @param file
+     * @return
+     */
+    public String getMD5(MultipartFile file) {
+
+        try {
+            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) {
+        }
+        return null;
+    }
+
 }

--
Gitblit v1.9.3