2021-03-15更新

2024-04-21 12:32:10 +00:00 · 2021-03-15 10:15:41 +08:00
parent ee52957d31
commit 867bde24be
145 changed files with 14644 additions and 25602 deletions
@@ -3,10 +3,7 @@ package cn.hellohao.service.impl;
 import cn.hellohao.pojo.Keys;
 import cn.hellohao.pojo.ReturnImage;
 import cn.hellohao.pojo.UploadConfig;
-import cn.hellohao.utils.DateUtils;
-import cn.hellohao.utils.DeleImg;
-import cn.hellohao.utils.ImgUrlUtil;
-import cn.hellohao.utils.Print;
+import cn.hellohao.utils.*;
 import com.netease.cloud.auth.BasicCredentials;
 import com.netease.cloud.auth.Credentials;
 import com.netease.cloud.services.nos.NosClient;
@@ -18,9 +15,7 @@ import com.qcloud.cos.auth.BasicCOSCredentials;
 import com.qcloud.cos.auth.COSCredentials;
 import com.qcloud.cos.exception.CosClientException;
 import com.qcloud.cos.exception.CosServiceException;
-import com.qcloud.cos.model.Bucket;
-import com.qcloud.cos.model.PutObjectRequest;
-import com.qcloud.cos.model.PutObjectResult;
+import com.qcloud.cos.model.*;
 import com.qcloud.cos.region.Region;
 import org.springframework.stereotype.Service;
 import org.springframework.web.multipart.MultipartFile;
