疯狂的狮子li
2022-02-10 1273421498f5e2afdf5bfef50b9358c45458fefd
ruoyi-oss/src/main/java/com/ruoyi/oss/service/impl/QiniuOssStrategy.java
@@ -12,6 +12,7 @@
import com.ruoyi.oss.exception.OssException;
import com.ruoyi.oss.properties.OssProperties;
import com.ruoyi.oss.service.abstractd.AbstractOssStrategy;
import org.springframework.stereotype.Component;
import java.io.InputStream;
@@ -20,108 +21,107 @@
 *
 * @author Lion Li
 */
@Component
public class QiniuOssStrategy extends AbstractOssStrategy {
   private UploadManager uploadManager;
   private BucketManager bucketManager;
   private Auth auth;
    private UploadManager uploadManager;
    private BucketManager bucketManager;
    private Auth auth;
   @Override
   public void init(OssProperties cloudStorageProperties) {
      properties = cloudStorageProperties;
      try {
         Configuration config = new Configuration(getRegion(properties.getRegion()));
         // https设置
         config.useHttpsDomains = false;
         config.useHttpsDomains = "Y".equals(properties.getIsHttps());
         uploadManager = new UploadManager(config);
         auth = Auth.create(properties.getAccessKey(), properties.getSecretKey());
         String bucketName = properties.getBucketName();
         bucketManager = new BucketManager(auth, config);
         if (!ArrayUtil.contains(bucketManager.buckets(), bucketName)) {
            bucketManager.createBucket(bucketName, properties.getRegion());
         }
      } catch (Exception e) {
         throw new OssException("七牛云存储配置错误! 请检查系统配置:[" + e.getMessage() + "]");
      }
   }
    @Override
    public void init(OssProperties ossProperties) {
        super.init(ossProperties);
        try {
            Configuration config = new Configuration(getRegion(properties.getRegion()));
            // https设置
            config.useHttpsDomains = false;
            config.useHttpsDomains = "Y".equals(properties.getIsHttps());
            uploadManager = new UploadManager(config);
            auth = Auth.create(properties.getAccessKey(), properties.getSecretKey());
            bucketManager = new BucketManager(auth, config);
            createBucket();
        } catch (Exception e) {
            throw new OssException("七牛云存储配置错误! 请检查系统配置:[" + e.getMessage() + "]");
        }
        isInit = true;
    }
   @Override
   public void createBucket() {
      try {
         String bucketName = properties.getBucketName();
         if (ArrayUtil.contains(bucketManager.buckets(), bucketName)) {
            return;
         }
         bucketManager.createBucket(bucketName, properties.getRegion());
      } catch (Exception e) {
         throw new OssException("创建Bucket失败, 请核对七牛云配置信息:[" + e.getMessage() + "]");
      }
   }
    @Override
    public void createBucket() {
        try {
            String bucketName = properties.getBucketName();
            if (ArrayUtil.contains(bucketManager.buckets(), bucketName)) {
                return;
            }
            bucketManager.createBucket(bucketName, properties.getRegion());
        } catch (Exception e) {
            throw new OssException("创建Bucket失败, 请核对七牛云配置信息:[" + e.getMessage() + "]");
        }
    }
   @Override
   public String getServiceType() {
      return OssEnumd.QINIU.getValue();
   }
    @Override
    public OssEnumd getServiceType() {
        return OssEnumd.QINIU;
    }
   @Override
   public UploadResult upload(byte[] data, String path, String contentType) {
      try {
    @Override
    public UploadResult upload(byte[] data, String path, String contentType) {
        try {
            String token = auth.uploadToken(properties.getBucketName());
            Response res = uploadManager.put(data, path, token, null, contentType, false);
         if (!res.isOK()) {
            throw new RuntimeException("上传七牛出错:" + res.error);
         }
      } catch (Exception e) {
         throw new OssException("上传文件失败,请核对七牛配置信息:[" + e.getMessage() + "]");
      }
      return new UploadResult().setUrl(getEndpointLink() + "/" + path).setFilename(path);
   }
            if (!res.isOK()) {
                throw new RuntimeException("上传七牛出错:" + res.error);
            }
        } catch (Exception e) {
            throw new OssException("上传文件失败,请核对七牛配置信息:[" + e.getMessage() + "]");
        }
        return UploadResult.builder().url(getEndpointLink() + "/" + path).filename(path).build();
    }
   @Override
   public void delete(String path) {
      try {
         path = path.replace(getEndpointLink() + "/", "");
         Response res = bucketManager.delete(properties.getBucketName(), path);
         if (!res.isOK()) {
            throw new RuntimeException("删除七牛文件出错:" + res.error);
         }
      } catch (Exception e) {
         throw new OssException(e.getMessage());
      }
   }
    @Override
    public void delete(String path) {
        try {
            path = path.replace(getEndpointLink() + "/", "");
            Response res = bucketManager.delete(properties.getBucketName(), path);
            if (!res.isOK()) {
                throw new RuntimeException("删除七牛文件出错:" + res.error);
            }
        } catch (Exception e) {
            throw new OssException(e.getMessage());
        }
    }
   @Override
   public UploadResult uploadSuffix(byte[] data, String suffix, String contentType) {
      return upload(data, getPath(properties.getPrefix(), suffix), contentType);
   }
    @Override
    public UploadResult uploadSuffix(byte[] data, String suffix, String contentType) {
        return upload(data, getPath(properties.getPrefix(), suffix), contentType);
    }
   @Override
   public UploadResult uploadSuffix(InputStream inputStream, String suffix, String contentType) {
      return upload(inputStream, getPath(properties.getPrefix(), suffix), contentType);
   }
    @Override
    public UploadResult uploadSuffix(InputStream inputStream, String suffix, String contentType) {
        return upload(inputStream, getPath(properties.getPrefix(), suffix), contentType);
    }
   @Override
   public String getEndpointLink() {
      return properties.getEndpoint();
   }
    @Override
    public String getEndpointLink() {
        return properties.getEndpoint();
    }
   private Region getRegion(String region) {
      switch (region) {
         case "z0":
            return Region.region0();
         case "z1":
            return Region.region1();
         case "z2":
            return Region.region2();
         case "na0":
            return Region.regionNa0();
         case "as0":
            return Region.regionAs0();
         default:
            return Region.autoRegion();
      }
   }
    private Region getRegion(String region) {
        switch (region) {
            case "z0":
                return Region.region0();
            case "z1":
                return Region.region1();
            case "z2":
                return Region.region2();
            case "na0":
                return Region.regionNa0();
            case "as0":
                return Region.regionAs0();
            default:
                return Region.autoRegion();
        }
    }
}