From 46251c20b66bb1ca05058ae63a92a195e5543b90 Mon Sep 17 00:00:00 2001
From: zhanghua <314079846@qq.com>
Date: 星期四, 02 十一月 2023 21:27:37 +0800
Subject: [PATCH] 优化

---
 ycl-common/src/main/java/com/ycl/service/oss/impl/OssServiceImpl.java |   93 ++++++++++++++++++++++++++++++++++++++++++----
 1 files changed, 85 insertions(+), 8 deletions(-)

diff --git a/ycl-common/src/main/java/com/ycl/service/oss/impl/OssServiceImpl.java b/ycl-common/src/main/java/com/ycl/service/oss/impl/OssServiceImpl.java
index 47a4f18..0ec8e5a 100644
--- a/ycl-common/src/main/java/com/ycl/service/oss/impl/OssServiceImpl.java
+++ b/ycl-common/src/main/java/com/ycl/service/oss/impl/OssServiceImpl.java
@@ -5,18 +5,28 @@
 import com.aliyun.oss.model.ObjectMetadata;
 import com.ycl.service.oss.OssService;
 import com.ycl.utils.ConstantPropertiesUtils;
+import com.ycl.utils.common.RandomUtils;
 import org.springframework.stereotype.Service;
 import org.springframework.web.multipart.MultipartFile;
 
 import java.io.ByteArrayInputStream;
+import java.io.ByteArrayOutputStream;
 import java.io.InputStream;
+import java.time.LocalDate;
+import java.time.LocalDateTime;
+import java.time.format.DateTimeFormatter;
 import java.util.Date;
-import java.util.UUID;
 
 @Service
 public class OssServiceImpl implements OssService {
     @Override
     public String uploadImages(MultipartFile file) {
+        DateTimeFormatter dtf2 = DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss");
+
+        System.out.println("寮�濮嬩笂浼�--" + dtf2.format(LocalDateTime.now()));
+        if (file == null) {
+            return "涓婁紶鏂囦欢涓虹┖";
+        }
         String endpoint = ConstantPropertiesUtils.END_POINT;
         String accessKeyId = ConstantPropertiesUtils.ACCESS_KEY_ID;
         String accessKeySecret = ConstantPropertiesUtils.ACCESS_KEY_SECRET;
@@ -38,13 +48,33 @@
             ossClient.shutdown();
             // 鎶婁笂浼犵殑鏂囦欢璺緞杩斿洖 锛堟墜鍔ㄦ嫾鎺ワ級
             // 杩欓噷璁剧疆鍥剧墖鏈夋晥鏃堕棿 鎴戣缃簡30骞�
-            Date expiration = new Date(System.currentTimeMillis() + 946080000 * 1000);
+            Date expiration = new Date(System.currentTimeMillis() + (long) 946080000 * 1000);
             String url = ossClient.generatePresignedUrl(bucketName, fileName, expiration).toString();
+
+            System.out.println("涓婁紶缁撴潫--" + dtf2.format(LocalDateTime.now()));
             return url;
         } catch (Exception e) {
-            e.printStackTrace();
+            System.out.println("uploadImages涓婁紶鍥剧墖澶辫触锛�");
+            // e.printStackTrace();
             return null;
         }
+    }
+
+    private byte[] readInputStream(InputStream inStream) throws Exception {
+        ByteArrayOutputStream outStream = new ByteArrayOutputStream();
+        //鍒涘缓涓�涓狟uffer瀛楃涓�
+        byte[] buffer = new byte[1024];
+        //姣忔璇诲彇鐨勫瓧绗︿覆闀垮害锛屽鏋滀负-1锛屼唬琛ㄥ叏閮ㄨ鍙栧畬姣�
+        int len = 0;
+        //浣跨敤涓�涓緭鍏ユ祦浠巄uffer閲屾妸鏁版嵁璇诲彇鍑烘潵
+        while ((len = inStream.read(buffer)) != -1) {
+            //鐢ㄨ緭鍑烘祦寰�buffer閲屽啓鍏ユ暟鎹紝涓棿鍙傛暟浠h〃浠庡摢涓綅缃紑濮嬭锛宭en浠h〃璇诲彇鐨勯暱搴�
+            outStream.write(buffer, 0, len);
+        }
+        //鍏抽棴杈撳叆娴�
+        inStream.close();
+        //鎶妎utStream閲岀殑鏁版嵁鍐欏叆鍐呭瓨
+        return outStream.toByteArray();
     }
 
     @Override
@@ -71,34 +101,81 @@
         String accessKeySecret = ConstantPropertiesUtils.ACCESS_KEY_SECRET;
         String bucketName = ConstantPropertiesUtils.BUCKET_NAME;
 
