From 21173f7ae60b1c598720c87ab966ea42b3d72c6a Mon Sep 17 00:00:00 2001
From: luxiaotao1123 <t1341870251@63.com>
Date: 星期一, 28 二月 2022 13:12:57 +0800
Subject: [PATCH] #

---
 src/main/java/com/zy/sc/common/service/OssService.java |   36 ++++--------------------------------
 1 files changed, 4 insertions(+), 32 deletions(-)

diff --git a/src/main/java/com/zy/sc/common/service/OssService.java b/src/main/java/com/zy/sc/common/service/OssService.java
index ebda706..cb573e7 100644
--- a/src/main/java/com/zy/sc/common/service/OssService.java
+++ b/src/main/java/com/zy/sc/common/service/OssService.java
@@ -12,14 +12,12 @@
 import org.springframework.beans.factory.annotation.Value;
 import org.springframework.stereotype.Service;
 
-import java.io.File;
-import java.io.FileInputStream;
-import java.io.FileNotFoundException;
 import java.io.InputStream;
 import java.net.URL;
 import java.security.SecureRandom;
 import java.text.SimpleDateFormat;
 import java.util.Date;
+import java.util.Objects;
 
 /**
  * 闃块噷浜慜SS鏈嶅姟绫�
@@ -55,21 +53,16 @@
      * 涓婁紶OSS鏂囦欢
      * @return the url
      */
-    public String upload(InputStream inputStream, String suffix){
-        if(suffix.trim().equals("file")){
-            suffix="jpg";
-        }
+    public String upload(InputStream inputStream, String fileName){
         OSSClient ossClient = new OSSClient(OSS_ENDPOINT, accessKeyId, accessKeySecret);
-        String name = "";
         try {
             if (!ossClient.doesBucketExist(bucket)) {
                 ossClient.createBucket(bucket);
                 log.info("鎮ㄧ殑Bucket涓嶅瓨鍦紝鍒涘缓Bucket锛歿}",bucket);
             }
-            name = String.valueOf(snowflakeIdWorker.nextId()).concat(".").concat(suffix);
-            // 娣诲姞 ContentType
+            String suffix = Objects.requireNonNull(fileName).substring(fileName.lastIndexOf(".") + 1);
+            String name = String.valueOf(snowflakeIdWorker.nextId()).concat(".").concat(suffix);
             ObjectMetadata objectMetadata = new ObjectMetadata();
-            objectMetadata.setContentType("image/jpg");
             ossClient.putObject(bucket, name, inputStream, objectMetadata);
             Date expiration = new Date(new Date().getTime() + 3600L * 1000 * 24 * 365 * 100);
             URL url = ossClient.generatePresignedUrl(bucket, name, expiration);
@@ -82,10 +75,6 @@
             ossClient.shutdown();
         }
         throw new CoolException("500-涓婁紶澶辫触");
-    }
-
-    public String upload(InputStream inputStream) {
-        return upload(inputStream, ".jpg");
     }
 
     /**
@@ -120,23 +109,6 @@
             ossClient.shutdown();
         }
         return true;
-    }
-
-    public static void main(String[] args) {
-        File file = new File("E:\\tmp\\v2-fbbb97b977b5cebc66dc3cefab0ac981_r.jpg");
-        try {
-            InputStream in = new FileInputStream(file);
-            OssService ossService = new OssService();
-            ossService.setAccessKeyId("LTAI4GDzr6ioSHuRw2mk22ug");
-            ossService.setAccessKeySecret("84CHL7tF21LbU1qpaP0jn9mIAZP9bv");
-            ossService.setBucket("tjdt");
-            ossService.setEndpoint("http://oss-cn-hangzhou.aliyuncs.com");
-            ossService.setSnowflakeIdWorker(new SnowflakeIdWorker());
-            System.out.println(ossService.upload(in));
-
-        } catch (FileNotFoundException e) {
-            e.printStackTrace();
-        }
     }
 
     public String uploadFile(InputStream inputStream, String fileName, Long len, String suffix){

--
Gitblit v1.9.1