update OSS模块 更新 阿里云 腾讯云 相关代码 重构部分代码结构

疯狂的狮子li

2021-07-23 7114fc27207ea8f8aca86dd91ff2acfe470c9349

 ruoyi-oss/src/main/java/com/ruoyi/oss/service/impl/AliyunCloudStorageServiceImpl.java

@@ -1,8 +1,13 @@
package com.ruoyi.oss.service.impl;

import cn.hutool.core.util.StrUtil;
import com.aliyun.oss.ClientConfiguration;
import com.aliyun.oss.OSSClient;
import com.aliyun.oss.common.auth.DefaultCredentialProvider;
import com.aliyun.oss.model.CannedAccessControlList;
import com.aliyun.oss.model.CreateBucketRequest;
import com.aliyun.oss.model.ObjectMetadata;
import com.aliyun.oss.model.PutObjectRequest;
import com.ruoyi.oss.entity.UploadResult;
import com.ruoyi.oss.enumd.CloudServiceEnumd;
import com.ruoyi.oss.exception.OssException;
@@ -39,8 +44,24 @@
            this.properties.getAccessKeyId(),
            this.properties.getAccessKeySecret());
         client = new OSSClient(this.properties.getEndpoint(), credentialProvider, configuration);
         createBucket();
      } catch (Exception e) {
         throw new IllegalArgumentException("阿里云存储配置错误! 请检查系统配置!");
      }
   }

   @Override
   public void createBucket() {
      try {
         String bucketName = properties.getBucketName();
         if (client.doesBucketExist(bucketName)) {
            return;
         }
         CreateBucketRequest createBucketRequest = new CreateBucketRequest(bucketName);
         createBucketRequest.setCannedACL(CannedAccessControlList.PublicRead);
         client.createBucket(createBucketRequest);
      } catch (Exception e) {
         throw new OssException("创建Bucket失败, 请核对阿里云配置信息");
      }
   }

@@ -57,35 +78,51 @@
   @Override
   public UploadResult upload(InputStream inputStream, String path, String contentType) {
      try {
         client.putObject(this.properties.getBucketName(), path, inputStream);
         ObjectMetadata metadata = new ObjectMetadata();
         metadata.setContentType(contentType);
         client.putObject(new PutObjectRequest(properties.getBucketName(), path, inputStream, metadata));
      } catch (Exception e) {
         throw new OssException("上传文件失败，请检查配置信息");
         throw new OssException("上传文件失败，请检查阿里云配置信息");
      }
      return new UploadResult().setUrl(properties.getEndpoint() + "/" + path).setFilename(path);
      return new UploadResult().setUrl(getEndpointLink() + "/" + path).setFilename(path);
   }

   @Override
   public void delete(String path) {
      path = path.replace(this.properties.getEndpoint() + "/", "");
      path = path.replace(getEndpointLink() + "/", "");
      try {
         client.deleteObject(this.properties.getBucketName(), path);
         client.deleteObject(properties.getBucketName(), path);
      } catch (Exception e) {
         throw new OssException("上传文件失败，请检查配置信息");
         throw new OssException("上传文件失败，请检查阿里云配置信息");
      }
   }

   @Override
   public UploadResult uploadSuffix(byte[] data, String suffix, String contentType) {
      return upload(data, getPath(this.properties.getPrefix(), suffix), contentType);
      return upload(data, getPath(properties.getPrefix(), suffix), contentType);
   }

   @Override
   public UploadResult uploadSuffix(InputStream inputStream, String suffix, String contentType) {
      return upload(inputStream, getPath(this.properties.getPrefix(), suffix), contentType);
      return upload(inputStream, getPath(properties.getPrefix(), suffix), contentType);
   }

   @Override
   public void afterPropertiesSet() throws Exception {
      OssFactory.register(getServiceType(),this);
      OssFactory.register(getServiceType(), this);
   }

   @Override
   public String getEndpointLink() {
      String endpoint = properties.getEndpoint();
      StringBuilder sb = new StringBuilder(endpoint);
      if (StrUtil.containsAnyIgnoreCase(endpoint, "http://")) {
         sb.insert(7, properties.getBucketName() + ".");
      } else if (StrUtil.containsAnyIgnoreCase(endpoint, "https://")) {
         sb.insert(8, properties.getBucketName() + ".");
      } else {
         throw new OssException("Endpoint配置错误");
      }
      return sb.toString();
   }
}