+        DateTimeFormatter dtf2 = DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss");
+
+        System.out.println("鍑嗗涓婁紶--" + dtf2.format(LocalDateTime.now()));
         // 鍒涘缓OSSClient瀹炰緥銆�
         OSS ossClient = new OSSClientBuilder().build(endpoint, accessKeyId, accessKeySecret);
         // 涓婁紶鏂囦欢娴�
         try {
             // 鑾峰彇鏂囦欢鐨勫悕绉�
-            String fileName = "sczf/" + String.valueOf(UUID.randomUUID());
+            LocalDate date = LocalDate.now();
+            DateTimeFormatter formatter = DateTimeFormatter.ofPattern("yyyyMMdd");
+            String fileName = "sczf/" + date.format(formatter) + RandomUtils.generateRandomInt(8) + "." + ext;
+
             ObjectMetadata objectMetadata = new ObjectMetadata();
             objectMetadata.setContentType(getcontentType(ext));
+
+            System.out.println("寮�濮嬩笂浼�--" + dtf2.format(LocalDateTime.now()));
+
+            ossClient.putObject(bucketName, fileName, inputStream, objectMetadata);
+            ossClient.shutdown();
+
+            Date expiration = new Date(System.currentTimeMillis() + (long) 946080000 * 1000);
+
+            String url = ossClient.generatePresignedUrl(bucketName, fileName, expiration).toString();
+//            System.out.println("oss涓婁紶鎴愬姛锛�" + url);
+
+            System.out.println("涓婁紶瀹屾垚--" + dtf2.format(LocalDateTime.now()));
+            return url;
+        } catch (Exception e) {
+            System.out.println("oss寮傚父锛�" + e.getMessage());
+            // e.printStackTrace();
+            return this.uploadImages(inputStream, ext);
+        }
+    }
+
+    @Override
+    public String uploadVideo(MultipartFile file) {
+        if (file == null) {
+            return "涓婁紶瑙嗛涓虹┖";
+        }
+        String endpoint = ConstantPropertiesUtils.END_POINT;
+        String accessKeyId = ConstantPropertiesUtils.ACCESS_KEY_ID;
+        String accessKeySecret = ConstantPropertiesUtils.ACCESS_KEY_SECRET;
+        String bucketName = ConstantPropertiesUtils.BUCKET_NAME;
+        // 鍒涘缓OSSClient瀹炰緥
+        OSS ossClient = new OSSClientBuilder().build(endpoint, accessKeyId, accessKeySecret);
+        // 涓婁紶鏂囦欢娴�
+        try {
+            // 鑾峰彇鏂囦欢鐨勫悕绉�
+            String fileName = "sczf/" + file.getOriginalFilename();
+            ObjectMetadata objectMetadata = new ObjectMetadata();
+            objectMetadata.setContentType(getcontentType(fileName.substring(fileName.lastIndexOf("."))));
             // 璋冪敤oss鐨勬柟娉曞疄鐜伴暱浼�
             // 绗竴涓弬鏁� bucketName
             // 绗簩涓弬鏁� 涓婁紶鍒皁ss鐨勬枃浠惰矾寰勫拰鏂囦欢鍚嶇О
-            ossClient.putObject(bucketName, fileName, inputStream, objectMetadata);
+            ossClient.putObject(bucketName, fileName, new ByteArrayInputStream(file.getBytes()), objectMetadata);
             // 鍏抽棴OSSClient銆�
             ossClient.shutdown();
             // 鎶婁笂浼犵殑鏂囦欢璺緞杩斿洖 锛堟墜鍔ㄦ嫾鎺ワ級
             // 杩欓噷璁剧疆鍥剧墖鏈夋晥鏃堕棿 鎴戣缃簡30骞�
-            Date expiration = new Date(System.currentTimeMillis() + 946080000 * 1000);
+            Date expiration = new Date(System.currentTimeMillis() + (long) 946080000 * 1000);
             String url = ossClient.generatePresignedUrl(bucketName, fileName, expiration).toString();
-            // System.out.println("oss鎴愬姛锛�" + url);
+
             return url;
         } catch (Exception e) {
-            System.out.println("oss寮傚父锛�" + e.getMessage());
+            System.out.println("uploadImages涓婁紶瑙嗛澶辫触锛�");
             // e.printStackTrace();
             return null;
         }
     }
 
+
     public static String getcontentType(String FilenameExtension) {
+        if (FilenameExtension.equalsIgnoreCase(".mp4")||FilenameExtension.equalsIgnoreCase(".mov")) {
+            return "video/mp4";
+        }
         if (FilenameExtension.equalsIgnoreCase(".bmp")) {
             return "image/bmp";
         }

--
Gitblit v1.8.0