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 |  117 ++++++++++++++++++++++++++++++++++++++++++++--------------
 1 files changed, 89 insertions(+), 28 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 bc84db9..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
@@ -9,12 +9,18 @@
 import org.springframework.stereotype.Service;
 import org.springframework.web.multipart.MultipartFile;
 
-import java.io.ByteArrayInputStream;
-import java.io.ByteArrayOutputStream;
-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;
 
 @Service
@@ -38,12 +44,20 @@
         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();
             // 鎶婁笂浼犵殑鏂囦欢璺緞杩斿洖 锛堟墜鍔ㄦ嫾鎺ワ級
@@ -58,6 +72,60 @@
             // 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 {
@@ -101,6 +169,9 @@
         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);
         // 涓婁紶鏂囦欢娴�
@@ -109,32 +180,21 @@
             LocalDate date = LocalDate.now();
             DateTimeFormatter formatter = DateTimeFormatter.ofPattern("yyyyMMdd");
             String fileName = "sczf/" + date.format(formatter) + RandomUtils.generateRandomInt(8) + "." + ext;
-            System.out.println("鏂囦欢鍦板潃锛�" + fileName);
+
             ObjectMetadata objectMetadata = new ObjectMetadata();
             objectMetadata.setContentType(getcontentType(ext));
-            // 璋冪敤oss鐨勬柟娉曞疄鐜伴暱浼�
-            // 绗竴涓弬鏁� bucketName
-            // 绗簩涓弬鏁� 涓婁紶鍒皁ss鐨勬枃浠惰矾寰勫拰鏂囦欢鍚嶇О
-            ossClient.putObject(bucketName, fileName, inputStream, objectMetadata);
-            // 鍏抽棴OSSClient銆�
-            ossClient.shutdown();
-            // 鎶婁笂浼犵殑鏂囦欢璺緞杩斿洖 锛堟墜鍔ㄦ嫾鎺ワ級
-            // 杩欓噷璁剧疆鍥剧墖鏈夋晥鏃堕棿 鎴戣缃簡30骞�
-            Date expiration = new Date(System.currentTimeMillis() + (long) 946080000 * 1000);
-            String url = ossClient.generatePresignedUrl(bucketName, fileName, expiration).toString();
-            System.out.println("oss涓婁紶鎴愬姛锛�" + url);
 
-            // URL urlO = new URL(url);
-            // HttpURLConnection conn = (HttpURLConnection) urlO.openConnection();
-            // conn.setRequestMethod("GET");
-            // conn.setConnectTimeout(10 * 1000);
-            // conn.setRequestProperty("User-Agent", "Mozilla/4.0 (compatible; MSIE 5.0; Windows NT; DigExt)");
-            //
-            // InputStream inputStream2 = conn.getInputStream();// 閫氳繃杈撳叆娴佽幏鍙栧浘鐗囨暟鎹�
-            //
-            // byte[] data = readInputStream(inputStream2);
-            //
-            // System.out.println("鑾峰彇鍥剧墖鎴愬姛:" + url);
+            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());
@@ -179,8 +239,9 @@
         }
     }
 
+
     public static String getcontentType(String FilenameExtension) {
-        if (FilenameExtension.equalsIgnoreCase(".mp4")||FilenameExtension.equalsIgnoreCase(".mov")) {
+        if (FilenameExtension.equalsIgnoreCase(".mp4") || FilenameExtension.equalsIgnoreCase(".mov")) {
             return "video/mp4";
         }
         if (FilenameExtension.equalsIgnoreCase(".bmp")) {

--
Gitblit v1.8.0