From bc2da7908a227c09e5cc7b6d8dab3e9c94b784a1 Mon Sep 17 00:00:00 2001
From: zhanghua <314079846@qq.com>
Date: 星期二, 12 十二月 2023 22:18:18 +0800
Subject: [PATCH] 图片压缩及bug修改

---
 ycl-common/src/main/java/com/ycl/service/oss/impl/OssServiceImpl.java |  167 ++++++++++++++++++++++++++++++++++++++++++++++++++++---
 1 files changed, 156 insertions(+), 11 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..3a24a08 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,34 @@
 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.InputStream;
+import javax.imageio.IIOImage;
+import javax.imageio.ImageIO;
+import javax.imageio.ImageWriteParam;
+import javax.imageio.ImageWriter;
+import javax.imageio.stream.ImageOutputStream;
+import java.awt.*;
+import java.awt.image.BufferedImage;
+import java.io.*;
+import java.time.LocalDate;
+import java.time.LocalDateTime;
+import java.time.format.DateTimeFormatter;
+import java.util.Arrays;
 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;
@@ -28,23 +44,105 @@
         try {
             // 鑾峰彇鏂囦欢鐨勫悕绉�
             String fileName = "sczf/" + file.getOriginalFilename();
+            byte[] newImg = file.getBytes();
+            String originalFilename = file.getOriginalFilename();
+            String fileType = originalFilename.substring(originalFilename.lastIndexOf("."));
+            if (".png".equalsIgnoreCase(fileType) || ".jpg".equalsIgnoreCase(fileType) || ".jpeg".equalsIgnoreCase(fileType)) {
+                if (file.getSize() > 2097152) {
+                    newImg = compressUnderSize(file.getBytes(), 2097152);
+                }
+            }
             ObjectMetadata objectMetadata = new ObjectMetadata();
             objectMetadata.setContentType(getcontentType(fileName.substring(fileName.lastIndexOf("."))));
             // 璋冪敤oss鐨勬柟娉曞疄鐜伴暱浼�
             // 绗竴涓弬鏁� bucketName
             // 绗簩涓弬鏁� 涓婁紶鍒皁ss鐨勬枃浠惰矾寰勫拰鏂囦欢鍚嶇О
-            ossClient.putObject(bucketName, fileName, new ByteArrayInputStream(file.getBytes()), objectMetadata);
+            ossClient.putObject(bucketName, fileName, new ByteArrayInputStream(newImg), 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("涓婁紶缁撴潫--" + dtf2.format(LocalDateTime.now()));
             return url;
         } catch (Exception e) {
-            e.printStackTrace();
+            System.out.println("uploadImages涓婁紶鍥剧墖澶辫触锛�");
+            // e.printStackTrace();
             return null;
         }
+    }
+
+    /**
+     * 鎸夌収 瀹介珮 姣斾緥鍘嬬缉
+     *
+     * @param imgIs 寰呭帇缂╁浘鐗囪緭鍏ユ祦
+     * @param scale 鍘嬬缉鍒诲害
+     * @param out   杈撳嚭
+     * @return 鍘嬬缉鍚庡浘鐗囨暟鎹�
+     * @throws IOException 鍘嬬缉鍥剧墖杩囩▼涓嚭閿�
+     */
+    public byte[] compress(byte[] srcImgData, double scale) throws IOException {
+        BufferedImage bi = ImageIO.read(new ByteArrayInputStream(srcImgData));
+        int width = (int) (bi.getWidth() * scale); // 婧愬浘瀹藉害
+        int height = (int) (bi.getHeight() * scale); // 婧愬浘楂樺害
+
+        Image image = bi.getScaledInstance(width, height, Image.SCALE_SMOOTH);
+        BufferedImage tag = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
+
+        Graphics g = tag.getGraphics();
+        g.setColor(Color.RED);
+        g.drawImage(image, 0, 0, null); // 缁樺埗澶勭悊鍚庣殑鍥�
+        g.dispose();
+
+        ByteArrayOutputStream bOut = new ByteArrayOutputStream();
+        ImageIO.write(tag, "JPEG", bOut);
+
+        return bOut.toByteArray();
+    }
+
+    /**
+     * 灏嗗浘鐗囧帇缂╁埌鎸囧畾澶у皬浠ュ唴
+     *
+     * @param srcImgData 婧愬浘鐗囨暟鎹�
+     * @param maxSize    鐩殑鍥剧墖澶у皬
+     * @return 鍘嬬缉鍚庣殑鍥剧墖鏁版嵁
+     */
+    public byte[] compressUnderSize(byte[] srcImgData, long maxSize) {
+        double scale = 0.9;
+        byte[] imgData = Arrays.copyOf(srcImgData, srcImgData.length);
+
+        if (imgData.length > maxSize) {
+            do {
+                try {
+                    imgData = compress(imgData, scale);
+
+                } catch (IOException e) {
+                    throw new IllegalStateException("鍘嬬缉鍥剧墖杩囩▼涓嚭閿欙紝璇峰強鏃惰仈绯荤鐞嗗憳锛�", e);
+                }
+
+            } while (imgData.length > maxSize);
+        }
+
+        return imgData;
+    }
+
+    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 +169,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