@@ -36,94 +31,91 @@ public class COSImageupload {
    static Keys key;

    public Map<ReturnImage, Integer> ImageuploadCOS(Map<String, MultipartFile> fileMap, String username,
-                                                    Map<String, String> fileMap2,Integer setday) throws Exception {
+                                                    Map<String, String> fileMap2,Integer setday) {
        // 要上传文件的路径
        if(fileMap2==null){
            File file = null;
            Map<ReturnImage, Integer> ImgUrl = new HashMap<>();
-            for (Map.Entry<String, MultipartFile> entry : fileMap.entrySet()) {
-                String uuid = UUID.randomUUID().toString().replace("-", "").toLowerCase().substring(0,5);//生成一个没有-的uuid，然后取前5位
-                java.text.DateFormat format1 = new java.text.SimpleDateFormat("MMddhhmmss");
-                String times = format1.format(new Date());
-                file = changeFile(entry.getValue());
-                try {
-                    // 指定要上传到的存储桶
-                    String bucketName = BarrelName;
-                    // 指定要上传到 COS 上对象键
-                    String userkey =username + "/" + uuid+times + "." + entry.getKey();
-                    PutObjectRequest putObjectRequest = new PutObjectRequest(bucketName, userkey, file);
-                    PutObjectResult putObjectResult = cosClient.putObject(putObjectRequest);
-                    ReturnImage returnImage = new ReturnImage();
-                    returnImage.setImgname(entry.getValue().getOriginalFilename());
-                    returnImage.setImgurl(key.getRequestAddress() + "/" + userkey);
-                    ImgUrl.put(returnImage, (int) (entry.getValue().getSize()));
-                    if(setday>0) {
-                        String deleimg = DateUtils.plusDay(setday);
-                        DeleImg.charu(username + "/" + uuid + times + "." + entry.getKey() + "|" + deleimg + "|" + "6");
+            try {
+                for (Map.Entry<String, MultipartFile> entry : fileMap.entrySet()) {
+                    String uuid = UUID.randomUUID().toString().replace("-", "").toLowerCase().substring(0,5);//生成一个没有-的uuid，然后取前5位
+                    java.text.DateFormat format1 = new java.text.SimpleDateFormat("MMddhhmmss");
+                    String times = format1.format(new Date());
+                    file = SetFiles.changeFile(entry.getValue());
+                    try {
+                        // 指定要上传到的存储桶
+                        String bucketName = BarrelName;
+                        // 指定要上传到 COS 上对象键
+                        String userkey =username + "/" + uuid+times + "." + entry.getKey();
+                        PutObjectRequest putObjectRequest = new PutObjectRequest(bucketName, userkey, file);
+                        PutObjectResult putObjectResult = cosClient.putObject(putObjectRequest);
+                        ReturnImage returnImage = new ReturnImage();
+                        returnImage.setImgname(userkey);//entry.getValue().getOriginalFilename()
+                        returnImage.setImgurl(key.getRequestAddress() + "/" + userkey);
+                        ImgUrl.put(returnImage, (int) (entry.getValue().getSize()));
+                        if(setday>0) {
+                            String deleimg = DateUtils.plusDay(setday);
+                            DeleImg.charu(username + "/" + uuid + times + "." + entry.getKey() + "|" + deleimg + "|" + "6");
+                        }
+                    } catch (CosServiceException serverException) {
+                        serverException.printStackTrace();
+                    } catch (CosClientException clientException) {
+                        clientException.printStackTrace();
                    }
-                } catch (CosServiceException serverException) {
-                    serverException.printStackTrace();
-                } catch (CosClientException clientException) {
-                    clientException.printStackTrace();
                }
+            }catch(Exception e){
+                e.printStackTrace();
+                ImgUrl.put(null, 500);
            }
            return ImgUrl;
        }else{
            Map<ReturnImage, Integer> ImgUrl = new HashMap<>();
            ObjectMetadata meta = new ObjectMetadata();
            meta.setHeader("Content-Disposition", "inline");
-            for (Map.Entry<String, String> entry : fileMap2.entrySet()) {
-                String uuid = UUID.randomUUID().toString().replace("-", "").toLowerCase().substring(0,5);//生成一个没有-的uuid，然后取前5位
-                java.text.DateFormat format1 = new java.text.SimpleDateFormat("MMddhhmmss");
-                String times = format1.format(new Date());
-                String imgurl = entry.getValue();
-                File file = new File(imgurl);
-                FileInputStream fileInputStream = new FileInputStream(file);
-                try {
-                    String userkey =username + "/" + uuid+times + "." + entry.getKey();
-                    PutObjectRequest putObjectRequest = new PutObjectRequest(BarrelName, userkey, file);
-                    PutObjectResult putObjectResult = cosClient.putObject(putObjectRequest);
-                    ReturnImage returnImage = new ReturnImage();
-                    returnImage.setImgurl(key.getRequestAddress() + "/" + userkey);
-                    ImgUrl.put(returnImage, ImgUrlUtil.getFileSize2(new File(imgurl)));
-                    if(setday>0) {
-                        String deleimg = DateUtils.plusDay(setday);
-                        DeleImg.charu(username + "/" + uuid + times + "." + entry.getKey() + "|" + deleimg + "|" + "6");
+            try {
+                for (Map.Entry<String, String> entry : fileMap2.entrySet()) {
+                    String uuid = UUID.randomUUID().toString().replace("-", "").toLowerCase().substring(0,5);//生成一个没有-的uuid，然后取前5位
+                    java.text.DateFormat format1 = new java.text.SimpleDateFormat("MMddhhmmss");
+                    String times = format1.format(new Date());
+                    String imgurl = entry.getValue();
+                    File file = new File(imgurl);
+                    FileInputStream fileInputStream = new FileInputStream(file);
+                    try {
+                        String userkey =username + "/" + uuid+times + "." + entry.getKey();
+                        PutObjectRequest putObjectRequest = new PutObjectRequest(BarrelName, userkey, file);
+                        PutObjectResult putObjectResult = cosClient.putObject(putObjectRequest);
+                        ReturnImage returnImage = new ReturnImage();
+                        returnImage.setImgurl(key.getRequestAddress() + "/" + userkey);
+                        ImgUrl.put(returnImage, ImgUrlUtil.getFileSize2(new File(imgurl)));
+                        if(setday>0) {
+                            String deleimg = DateUtils.plusDay(setday);
+                            DeleImg.charu(username + "/" + uuid + times + "." + entry.getKey() + "|" + deleimg + "|" + "6");
+                        }
+                        boolean bb= new File(imgurl).getAbsoluteFile().delete();
+                        Print.Normal("删除情况"+bb);
+                    } catch (Exception e) {
+                        System.err.println("上传报错:" + e.getMessage());
+                    }
+                    if(fileInputStream!=null){
+                        fileInputStream.close();
+                        Print.Normal("流已经关闭");
                    }
-                    boolean bb= new File(imgurl).getAbsoluteFile().delete();
-                    Print.Normal("删除情况"+bb);
-                } catch (Exception e) {
-                    System.err.println("上传报错:" + e.getMessage());
-                }
-                if(fileInputStream!=null){
-                    fileInputStream.close();
-                    Print.Normal("流已经关闭");
                }
+            }catch(Exception e){
+                ImgUrl.put(null, 500);
            }
            cosClient.shutdown();
            return ImgUrl;
        }
    }

-    // 转换文件方法
-    private File changeFile(MultipartFile multipartFile) throws Exception {
-        // 获取文件名
-        String fileName = multipartFile.getOriginalFilename();
-        // 获取文件后缀
-        String prefix = fileName.substring(fileName.lastIndexOf("."));
-        // todo 修改临时文件文件名
-        File file = File.createTempFile(fileName, prefix);
-        multipartFile.transferTo(file);
-        return file;
-    }

-    //初始化网易NOS对象存储
    public static Integer Initialize(Keys k) {
        int ret = -1;
        if(k.getEndpoint()!=null && k.getAccessSecret()!=null && k.getEndpoint()!=null
                && k.getBucketname()!=null && k.getRequestAddress()!=null ) {
-            if(!k.getEndpoint().equals("") && !k.getAccessSecret().equals("") && !k.getEndpoint().equals("")
-                    && !k.getBucketname().equals("") && !k.getRequestAddress().equals("") ) {
+            if (!k.getEndpoint().equals("") && !k.getAccessSecret().equals("") && !k.getEndpoint().equals("")
+                    && !k.getBucketname().equals("") && !k.getRequestAddress().equals("")) {
                String secretId = k.getAccessKey();
                String secretKey = k.getAccessSecret();
                COSCredentials cred = new BasicCOSCredentials(secretId, secretKey);
@@ -131,8 +123,20 @@ public class COSImageupload {
                ClientConfig clientConfig = new ClientConfig(region);
                cosClient = new COSClient(cred, clientConfig);
                BarrelName = k.getBucketname();
-                key = k;
-                ret = 1;
+
+                ListObjectsRequest listObjectsRequest = new ListObjectsRequest();
+                listObjectsRequest.setBucketName(k.getBucketname());
+                listObjectsRequest.setDelimiter("/");
+                listObjectsRequest.setMaxKeys(1);
+                ObjectListing objectListing = null;
+                try {
+                    objectListing = cosClient.listObjects(listObjectsRequest);
+                    key = k;
+                    ret = 1;
+                } catch (Exception e) {
+                    System.out.println("COS - Waiting for configuration。");
+                    ret = -1;
+                }
            }
        }
        return ret;
@@ -148,7 +152,7 @@ public class COSImageupload {
            String uuid = UUID.randomUUID().toString().replace("-", "").toLowerCase().substring(0,5);//生成一个没有-的uuid，然后取前5位
            java.text.DateFormat format1 = new java.text.SimpleDateFormat("MMddhhmmss");
            String times = format1.format(new Date());
-            file = changeFile(entry.getValue());
+            file = SetFiles.changeFile_c(entry.getValue());
            try {
                ReturnImage returnImage = new ReturnImage();
                if(entry.getValue().getSize()/1024<=uploadConfig.getFilesizeuser()*1024){